21 lines
688 B
Python
21 lines
688 B
Python
from fastapi import APIRouter
|
|
from ..core.items import ChatRequest
|
|
from ..tools.get_all_role_and_chat import get_all_role_and_chat
|
|
from ..core.models import chat_history
|
|
|
|
router = APIRouter()
|
|
|
|
# 1. 从本地读取所有的data内容
|
|
@router.get("/tool_bar/get_all_role_and_chat")
|
|
def get_all_role_and_chat_endpoint():
|
|
# 正确调用函数并返回结果
|
|
return get_all_role_and_chat()
|
|
|
|
# 2. 根据rolename和chatname读取特定聊天记录
|
|
@router.post("/chat_box/get_chat_history")
|
|
async def get_chat_history_endpoint(role_name: str, chat_name: str):
|
|
# 实例化工具类
|
|
reader = chat_history.load_from_file(role_name, chat_name)
|
|
|
|
return reader.to_chatbox_format()
|