exaai-agent 2.0.8__tar.gz → 2.0.9__tar.gz

{exaai_agent-2.0.8 → exaai_agent-2.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: exaai-agent
-Version: 2.0.8
+Version: 2.0.9
 Summary: ExaAi - Advanced AI Security Agent for Comprehensive Penetration Testing
 License: Apache-2.0
 License-File: LICENSE

{exaai_agent-2.0.8 → exaai_agent-2.0.9}/exaaiagnt/llm/llm.py

@@ -17,6 +17,7 @@ from litellm.utils import supports_prompt_caching
 
 from exaaiagnt.llm.config import LLMConfig
 from exaaiagnt.llm.memory_compressor import MemoryCompressor
+from exaaiagnt.llm.request_queue import get_global_queue
 from exaaiagnt.llm.llm_traffic_controller import get_traffic_controller, RequestPriority
 from exaaiagnt.llm.utils import _truncate_to_first_function, parse_tool_invocations
 from exaaiagnt.prompts import load_prompt_modules

{exaai_agent-2.0.8 → exaai_agent-2.0.9}/exaaiagnt/llm/llm_traffic_controller.py

@@ -73,8 +73,8 @@ class AdaptiveLLMController:
         # Rate limiting state
         self._last_request_time = 0.0
         self._consecutive_rate_limits = 0
-        self._base_delay = 2.0  # Base delay between requests
-        self._current_delay = 2.0
+        self._base_delay = 4.0  # Base delay between requests
+        self._current_delay = 4.0
         self._max_delay = 30.0
         
         # Statistics
@@ -88,8 +88,8 @@ class AdaptiveLLMController:
         self._tool_execution_lock = asyncio.Lock()
         
         # Configuration
-        self._max_retries = 3
-        self._rate_limit_wait = 6.0  # Wait time after rate limit
+        self._max_retries = 10  # Increased for Gemini
+        self._rate_limit_wait = 15.0  # Longer wait after rate limit
         self._enable_verbose_logging = False
         
         self._initialized = True

{exaai_agent-2.0.8 → exaai_agent-2.0.9}/exaaiagnt/llm/request_queue.py

@@ -65,8 +65,8 @@ class LLMRequestQueue:
     
     def __init__(
         self,
-        max_concurrent: int = 4,
-        delay_between_requests: float = 2.0,
+        max_concurrent: int = 2,
+        delay_between_requests: float = 4.0,
         request_timeout: int = 300,
     ):
         # Load configuration from environment
@@ -161,8 +161,8 @@ class LLMRequestQueue:
             self._semaphore.release()
 
     @retry(
-        stop=stop_after_attempt(10),  # Increased from 7 to 10
-        wait=wait_exponential(multiplier=2, min=5, max=120),  # Better backoff
+        stop=stop_after_attempt(15),  # Increased for rate limits
+        wait=wait_exponential(multiplier=3, min=10, max=180),  # Longer delays
         retry=retry_if_exception(should_retry_exception),
         before_sleep=before_sleep_log(logger, logging.WARNING),
         reraise=True,

{exaai_agent-2.0.8 → exaai_agent-2.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "exaai-agent"
-version = "2.0.8"
+version = "2.0.9"
 description = "ExaAi - Advanced AI Security Agent for Comprehensive Penetration Testing"
 authors = ["ExaAi Team <mahmedhleli@gmail.com>"]
 readme = "README.md"