mirror of
https://github.com/Kakune55/PyGetGPT.git
synced 2025-05-06 18:29:24 +08:00
21 lines
510 B
Python
21 lines
510 B
Python
from fastapi import APIRouter
|
|
from model.util import InputData, getModels, getModelsInfo, requestModel
|
|
from pydantic import BaseModel
|
|
|
|
router = APIRouter()
|
|
|
|
class ChatRequest(BaseModel):
|
|
model: str
|
|
message: str
|
|
@router.post("/api/chat")
|
|
def chat(req: ChatRequest):
|
|
model_name = req.model
|
|
input_data = InputData(req.message)
|
|
#调用模型
|
|
ret = requestModel(model_name, input_data)
|
|
return ret
|
|
|
|
|
|
@router.get("/api/chat/models")
|
|
def get_models():
|
|
return {"models": getModelsInfo()} |