Improve error handling

Author: guerler

lib/galaxy/webapps/galaxy/api/plugins.py

@@ -203,20 +203,27 @@ async def _open_ai_adapter(
         try:
             client = AsyncOpenAI(**client_kwargs)
         except Exception as e:
-            log.debug("Failed to initialize OpenAI client", exc_info=e)
+            log.debug("Failed to initialize OpenAI client.", exc_info=e)
             return self.create_error("Failed to initialize AI client.", 500)
 
         # Connect to ai provider
         log.info(f"Proxying to {ai_model}, tokens: {max_tokens}")
-        response = await client.chat.completions.create(
-            max_tokens=max_tokens,
-            messages=messages,
-            model=ai_model,
-            stream=stream,
-            temperature=TEMPERATURE,
-            tools=tools,
-            top_p=TOP_P,
-        )
+        try:
+            response = await client.chat.completions.create(
+                max_tokens=max_tokens,
+                messages=messages,
+                model=ai_model,
+                stream=stream,
+                temperature=TEMPERATURE,
+                tools=tools,
+                top_p=TOP_P,
+            )
+        except Exception as e:
+            log.debug("Failed to complete OpenAI request.", exc_info=e)
+            status_code = getattr(e, "status_code", 500)
+            if hasattr(e, "body") and isinstance(e.body, dict):
+                return JSONResponse(status_code=status_code, content=dict(error=e.body))
+            return self.create_error("Failed to complete OpenAI request.", status_code)
 
         # Parse response
         if stream: