cite-agent 1.3.3__tar.gz → 1.3.5__tar.gz

{cite_agent-1.3.3/cite_agent.egg-info → cite_agent-1.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cite-agent
-Version: 1.3.3
+Version: 1.3.5
 Summary: Terminal AI assistant for academic research with citation verification
 Home-page: https://github.com/Spectating101/cite-agent
 Author: Cite-Agent Team

cite_agent-1.3.5/cite_agent/__version__.py

@@ -0,0 +1 @@
+__version__ = "1.3.5"

{cite_agent-1.3.3 → cite_agent-1.3.5}/cite_agent/enhanced_ai_agent.py

@@ -1696,6 +1696,62 @@ class EnhancedNocturnalAgent:
                         tokens_used=detail.get('tokens_used_today', 0)
                     )
                 
+                elif response.status == 503:
+                    # Backend AI service temporarily unavailable (Cerebras/Groq rate limited)
+                    # Auto-retry with exponential backoff
+                    max_retries = 3
+                    retry_delays = [5, 15, 30]  # seconds
+                    
+                    for retry_num in range(max_retries):
+                        delay = retry_delays[retry_num]
+                        print(f"\n⏳ Hit rate limit. Waiting {delay} seconds before retry {retry_num + 1}/{max_retries}...")
+                        
+                        # Wait with countdown
+                        import asyncio
+                        for remaining in range(delay, 0, -1):
+                            print(f"\r⏱️  Retrying in {remaining}s...", end='', flush=True)
+                            await asyncio.sleep(1)
+                        print("\r🔄 Retrying now...         ")
+                        
+                        # Retry the request
+                        async with self.session.post(url, json=payload, headers=headers, timeout=60) as retry_response:
+                            if retry_response.status == 200:
+                                # Success after retry!
+                                data = await retry_response.json()
+                                response_text = data.get('response', '')
+                                tokens = data.get('tokens_used', 0)
+                                
+                                all_tools = tools_used or []
+                                all_tools.append("backend_llm")
+                                
+                                self.workflow.save_query_result(
+                                    query=query,
+                                    response=response_text,
+                                    metadata={
+                                        "tools_used": all_tools,
+                                        "tokens_used": tokens,
+                                        "model": data.get('model'),
+                                        "provider": data.get('provider'),
+                                        "retries": retry_num + 1
+                                    }
+                                )
+                                
+                                return ChatResponse(
+                                    response=response_text,
+                                    tokens_used=tokens,
+                                    model_used=data.get('model'),
+                                    sources=data.get('sources', [])
+                                )
+                            elif retry_response.status != 503:
+                                # Different error, stop retrying
+                                break
+                    
+                    # All retries failed
+                    return ChatResponse(
+                        response="❌ Service still unavailable after retries. Please try again in a few minutes.",
+                        error_message="Service temporarily unavailable after retries"
+                    )
+                
                 elif response.status == 200:
                     data = await response.json()
                     response_text = data.get('response', '')

{cite_agent-1.3.3 → cite_agent-1.3.5/cite_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cite-agent
-Version: 1.3.3
+Version: 1.3.5
 Summary: Terminal AI assistant for academic research with citation verification
 Home-page: https://github.com/Spectating101/cite-agent
 Author: Cite-Agent Team

{cite_agent-1.3.3 → cite_agent-1.3.5}/setup.py

@@ -7,7 +7,7 @@ long_description = readme_path.read_text() if readme_path.exists() else "Termina
 
 setup(
     name="cite-agent",
-    version="1.3.3",
+    version="1.3.5",
     author="Cite-Agent Team",
     author_email="contact@citeagent.dev",
     description="Terminal AI assistant for academic research with citation verification",

cite_agent-1.3.3/cite_agent/__version__.py

@@ -1 +0,0 @@
-__version__ = "1.3.3"