CommencezCommencez gratuitement

Point de terminaison POST pour l'enregistrement de modèles

Le point de terminaison GET que vous avez créé plus tôt permet aux utilisateurs de récupérer l'information sur les modèles existants, mais vous avez maintenant besoin d'un moyen pour que les membres autorisés de l'équipe puissent enregistrer de nouveaux modèles ou mettre à jour l'information sur des 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 existants. Ce point de terminaison enregistrera l'information sur les modèles sur le serveur.

Cette activité fait partie du cours

Déployer l'IA en production avec FastAPI

Voir le cours

Instructions de l’exercice

  • Créez un point de terminaison de requête POST à "/register-model".
  • Définissez le paramètre de fonction model_info pour accepter et valider l'information du modèle entrant au moyen 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 ce code d’exemple.

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