From 844547e04525155ac7458c07a4d3cc7d92d30268 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Wed, 25 Jun 2025 20:34:26 +0800
Subject: [PATCH] 'commit'

---
 dsRag/Tools/T4_DocxProcessor.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/dsRag/Tools/T4_DocxProcessor.py b/dsRag/Tools/T4_DocxProcessor.py
index 67a9fc7b..58b25e43 100644
--- a/dsRag/Tools/T4_DocxProcessor.py
+++ b/dsRag/Tools/T4_DocxProcessor.py
@@ -85,10 +85,20 @@ PROMPT_TEMPLATE = """
 {text_chunk}
 """
 
-def ask_llm(text_chunk):
-    """向大模型提问并获取响应"""
-    prompt = PROMPT_TEMPLATE.format(text_chunk=text_chunk)
-    return call_qwen_plus(prompt)
+def ask_llm(text_chunk, is_final=False):
+    """调用大模型并处理响应"""
+    prompt = PROMPT_TEMPLATE.format(text_chunk=text_chunk, is_final=is_final)
+    
+    def stream_callback(chunk):
+        """SSE流式回调函数"""
+        print(chunk, end='', flush=True)
+    
+    try:
+        response = call_qwen_plus(prompt, stream_callback=stream_callback)
+        return response
+    except Exception as e:
+        print(f"调用大模型出错: {str(e)}")
+        return None
 
 def process_document(input_path, output_dir):
     """处理文档主函数"""