exaai-agent 2.0.8__py3-none-any.whl → 2.0.9__py3-none-any.whl

{exaai_agent-2.0.8.dist-info → exaai_agent-2.0.9.dist-info}/METADATA

@@ -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.dist-info → exaai_agent-2.0.9.dist-info}/RECORD

@@ -33,11 +33,11 @@ exaaiagnt/interface/utils.py,sha256=xp6eDOC8C0c3cjt791S_jBDs1B-xp_ydIb74QnMLEt8,
 exaaiagnt/llm/__init__.py,sha256=hUVixjSSIUtwIP2I5D_9e6Kdxhhunnajgxx_2DEYNww,1095
 exaaiagnt/llm/config.py,sha256=HQ0skwQxtHwiDLDWBCU1Fp4UoQ8tbrTNQw9s7JGVaiY,3303
 exaaiagnt/llm/fallback.py,sha256=oPS0PGRxEHnyyBgS4yP9zdwSf4JFJh4dYZ3g8OFwWEE,11413
-exaaiagnt/llm/llm.py,sha256=EEPte4O37YVhCrJleHS8uU5eaYw_WDXKIEwlK2USJgE,18606
-exaaiagnt/llm/llm_traffic_controller.py,sha256=HZ0OZcbfr5XLVSpuW8EQLn3gQht3rjY1n6SGRFpBZa0,12187
+exaaiagnt/llm/llm.py,sha256=D9VnU73ffz7GhotGqQ5RZDufQ8xPKz8aouU23oW0kPY,18664
+exaaiagnt/llm/llm_traffic_controller.py,sha256=HkblT8OI3RcUdxT9kvyeNsUhowAGf1FZkhid_PdS998,12215
 exaaiagnt/llm/memory_compressor.py,sha256=_At7e5QlDv2vrUDUJMEwm4CjNJ2uGYQsOBhiHiQvVr8,7054
 exaaiagnt/llm/output_processor.py,sha256=JC3TtzYj9DJhJRuKzz_VV3WIwAyYhUdZeY1N9c4SzVw,13568
-exaaiagnt/llm/request_queue.py,sha256=niwJVzWyvNcOl9uVYABO55c6ZReMxY_GIUhKXh946dI,9195
+exaaiagnt/llm/request_queue.py,sha256=2H-jEcI1krY8U8oMoULu-1y6kEV84enI36QLUkTLKm0,9198
 exaaiagnt/llm/utils.py,sha256=0Z0r6qo9IfUSOJt5FJsq3X-veXrA8A09frc2VIy-aS4,2540
 exaaiagnt/prompts/README.md,sha256=Svgjx8pO1W0aVzC9z0Fxs88-NBA1whEqU8zrGBDqO0M,3813
 exaaiagnt/prompts/__init__.py,sha256=nUs1powNe_6sWmM5_KkAIxZj97t5IiMMgNrL39WwUjg,5042
@@ -143,8 +143,8 @@ exaaiagnt/tools/waf_bypass.py,sha256=71oPWnDHjn2EHi6I1SluZCKfqfXkA5j61oIkL5kNoSw
 exaaiagnt/tools/web_search/__init__.py,sha256=m5PCHXqeNVraLRLNIbh54Z2N4Y_75d-ftqwyq3dbCd0,70
 exaaiagnt/tools/web_search/web_search_actions.py,sha256=jmlN2uIq8lRbhRnyaMQkC-44jhpfkLQZ_byYNlNNlOY,3111
 exaaiagnt/tools/web_search/web_search_actions_schema.xml,sha256=Ihc3Gv4LaPI_MzBbwZOt3y4pwg9xmtl8KfPNvFihEP4,4805
-exaai_agent-2.0.8.dist-info/METADATA,sha256=DQPnhLMYY4bBopzTdAK7M_jMZS1jcZZrIyx-k8pCqSQ,11959
-exaai_agent-2.0.8.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-exaai_agent-2.0.8.dist-info/entry_points.txt,sha256=iLSTRDSjN8Zyc2Wo6WXmr4MtyEWkdqtZ1j_Hx73MiUs,137
-exaai_agent-2.0.8.dist-info/licenses/LICENSE,sha256=RV6IGl0sWdfbbtJmjPr1w_qwGyGt2jv02PXsAzN_kNs,11460
-exaai_agent-2.0.8.dist-info/RECORD,,
+exaai_agent-2.0.9.dist-info/METADATA,sha256=LCoSJoucmT-AzVO6DLNcH8PNZEZLP6k45QJaMNO-6fc,11959
+exaai_agent-2.0.9.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+exaai_agent-2.0.9.dist-info/entry_points.txt,sha256=iLSTRDSjN8Zyc2Wo6WXmr4MtyEWkdqtZ1j_Hx73MiUs,137
+exaai_agent-2.0.9.dist-info/licenses/LICENSE,sha256=RV6IGl0sWdfbbtJmjPr1w_qwGyGt2jv02PXsAzN_kNs,11460
+exaai_agent-2.0.9.dist-info/RECORD,,

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

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

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,