diff --git a/src/main/java/com/twilio/oai/StringHelper.java b/src/main/java/com/twilio/oai/StringHelper.java index 44f047564..0f38ef18d 100644 --- a/src/main/java/com/twilio/oai/StringHelper.java +++ b/src/main/java/com/twilio/oai/StringHelper.java @@ -48,4 +48,20 @@ private String convertFirstChar(final String inputWord, final UnaryOperator= 200 && code <= 399); + } catch (NumberFormatException e) { + // Handle case where responseCode is not a valid integer + return false; + } + } } diff --git a/src/main/java/com/twilio/oai/api/CsharpApiResourceBuilder.java b/src/main/java/com/twilio/oai/api/CsharpApiResourceBuilder.java index 464781f7f..da7660f95 100644 --- a/src/main/java/com/twilio/oai/api/CsharpApiResourceBuilder.java +++ b/src/main/java/com/twilio/oai/api/CsharpApiResourceBuilder.java @@ -210,18 +210,20 @@ private List getNonPathParams(List allParams public ApiResourceBuilder updateResponseModel(Resolver codegenPropertyIResolver, Resolver codegenModelResolver) { List responseModels = new ArrayList<>(); codegenOperationList.forEach(codegenOperation -> { - codegenOperation.responses.forEach(response -> { - String modelName = response.dataType; - if (response.dataType != null && response.dataType.startsWith(EnumConstants.CsharpDataTypes.LIST.getValue())){ - modelName = response.baseType; - } - Optional responseModel = Utility.getModel(allModels, modelName, recordKey, codegenOperation); - if ((responseModel == null) || responseModel.isEmpty() || (Integer.parseInt(response.code) >= 400)) { - return; - } - codegenModelResolver.resolve(responseModel.get(), this); - responseModels.add(responseModel.get()); - }); + codegenOperation.responses.stream() + .filter(response -> StringHelper.isSuccess(response.code.trim())) + .forEach(response -> { + String modelName = response.dataType; + if (response.dataType != null && response.dataType.startsWith(EnumConstants.CsharpDataTypes.LIST.getValue())){ + modelName = response.baseType; + } + Optional responseModel = Utility.getModel(allModels, modelName, recordKey, codegenOperation); + if ((responseModel == null) || responseModel.isEmpty() || (Integer.parseInt(response.code) >= 400)) { + return; + } + codegenModelResolver.resolve(responseModel.get(), this); + responseModels.add(responseModel.get()); + }); }); this.apiResponseModels = getDistinctResponseModel(responseModels); return this; diff --git a/src/main/java/com/twilio/oai/api/JavaApiResourceBuilder.java b/src/main/java/com/twilio/oai/api/JavaApiResourceBuilder.java index c342a9559..ef61fcb5b 100644 --- a/src/main/java/com/twilio/oai/api/JavaApiResourceBuilder.java +++ b/src/main/java/com/twilio/oai/api/JavaApiResourceBuilder.java @@ -247,7 +247,7 @@ public ApiResourceBuilder updateResponseModel(Resolver codegenP jsonRequestBodyResolver.setResourceName(resourceName); co.responses .stream() - .filter(response -> SUCCESS.test(Integer.parseInt(response.code.trim()))) + .filter(response -> StringHelper.isSuccess(response.code.trim())) .map(response -> { if (response.dataType != null && response.dataType.startsWith(EnumConstants.JavaDataTypes.LIST.getValue())) { return response.baseType;