bstraehle commited on
Commit
4e47816
·
verified ·
1 Parent(s): 423767f

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +2 -7
crew.py CHANGED
@@ -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.0-flash-thinking-exp-01-21"
29
  CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
30
  CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
31
 
@@ -210,12 +210,7 @@ def run_crew(question, file_path):
210
 
211
  response = client.models.generate_content(
212
  model=DOCUMENT_ANALYSIS_MODEL,
213
- contents=contents,
214
- config=types.GenerateContentConfig(
215
- thinking_config=types.ThinkingConfig(
216
- thinking_budget=8196
217
- )
218
- )
219
  )
220
 
221
  return response.text
 
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
 
 
210
 
211
  response = client.models.generate_content(
212
  model=DOCUMENT_ANALYSIS_MODEL,
213
+ contents=contents
 
 
 
 
 
214
  )
215
 
216
  return response.text