15 lines
871 B
Python
15 lines
871 B
Python
from database.milvus_api import Milvus_Database
|
|
from modelsAPI.model_api import get_embeddings
|
|
|
|
async def autorag_flow(query,milvus_db,collection_name,top_k=5):
|
|
query_vector = get_embeddings(list(query))
|
|
search_results = milvus_db.search(collection_name=collection_name,query_vector=query_vector, top_k=top_k)
|
|
|
|
search_prompt="""你作为一个由中科院的资深文档问答专家，请你从以下文本信息中选取与用户问题相关的内容作为参考:
|
|
-----
|
|
文本信息：{context}
|
|
-----
|
|
请你仔细思考后用中文回答用户的问题，如果文本中没有关于问题的答案，则直接回答 文档中没有相关内容
|
|
用户提问：{query}
|
|
你的回答："""
|
|
|