Coool2 commited on
Commit
fa0097a
·
verified ·
1 Parent(s): c0ddcb5

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -549,7 +549,7 @@ class EnhancedGAIAAgent:
549
  verbose = True
550
  )
551
 
552
- def format_gaia_answer(self, raw_response: str, original_question: str) -> str:
553
  """
554
  Post-process the agent response to extract the exact GAIA format answer
555
  """
@@ -617,7 +617,7 @@ class EnhancedGAIAAgent:
617
  raw_response = await self.coordinator.run(ctx=ctx, user_msg=context_prompt)
618
 
619
  # Post-process to extract exact GAIA format
620
- formatted_answer = self.format_gaia_answer(str(raw_response), question)
621
 
622
  print(f"Raw response: {raw_response}")
623
  print(f"Formatted answer: {formatted_answer}")
 
549
  verbose = True
550
  )
551
 
552
+ async def format_gaia_answer(self, raw_response: str, original_question: str) -> str:
553
  """
554
  Post-process the agent response to extract the exact GAIA format answer
555
  """
 
617
  raw_response = await self.coordinator.run(ctx=ctx, user_msg=context_prompt)
618
 
619
  # Post-process to extract exact GAIA format
620
+ formatted_answer = await self.format_gaia_answer(str(raw_response), question)
621
 
622
  print(f"Raw response: {raw_response}")
623
  print(f"Formatted answer: {formatted_answer}")