Update app.py

app.py

@@ -13,29 +13,29 @@ import base64
 from smolagents import load_tool
 from typing import Any
 
-class DuckDuckGoSearchTool(Tool):
-    name = "web_search"
-    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
-    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
-    output_type = "string"
-
-    def __init__(self, max_results=10, **kwargs):
-        super().__init__()
-        self.max_results = max_results
-        try:
-            from duckduckgo_search import DDGS
-        except ImportError as e:
-            raise ImportError(
-                "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
-            ) from e
-        self.ddgs = DDGS(**kwargs)
+# class DuckDuckGoSearchTool(Tool):
+#     name = "web_search"
+#     description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
+#     inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
+#     output_type = "string"
+
+#     def __init__(self, max_results=10, **kwargs):
+#         super().__init__()
+#         self.max_results = max_results
+#         try:
+#             from duckduckgo_search import DDGS
+#         except ImportError as e:
+#             raise ImportError(
+#                 "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
+#             ) from e
+#         self.ddgs = DDGS(**kwargs)
 
-    def forward(self, query: str) -> str:
-        results = self.ddgs.text(query, max_results=self.max_results)
-        if len(results) == 0:
-            raise Exception("No results found! Try a less restrictive/shorter query.")
-        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
-        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+#     def forward(self, query: str) -> str:
+#         results = self.ddgs.text(query, max_results=self.max_results)
+#         if len(results) == 0:
+#             raise Exception("No results found! Try a less restrictive/shorter query.")
+#         postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
+#         return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
 
 
 model = InferenceClientModel("qwen/Qwen2.5-0.5B-Instruct",
@@ -112,11 +112,15 @@ tools = [
         name="image-captioning",
         description="Generates a caption for an input image."
     ),
-    # load_tool("duckduckgo-search", trust_remote_code=True),
     DuckDuckGoSearchTool(),
-    load_tool("wikipedia", trust_remote_code=True),
-    load_tool("python", trust_remote_code=True),
-    load_tool("user-input", trust_remote_code=True),
+    WikipediaSearchTool(),
+    PythonInterpreterTool(),
+    UserInputTool(),
+    # load_tool("duckduckgo-search", trust_remote_code=True),
+    # DuckDuckGoSearchTool(),
+    # load_tool("wikipedia", trust_remote_code=True),
+    # load_tool("python", trust_remote_code=True),
+    # load_tool("user-input", trust_remote_code=True),
 ]
 # # ---------------------- AGENT SETUP ---------------------- #
 # agent = CodeAgent(