2024-03-27 10:12:28 +08:00

24 lines
764 B
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

from typing import Literal
from fastapi import APIRouter, Body
from chatchat.server.utils import get_server_configs, get_prompt_template
server_router = APIRouter(prefix="/server", tags=["Server State"])
# 服务器相关接口
server_router.post("/configs",
summary="获取服务器原始配置信息",
)(get_server_configs)
@server_router.post("/get_prompt_template",
summary="获取服务区配置的 prompt 模板")
def get_server_prompt_template(
type: Literal["llm_chat", "knowledge_base_chat"]=Body("llm_chat", description="模板类型可选值llm_chatknowledge_base_chat"),
name: str = Body("default", description="模板名称"),
) -> str:
return get_prompt_template(type=type, name=name)