Update crew.py
Browse files
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.
|
| 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
|