Ponto final POST para registro do modelo
Embora o endpoint GET que você criou anteriormente permita que os usuários recuperem informações sobre modelos existentes, agora você precisa de uma maneira para que os membros autorizados da equipe registrem novos modelos ou atualizem informações sobre os existentes.
Você precisa criar um endpoint POST
que permita aos membros da equipe registrar novos modelos ou atualizar os já existentes.
Esse endpoint vai guardar as informações do modelo no servidor.
Este exercício faz parte do curso
Implementando IA na produção com FastAPI
Instruções do exercício
- Crie um ponto final de solicitação “
POST
” em"/register-model"
. - Defina o parâmetro da função
model_info
para aceitar e validar as informações do modelo recebidas usando o modelo PydanticModelInfo
. - Converta o objeto “
ModelInfo
” em um dicionário e guarde-o em “model_db
”.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
model_db = {}
class ModelInfo(BaseModel):
model_id: int
model_name: str
description: str
# Create the POST request endpoint
@app.____("/register-model", status_code=201)
# Pass the model info from the request as function parameter
def register_model(____: ModelInfo):
# Add new model's information dictionary to the model database
model_db[model_info.model_id] = model_info.____
return {"message": "Model registered successfully", "model": model_info}, 201