Update crew.py

crew.py

@@ -21,6 +21,8 @@ from utils import read_file_json, is_ext
 MANAGER_MODEL      = "gpt-4.5-preview"
 AGENT_MODEL        = "gpt-4.1-mini"
 
+STAGEHAND_MODEL    = "gpt-4.1"
+
 FINAL_ANSWER_MODEL = "gemini-2.5-pro-preview-03-25"
 
 # LLM evaluation
@@ -38,15 +40,15 @@ tracer_provider = register(
 CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
 @CrewBase
-class GAIACrew():
+class GAIACrew(openai_api_key):
     agents: List[BaseAgent]
     tasks: List[Task]
 
     with StagehandTool(
         api_key=os.environ["BROWSERBASE_API_KEY"],
         project_id=os.environ["BROWSERBASE_PROJECT_ID"],
-        model_api_key=os.environ["OPENAI_API_KEY"],
-        model_name="gpt-4.1"
+        model_api_key=openai_api_key,
+        model_name=STAGEHAND_MODEL
     ) as web_browser_tool:
         web_browser_agent = Agent(
             role="Web Browser Agent",
@@ -182,7 +184,7 @@ class GAIACrew():
             verbose=True
         )
 
-def run_crew(question, file_path):
+def run_crew(question, file_path, openai_api_key):
     final_question = question
     
     if file_path:
@@ -192,7 +194,7 @@ def run_crew(question, file_path):
         else:
             final_question = f"{question} File path: {file_path}."
 
-    answer = GAIACrew().crew().kickoff(inputs={"question": final_question})
+    answer = GAIACrew(openai_api_key).crew().kickoff(inputs={"question": final_question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(answer))
 
     print(f"=> Initial question: {question}")