Update crew.py
Browse files
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
|
| 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-
|
| 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-
|
| 29 |
-
CODE_EXECUTION_MODEL = "gemini-2.5-
|
| 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
|
| 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,
|