GPTRouter 0.3.9__tar.gz → 0.4.1__tar.gz

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/PKG-INFO

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

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/gpt_router/client.py

@@ -118,6 +118,7 @@ class GPTRouterClient:
                 "historyId": (
                     str(metadata["history_id"]) if metadata.get("history_id") else None
                 ),
+                "appId": metadata.get("appId"),
             }
         )
 

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/gpt_router/enums.py

@@ -50,6 +50,9 @@ class ModelsEnum(Enum):
     STABLE_DIFFUSION_V6 = "stable-diffusion-v1-6"
 
     REPLICATE_FLUX = "black-forest-labs/flux-dev"
+    REPLICATE_FLUX_PRO = "black-forest-labs/flux-pro"
+    REPLICATE_FLUX_11_PRO = "black-forest-labs/flux-1.1-pro"
+    REPLICATE_FLUX_SCHNELL = "black-forest-labs/flux-schnell"
 
     OPENAI_EMBEDDINGS_ADA_002 = "text-embedding-ada-002"
     AZURE_OPENAI_EMBEDDINGS_ADA_002 = "text-embedding-ada-002"
@@ -164,6 +167,15 @@ class GPTRouterEngines(Enum):
     REPLICATE_FLUX = ModelDetails(
         ProvidersEnum.REPLICATE.value, "black-forest-labs/flux-dev"
     )
+    REPLICATE_FLUX_PRO = ModelDetails(
+        ProvidersEnum.REPLICATE.value, "black-forest-labs/flux-pro"
+    )
+    REPLICATE_FLUX_11_PRO = ModelDetails(
+        ProvidersEnum.REPLICATE.value, "black-forest-labs/flux-1.1-pro"
+    )
+    REPLICATE_FLUX_SCHNELL = ModelDetails(
+        ProvidersEnum.REPLICATE.value, "black-forest-labs/flux-schnell"
+    )
 
     LLAMA_3_70B_NITRO = ModelDetails(
         ProvidersEnum.OPENROUTER, "meta-llama/llama-3-70b-instruct:nitro"

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/gpt_router/models.py

@@ -13,6 +13,7 @@ class GPTRouterMetadata(BaseModel):
     tag: Optional[str] = None
     history_id: Optional[str] = None
     created_by_user_id: Optional[str] = None
+    appId: Optional[str] = None
 
 class GenerationParams(BaseModel):
     messages: Optional[List[Any]] = None
@@ -32,6 +33,7 @@ class GenerationParams(BaseModel):
     tag: Optional[str] = None
     user_id: Optional[str] = None
     history_id: Optional[str] = None
+    appId: Optional[str] = None
     metadata: Optional[dict] = None
     raw_event: Optional[bool] = None
 
@@ -89,6 +91,7 @@ class ImageGenerationRequest(BaseModel):
     num_images: int = Field(alias="numImages", default=1)
     width: Optional[int] = Field(None, alias="width")
     height: Optional[int] = Field(None, alias="height")
+    num_inference_steps: Optional[int] = Field(None, alias="numInferenceSteps")
 
     class Config:
         allow_population_by_field_name = True

{GPTRouter-0.3.9 → GPTRouter-0.4.1}/setup.py

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