Update app.py

app.py

@@ -119,7 +119,25 @@ class BasicAgent:
             raise ValueError("ANTHROPIC_API_KEY not found in environment variables.")
         
         self.model_name = "claude-3-haiku-20240307"
-        self.chat_model = ChatAnthropic(model=self.model_name, anthropic_api_key=API_KEY)
+        
+        # Test API key and connection
+        print("Testing Anthropic API connection...")
+        try:
+            self.chat_model = ChatAnthropic(
+                model=self.model_name, 
+                anthropic_api_key=API_KEY,
+                timeout=30,
+                max_retries=2
+            )
+            
+            # Test with a simple query
+            test_response = self.chat_model.invoke("Hello, are you working?")
+            print(f"✅ API connection successful! Test response: {test_response.content[:50]}...")
+            
+        except Exception as e:
+            print(f"❌ API connection failed: {e}")
+            raise ValueError(f"Failed to initialize Claude model: {e}")
+        
         self.rate_limiter = RateLimiter()
         
         # Initialize tools
@@ -143,8 +161,13 @@ class BasicAgent:
                 # Create a comprehensive prompt for Claude
                 prompt = self._create_prompt(question)
                 
-                # Get response from Claude
-                response = self.chat_model.invoke(prompt)
+                print(f"Attempt {attempt + 1}: Sending request to Claude...")
+                
+                # Get response from Claude with timeout
+                response = self.chat_model.invoke(
+                    prompt,
+                    config={"timeout": 30}  # 30 second timeout
+                )
                 agent_answer = response.content
                 
                 print(f"Agent returning answer: {agent_answer[:100]}...")
@@ -152,20 +175,39 @@ class BasicAgent:
                 
             except Exception as e:
                 error_msg = str(e)
-                print(f"Attempt {attempt + 1} failed: {error_msg}")
+                error_type = type(e).__name__
+                print(f"Attempt {attempt + 1} failed with {error_type}: {error_msg}")
                 
-                # Check if it's a rate limit error
-                if "rate limit" in error_msg.lower() or "429" in error_msg:
+                # Check specific error types
+                if "connection" in error_msg.lower() or "timeout" in error_msg.lower():
+                    if attempt < max_retries - 1:
+                        wait_time = (attempt + 1) * 10  # Progressive backoff for connection issues
+                        print(f"Connection issue detected. Waiting {wait_time} seconds before retry...")
+                        time.sleep(wait_time)
+                        continue
+                    else:
+                        return f"CONNECTION_ERROR: {error_msg}"
+                        
+                elif "rate limit" in error_msg.lower() or "429" in error_msg:
                     if attempt < max_retries - 1:
-                        wait_time = (attempt + 1) * 30  # Progressive backoff
+                        wait_time = (attempt + 1) * 30  # Longer wait for rate limits
                         print(f"Rate limit hit. Waiting {wait_time} seconds before retry...")
                         time.sleep(wait_time)
                         continue
                     else:
                         return f"RATE_LIMIT_ERROR: {error_msg}"
+                        
+                elif "authentication" in error_msg.lower() or "api key" in error_msg.lower():
+                    return f"AUTH_ERROR: {error_msg}"
+                    
                 else:
-                    # For other errors, return immediately
-                    return f"AGENT_ERROR: {error_msg}"
+                    if attempt < max_retries - 1:
+                        wait_time = (attempt + 1) * 5  # Short wait for other errors
+                        print(f"Unknown error. Waiting {wait_time} seconds before retry...")
+                        time.sleep(wait_time)
+                        continue
+                    else:
+                        return f"AGENT_ERROR: {error_msg}"
         
         return "MAX_RETRIES_EXCEEDED"
     
@@ -304,9 +346,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None, progress=gr.Progress()):
                 submitted_answer = agent(question_text)
 
             # Check if the answer indicates an error
-            if submitted_answer.startswith(("RATE_LIMIT_ERROR", "AGENT_ERROR", "MAX_RETRIES_EXCEEDED")):
+            if submitted_answer.startswith(("RATE_LIMIT_ERROR", "AGENT_ERROR", "MAX_RETRIES_EXCEEDED", "CONNECTION_ERROR", "AUTH_ERROR")):
                 print(f"Error processing task {task_id}: {submitted_answer}")
                 results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+                
+                # For authentication errors, stop processing
+                if submitted_answer.startswith("AUTH_ERROR"):
+                    print("Authentication error detected. Stopping processing.")
+                    break
+                    
                 # Don't add to answers_payload for submission if it's an error
                 continue