bstraehle commited on
Commit
8632652
·
verified ·
1 Parent(s): 40edc1e

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +6 -6
crew.py CHANGED
@@ -15,18 +15,18 @@ from util import read_file, get_final_answer
15
 
16
  ## LLMs
17
 
18
- MANAGER_MODEL = "gpt-4.1-mini"
19
- AGENT_MODEL = "gpt-4.1-mini"
20
 
21
  FINAL_ANSWER_MODEL = "gemini-2.5-flash-preview-04-17"
22
 
23
  WEB_SEARCH_MODEL = "gemini-2.5-flash-preview-04-17"
24
- IMAGE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
25
  AUDIO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
26
  VIDEO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
27
  YOUTUBE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
28
- CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
29
- CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
30
 
31
  # LLM evaluation
32
 
@@ -322,7 +322,7 @@ def run_crew(question, file_path):
322
  manager_agent = Agent(
323
  role="Manager Agent",
324
  goal="Try to answer the following question. If needed, delegate to one or more of your coworkers for help. "
325
- "If there is no good coworker, delegate to the Python Coding Agent to implement a tool for the task. "
326
  "Question: {question}",
327
  backstory="As an expert manager assistant, you answer the question.",
328
  allow_delegation=True,
 
15
 
16
  ## LLMs
17
 
18
+ MANAGER_MODEL = "gpt-4.1"
19
+ AGENT_MODEL = "gpt-4.1"
20
 
21
  FINAL_ANSWER_MODEL = "gemini-2.5-flash-preview-04-17"
22
 
23
  WEB_SEARCH_MODEL = "gemini-2.5-flash-preview-04-17"
24
+ IMAGE_ANALYSIS_MODEL = "gemini-2.5-pro-preview-03-25"
25
  AUDIO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
26
  VIDEO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
27
  YOUTUBE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
28
+ CODE_GENERATION_MODEL = "gemini-2.5-pro-preview-03-25"
29
+ CODE_EXECUTION_MODEL = "gemini-2.5-pro-preview-03-25"
30
 
31
  # LLM evaluation
32
 
 
322
  manager_agent = Agent(
323
  role="Manager Agent",
324
  goal="Try to answer the following question. If needed, delegate to one or more of your coworkers for help. "
325
+ "If there is no good coworker, delegate to the Code Generation Agent to implement a tool for the task. "
326
  "Question: {question}",
327
  backstory="As an expert manager assistant, you answer the question.",
328
  allow_delegation=True,