From cd2f479bb44c916948e80d7bf9a47f3846ed8c26 Mon Sep 17 00:00:00 2001
From: zstar <65890619+zstar1003@users.noreply.github.com>
Date: Mon, 9 Jun 2025 18:46:51 +0800
Subject: [PATCH] =?UTF-8?q?chore(dialog):=20=E8=B0=83=E6=95=B4=E7=9B=B8?=
 =?UTF-8?q?=E4=BC=BC=E5=BA=A6=E9=98=88=E5=80=BC=E4=BB=A5=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E5=8C=B9=E9=85=8D=E6=95=88=E6=9E=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 dialog_service.py 中设置更小的相似度阈值（从 0.1 调整到 0.01）
- 更新 docker-compose.yml 和 docker-compose_gpu.yml，挂载 dialog_service.py 文件
---
 api/db/services/dialog_service.py | 4 +++-
 docker/docker-compose.yml         | 1 +
 docker/docker-compose_gpu.yml     | 1 +
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/db/services/dialog_service.py b/api/db/services/dialog_service.py
index e56b131..a5d15c3 100644
--- a/api/db/services/dialog_service.py
+++ b/api/db/services/dialog_service.py
@@ -495,8 +495,10 @@ def ask(question, kb_ids, tenant_id):
     max_tokens = chat_mdl.max_length
     # 获取所有知识库的租户ID并去重
     tenant_ids = list(set([kb.tenant_id for kb in kbs]))
+    # 设置更小的相似度阈值以适配更好的效果(原始值0.1)
+    similarity_threshold = 0.01
     # 调用检索器检索相关文档片段
-    kbinfos = retriever.retrieval(question, embd_mdl, tenant_ids, kb_ids, 1, 12, 0.1, 0.3, aggs=False, rank_feature=label_question(question, kbs))
+    kbinfos = retriever.retrieval(question, embd_mdl, tenant_ids, kb_ids, 1, 12, similarity_threshold, 0.3, aggs=False, rank_feature=label_question(question, kbs))
     # 将检索结果格式化为提示词，并确保不超过模型最大token限制
     knowledges = kb_prompt(kbinfos, max_tokens)
     prompt = """
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index ee9aec8..11dd245 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -18,6 +18,7 @@ services:
       - ./nginx/proxy.conf:/etc/nginx/proxy.conf
       - ./nginx/nginx.conf:/etc/nginx/nginx.conf
       - ../api/db/services/database.py:/ragflow/api/db/services/database.py
+      - ../api/db/services/dialog_service.py:/ragflow/api/db/services/dialog_service.py
     env_file: .env
     environment:
       - TZ=${TIMEZONE}
diff --git a/docker/docker-compose_gpu.yml b/docker/docker-compose_gpu.yml
index 9dcd6bd..a1c0ed4 100644
--- a/docker/docker-compose_gpu.yml
+++ b/docker/docker-compose_gpu.yml
@@ -18,6 +18,7 @@ services:
       - ./nginx/proxy.conf:/etc/nginx/proxy.conf
       - ./nginx/nginx.conf:/etc/nginx/nginx.conf
       - ../api/db/services/database.py:/ragflow/api/db/services/database.py
+      - ../api/db/services/dialog_service.py:/ragflow/api/db/services/dialog_service.py
     env_file: .env
     environment:
       - TZ=${TIMEZONE}