ПочатиПочніть безкоштовно

Endpoint POST для реєстрації моделі

GET-ендпоінт, який ви створили раніше, дозволяє користувачам отримувати інформацію про наявні моделі. Тепер вам потрібен спосіб, щоб уповноважені члени команди могли реєструвати нові моделі або оновлювати дані про наявні.

Вам потрібно створити POST-ендпоінт, який дозволить членам команди реєструвати нові моделі або оновлювати наявні. Цей ендпоінт зберігатиме інформацію про модель на сервері.

Ця вправа є частиною курсу

Розгортання AI у продакшні з FastAPI

Переглянути курс

Інструкції до вправи

  • Створіть endpoint для запиту POST за адресою "/register-model".
  • Оголосіть параметр функції model_info, щоб приймати й валідувати вхідну інформацію про модель за допомогою Pydantic-моделі ModelInfo.
  • Перетворіть об'єкт ModelInfo на словник і збережіть його в model_db.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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
Редагувати та запускати код