AAP-36462: [r11]: Add support for HTTP422 response (#1672)

ansible · Nov 25, 2024 · 06f3806 · 06f3806
1 parent fd6d8fd
commit 06f3806
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/src/features/lightspeed/errors.ts b/src/features/lightspeed/errors.ts
@@ -184,6 +184,20 @@ ERRORS.addError(
   ),
 );
 ERRORS.addError(400, new Error("error__feedback_validation"));
+ERRORS.addError(
+  400,
+  new Error(
+    "error__wca_inference_failure",
+    "IBM watsonx Code Assistant inference failed. Please check your input and try again.",
+  ),
+);
+ERRORS.addError(
+  400,
+  new Error(
+    "error__wca_validation_failure",
+    "IBM watsonx Code Assistant failed to validate the response from the model. Please check your input and try again.",
+  ),
+);
 
 ERRORS.addError(
   403,

diff --git a/test/units/lightspeed/utils/handleApiError.test.ts b/test/units/lightspeed/utils/handleApiError.test.ts
@@ -150,6 +150,34 @@ describe("testing the error handling", () => {
     );
     assert.equal(error.message, "A field was invalid.");
   });
+
+  it("err WCA inference failure", () => {
+    const error = mapError(
+      createError(400, {
+        code: "error__wca_inference_failure",
+        message:
+          "IBM watsonx Code Assistant inference failed. Please check your input and try again.",
+      }),
+    );
+    assert.equal(
+      error.message,
+      "IBM watsonx Code Assistant inference failed. Please check your input and try again.",
+    );
+  });
+
+  it("err WCA validation failure", () => {
+    const error = mapError(
+      createError(400, {
+        code: "error__wca_validation_failure",
+        detail: { reason: "error 1" },
+      }),
+    );
+    assert.equal(
+      error.message,
+      "IBM watsonx Code Assistant failed to validate the response from the model. Please check your input and try again.",
+    );
+    assert.equal(error.detail, "reason: error 1");
+  });
   // =================================
 
   // =================================