GPTRouter 1.0.0__tar.gz → 1.0.2__tar.gz

{GPTRouter-1.0.0 → GPTRouter-1.0.2}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-1.0.0 → GPTRouter-1.0.2}/PKG-INFO

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

{GPTRouter-1.0.0 → GPTRouter-1.0.2}/gpt_router/models.py

@@ -1,7 +1,7 @@
 from uuid import UUID
 from pydantic import BaseModel, Field, root_validator, validator
 from enum import Enum
-from typing import Dict, List, Optional, Any, Union
+from typing import Dict, List, Literal, Optional, Any, Union
 from humps import camelize
 
 
@@ -32,8 +32,21 @@ class ExtendedThinking(BaseModel):
     budget_tokens: Optional[int] = 1024
 
 
+class UserLocationApproximate(BaseModel):
+    city: Optional[str] = None
+    country: Optional[str] = None
+    region: Optional[str] = None
+    timezone: Optional[str] = None
+
+
+class UserLocation(BaseModel):
+    type: Optional[str] = None
+    approximate: Optional[UserLocationApproximate] = None
+
+
 class WebSearchOptions(BaseModel):
     search_context_size: Optional[SearchContextSize] = None
+    user_location: Optional[Union[UserLocation, dict]] = None
 
 
 class GenerationParams(BaseModel):
@@ -62,6 +75,7 @@ class GenerationParams(BaseModel):
     colasce_into_thinking_tag: Optional[bool] = None
     use_response_api: Optional[bool] = None
     web_search_options: Optional[WebSearchOptions] = None
+    reasoning_effort: Optional[Literal["low", "medium", "high"]] = None
 
 
 class ModelGenerationRequest(BaseModel):
@@ -119,6 +133,10 @@ class ImageGenerationRequest(BaseModel):
     width: Optional[int] = Field(None, alias="width")
     height: Optional[int] = Field(None, alias="height")
     num_inference_steps: Optional[int] = Field(None, alias="numInferenceSteps")
+    app_id: Optional[str] = Field(None, alias="appId")
+    user_id: Optional[str] = Field(None, alias="createdByUserId")
+    history_id: Optional[str] = Field(None, alias="historyId")
+    feature: Optional[str] = Field(None, alias="feature")
 
     class Config:
         allow_population_by_field_name = True
@@ -151,11 +169,11 @@ class ImageEditRequest(BaseModel):
     response_format: Optional[str] = Field(alias="responseFormat", default=None)
     size: Optional[str] = Field(default="1024x1024")
     user: Optional[str] = Field(default=None)
-    
+
     class Config:
         allow_population_by_field_name = True
         protected_namespaces = ()
-        
+
     def dict(self, *args, **kwargs):
         base_dict = super().dict(*args, **kwargs, exclude_none=True, by_alias=True)
         return base_dict

{GPTRouter-1.0.0 → GPTRouter-1.0.2}/setup.py

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