从发送到 FastAPI 应用程序的请求正文中提取列表

Question

我有一个 FastAPI 应用程序，它根据发送给它的 HTTP POST 数据进行预测。代码如下

import tensorflow as tf
from typing import List
from fastapi import FastAPI, Response
from pydantic import BaseModel

class ItemList(BaseModel):
    instances: List[List[float]]

app = FastAPI()

model = tf.keras.models.load_model('trainedmodel/1')

@app.post('/v1/models/my_model:predict')
def predict(inputdata: ItemList):
    # Extract [[1.5,1.65,2,0.5,-2,1.5,0.1,2,0.2]]
    result = {
                 "predictions": [
                         model.predict(modelInput).flatten().tolist()
                 ]
    }
    return result

对应用的请求类型为application/json，请求正文包含{"instances":[[1.5,1.65,2,0.5,-2,1.5,0.1,2,0.2]]}

我需要从请求正文中提取[[1.5,1.65,2,0.5,-2,1.5,0.1,2,0.2]]，以便将其输入到模型中（如上述代码中的modelInput）。在 Flask 中，我使用了request.json['instances'] 这样做。有人可以告诉我如何在 FastAPI 中做同样的事情吗？

Answer 1

要访问请求正文，您只需使用inputdata.instances

class ItemList(BaseModel):
    instances: List[List[float]]

@app.post('/v1/models/my_model:predict')
def predict(inputdata: ItemList):
    # Extract [[1.5,1.65,2,0.5,-2,1.5,0.1,2,0.2]]
    extracted_list = inputdata.instances