Spaces:
Paused
Paused
Update crew.py
Browse files
crew.py
CHANGED
|
@@ -16,7 +16,7 @@ from util import read_file, read_docx, is_ext
|
|
| 16 |
## LLMs
|
| 17 |
|
| 18 |
MANAGER_MODEL = "gpt-4.5-preview"
|
| 19 |
-
AGENT_MODEL = "gpt-4.
|
| 20 |
|
| 21 |
FINAL_ANSWER_MODEL = "gemini-2.5-pro-preview-03-25"
|
| 22 |
|
|
@@ -25,7 +25,7 @@ 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 |
-
DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-
|
| 29 |
CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
|
| 30 |
CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
|
| 31 |
|
|
@@ -201,6 +201,9 @@ def run_crew(question, file_path):
|
|
| 201 |
if is_ext(file_path, ".docx"):
|
| 202 |
file_data = read_docx(file_path)
|
| 203 |
contents = [f"{question}\n{file_data}"]
|
|
|
|
|
|
|
|
|
|
| 204 |
else:
|
| 205 |
file = client.files.upload(file=file_path)
|
| 206 |
contents = [file, question]
|
|
|
|
| 16 |
## LLMs
|
| 17 |
|
| 18 |
MANAGER_MODEL = "gpt-4.5-preview"
|
| 19 |
+
AGENT_MODEL = "gpt-4.5-preview"
|
| 20 |
|
| 21 |
FINAL_ANSWER_MODEL = "gemini-2.5-pro-preview-03-25"
|
| 22 |
|
|
|
|
| 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 |
+
DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
|
| 29 |
CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
|
| 30 |
CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
|
| 31 |
|
|
|
|
| 201 |
if is_ext(file_path, ".docx"):
|
| 202 |
file_data = read_docx(file_path)
|
| 203 |
contents = [f"{question}\n{file_data}"]
|
| 204 |
+
print("###")
|
| 205 |
+
print(contents[0])
|
| 206 |
+
print("###")
|
| 207 |
else:
|
| 208 |
file = client.files.upload(file=file_path)
|
| 209 |
contents = [file, question]
|