forked from killua/TakwayPlatform
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
|
from fastapi import APIRouter, HTTPException, status, WebSocket
|
||
|
|
from ..controllers.chat import *
|
||
|
|
from fastapi import Depends
|
||
|
|
from ..dependencies.database import get_db
|
||
|
|
from ..dependencies.redis import get_redis
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
#创建新聊天接口
|
||
|
|
@router.post("/chats", response_model=ChatCreateResponse)
|
||
|
|
async def create_chat(chat: ChatCreateRequest, db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
response = await create_chat_handler(chat, db, redis)
|
||
|
|
return response
|
||
|
|
|
||
|
|
#删除聊天接口
|
||
|
|
@router.delete("/chats/{user_character_id}", response_model=ChatDeleteResponse)
|
||
|
|
async def delete_chat(user_character_id: int, db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
response = await delete_chat_handler(user_character_id, db, redis)
|
||
|
|
return response
|
||
|
|
|
||
|
|
#非流式聊天
|
||
|
|
@router.post("/chats/non-streaming", response_model=ChatNonStreamResponse)
|
||
|
|
async def non_streaming_chat(chat: ChatNonStreamRequest,db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN,detail="this api is not available")
|
||
|
|
response = await non_streaming_chat_handler(chat, db, redis)
|
||
|
|
return response
|
||
|
|
|
||
|
|
|
||
|
|
#流式聊天_单次
|
||
|
|
@router.websocket("/chat/streaming/temporary")
|
||
|
|
async def streaming_chat(ws: WebSocket,db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
await ws.accept()
|
||
|
|
await streaming_chat_temporary_handler(ws,db,redis)
|
||
|
|
|
||
|
|
|
||
|
|
#流式聊天_持续
|
||
|
|
@router.websocket("/chat/streaming/lasting")
|
||
|
|
async def streaming_chat(ws: WebSocket,db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
await ws.accept()
|
||
|
|
await streaming_chat_lasting_handler(ws,db,redis)
|
||
|
|
|
||
|
|
|
||
|
|
#语音通话
|
||
|
|
@router.websocket("/chat/voice_call")
|
||
|
|
async def voice_chat(ws: WebSocket,db=Depends(get_db), redis=Depends(get_redis)):
|
||
|
|
await ws.accept()
|
||
|
|
await voice_call_handler(ws,db,redis)
|