Update app.py

app.py

@@ -9,7 +9,6 @@ from smolagents.agent_types import AgentImage
 import uuid
 import requests
 import os
-from smolagents import ToolCallingAgent
 
 CRICKET_API_KEY = os.getenv("CRICKET_API_KEY")
 
@@ -145,37 +144,6 @@ def get_today_matches() -> str:
     
     return result
 
-@tool
-def wiki_search(query: str) -> dict:
-    """
-    Fetches a short summary from Wikipedia for a given topic.
-
-    Args:
-        query: The topic or keyword to search on Wikipedia.
-
-    Returns:
-        A dictionary containing the page title, summary, and URL.
-    """
-
-    encoded_query = urllib.parse.quote(query)
-    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{encoded_query}"
-
-    try:
-        response = requests.get(url, timeout=10)
-
-        if response.status_code != 200:
-            return {"error": "No Wikipedia page found."}
-
-        data = response.json()
-
-        return {
-            "title": data.get("title"),
-            "summary": data.get("extract"),
-            "url": data.get("content_urls", {}).get("desktop", {}).get("page"),
-        }
-
-    except Exception as e:
-        return {"error": str(e)}
 
 final_answer = FinalAnswerTool()
 
@@ -194,9 +162,9 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
     
-agent = ToolCallingAgent(
+agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches, wiki_search], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,