aient 1.1.81__py3-none-any.whl → 1.1.82__py3-none-any.whl

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.81.dist-info → aient-1.1.82.dist-info}/METADATA

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

{aient-1.1.81.dist-info → aient-1.1.82.dist-info}/RECORD

@@ -4,7 +4,7 @@ aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 aient/core/models.py,sha256=KMlCRLjtq1wQHZTJGqnbWhPS2cHq6eLdnk7peKDrzR8,7490
 aient/core/request.py,sha256=QnDhyrjzcJOEQU2oauMQi_HHMRR5NxdkrX7nn5JMwTc,76675
 aient/core/response.py,sha256=ye6Ie5HevXVcH3X5V5BoOC5yDJMBKTKopWQzsCNs008,34977
-aient/core/utils.py,sha256=9r-Z4KqZXXBm7bUkPx-vOYqgn4CEkG-cRRYz_tl0QUg,28851
+aient/core/utils.py,sha256=BwC23v1i__-HxF5vHOxZUfDOIIOsnqsiIW1HhKyBVbk,28934
 aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
 aient/core/test/test_geminimask.py,sha256=HFX8jDbNg_FjjgPNxfYaR-0-roUrOO-ND-FVsuxSoiw,13254
 aient/core/test/test_image.py,sha256=_T4peNGdXKBHHxyQNx12u-NTyFE8TlYI6NvvagsG2LE,319
@@ -30,8 +30,8 @@ aient/plugins/write_file.py,sha256=Jt8fOEwqhYiSWpCbwfAr1xoi_BmFnx3076GMhuL06uI,3
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=VqtK4RFEx7KxkmcqG3lFDS1DxoNlFFGErEjopVcc8IE,40974
-aient-1.1.81.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.81.dist-info/METADATA,sha256=EhWleC1wNEH3zGSYyqd6yliP2Zb_ANww70H37uCwhCU,4842
-aient-1.1.81.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.1.81.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.81.dist-info/RECORD,,
+aient-1.1.82.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.82.dist-info/METADATA,sha256=WU1r-B44ZrSmBLiMxqvvltgSpGMxUFn9TEVWvfeZeI0,4842
+aient-1.1.82.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.1.82.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.82.dist-info/RECORD,,