wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 打印milvus 检索日志

Browse Source
This commit is contained in:
martsforever
2025-12-03 00:39:42 +08:00
parent 4b89b2a9d6
commit 829d723dfb
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/utils/milvus_utils.py
+3
View File
@@ -95,6 +95,9 @@ class MilvusService:
retriever = vector_index.as_retriever(filters=filters, similarity_top_k=param.top_k) retriever = vector_index.as_retriever(filters=filters, similarity_top_k=param.top_k)
result_nodes = await retriever.aretrieve(param.question) result_nodes = await retriever.aretrieve(param.question)
[print(node.node.metadata.get('name', None), node.score) for node in result_nodes]
result_nodes = [node for node in result_nodes if node.score >= 0.3] result_nodes = [node for node in result_nodes if node.score >= 0.3]
chain = ChatPromptTemplate.from_template(""" chain = ChatPromptTemplate.from_template("""