Update tools/ai_tools.py

tools/ai_tools.py

@@ -48,6 +48,34 @@ class AITools():
             return response.text
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
+
+    @tool("Web Browser Tool")
+    def web_browser_tool(question: str, url: str) -> str:
+        """Given a question only, search the web to answer the question.
+    
+           Args:
+               question (str): Question to answer
+                
+           Returns:
+               str: Answer to the question
+                
+           Raises:
+               RuntimeError: If processing fails"""
+        try:
+            stagehand_tool = StagehandTool(
+                api_key=os.environ["BROWSERBASE_API_KEY"],
+                project_id=os.environ["BROWSERBASE_PROJECT_ID"],
+                model_api_key=os.environ["ANTHROPIC_API_KEY"],
+                model_name=AvailableModel.CLAUDE_3_7_SONNET_LATEST
+            )
+    
+            return result = stagehand_tool.run(
+                instruction=question, 
+                url=url,
+                command_type="act"
+            )
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str: