diff --git a/app/utils/milvus_utils.py b/app/utils/milvus_utils.py index 18cb8e3..66c99e4 100644 --- a/app/utils/milvus_utils.py +++ b/app/utils/milvus_utils.py @@ -111,8 +111,17 @@ class MilvusService: 用户的问题是:{question} """) | create_llm(disable_streaming=True) | StrOutputParser() - chain_input = {"context": "\n".join([node.node.text for node in result_nodes]), "question": param.question} + rag_context = "\n-----------------分隔符------------------\n".join([ + f""" +文件名:{node.node.metadata.get('name', "未命名文件.txt")} +文件内容: +{node.node.text} + """ for node in result_nodes + ]) + chain_input = {"context": rag_context, "question": param.question} + + print("chain_input", chain_input) relative_content = await chain.ainvoke(chain_input) # print("milvus search answer", relative_content)