rag_search.py
1 """Agent with RAG search.""" 2 3 import asyncio 4 5 from examples.rag.rag_pipeline import DOC_PATH, QUESTION 6 from metagpt.logs import logger 7 from metagpt.rag.engines import SimpleEngine 8 from metagpt.roles import Sales 9 10 11 async def search(): 12 """Agent with RAG search.""" 13 14 store = SimpleEngine.from_docs(input_files=[DOC_PATH]) 15 role = Sales(profile="Sales", store=store) 16 result = await role.run(QUESTION) 17 logger.info(result) 18 19 20 if __name__ == "__main__": 21 asyncio.run(search())