From 1c8036cb8de6731f0b0385a6a08d94b20215ca34 Mon Sep 17 00:00:00 2001
From: Miyamizu-MitsuhaSang <2510681107@qq.com>
Date: Fri, 7 Nov 2025 14:52:30 +0800
Subject: [PATCH] =?UTF-8?q?=E5=89=8D=E7=AB=AF=E9=80=82=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/ai_assist/routes.py         | 5 ++++-
 app/api/article_director/service.py | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/api/ai_assist/routes.py b/app/api/ai_assist/routes.py
index 3fd91ea..28af951 100644
--- a/app/api/ai_assist/routes.py
+++ b/app/api/ai_assist/routes.py
@@ -7,6 +7,7 @@ from starlette.requests import Request
 from app.api.ai_assist import service
 from app.api.ai_assist.ai_schemas import AIAnswerResponse, AIAnswerOut, AIQuestionRequest
 from app.api.ai_assist.utils.redis_memory import get_chat_history, save_message, clear_chat_history
+from app.api.article_director.service import reply_process
 from app.models import User
 from app.utils.security import get_current_user
 from settings import settings
@@ -85,9 +86,11 @@ async def dict_exp(
             await save_message(redis, user_id, word, "user", question)
             await save_message(redis, user_id, word, "assistant", answer)
 
+            answer = await reply_process(answer)
+
             return AIAnswerOut(
                 word=word,
-                answer=ai_resp.get_answer(),
+                answer=answer,
                 model=ai_resp.model,
                 tokens_used=ai_resp.usage.total_tokens if ai_resp.usage else None
             )
diff --git a/app/api/article_director/service.py b/app/api/article_director/service.py
index f2cf37c..86eec08 100644
--- a/app/api/article_director/service.py
+++ b/app/api/article_director/service.py
@@ -78,4 +78,5 @@ async def reply_process(reply: str) -> str:
     :return:
     """
     reply.replace("**", "")
+    reply.replace("---", "")
     return reply