Merge pull request #2 from Infuzu/dev

feat: enhance error handling and type annotations

Author: infuzu-yidisprei

main.py

@@ -1,18 +1,18 @@
-from src.infuzu import create_chat_completion, ChatCompletionsHandlerRequestMessage
+from src.infuzu import (create_chat_completion, ChatCompletionsHandlerRequestMessage)
 from dotenv import load_dotenv
 
 
 load_dotenv()
 
 
-messages = [
+messages: list[ChatCompletionsHandlerRequestMessage] = [
     ChatCompletionsHandlerRequestMessage(role="system", content="You are a helpful assistant."),
     ChatCompletionsHandlerRequestMessage(role="user", content="What is the capital of France?"),
 ]
 
 
 try:
-    response = create_chat_completion(messages=messages)
+    response: dict[str, any] = create_chat_completion(messages=messages)
     print(response)
 except Exception as e:
     print(f"Error: {e}")

src/infuzu/__init__.py

@@ -1,7 +1,14 @@
-from .api_client import create_chat_completion, ChatCompletionsHandlerRequestMessage, ChatCompletionsRequestContentPart
+from .api_client import (
+    create_chat_completion, ChatCompletionsHandlerRequestMessage, ChatCompletionsRequestContentPart
+)
+from .errors import (InfuzuAPIError, APIWarning, APIError)
 
 __all__: list[str] = [
     "create_chat_completion",
     "ChatCompletionsRequestContentPart",
-    "ChatCompletionsHandlerRequestMessage"
+    "ChatCompletionsHandlerRequestMessage",
+
+    "InfuzuAPIError",
+    "APIWarning",
+    "APIError",
 ]

src/infuzu/api_client.py

@@ -122,6 +122,5 @@ def create_chat_completion(
 
         response.raise_for_status()
         return response.json()
-
     except httpx.HTTPStatusError as e:
         raise InfuzuAPIError(e)

src/infuzu/errors.py

@@ -1,6 +1,46 @@
+import logging
+from json import JSONDecodeError
+from typing import Optional
 import httpx
+from pydantic import BaseModel
 
 
-class InfuzuAPIError(Exception):
-    def __init__(self, base_error: httpx.HTTPError) -> None:
-        super().__init__(f"HTTP Error: {str(base_error)}")
+logger: logging.Logger = logging.getLogger(__name__)
+
+
+class APIError(BaseModel):
+    code: Optional[str] = None
+    message: Optional[str] = None
+
+    class Config:
+        extra: str = "allow"
+
+
+class APIWarning(BaseModel):
+    code: Optional[str] = None
+    message: Optional[str] = None
+
+    class Config:
+        extra: str = "allow"
+
+
+class InfuzuAPIError(httpx.HTTPStatusError):
+    def __init__(self, base_error: httpx.HTTPStatusError) -> None:
+        super().__init__(
+            f"Infuzu API Error: {base_error}", request=base_error.request, response=base_error.response
+        )
+
+        try:
+            self.response_json: dict[str, any] = self.response.json()
+        except JSONDecodeError as e:
+            logger.warning(f"Failed to decode JSON response: {e}")
+            self.response_json: dict[str, any] = {}
+
+        self.results: dict[str, any] = self.response_json.get("results", {})
+
+        self.errors: list[APIError] = [APIError(**error) for error in self.response_json.get("errors", [])]
+
+        self.warnings: list[APIWarning] = [APIWarning(**warning) for warning in self.response_json.get("warnings", [])]
+
+    def __str__(self) -> str:
+        return f"{super().__str__()} | Errors: {self.errors} | Warnings: {self.warnings} | Results: {self.results}"