bstraehle commited on
Commit
9cfd8ad
·
verified ·
1 Parent(s): fe0e860

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +5 -2
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.1-mini"
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-pro-preview-03-25"
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]