BBIT-Kai
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
from models.ChatRequest import ChatRequest
|
|
from models.BaseResponse import BaseResponse
|
|
import uuid
|
|
import db.postgres as db
|
|
import uuid
|
|
import threading
|
|
from fastapi import APIRouter, Depends
|
|
from uuid import UUID
|
|
from config.security import get_user_id_from_token
|
|
router = APIRouter()
|
|
from llm.chatLLM import get_chat_response
|
|
from llm.titleChain import get_title
|
|
from llm.dataLLM import get_graph_output
|
|
|
|
# def async_db_task(func, *args, **kwargs):
|
|
# """将数据库操作放到后台线程执行"""
|
|
# threading.Thread(target=func, args=args, kwargs=kwargs, daemon=True).start()
|
|
|
|
@router.get("/history")
|
|
def getHistory(sessionId: str):
|
|
return BaseResponse(data=db.get_history(sessionId))
|
|
|
|
@router.get("/aiList")
|
|
def getAiList(user_id: UUID = Depends(get_user_id_from_token)):
|
|
if not user_id:
|
|
return {"error": "userId is required"}
|
|
return BaseResponse(data=db.get_all_ai(user_id))
|
|
|
|
@router.get("/sessions")
|
|
def getSessions(user_id: UUID = Depends(get_user_id_from_token)):
|
|
if not user_id:
|
|
return {"error": "userId is required"}
|
|
return BaseResponse(data=db.get_sessions(user_id))
|
|
|
|
@router.post("/chat")
|
|
def chat(req: ChatRequest, user_id: UUID = Depends(get_user_id_from_token)):
|
|
if not user_id:
|
|
return {"error": "userId is required"}
|
|
if not req.aiId:
|
|
return {"error": "aiId is required"}
|
|
sessionName = get_title(req.userInput)
|
|
# 如果没有 sessionId 就新建
|
|
if not req.sessionId:
|
|
isNewSession = True
|
|
req.sessionId = str(uuid.uuid4())
|
|
db.insert_session(user_id,req.aiId, req.sessionId, sessionName)
|
|
else:
|
|
isNewSession = False
|
|
db.update_session_updated_at(req.sessionId)
|
|
|
|
# 插入用户消息
|
|
db.insert_message(req.sessionId, False, req.userInput)
|
|
|
|
# 调用 LLM
|
|
if req.aiId == "9d157dd1-921b-c768-5b90-3e903b50f6f9":
|
|
# 数据专家AI
|
|
answer = get_graph_output(aiRole=db.get_ai_personality(req.aiId),history=db.get_history(req.sessionId), userInput= req.userInput)
|
|
else:
|
|
answer = get_chat_response(aiRole=db.get_ai_personality(req.aiId),history=db.get_history(req.sessionId), userInput= req.userInput).content
|
|
# 插入 AI 回复
|
|
db.insert_message(req.sessionId, True, answer)
|
|
|
|
return BaseResponse(data={"sessionName":sessionName,"isNewSession":isNewSession,"content":answer,"sessionId": req.sessionId}) |