Update app.py

app.py

@@ -12,13 +12,37 @@ from smolagents.models import OpenAIServerModel
 from smolagents import Tool
 from wikipedia_searcher import WikipediaSearcher
 
-wikipedia_search_tool = Tool(
-    name="wikipedia_search",
-    func=WikipediaSearcher().search,
-    description="Search Wikipedia for a given query.",
-    inputs={"query": "string"},
-    output_type="string"
-)
+#wikipedia_search_tool = Tool(
+#    name="wikipedia_search",
+#    func=WikipediaSearcher().search,
+#    description="Search Wikipedia for a given query.",
+#    inputs={"query": "string"},
+#    output_type="string"
+#)
+
+
+
+class WikipediaSearchTool(Tool):
+    def __init__(self):
+        super().__init__(
+            name="wikipedia_search",
+            description="Search Wikipedia for a given query.",
+            inputs={"query": "string"},
+            output_type="string",
+        )
+        self.searcher = WikipediaSearcher()
+
+    def __call__(self, inputs: dict) -> str:
+        query = inputs.get("query", "")
+        return self.searcher.search(query)
+
+wikipedia_search_tool = WikipediaSearchTool()
+
+
+
+
+
+
 
 
 # Define the system prompt