aient 1.1.76__tar.gz → 1.1.77__tar.gz

{aient-1.1.76 → aient-1.1.77}/PKG-INFO

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

{aient-1.1.76 → aient-1.1.77}/aient/models/chatgpt.py

@@ -33,6 +33,10 @@ class ModelNotFoundError(Exception):
     """Custom exception for model not found (404) errors."""
     pass
 
+class RateLimitError(Exception):
+    """Custom exception for rate limit (429) errors."""
+    pass
+
 class TaskComplete(Exception):
     """Exception-like signal to indicate the task is complete."""
     def __init__(self, message):
@@ -778,6 +782,8 @@ class chatgpt(BaseLLM):
                             raise APITimeoutError("Response timeout")
                         if "HTTP Error', 'status_code': 404" in processed_chunk:
                             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}")
                     yield processed_chunk
                     index += 1
 
@@ -791,6 +797,9 @@ class chatgpt(BaseLLM):
             except APITimeoutError:
                 self.logger.warning("API response timeout (524), retrying...")
                 continue
+            except RateLimitError as e:
+                self.logger.warning(f"{e}, retrying...")
+                continue
             except ValidationError as e:
                 self.logger.warning(f"Validation failed: {e}. Retrying with corrective prompt.")
                 need_done_prompt = [
@@ -941,4 +950,4 @@ class chatgpt(BaseLLM):
             {"role": "system", "content": self.system_prompt},
         ]
         self.tokens_usage[convo_id] = 0
-        self.current_tokens[convo_id] = 0
+        self.current_tokens[convo_id] = 0

{aient-1.1.76 → aient-1.1.77}/aient.egg-info/PKG-INFO

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

{aient-1.1.76 → aient-1.1.77}/pyproject.toml

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