feat: 优化知识库召回接口,将文件名也作为上下文作为召回

This commit is contained in:
martsforever
2025-12-02 22:12:29 +08:00
parent 0288d17525
commit 4b89b2a9d6
+10 -1
View File
@@ -111,8 +111,17 @@ class MilvusService:
用户的问题是:{question} 用户的问题是:{question}
""") | create_llm(disable_streaming=True) | StrOutputParser() """) | 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) relative_content = await chain.ainvoke(chain_input)
# print("milvus search answer", relative_content) # print("milvus search answer", relative_content)