Endpoint POST untuk pendaftaran model
Meskipun endpoint GET yang Anda buat sebelumnya memungkinkan pengguna mengambil informasi tentang model yang sudah ada, kini Anda memerlukan cara bagi anggota tim yang berwenang untuk mendaftarkan model baru atau memperbarui informasi model yang ada.
Anda perlu membuat endpoint POST yang memungkinkan anggota tim mendaftarkan model baru atau memperbarui model yang sudah ada.
Endpoint ini akan menyimpan informasi model di server.
Latihan ini merupakan bagian dari kursus
Menerapkan AI ke Produksi dengan FastAPI
Instruksi latihan
- Buat endpoint permintaan
POSTdi"/register-model". - Definisikan parameter fungsi
model_infountuk menerima dan memvalidasi informasi model yang masuk menggunakan model PydanticModelInfo. - Konversikan objek
ModelInfomenjadi dictionary dan simpan dimodel_db.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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