feat:增加查询用户历史消息功能
This commit is contained in:
19
router.py
19
router.py
@@ -1,8 +1,10 @@
|
||||
from typing import List
|
||||
|
||||
from fastapi import APIRouter
|
||||
from fastapi.responses import StreamingResponse
|
||||
|
||||
from model import ChatRequest
|
||||
from services.agent_service import get_messages, clear_messages, search_recipes
|
||||
from model import ChatRequest, SessionMessageResponse
|
||||
from services.agent_service import get_messages, clear_messages, search_recipes, get_user_session_messages
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@@ -16,11 +18,16 @@ async def chat_endpoint(request: ChatRequest):
|
||||
)
|
||||
|
||||
|
||||
@router.get("/chat/messages")
|
||||
async def get_chat_messages(thread_id: str):
|
||||
@router.get("/chat/messages/session", response_model=List[SessionMessageResponse])
|
||||
def get_session_messages(username: str):
|
||||
"""获取历史消息"""
|
||||
messages = get_messages(thread_id)
|
||||
return {"messages": messages}
|
||||
return get_user_session_messages(username)
|
||||
|
||||
|
||||
@router.get("/chat/messages")
|
||||
def get_chat_messages(thread_id: str):
|
||||
"""获取历史消息"""
|
||||
return get_messages(thread_id)
|
||||
|
||||
|
||||
@router.delete("/chat/messages")
|
||||
|
||||
Reference in New Issue
Block a user