feat: 优化知识库召回接口,将文件名也作为上下文作为召回
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user