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}
                            你的回答："""
    