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
|