aient 1.1.79__py3-none-any.whl → 1.1.80__py3-none-any.whl

aient/core/request.py

@@ -1952,40 +1952,21 @@ async def get_embedding_payload(request, engine, provider, api_key=None):
     headers = {
         "Content-Type": "application/json",
     }
+    if api_key:
+        headers['Authorization'] = f"Bearer {api_key}"
 
     url = provider['base_url']
-    parsed_url = urlparse(url)
-    if "embedding-00" in original_model and "127.0.0.1" not in url and \
-    (parsed_url.path.endswith("/v1beta") or \
-    parsed_url.path.endswith("/v1") or \
-    (parsed_url.netloc == 'generativelanguage.googleapis.com' and "openai/chat/completions" not in parsed_url.path)):
-        if api_key:
-            headers['x-goog-api-key'] = f"{api_key}"
-        parsed_url = urllib.parse.urlparse(url)
-        url = f"{parsed_url.scheme}://{parsed_url.netloc}{parsed_url.path.split('/models')[0].rstrip('/')}/models/{original_model}:embedContent"
-        payload = {
-            "content": {
-                "parts": [
-                    {
-                        "text": request.input
-                    }
-                ]
-            }
-        }
-    else:
-        if api_key:
-            headers['Authorization'] = f"Bearer {api_key}"
-        url = BaseAPI(url).embeddings
-        payload = {
-            "input": request.input,
-            "model": original_model,
-        }
+    url = BaseAPI(url).embeddings
+    payload = {
+        "input": request.input,
+        "model": original_model,
+    }
 
-        if request.encoding_format:
-            if url.startswith("https://api.jina.ai"):
-                payload["embedding_type"] = request.encoding_format
-            else:
-                payload["encoding_format"] = request.encoding_format
+    if request.encoding_format:
+        if url.startswith("https://api.jina.ai"):
+            payload["embedding_type"] = request.encoding_format
+        else:
+            payload["encoding_format"] = request.encoding_format
 
     return url, headers, payload
 

aient/core/utils.py

@@ -63,11 +63,14 @@ class BaseAPI:
         else:
             self.audio_speech: str = urlunparse(parsed_url[:2] + (before_v1 + "audio/speech",) + ("",) * 3)
 
-        if parsed_url.hostname == "generativelanguage.googleapis.com":
+        if parsed_url.path.endswith("/v1beta") or \
+        parsed_url.path.endswith("/v1") or \
+        (parsed_url.netloc == 'generativelanguage.googleapis.com' and "openai/chat/completions" not in parsed_url.path):
+            before_v1 = parsed_url.path.split("/v1")[0]
             self.base_url = api_url
             self.v1_url = api_url
             self.chat_url = api_url
-            self.embeddings = api_url
+            self.embeddings = urlunparse(parsed_url[:2] + (before_v1 + "/v1beta/embeddings",) + ("",) * 3)
 
 def get_engine(provider, endpoint=None, original_model=""):
     parsed_url = urlparse(provider['base_url'])

aient/models/chatgpt.py

@@ -37,6 +37,10 @@ class RateLimitError(Exception):
     """Custom exception for rate limit (429) errors."""
     pass
 
+class HTTPError(Exception):
+    """Custom exception for HTTP 500 errors."""
+    pass
+
 class ConfigurationError(Exception):
     """Custom exception for configuration errors."""
     pass
@@ -793,6 +797,8 @@ class chatgpt(BaseLLM):
                             raise ModelNotFoundError(f"Model: {model or self.engine} not found!")
                         if "HTTP Error', 'status_code': 429" in processed_chunk:
                             raise RateLimitError(f"Rate limit exceeded for model: {model or self.engine}")
+                        if "HTTP Error', 'status_code': " in processed_chunk:
+                            raise HTTPError(f"HTTP Error: {processed_chunk}")
                     yield processed_chunk
                     index += 1
 
@@ -806,6 +812,9 @@ class chatgpt(BaseLLM):
             except APITimeoutError:
                 self.logger.warning("API response timeout (524), retrying...")
                 continue
+            except HTTPError as e:
+                self.logger.warning(f"{e}, retrying...")
+                continue
             except RateLimitError as e:
                 self.logger.warning(f"{e}, retrying...")
                 continue

{aient-1.1.79.dist-info → aient-1.1.80.dist-info}/METADATA

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

{aient-1.1.79.dist-info → aient-1.1.80.dist-info}/RECORD

@@ -2,9 +2,9 @@ aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 aient/core/models.py,sha256=KMlCRLjtq1wQHZTJGqnbWhPS2cHq6eLdnk7peKDrzR8,7490
-aient/core/request.py,sha256=nSLKNI-T6lI4yv873eQUElz5_fkJVy1LbOy27NiJPNk,77512
+aient/core/request.py,sha256=QnDhyrjzcJOEQU2oauMQi_HHMRR5NxdkrX7nn5JMwTc,76675
 aient/core/response.py,sha256=ye6Ie5HevXVcH3X5V5BoOC5yDJMBKTKopWQzsCNs008,34977
-aient/core/utils.py,sha256=wCPw4upG4-ciBxGyEXeTEHzxTbm38rVHrr4PIu8hDW8,28579
+aient/core/utils.py,sha256=9r-Z4KqZXXBm7bUkPx-vOYqgn4CEkG-cRRYz_tl0QUg,28851
 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
@@ -12,7 +12,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 aient/models/__init__.py,sha256=ZTiZgbfBPTjIPSKURE7t6hlFBVLRS9lluGbmqc1WjxQ,43
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=-nnihYnx-vHZMqeVO9ljjt3k4FcD3n-iMk4tT-10nRQ,7232
-aient/models/chatgpt.py,sha256=S2KLZEfJLoRh9u60C4AEjOR7yf5iyqMmvSYt9tr8ldY,45256
+aient/models/chatgpt.py,sha256=hMYVxN6BuAWKoZnL73z60dUnXmtXVul9kFhdCeOEkW8,45615
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=TGgZ5SnNKZ8MmdznrZ-TEq7s2ulhAAwTSKH89bci3dA,7079
@@ -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.79.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.79.dist-info/METADATA,sha256=zphfUbkd-RKheXgm26wDs91hdp_ECfcV2VB816IQXzw,4842
-aient-1.1.79.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.1.79.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.79.dist-info/RECORD,,
+aient-1.1.80.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.80.dist-info/METADATA,sha256=Ix22MRtZoTcO4BewQWGer393FlWC5TRYbqOoyRDrYpo,4842
+aient-1.1.80.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.1.80.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.80.dist-info/RECORD,,