Update helpers/ai_client.py

helpers/ai_client.py

@@ -2,6 +2,11 @@
 import requests
 import os
 from typing import Optional, Dict, Any
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class AIClient:
     def __init__(self):
@@ -33,7 +38,7 @@ class AIClient:
             Dict[str, Any]: The JSON response from the LLM API.
 
         Raises:
-            HTTPException: If the API request fails.
+            Exception: If the API request fails or the response is invalid.
         """
         if api_key is None:
             api_key = self.api_key
@@ -52,9 +57,20 @@ class AIClient:
             "Content-Type": "application/json"
         }
 
-        # Use requests to call the external API
-        response = requests.post(self.llm_api_url, json=payload, headers=headers)
-        if response.status_code != 200:
-            raise Exception(f"Error from LLM API: {response.status_code} - {response.text}")
+        try:
+            # Use requests to call the external API
+            response = requests.post(self.llm_api_url, json=payload, headers=headers)
+            logger.info(f"API Response Status Code: {response.status_code}")
+            logger.info(f"API Response Content: {response.text}")
+
+            # Check if the response is valid JSON
+            if response.status_code != 200:
+                raise Exception(f"Error from LLM API: {response.status_code} - {response.text}")
 
-        return response.json()
+            return response.json()
+        except requests.exceptions.JSONDecodeError as e:
+            logger.error(f"Failed to decode JSON response: {e}")
+            raise Exception(f"Invalid response from LLM API: {response.text}")
+        except Exception as e:
+            logger.error(f"Error in send_prompt: {e}")
+            raise