susmitsil commited on
Commit
682c9cd
·
verified ·
1 Parent(s): 08228d1
Files changed (1) hide show
  1. 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