From 9683aeeafa2f1067ef061b34124a1c362df07e5e Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 03 四月 2025 14:10:13 +0800
Subject: [PATCH] rg配置修改

---
 app/service/files.py |   28 +++++++++++++++++++++++++++-
 1 files changed, 27 insertions(+), 1 deletions(-)

diff --git a/app/service/files.py b/app/service/files.py
index ad0e93e..69f21ed 100644
--- a/app/service/files.py
+++ b/app/service/files.py
@@ -1,8 +1,11 @@
+import json
+
 import fitz
 import io
 from docx import Document
 from dashscope import get_tokenizer  # dashscope鐗堟湰 >= 1.14.0
 
+from app.models import ComplexChatSessionDao
 from app.service.auth import decode_access_token
 
 
@@ -52,4 +55,27 @@
             '.docx'):
         text = await read_word(file)
 
-    return await get_str_token(text)
\ No newline at end of file
+    return await get_str_token(text)
+
+
+async def service_chat_message(db, message_id: str):
+    message = await ComplexChatSessionDao(db).get_session_by_id(message_id)
+    content = ""
+    title = ""
+    if message:
+        content = message.content
+        title= json.loads(message.query).get("query")
+    return title, content
+
+
+async def generate_word_document(title, content):
+    doc = Document()
+    # 娣诲姞鏍囬
+    doc.add_heading(title, level=1)
+
+    # 灏嗗唴瀹规寜娈佃惤鍒嗗壊骞跺啓鍏ユ枃妗�
+    for paragraph in content.split('\n'):
+        # print("--------------:", paragraph)
+        doc.add_paragraph(paragraph)
+
+    return doc
\ No newline at end of file

--
Gitblit v1.8.0