aient 1.1.80__tar.gz → 1.1.82__tar.gz

{aient-1.1.80 → aient-1.1.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.1.80
+Version: 1.1.82
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.1.80 → aient-1.1.82}/aient/core/utils.py

@@ -169,7 +169,7 @@ def get_proxy(proxy, client_config = {}):
             }
     return client_config
 
-def update_initial_model(provider):
+async def update_initial_model(provider):
     try:
         engine, stream_mode = get_engine(provider, endpoint=None, original_model="")
         # print("engine", engine, provider)
@@ -181,8 +181,8 @@ def update_initial_model(provider):
             before_v1 = api_url.split("/v1beta")[0]
             url = before_v1 + "/v1beta/models"
             params = {"key": api}
-            with httpx.Client(**client_config) as client:
-                response = client.get(url, params=params)
+            async with httpx.AsyncClient(**client_config) as client:
+                response = await client.get(url, params=params)
 
             original_models = response.json()
             if original_models.get("error"):
@@ -202,11 +202,11 @@ def update_initial_model(provider):
                 headers = {"x-api-key": api, "anthropic-version": "2023-06-01"}
             else:
                 headers = {"Authorization": f"Bearer {api}"}
-            response = httpx.get(
-                endpoint_models_url,
-                headers=headers,
-                **client_config
-            )
+            async with httpx.AsyncClient(**client_config) as client:
+                response = await client.get(
+                    endpoint_models_url,
+                    headers=headers,
+                )
             models = response.json()
             if models.get("error"):
                 logger.error({"error": models.get("error"), "endpoint": endpoint_models_url, "api": api})

{aient-1.1.80 → aient-1.1.82}/aient/models/chatgpt.py

@@ -809,6 +809,9 @@ class chatgpt(BaseLLM):
                 return # Stop iteration
             except httpx.RemoteProtocolError:
                 continue
+            except httpx.ReadError:
+                self.logger.warning(f"{e}, retrying...")
+                continue
             except APITimeoutError:
                 self.logger.warning("API response timeout (524), retrying...")
                 continue

{aient-1.1.80 → aient-1.1.82}/aient.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.1.80
+Version: 1.1.82
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.1.80 → aient-1.1.82}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aient"
-version = "1.1.80"
+version = "1.1.82"
 description = "Aient: The Awakening of Agent."
 readme = "README.md"
 requires-python = ">=3.11"