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