GPTRouter 0.2.2__tar.gz → 0.2.4__tar.gz

{GPTRouter-0.2.2 → GPTRouter-0.2.4}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-0.2.2 → GPTRouter-0.2.4}/PKG-INFO

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

{GPTRouter-0.2.2 → GPTRouter-0.2.4}/gpt_router/client.py

@@ -3,7 +3,7 @@ from pydantic import parse_obj_as
 import logging
 import json
 import httpx
-from typing import Generator, List, Union
+from typing import Generator, List, Optional, Union
 
 from gpt_router.models import (
     ImageGenerationRequest,
@@ -206,6 +206,9 @@ class GPTRouterClient:
         *,
         ordered_generation_requests: List[ModelGenerationRequest],
         is_stream=False,
+        tag: Optional[str] = None,
+        user_id: Optional[str] = None,
+        history_id: Optional[str] = None,
     ) -> GenerationResponse:
         api_path = "/v1/generate"
         api_method = "POST"
@@ -215,6 +218,9 @@ class GPTRouterClient:
                 request.dict(exclude_none=True, by_alias=True)
                 for request in ordered_generation_requests
             ],
+            "tag": tag,
+            "createdByUserId": user_id,
+            "historyId": history_id,
         }
         if is_stream:
             return self.astream_events(

{GPTRouter-0.2.2 → GPTRouter-0.2.4}/gpt_router/models.py

@@ -21,6 +21,7 @@ class GenerationParams(BaseModel):
     functions: Optional[List[Any]] = None
     function_call: Optional[dict] = None
     tools: Optional[List[Any]] = None
+    response_format: Optional[Dict[str, Any]] = None
 
 
 class ModelGenerationRequest(BaseModel):

{GPTRouter-0.2.2 → GPTRouter-0.2.4}/setup.py

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