GPTRouter 0.2.7__tar.gz → 0.2.9__tar.gz

{GPTRouter-0.2.7 → GPTRouter-0.2.9}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-0.2.7 → GPTRouter-0.2.9}/PKG-INFO

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

{GPTRouter-0.2.7 → GPTRouter-0.2.9}/gpt_router/client.py

@@ -181,6 +181,7 @@ class GPTRouterClient:
         tag: Optional[str] = None,
         user_id: Optional[str] = None,
         history_id: Optional[str] = None,
+        metadata: Optional[dict] = None,
     ) -> Union[GenerationResponse, Generator[ChunkedGenerationResponse]]:
         api_path = "/v1/generate"
         api_method = "POST"
@@ -193,6 +194,7 @@ class GPTRouterClient:
             "tag": tag,
             "createdByUserId": user_id,
             "historyId": history_id,
+            "metadata": metadata,
         }
         if is_stream:
             return self.stream_events(
@@ -215,6 +217,7 @@ class GPTRouterClient:
         tag: Optional[str] = None,
         user_id: Optional[str] = None,
         history_id: Optional[str] = None,
+        metadata: Optional[dict] = None,
     ) -> GenerationResponse:
         api_path = "/v1/generate"
         api_method = "POST"
@@ -227,6 +230,7 @@ class GPTRouterClient:
             "tag": tag,
             "createdByUserId": user_id,
             "historyId": history_id,
+            "metadata": metadata,
         }
         if is_stream:
             return self.astream_events(
@@ -255,8 +259,8 @@ class GPTRouterClient:
         )
         generated_images = api_response.get("response", [])
         if isinstance(generated_images, dict):
-            generated_images = generated_images.get('artifacts', [])
-        
+            generated_images = generated_images.get("artifacts", [])
+
         return [
             parse_obj_as(ImageGenerationResponse, generated_img)
             for generated_img in generated_images

{GPTRouter-0.2.7 → GPTRouter-0.2.9}/gpt_router/enums.py

@@ -91,6 +91,8 @@ class GPTRouterEngines(Enum):
     )
 
     OPENAI_GPT_4_O = ModelDetails(ProvidersEnum.CHAT_OPENAI.value, "gpt-4o")
+    
+    OPENAI_GPT_4_O_MINI = ModelDetails(ProvidersEnum.CHAT_OPENAI.value, "gpt-4o-mini")
 
     OPENAI_GPT_4_32K_0613 = ModelDetails(
         ProvidersEnum.CHAT_OPENAI.value, "gpt-4-32k-0613"

{GPTRouter-0.2.7 → GPTRouter-0.2.9}/setup.py

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