Spaces:
Sleeping
Sleeping
Update
Browse files- gemini_agent.py +3 -5
gemini_agent.py
CHANGED
|
@@ -335,11 +335,7 @@ class GeminiAgent:
|
|
| 335 |
|
| 336 |
# Initialize agent
|
| 337 |
self.agent = self._setup_agent()
|
| 338 |
-
|
| 339 |
-
# Load answer bank
|
| 340 |
-
self._load_answer_bank()
|
| 341 |
-
|
| 342 |
-
|
| 343 |
|
| 344 |
def run(self, query: str) -> str:
|
| 345 |
"""Run the agent on a query with incremental retries."""
|
|
@@ -361,6 +357,8 @@ class GeminiAgent:
|
|
| 361 |
continue
|
| 362 |
return f"Error processing query after {max_retries} attempts: {str(e)}"
|
| 363 |
|
|
|
|
|
|
|
| 364 |
def _clean_response(self, response: str) -> str:
|
| 365 |
"""Clean up the response from the agent."""
|
| 366 |
# Remove any tool invocation artifacts
|
|
|
|
| 335 |
|
| 336 |
# Initialize agent
|
| 337 |
self.agent = self._setup_agent()
|
| 338 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
| 339 |
|
| 340 |
def run(self, query: str) -> str:
|
| 341 |
"""Run the agent on a query with incremental retries."""
|
|
|
|
| 357 |
continue
|
| 358 |
return f"Error processing query after {max_retries} attempts: {str(e)}"
|
| 359 |
|
| 360 |
+
print("Agent processed all queries!")
|
| 361 |
+
|
| 362 |
def _clean_response(self, response: str) -> str:
|
| 363 |
"""Clean up the response from the agent."""
|
| 364 |
# Remove any tool invocation artifacts
|