Update cool_agent.py

cool_agent.py

@@ -17,33 +17,10 @@ from smolagents import (
     CodeAgent,
     LiteLLMModel,
     ToolCallingAgent,
-    Tool,
 )
 import threading
 
 
-class WikiSearchTool(Tool):
-    name = "wiki_search"
-    description = "Search Wikipedia for a query and return maximum 2 results."
-    inputs = {
-        "query": {"type": "string", "description": "The search query to perform."},
-    }
-    output_type = "dict"
-
-    def __init__(self):
-        super().__init__(self)
-
-    def forward(self, query: str) -> dict:
-        search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
-        formatted_search_docs = "\n\n---\n\n".join(
-            [
-                f'<Document source="{doc.metadata["source"]}" page="'
-                f'{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
-                for doc in search_docs
-            ])
-        return {"wiki_results": formatted_search_docs}
-
-
 AUTHORIZED_IMPORTS = [
     "requests",
     "zipfile",
@@ -115,7 +92,6 @@ def create_agent(model_id="gpt-4o-mini"):
         FindNextTool(browser),
         ArchiveSearchTool(browser),
         TextInspectorTool(model, text_limit),
-        WikiSearchTool(),
     ]
     text_webbrowser_agent = ToolCallingAgent(
         model=model,