Spaces:
Sleeping
Sleeping
Update gemini_agent.py
Browse files- gemini_agent.py +1 -1
gemini_agent.py
CHANGED
|
@@ -116,7 +116,7 @@ class GEMINI_AGENT:
|
|
| 116 |
base_sleep = 1
|
| 117 |
for attempt in range(max_retries):
|
| 118 |
try:
|
| 119 |
-
response = self.app.invoke({"
|
| 120 |
final_ans = self.extract_after_final_answer(response)
|
| 121 |
time.sleep(60) # avoid rate limit
|
| 122 |
return final_ans
|
|
|
|
| 116 |
base_sleep = 1
|
| 117 |
for attempt in range(max_retries):
|
| 118 |
try:
|
| 119 |
+
response = self.app.invoke({"messages": messages, "input_file": None})
|
| 120 |
final_ans = self.extract_after_final_answer(response)
|
| 121 |
time.sleep(60) # avoid rate limit
|
| 122 |
return final_ans
|