Spaces:
Paused
Paused
Update crew.py
Browse files
crew.py
CHANGED
|
@@ -216,7 +216,12 @@ def run_crew(question, file_path):
|
|
| 216 |
|
| 217 |
response = client.models.generate_content(
|
| 218 |
model=DOCUMENT_ANALYSIS_MODEL,
|
| 219 |
-
contents=contents
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 220 |
)
|
| 221 |
|
| 222 |
return response.text
|
|
|
|
| 216 |
|
| 217 |
response = client.models.generate_content(
|
| 218 |
model=DOCUMENT_ANALYSIS_MODEL,
|
| 219 |
+
contents=contents,
|
| 220 |
+
config=types.GenerateContentConfig(
|
| 221 |
+
thinking_config=types.ThinkingConfig(
|
| 222 |
+
thinking_budget=8196
|
| 223 |
+
)
|
| 224 |
+
)
|
| 225 |
)
|
| 226 |
|
| 227 |
return response.text
|