Update app.py

app.py

@@ -6,14 +6,7 @@ from typing import List, Dict, Any, Union
 import gradio as gr
 from gradio.components.chatbot import ChatMessage
 from anthropic import Anthropic
-
-# Try to import smolagents MCP client, fallback to basic implementation
-try:
-    from smolagents import MCPClient
-    SMOLAGENTS_AVAILABLE = True
-except ImportError:
-    SMOLAGENTS_AVAILABLE = False
-    print("Warning: smolagents not available. Please install with: pip install smolagents")
+from smolagents import MCPClient
 
 class MCPClientWrapper:
     def __init__(self):
@@ -39,12 +32,10 @@ class MCPClientWrapper:
             
             # Make a small test request to validate the key
             test_response = test_client.messages.create(
-                model="claude-3-5-sonnet-20241022",
-                max_tokens=10,
+                model="claude-sonnet-4-20250514",
                 messages=[{"role": "user", "content": "Hi"}]
             )
             
-            # If we get here, the key is valid
             self.api_key = api_key
             self.anthropic = test_client
             return "✅ Anthropic API key validated and set successfully"
@@ -332,7 +323,6 @@ def gradio_interface():
         msg.submit(client.process_message, [msg, chatbot], [chatbot, msg])
         clear_btn.click(lambda: [], None, chatbot)
         
-        
     return demo
 
 if __name__ == "__main__":