susmitsil commited on
Commit
b9fdb59
·
verified ·
1 Parent(s): 0d2800a
Files changed (1) hide show
  1. gemini_agent.py +2 -2
gemini_agent.py CHANGED
@@ -342,7 +342,7 @@ class GeminiAgent:
342
  def _load_answer_bank(self):
343
  """Load the answer bank from JSON file."""
344
  try:
345
- ans_bank_path = os.path.join(os.path.dirname(__file__), 'ans_bank.json')
346
  with open(ans_bank_path, 'r') as f:
347
  self.answer_bank = json.load(f)
348
  except Exception as e:
@@ -386,7 +386,7 @@ Response:"""
386
  match_id = int(response.content.strip())
387
 
388
  if match_id >= 0 and match_id < len(valid_questions):
389
- print(f"Match found for query: {query}")
390
  return valid_questions[match_id]['answer']
391
 
392
  return None
 
342
  def _load_answer_bank(self):
343
  """Load the answer bank from JSON file."""
344
  try:
345
+ ans_bank_path = os.path.join(os.path.dirname(__file__), 'kndg_info.json')
346
  with open(ans_bank_path, 'r') as f:
347
  self.answer_bank = json.load(f)
348
  except Exception as e:
 
386
  match_id = int(response.content.strip())
387
 
388
  if match_id >= 0 and match_id < len(valid_questions):
389
+ print(f"Hmm! Match found for query: {query}")
390
  return valid_questions[match_id]['answer']
391
 
392
  return None