Update tools.py

tools.py

@@ -1,11 +1,25 @@
-from tools import DuckDuckGoSearchTool
+from smolagents import DuckDuckGoSearchTool
 from smolagents import Tool
 import random
 from huggingface_hub import list_models
 
 
 # Initialize the DuckDuckGo search tool
-search_tool = DuckDuckGoSearchTool()
+class InternetSearchTool(Tool):
+    name = "internet_search"
+    description = "Searches the internet for the answer to any question input. Returns a long batch of textual information related to the query."
+    inputs = {
+        "question": {
+            "type": "string",
+            "description": "The question to retrieve an answer for."
+        }
+    }
+    output_type = "string"
+
+    def forward(self, question: str):
+        search_tool = DuckDuckGoSearchTool()
+        result = search_tool(question)
+        return result
 
 
 class WeatherInfoTool(Tool):