Coool2 commited on
Commit
05f03a7
·
verified ·
1 Parent(s): a397179

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +15 -1
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())