Endpoint POST do rejestracji modelu
Wcześniej utworzony endpoint GET pozwala użytkownikom pobierać informacje o istniejących modelach. Teraz potrzebujesz sposobu, aby uprawnieni członkowie zespołu mogli rejestrować nowe modele lub aktualizować dane o istniejących.
Utwórz endpoint POST, który umożliwi członkom zespołu rejestrowanie nowych modeli oraz aktualizowanie już istniejących.
Ten endpoint będzie przechowywał informacje o modelach na serwerze.
To ćwiczenie jest częścią kursu
Wdrażanie AI na produkcję z FastAPI
Instrukcje do ćwiczenia
- Utwórz endpoint żądania
POSTpod adresem"/register-model". - Zdefiniuj parametr funkcji
model_info, aby przyjmował i walidował przychodzące dane o modelu przy użyciu modelu PydanticModelInfo. - Przekonwertuj obiekt
ModelInfona słownik i zapisz go wmodel_db.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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