Update tools/ai_tools.py

tools/ai_tools.py

@@ -4,6 +4,7 @@ from crewai.tools import tool
 from crewai_tools import StagehandTool
 from google import genai
 from google.genai import types
+from stagehand.schemas import AvailableModel
 from utils import (
     read_file_json, 
     read_docx_text, 
@@ -50,12 +51,13 @@ class AITools():
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Web Browser Tool")
-    def web_browser_tool(question: str, url: str) -> str:
-        """Given a question and URL, load the URL and act, extract, navigate, and observe to answer the question.
+    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.
     
            Args:
                question (str): Question about a URL
                url (str): The URL
+               action (str): Act, extract, or observe
                 
            Returns:
                str: Answer to the question
@@ -73,7 +75,7 @@ class AITools():
             return stagehand_tool.run(
                 instruction=question, 
                 url=url,
-                command_type="act"
+                command_type=action
             )
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")