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