/ examples / rag / rag_search.py
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())