前端适配

This commit is contained in:
Miyamizu-MitsuhaSang 2025-11-07 14:52:30 +08:00
parent ab437dc982
commit 1c8036cb8d
2 changed files with 5 additions and 1 deletions

View File

@ -7,6 +7,7 @@ from starlette.requests import Request
from app.api.ai_assist import service from app.api.ai_assist import service
from app.api.ai_assist.ai_schemas import AIAnswerResponse, AIAnswerOut, AIQuestionRequest 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.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.models import User
from app.utils.security import get_current_user from app.utils.security import get_current_user
from settings import settings 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, "user", question)
await save_message(redis, user_id, word, "assistant", answer) await save_message(redis, user_id, word, "assistant", answer)
answer = await reply_process(answer)
return AIAnswerOut( return AIAnswerOut(
word=word, word=word,
answer=ai_resp.get_answer(), answer=answer,
model=ai_resp.model, model=ai_resp.model,
tokens_used=ai_resp.usage.total_tokens if ai_resp.usage else None tokens_used=ai_resp.usage.total_tokens if ai_resp.usage else None
) )

View File

@ -78,4 +78,5 @@ async def reply_process(reply: str) -> str:
:return: :return:
""" """
reply.replace("**", "") reply.replace("**", "")
reply.replace("---", "")
return reply return reply