107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
"""
|
|
每次调用 article-director/article 接口时都要同时调用reset以清空 redis 中的上下文
|
|
"""
|
|
from typing import Literal, Dict, Tuple
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from starlette.requests import Request
|
|
|
|
from app.api.article_director import service
|
|
from app.api.article_director.article_schemas import UserArticleRequest, UserQuery
|
|
from app.models import User
|
|
from app.utils.security import get_current_user
|
|
|
|
article_router = APIRouter()
|
|
|
|
|
|
@article_router.post("/article-director/article")
|
|
async def article_director(
|
|
request: Request,
|
|
upload_article: UserArticleRequest,
|
|
lang: Literal["en-US", "fr-FR", "ja-JP"] = "fr-FR",
|
|
user: Tuple[User, Dict] = Depends(get_current_user)
|
|
):
|
|
"""
|
|
文本形式接口，即直接从文本框中获取
|
|
每次调用本接口的同时都要同时调用reset接口
|
|
:param upload_article:
|
|
:param lang:
|
|
:return:
|
|
"""
|
|
redis = request.app.state.redis
|
|
# print(upload_article)
|
|
|
|
article_lang = "法语" if lang == "fr-FR" else "日语"
|
|
|
|
user_id = user[0].id
|
|
article = upload_article.content
|
|
|
|
# 读取历史对话
|
|
session = await service.get_session(redis_client=redis, user_id=user_id)
|
|
|
|
# 追加用户输入
|
|
user_prompt = service.set_user_prompt(upload_article, article_lang=article_lang)
|
|
session.append({"role": "user", "content": user_prompt})
|
|
|
|
# 调用 EduChat 模型
|
|
completion = service.chat_ecnu_request(session)
|
|
|
|
# 取出回答内容
|
|
assistant_reply = completion.choices[0].message.content
|
|
|
|
assistant_reply = await service.reply_process(assistant_reply)
|
|
|
|
# 保存模型回复
|
|
session.append({"role": "assistant", "content": assistant_reply})
|
|
|
|
# 存入 Redis
|
|
await service.save_session(redis, user_id, session)
|
|
|
|
return {
|
|
"reply": assistant_reply,
|
|
"tokens": completion.usage.total_tokens,
|
|
"conversation_length": len(session),
|
|
}
|
|
|
|
|
|
@article_router.post("/article-director/question", description="用户进一步询问")
|
|
async def further_question(
|
|
request: Request,
|
|
user_prompt: UserQuery,
|
|
user: Tuple[User, Dict] = Depends(get_current_user)
|
|
):
|
|
redis = request.app.state.redis
|
|
|
|
user_id = user[0].id
|
|
|
|
# 读取历史对话
|
|
session = await service.get_session(redis_client=redis, user_id=user_id)
|
|
|
|
# 追加用户输入
|
|
session.append({"role": "user", "content": user_prompt.query})
|
|
|
|
# 调用 EduChat 模型
|
|
completion = service.chat_ecnu_request(session)
|
|
|
|
# 取出回答内容
|
|
assistant_reply = completion.choices[0].message.content
|
|
|
|
# 保存模型回复
|
|
session.append({"role": "assistant", "content": assistant_reply})
|
|
|
|
# 存入 Redis
|
|
await service.save_session(redis, user_id, session)
|
|
|
|
return {
|
|
"reply": assistant_reply,
|
|
"tokens": completion.usage.total_tokens,
|
|
"conversation_length": len(session),
|
|
}
|
|
|
|
@article_router.post("/article-director/reset", description="重置上下文")
|
|
async def reset_conversation(request: Request, user: Tuple[User, Dict] = Depends(get_current_user)):
|
|
user_id = user[0].id
|
|
redis = request.app.state.redis
|
|
await service.reset_session(redis, user_id)
|
|
return {"message": f"已重置用户 {user_id} 的作文对话记录"}
|