Punto final POST para el registro del modelo
Aunque el punto final GET creado anteriormente permite a los usuarios recuperar información sobre los modelos existentes, ahora necesitáis una forma de que los miembros autorizados del equipo puedan registrar nuevos modelos o actualizar la información sobre los existentes.
Debes crear un punto final POST
que permita a los miembros del equipo registrar nuevos modelos o actualizar los existentes.
Este punto final almacenará la información del modelo en el servidor.
Este ejercicio forma parte del curso
Implementación de IA en producción con FastAPI
Instrucciones del ejercicio
- Crea un punto final de solicitud «
POST
» en"/register-model"
. - Define el parámetro de función
model_info
para aceptar y validar la información del modelo entrante utilizando el modelo PydanticModelInfo
. - Convierte el objeto «
ModelInfo
» en un diccionario y guárdalo en «model_db
».
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
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