取得模型資訊的 GET 端點
你是機器學習團隊的一員,團隊已開發多個模型,分別用於情緒分析、產品分類與客戶流失預測等任務。你正在部署這些模型,需要建立一個端點,提供每個模型的基本資訊。
你的任務是實作一個 GET 端點,路徑為 /model-info/{model_id},用來擷取並回傳這些關鍵的模型資訊。
本練習屬於課程
使用 FastAPI 將 AI 佈署到生產環境
練習說明
- 在
"/model-info/{model_id}"建立一個GET端點,回傳指定模型的資訊。 - 端點應接受
model_id作為路徑參數。 - 若
model_id為0,則拋出HTTPException,並使用404狀態碼表示找不到該模型。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
from fastapi import FastAPI, HTTPException
app = FastAPI()
# Add model_id as a path parameter in the route
@app.get("/model-info/{____}")
# Pass on the model id as an argument
async def get_model_info(____: int):
if model_id == 0:
# Raise the right status code for not found
raise HTTPException(status_code=____, detail="Model not found")
model_info = get_model_details(id)
return {"model_id": model_id, "model_name": model_info}