Update custom_tools.py

custom_tools.py

@@ -3,6 +3,10 @@ from bs4 import BeautifulSoup
 from typing import List, Dict, Any
 from functools import lru_cache
 from smolagents import Tool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool
+from pydantic import BaseModel, Field
+
+class WebpageStructureInputs(BaseModel):
+    url: str = Field(description="The URL of the webpage to analyze.")
 
 class WebpageStructureAnalyzerTool(Tool):
     """
@@ -17,8 +21,9 @@ class WebpageStructureAnalyzerTool(Tool):
         "a webpage's layout *before* trying to write specific 'bs4' code "
         "to extract detailed information."
     )
+    inputs: Type[BaseModel] = WebpageStructureInputs
 
-    def run(self, url: str) -> str:
+    def run(self, inputs: WebpageStructureInputs) -> str:
         """
         Executes the webpage structure analysis.
 
@@ -29,7 +34,7 @@ class WebpageStructureAnalyzerTool(Tool):
             A string containing the structure summary or an error message.
         """
         # Ensure the core function is accessible here
-        return analyze_webpage_structure(url)
+        return analyze_webpage_structure(inputs.url)
 
 class CachedWebSearchTool(WebSearchTool):
     @lru_cache(maxsize=128)