Update tools/ai_tools.py

tools/ai_tools.py

@@ -51,13 +51,12 @@ class AITools():
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Web Browser Tool")
-    def web_browser_tool(question: str, url: str, action: str) -> str:
-        """Given a question, URL, and action, load the URL and act, extract, or observe to answer the question.
+    def web_browser_tool(question: str, url: str) -> str:
+        """Given a question and URL, load the URL and act, extract, or observe to answer the question.
     
            Args:
                question (str): Question about a URL
                url (str): The URL
-               action (str): Act, extract, or observe
                 
            Returns:
                str: Answer to the question
@@ -72,12 +71,9 @@ class AITools():
                 model_name=AvailableModel.CLAUDE_3_7_SONNET_LATEST
             )
 
-            print("===>" + os.environ.get("ANTHROPIC_API_KEY")[:5])
-            
             return stagehand_tool.run(
                 instruction=question, 
-                url=url,
-                command_type=action
+                url=url
             )
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")