POST endpoint for model registration
While the GET endpoint you created earlier allows users to retrieve information about existing models, you now need a way for authorized team members to register new models or update information about existing ones.
You need to create a POST endpoint that allows team members to register new models or update existing ones.
This endpoint will store model information on the server.
Bu egzersiz
Deploying AI into Production with FastAPI
kursunun bir parçasıdırEgzersiz talimatları
- Create a
POSTrequest endpoint at"/register-model". - Define the function parameter
model_infoto accept and validate the incoming model information using theModelInfoPydantic model. - Convert the
ModelInfoobject to a dictionary and store it in themodel_db.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
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