Update abstra-lib

abstra_internals/interface/sdk/ai/api.py

@@ -94,7 +94,7 @@ def prompt(
         function = _build_function_tool_call(format)
         tools.append({"type": "function", "function": function})
 
-    response = ai_repository.prompt(messages, tools, temperature).json()
+    response = ai_repository.prompt(messages, tools, temperature)
 
     if response.get("error"):
         raise Exception(response["error"])

abstra_internals/repositories/ai.py

@@ -1,3 +1,4 @@
+import json
 from abc import ABC, abstractmethod
 from typing import Any, List
 
@@ -12,30 +13,29 @@ def __init__(self, base_url: str) -> None:
         self.url = f"{base_url}/ai"
 
     @abstractmethod
-    def prompt(
-        self, messages: List[Any], tools: List[Any], temperature: float
-    ) -> requests.Response:
+    def prompt(self, messages: List[Any], tools: List[Any], temperature: float):
         raise NotImplementedError()
 
 
 class ProductionAiApiHttpClient(AiApiHttpClient):
-    def prompt(
-        self, messages: List[Any], tools: List[Any], temperature: float
-    ) -> requests.Response:
+    def prompt(self, messages: List[Any], tools: List[Any], temperature: float):
         body = {
             "messages": messages,
             "tools": tools,
             "temperature": temperature,
         }
-        r = requests.post(f"{self.url}/prompt", headers=SIDECAR_HEADERS, json=body)
-        r.raise_for_status()
-        return r
+        response = requests.post(
+            f"{self.url}/prompt", headers=SIDECAR_HEADERS, json=body
+        )
+        try:
+            response = response.json()
+            return response
+        except json.JSONDecodeError:
+            raise Exception(f"Error parsing JSON: {response.text}")
 
 
 class LocalAiApiHttpClient(AiApiHttpClient):
-    def prompt(
-        self, messages: List[Any], tools: List[Any], temperature: float
-    ) -> requests.Response:
+    def prompt(self, messages: List[Any], tools: List[Any], temperature: float):
         body = {
             "messages": messages,
             "tools": tools,
@@ -44,9 +44,12 @@ def prompt(
         headers = resolve_headers()
         if headers is None:
             raise Exception("You must be logged in to use AI")
-        r = requests.post(f"{self.url}/prompt", headers=headers, json=body)
-        r.raise_for_status()
-        return r
+        response = requests.post(f"{self.url}/prompt", headers=headers, json=body)
+        try:
+            response = response.json()
+            return response
+        except json.JSONDecodeError:
+            raise Exception(f"Error parsing JSON: {response.text}")
 
 
 def ai_api_http_client_factory() -> AiApiHttpClient: