diff --git a/app/utils/milvus_utils.py b/app/utils/milvus_utils.py index 62b4adc..ddc333e 100644 --- a/app/utils/milvus_utils.py +++ b/app/utils/milvus_utils.py @@ -95,6 +95,7 @@ class MilvusService: retriever = vector_index.as_retriever(filters=filters, similarity_top_k=param.top_k) result_nodes = await retriever.aretrieve(param.question) + result_nodes = [node for node in result_nodes if node.score >= 0.3] chain = ChatPromptTemplate.from_template(""" 你将收到一个用户问题和一组来自LlamaIndex的检索结果。你的任务是: @@ -124,6 +125,9 @@ class MilvusService: relative_content = await chain.ainvoke(chain_input) + # print("milvus search answer", relative_content) + # print("milvus search nodes", [node.node.text for node in result_nodes]) + return MilvusSearchResponse( answer=relative_content, nodes=[