Update crew.py

crew.py

@@ -7,12 +7,10 @@ import os
 import pandas as pd
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
-from crewai_tools import StagehandTool
 from google import genai
 from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
-from stagehand.schemas import AvailableModel
 from tools import add, subtract, multiply, divide, modulus
 from utils import read_file_json, read_docx_text, read_pptx_text, is_ext
 
@@ -50,12 +48,12 @@ tracer_provider = register(
 def run_crew(question, file_path):
     # Tools
 
-    web_browser_tool = StagehandTool(
-        api_key=os.environ["BROWSERBASE_API_KEY"],
-        project_id=os.environ["BROWSERBASE_PROJECT_ID"],
-        model_api_key="your-llm-api-key",
-        model_name=AvailableModel.CLAUDE_3_7_SONNET_LATEST,
-    )
+    #web_browser_tool = StagehandTool(
+    #    api_key=os.environ["BROWSERBASE_API_KEY"],
+    #    project_id=os.environ["BROWSERBASE_PROJECT_ID"],
+    #    model_api_key="your-llm-api-key",
+    #    model_name=AvailableModel.CLAUDE_3_7_SONNET_LATEST,
+    #)
     
     @tool("Web Search Tool")
     def web_search_tool(question: str) -> str:
@@ -334,16 +332,16 @@ def run_crew(question, file_path):
         verbose=True
     )
 
-    web_browser_agent = Agent(
-        role="Web Browser Agent",
-        goal="Given a question and URL, load the URL and act, extract, navigate, observe, and answer the question: {question}",
-        backstory="As an expert browser assistant, you load the URL and act, extract, navigate, observe to answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[web_browser_tool],
-        verbose=True
-    )
+    #web_browser_agent = Agent(
+    #    role="Web Browser Agent",
+    #    goal="Given a question and URL, load the URL and act, extract, navigate, observe, and answer the question: {question}",
+    #    backstory="As an expert browser assistant, you load the URL and act, extract, navigate, observe to answer the question.",
+    #    allow_delegation=False,
+    #    llm=AGENT_MODEL,
+    #    max_iter=2,
+    #    tools=[web_browser_tool],
+    #    verbose=True
+    #)
     
     image_analysis_agent = Agent(
         role="Image Analysis Agent",
@@ -468,7 +466,7 @@ def run_crew(question, file_path):
     
     crew = Crew(
         agents=[web_search_agent, 
-                web_browser_agent, 
+                #web_browser_agent, 
                 image_analysis_agent, 
                 audio_analysis_agent, 
                 video_analysis_agent,