Update agent.py
Browse files
agent.py
CHANGED
@@ -716,4 +716,18 @@ You are a general AI assistant. I will ask you a question. Report your thoughts,
|
|
716 |
return {
|
717 |
"total_documents": len(self.dynamic_qe_manager.documents),
|
718 |
"document_sources": [doc.metadata.get("source", "Unknown") for doc in self.dynamic_qe_manager.documents]
|
719 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
716 |
return {
|
717 |
"total_documents": len(self.dynamic_qe_manager.documents),
|
718 |
"document_sources": [doc.metadata.get("source", "Unknown") for doc in self.dynamic_qe_manager.documents]
|
719 |
+
}
|
720 |
+
|
721 |
+
import asyncio
|
722 |
+
|
723 |
+
async def main():
|
724 |
+
agent = EnhancedGAIAAgent()
|
725 |
+
question_data = {
|
726 |
+
"Question": "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.",
|
727 |
+
"task_id": ""
|
728 |
+
}
|
729 |
+
answer = await agent.solve_gaia_question(question_data)
|
730 |
+
print(f"Answer: {answer}")
|
731 |
+
|
732 |
+
if __name__ == '__main__':
|
733 |
+
asyncio.run(main())
|