Update tools/ai_tools.py

tools/ai_tools.py

@@ -4,7 +4,6 @@ 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, 
@@ -22,6 +21,8 @@ ARITHMETIC_MODEL        = "gemini-2.5-flash-preview-04-17"
 CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
 CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
+WEB_BROWSER_MODEL       = "claude-3-7-sonnet-latest"
+
 class AITools():
     @tool("Web Search Tool")
     def web_search_tool(question: str) -> str:
@@ -68,7 +69,7 @@ class AITools():
                 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
+                model_name=WEB_BROWSER_MODEL
             )
 
             print("###" + str(stagehand_tool))