Merge pull request #1 from Infuzu/dev

Dev

Author: infuzu-yidisprei

src/infuzu/api_client.py

@@ -2,6 +2,7 @@
 import os
 from typing import (Optional, Dict, Union, List)
 from pydantic import (BaseModel, validator)
+from .errors import InfuzuAPIError
 
 
 class ModelWeights(BaseModel):
@@ -10,6 +11,9 @@ class ModelWeights(BaseModel):
     start_latency: Optional[float] = None
     end_latency: Optional[float] = None
 
+    class Config:
+        extra: str = "allow"
+
 
 class InfuzuModelParams(BaseModel):
     llms: Optional[List[str]] = None
@@ -19,13 +23,19 @@ class InfuzuModelParams(BaseModel):
     max_input_cost: Optional[float] = None
     max_output_cost: Optional[float] = None
 
+    class Config:
+        extra: str = "allow"
+
 
 class ChatCompletionsRequestContentPart(BaseModel):
     type: str
     text: Optional[str] = None
     image_url: Optional[str] = None
     input_audio: Optional[str] = None
 
+    class Config:
+        extra: str = "allow"
+
     @validator("text", always=True)
     def check_content_fields(cls, value, values):
         if "type" in values:
@@ -42,6 +52,9 @@ class ChatCompletionsHandlerRequestMessage(BaseModel):
     role: str
     name: Optional[str] = None
 
+    class Config:
+        extra: str = "allow"
+
     @validator('role')
     def role_must_be_valid(cls, v):
         if v not in ('system', 'user', 'assistant'):
@@ -111,11 +124,4 @@ def create_chat_completion(
         return response.json()
 
     except httpx.HTTPStatusError as e:
-        print(f"HTTP Error: {e}")
-        raise
-    except httpx.RequestError as e:
-        print(f"Request Error: {e}")
-        raise
-    except Exception as e:
-        print(f"An unexpected error occurred: {e}")
-        raise
+        raise InfuzuAPIError(e)

src/infuzu/errors.py

@@ -0,0 +1,6 @@
+import httpx
+
+
+class InfuzuAPIError(Exception):
+    def __init__(self, base_error: httpx.HTTPError) -> None:
+        super().__init__(f"HTTP Error: {str(base_error)}")