CommencerCommencer gratuitement

Endpoint POST pour l’enregistrement de modèles

Alors que l’endpoint GET que vous avez créé plus tôt permet aux utilisateurs de récupérer des informations sur les modèles existants, 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 les informations des modèles existants.

Vous devez créer un endpoint POST qui permet aux membres de l’équipe d’enregistrer de nouveaux modèles ou de mettre à jour des modèles existants. Cet endpoint stockera les informations des modèles sur le serveur.

Cet exercice fait partie du cours

Déployer l’IA en production avec FastAPI

Afficher le cours

Instructions

  • Créez un endpoint de requête POST à "/register-model".
  • Définissez le paramètre de fonction model_info pour accepter et valider les informations du modèle entrant en utilisant le 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