POST-endpoint för modellregistrering
GET-endpointen du skapade tidigare låter användare hämta information om befintliga modeller. Nu behöver du ett sätt för behöriga teammedlemmar att registrera nya modeller eller uppdatera information om befintliga.
Du ska skapa en POST-endpoint som låter teammedlemmar registrera nya modeller eller uppdatera befintliga. Endpointen lagrar modellinformation på servern.
Den här övningen är en del av kursen
Driftsättning av AI i produktion med FastAPI
Övningsinstruktioner
- Skapa en
POST-endpoint på"/register-model". - Definiera funktionsparametern
model_infoför att ta emot och validera inkommande modellinformation med hjälp av Pydantic-modellenModelInfo. - Konvertera
ModelInfo-objektet till ett lexikon och lagra det imodel_db.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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