bstraehle commited on
Commit
9f41d52
·
verified ·
1 Parent(s): b2afbd4

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +1 -2
crew.py CHANGED
@@ -12,7 +12,6 @@ from google import genai
12
  from google.genai import types
13
  from openinference.instrumentation.crewai import CrewAIInstrumentor
14
  from phoenix.otel import register
15
- from stagehand.schemas import AvailableModel
16
  from tools import add, subtract, multiply, divide, modulus
17
  from utils import read_file_json, read_docx_text, read_pptx_text, is_ext
18
 
@@ -54,7 +53,7 @@ def run_crew(question, file_path):
54
  api_key=os.environ["BROWSERBASE_API_KEY"],
55
  project_id=os.environ["BROWSERBASE_PROJECT_ID"],
56
  model_api_key=os.environ["OPENAI_API_KEY"],
57
- model_name=AvailableModel.GPT_4O
58
  )
59
 
60
  @tool("Web Search Tool")
 
12
  from google.genai import types
13
  from openinference.instrumentation.crewai import CrewAIInstrumentor
14
  from phoenix.otel import register
 
15
  from tools import add, subtract, multiply, divide, modulus
16
  from utils import read_file_json, read_docx_text, read_pptx_text, is_ext
17
 
 
53
  api_key=os.environ["BROWSERBASE_API_KEY"],
54
  project_id=os.environ["BROWSERBASE_PROJECT_ID"],
55
  model_api_key=os.environ["OPENAI_API_KEY"],
56
+ model_name="gpt-4.1"
57
  )
58
 
59
  @tool("Web Search Tool")