From f6546da6a774d9149e441f58f4a7fc8a6c96ff5a Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Fri, 27 Jun 2025 09:31:25 +0800
Subject: [PATCH] 'commit'

---
 dsRag/Start.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/dsRag/Start.py b/dsRag/Start.py
index 072f9658..d9b158c6 100644
--- a/dsRag/Start.py
+++ b/dsRag/Start.py
@@ -34,6 +34,7 @@ logger.addHandler(handler)
 model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT)
 logger.info(f"模型加载成功，词向量维度: {model.vector_size}")
 
+
 # 将HTML文件转换为Word文件
 def html_to_word_pandoc(html_file, output_file):
     subprocess.run(['pandoc', html_file, '-o', output_file])
@@ -114,7 +115,6 @@ async def generate_stream(client, milvus_pool, collection_manager, query, docume
                             logger.info(f"ID: {hit.id}")
                             logger.info(f"标签: {record['tags']}")
                             logger.info(f"用户问题: {record['user_input']}")
-
                             logger.info(f"时间: {record['timestamp']}")
                             logger.info(f"距离: {hit.distance}")
                             logger.info("-" * 40)  # 分隔线
@@ -148,8 +148,11 @@ async def generate_stream(client, milvus_pool, collection_manager, query, docume
 """
 
         # 调用阿里云大模型
-        html_content = client.chat(prompt)
-        yield {"data": html_content}
+        if len(context) > 0:
+            html_content = client.chat(prompt)
+            yield {"data": html_content}
+        else:
+            yield {"data": "没有在知识库中找到相关的信息，无法回答此问题。"}
     except Exception as e:
         yield {"data": f"生成报告时出错: {str(e)}"}
     finally: