Update agent.py
Browse files
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}")
|