Update app.py

app.py

@@ -4,7 +4,7 @@ import traceback
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
+from smolagents import CodeAgent, tool
 from smolagents.models import OpenAIServerModel
 
 # Setup logging
@@ -19,32 +19,6 @@ if not GITHUB_TOKEN:
 GITHUB_ENDPOINT = "https://models.github.ai/inference"
 MODEL_ID = os.getenv("MODEL_ID", "openai/gpt-4o-mini")
 
-# Initialize search tool
-try:
-    search_tool_instance = DuckDuckGoSearchTool()
-    logger.info("DuckDuckGoSearchTool initialized successfully.")
-except Exception as e:
-    logger.error(f"Failed to initialize DuckDuckGoSearchTool: {e}")
-    search_tool_instance = None
-
-@tool
-def web_search(query: str) -> str:
-    """
-    Performs a web search using DuckDuckGo. Use this for general questions or current info.
-    
-    Args:
-        query (str): The search query string.
-    """
-    logger.info(f"Searching: '{query[:50]}...'")
-    if search_tool_instance is None:
-        return "Search Error: Tool not initialized."
-    try:
-        result = search_tool_instance(query=query)
-        return result[:3000] + "... (truncated)" if len(result) > 3000 else result
-    except Exception as e:
-        logger.exception("Search failed")
-        return f"Search Error: {e}"
-
 @tool
 def wikipedia_lookup(page_title: str) -> str:
     """
@@ -75,14 +49,13 @@ def wikipedia_lookup(page_title: str) -> str:
     except Exception as e:
         return f"Wikipedia Error: {e}"
 
-# Agent prompt
+# Agent prompt - updated to mention only Wikipedia tool
 REACT_INSTRUCTION_PROMPT = """You are a helpful assistant using tools to answer questions.
 Available Tools:
-- web_search(query: str): Searches the web. Use for general info or current events.
 - wikipedia_lookup(page_title: str): Looks up a specific English Wikipedia page. Use exact titles (e.g., 'Berlin').
 Follow these steps:
 1. Thought: Plan which tool to use and why.
-2. Action: Call ONE tool (e.g., web_search(query="...") or wikipedia_lookup(page_title="...")).
+2. Action: Call the tool (e.g., wikipedia_lookup(page_title="...")).
 3. Observation: Record the result.
 4. Thought: Analyze result. If answered, prepare final answer. If not, plan next step.
 5. Repeat Action/Observation/Thought until answered or determined impossible.
@@ -105,7 +78,7 @@ try:
     )
     
     agent = CodeAgent(
-        tools=[web_search, wikipedia_lookup],
+        tools=[wikipedia_lookup],  # Only Wikipedia tool
         model=llm_model
     )
     logger.info("Agent initialization complete")