CommencerCommencer gratuitement

Point de terminaison POST pour l'enregistrement du modèle

Bien que le point de terminaison GET que vous avez créé précédemment permette aux utilisateurs de récupérer des informations sur les modèles existants, vous devez maintenant trouver un moyen pour les membres autorisés de l'équipe d'enregistrer de nouveaux modèles ou de mettre à jour les informations sur les modèles existants.

Vous devez créer un point de terminaison POST qui permet aux membres de l'équipe d'enregistrer de nouveaux modèles ou de mettre à jour ceux qui existent déjà. Ce point de terminaison stockera les informations relatives au modèle sur le serveur.

Cet exercice fait partie du cours

Déployer l'IA en production avec FastAPI

Afficher le cours

Instructions

  • Veuillez créer un point de terminaison de demande d'POST à l'adresse "/register-model".
  • Définissez le paramètre de fonction model_info afin d'accepter et de valider les informations du modèle entrant à l'aide du modèle Pydantic ModelInfo.
  • Convertissez l'objet « ModelInfo » en dictionnaire et enregistrez-le dans « model_db ».

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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
Modifier et exécuter le code