Merge 3c3f5d6 into c2a6cf6

IBM · Nov 14, 2023 · ee8db27 · ee8db27
2 parents c2a6cf6 + 3c3f5d6
commit ee8db27
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/genai/exceptions/genai_exception.py b/src/genai/exceptions/genai_exception.py
@@ -17,7 +17,7 @@ def __init__(self, error: Union[Exception, Response, str, ErrorResponse]) -> Non
         elif isinstance(error, Response):
             try:
                 self.error = ErrorResponse(**error.json())
-                self.error_message = self.error.message
+                self.error_message = self.error.model_dump_json()
             except ValidationError:
                 self.error = error
                 self.error_message = str(error.content)

diff --git a/src/genai/services/async_generator.py b/src/genai/services/async_generator.py
@@ -121,7 +121,7 @@ async def _get_response_json(self, model, inputs, params, options):
             response_raw = await self._service_fn(model, inputs, params, options)
             response = response_raw.json()
             if response_raw and not response_raw.is_success:
-                raise Exception(response)
+                raise GenAiException(response)
         except Exception as ex:
             logger.error("Error in _get_response_json {}: {}".format(type(ex), str(ex)))
             if self.throw_on_error:

diff --git a/src/genai/utils/http_utils.py b/src/genai/utils/http_utils.py
@@ -55,7 +55,7 @@ async def handle_async_request(
 
                 if ex.response and "application/json" in ex.response.headers["Content-Type"]:
                     await ex.response.aread()
-                    raise GenAiException(ex.response)
+                    raise GenAiException(ex.response) from None
                 else:
                     raise ex