GPTRouter 0.4.3__tar.gz → 0.4.5__tar.gz

{GPTRouter-0.4.3 → GPTRouter-0.4.5}/GPTRouter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.4.3
+Version: 0.4.5
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.4.3 → GPTRouter-0.4.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.4.3
+Version: 0.4.5
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.4.3 → GPTRouter-0.4.5}/gpt_router/models.py

@@ -1,7 +1,7 @@
 from uuid import UUID
 from pydantic import BaseModel, Field
 
-from typing import List, Optional, Dict, Any
+from typing import List, Optional, Dict, Any, Union
 from humps import camelize
 
 
@@ -15,6 +15,7 @@ class GPTRouterMetadata(BaseModel):
     created_by_user_id: Optional[str] = None
     appId: Optional[str] = None
 
+
 class GenerationParams(BaseModel):
     messages: Optional[List[Any]] = None
     max_tokens: Optional[int] = None
@@ -28,6 +29,7 @@ class GenerationParams(BaseModel):
     functions: Optional[List[Any]] = None
     function_call: Optional[dict] = None
     tools: Optional[List[Any]] = None
+    tool_choice: Optional[Union[str, dict]] = None
     response_format: Optional[Dict[str, Any]] = None
     parallel_tool_calls: Optional[bool] = None
     tag: Optional[str] = None
@@ -111,7 +113,7 @@ class ImageGenerationRequest(BaseModel):
 class ImageGenerationResponse(BaseModel):
     url: str = None
     base64: str = None
-    finish_reason: Optional[str] = Field(default="SUCCESS", alias='finishReason')
+    finish_reason: Optional[str] = Field(default="SUCCESS", alias="finishReason")
 
 
 class ModelRouterGenerationRequest(BaseModel):
@@ -134,6 +136,7 @@ class ModelRouterGenerationRequest(BaseModel):
 
         return result
 
+
 class ModelRouterEmbeddingsGenerationParams(BaseModel):
     input: List[str]
 
@@ -141,6 +144,7 @@ class ModelRouterEmbeddingsGenerationParams(BaseModel):
 class ModelRouterEmbeddingsGenerationRequest(ModelRouterGenerationRequest):
     prompt_params: ModelRouterEmbeddingsGenerationParams = Field(alias="promptParams")
 
+
 class ModelRouterGenerationResponse(BaseModel):
     id: str
     choices: List[Choice]
@@ -150,9 +154,9 @@ class ModelRouterGenerationResponse(BaseModel):
     meta: Optional[Usage]
 
 
-
 class EmbeddingsChoice(BaseModel):
     embedding: List[float]
 
+
 class ModelRouterEmbeddingsGenerationResponse(BaseModel):
-    choices: List[EmbeddingsChoice]
+    choices: List[EmbeddingsChoice]

{GPTRouter-0.4.3 → GPTRouter-0.4.5}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name="GPTRouter",
-    version="0.4.3",
+    version="0.4.5",
     description="A Python package for working with GPTRouter APIs",
     long_description=long_description,
     long_description_content_type="text/markdown",