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
|