Update helpers/ai_client.py

helpers/ai_client.py

@@ -22,9 +22,9 @@ class AIClient:
         conversation_id: str = "string",
         user_id: str = "string",
         api_key: Optional[str] = None
-    ) -> Dict[str, Any]:
+    ) -> str:
         """
-        Sends a prompt to the LLM API and returns the response.
+        Sends a prompt to the LLM API and returns the response as text.
 
         Args:
             prompt (str): The user's input prompt.
@@ -35,10 +35,10 @@ class AIClient:
             api_key (str): API key for authentication.
 
         Returns:
-            Dict[str, Any]: The JSON response from the LLM API.
+            str: The text response from the LLM API.
 
         Raises:
-            Exception: If the API request fails or the response is invalid.
+            Exception: If the API request fails.
         """
         if api_key is None:
             api_key = self.api_key
@@ -57,20 +57,10 @@ class AIClient:
             "Content-Type": "application/json"
         }
 
-        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}")
+        # 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}")
 
-            # 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()
-        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
+        # Return the response as text
+        return response.text