GPTRouter 0.3.4__tar.gz → 0.3.5__tar.gz

{GPTRouter-0.3.4 → GPTRouter-0.3.5}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-0.3.4 → GPTRouter-0.3.5}/PKG-INFO

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

{GPTRouter-0.3.4 → GPTRouter-0.3.5}/gpt_router/client.py

@@ -95,7 +95,7 @@ class GPTRouterClient:
         payload = {k: v for k, v in payload.items() if v is not None}
         return payload
 
-    async def _async_api_call(self, *, path: str, method: str, payload: dict):
+    async def _async_api_call(self, *, path: str, method: str, payload: dict, **kwargs):
         try:
             async with httpx.AsyncClient() as client:
                 response = await client.request(
@@ -123,7 +123,7 @@ class GPTRouterClient:
             logger.error(f"Timeout error: {err}")
             raise GPTRouterApiTimeoutError("Api Request timed out")
 
-    def _api_call(self, *, path: str, method: str, payload: dict):
+    def _api_call(self, *, path: str, method: str, payload: dict, **kwargs):
         try:
             with httpx.Client() as client:
                 response = client.request(
@@ -151,9 +151,8 @@ class GPTRouterClient:
             logger.error(f"Timeout error: {err}")
             raise GPTRouterApiTimeoutError("Api Request timed out")
 
-    async def astream_events(
-        self, *, path: str, method: str, payload: dict, retry_count: int = 1
-    ):
+    async def astream_events(self, *, path: str, method: str, payload: dict, **kwargs):
+        retry_count = kwargs.get("stream_retry_count") or 1
         async_retrying = AsyncRetrying(
             stop=stop_after_attempt(retry_count),
             wait=wait_fixed(0.5),
@@ -226,8 +225,9 @@ class GPTRouterClient:
                 raise TimeoutError("Request timed out") from err
 
     def stream_events(
-        self, *, path: str, method: str, payload: dict, retry_count: int = 1
+        self, *, path: str, method: str, payload: dict, **kwargs
     ) -> Generator[ChunkedGenerationResponse]:
+        retry_count = kwargs.get("stream_retry_count") or 1
         retrying = Retrying(
             stop=stop_after_attempt(retry_count),
             wait=wait_fixed(0.5),
@@ -322,11 +322,13 @@ class GPTRouterClient:
                 path=api_path,
                 method=api_method,
                 payload=api_payload,
+                **kwargs,
             )
         result = self._api_call(
             path=api_path,
             method=api_method,
             payload=api_payload,
+            **kwargs,
         )
         return parse_obj_as(GenerationResponse, result)
 
@@ -353,11 +355,13 @@ class GPTRouterClient:
                 path=api_path,
                 method=api_method,
                 payload=api_payload,
+                **kwargs,
             )
         result = await self._async_api_call(
             path=api_path,
             method=api_method,
             payload=api_payload,
+            **kwargs,
         )
         return parse_obj_as(GenerationResponse, result)
 

{GPTRouter-0.3.4 → GPTRouter-0.3.5}/setup.py

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