bstraehle commited on
Commit
c8b5e97
·
verified ·
1 Parent(s): 4d1bbf7

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +6 -1
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