@@ -65,7 +65,17 @@ public static List<DetokenizeRequest> createDetokenizeBatches(DetokenizeRequest
6565 public static ErrorRecord createErrorRecord (Map <String , Object > recordMap , int indexNumber ) {
6666 ErrorRecord err = null ;
6767 if (recordMap != null ) {
68- int code = recordMap .containsKey ("http_code" ) ? (Integer ) recordMap .get ("http_code" ) : 500 ;
68+ int code = 500 ;
69+ if (recordMap .containsKey ("http_code" )) {
70+ code = (Integer ) recordMap .get ("http_code" );
71+ } else if (recordMap .containsKey ("httpCode" )) {
72+ code = (Integer ) recordMap .get ("httpCode" );
73+
74+ } else {
75+ if (recordMap .containsKey ("statusCode" )) {
76+ code = (Integer ) recordMap .get ("statusCode" );
77+ }
78+ }
6979 String message = recordMap .containsKey ("error" ) ? (String ) recordMap .get ("error" ) :
7080 recordMap .containsKey ("message" ) ? (String ) recordMap .get ("message" ) : "Unknown error" ;
7181 err = new ErrorRecord (indexNumber , message , code );
@@ -126,8 +136,8 @@ public static List<ErrorRecord> handleDetokenizeBatchException(
126136 Map <String , Object > responseBody = (Map <String , Object >) apiException .body ();
127137 int indexNumber = batchNumber * batchSize ;
128138 if (responseBody != null ) {
129- if (responseBody .containsKey ("records " )) {
130- Object recordss = responseBody .get ("records " );
139+ if (responseBody .containsKey ("response " )) {
140+ Object recordss = responseBody .get ("response " );
131141 if (recordss instanceof List ) {
132142 List <?> recordsList = (List <?>) recordss ;
133143 for (Object record : recordsList ) {
0 commit comments