bstraehle commited on
Commit
5a8d469
·
verified ·
1 Parent(s): 315329c

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +5 -2
crew.py CHANGED
@@ -200,11 +200,11 @@ def run_crew(question, file_path):
200
  client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
201
 
202
  ###
203
- #file = client.files.upload(file=file_path)
204
 
205
  response = client.models.generate_content(
206
  model=CODE_MODEL,
207
- contents=question,
208
  config=types.GenerateContentConfig(
209
  tools=[types.Tool(code_execution=types.ToolCodeExecution)]
210
  ),
@@ -212,10 +212,13 @@ def run_crew(question, file_path):
212
 
213
  for part in response.candidates[0].content.parts:
214
  if part.text is not None:
 
215
  print(part.text)
216
  if part.executable_code is not None:
 
217
  print(part.executable_code.code)
218
  if part.code_execution_result is not None:
 
219
  print(part.code_execution_result.output)
220
  return part.code_execution_result.output
221
  ###
 
200
  client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
201
 
202
  ###
203
+ file = client.files.upload(file=file_path)
204
 
205
  response = client.models.generate_content(
206
  model=CODE_MODEL,
207
+ contents=[file, question],
208
  config=types.GenerateContentConfig(
209
  tools=[types.Tool(code_execution=types.ToolCodeExecution)]
210
  ),
 
212
 
213
  for part in response.candidates[0].content.parts:
214
  if part.text is not None:
215
+ print("###")
216
  print(part.text)
217
  if part.executable_code is not None:
218
+ print("###")
219
  print(part.executable_code.code)
220
  if part.code_execution_result is not None:
221
+ print("###")
222
  print(part.code_execution_result.output)
223
  return part.code_execution_result.output
224
  ###