使用带有文件和查询参数的Python请求向FastAPI发送POST请求

Question

我正在使用FastAPI来提供一些ML模型，并且我有一个使用Python Requests模块的Streamlit基本UI。

我的服务之一是通过POST请求获得图像，它就像一个魅力。

服务器端

@app.post("/segformer")
async def segformer(file: Optional[UploadFile] = None):

｛BASE_URI｝/docs提供的卷曲

curl -X 'POST' \
  'http://localhost:8080/segformer' \
  -H 'accept: application/json' \
  -H 'Content-Type: multipart/form-data' \
  -F 'file=@image1.jpg;type=image/jpeg'

客户端使用Python请求

response = requests.post(f"{BASE_URI}/segformer", files=files)

一旦我想添加额外的参数，它就变得令人毛骨悚然。例如：

服务器端

@dataclass
class ModelInterface:
    model_name: str = "detr"
    tags: List[str] = Query(...)

@app.post("/detr")
async def detr(file: UploadFile = File(...), model: ModelInterface = Depends()):

curl -- 由 {BASE_URI}/docs 给出

curl -X 'POST' \
  'http://localhost:8080/detr?model_name=detr&tags=balloon&tags=dog' \
  -H 'accept: application/json' \
  -H 'Content-Type: multipart/form-data' \
  -F 'file=@image1.jpg;type=image/jpeg'

在那之前，一切正常。一旦我想使用Python＜code＞Requests＜/code＞转换该调用，我就会遇到一些问题，最终总是出现＜code＞/detr HTTP/1.1“400 Bad Request＜/code>错误。

以下是我尝试过的:

客户端使用Python请求

headers = {
    "Content-type": "multipart/form-data",
    "accept": "application/json"
}

payload = {
    "tags": ["balloon", "dog"]
}

response = requests.post(
    f"{BASE_URI}/detr",
    headers=headers,
    json=payload,  <- also *data*  
    files=files,
)

最后，问题似乎缩小了如何转换它：

curl -X 'POST' \
  'http://localhost:8080/detr?tags=balloon&tags=dog' \
  -H 'accept: application/json' \
  -H 'Content-Type: multipart/form-data' \
  -F 'file=@image1.jpg;type=image/jpeg'

到有效的Python请求调用！

我还遇到了以下FastAPI代码的问题：

@dataclass
class ModelInterface:
    model_name: str = "detr"
    tags: List[str] = None

@app.post("/detr2")
async def detr2(file: UploadFile = File(...), model: ModelInterface = Form(...)):
...

转换为curl命令：

curl -X 'POST' \
  'http://localhost:8080/detr2' \
  -H 'accept: application/json' \
  -H 'Content-Type: multipart/form-data' \
  -F 'file=@image1.jpg;type=image/jpeg' \
  -F 'model={
  "model_name": "detr",
  "tags": [
    "balloon", "dog"
  ]
}' 

失败，出现“POST/det2 HTTP/1.1”422不可处理实体错误

Answer 1

要在Python请求中传递查询参数，应该使用< code>params键。因此:

response = requests.post(url='<your_url_here>', params=payload)

此外，不需要在头中设置< code>Content-type，因为它会根据传递给< code>requests.post()的参数自动添加。这样做，请求将失败，因为除了< code>multipart/form-data之外，“< code>Content-type必须包括用于描述post主体中各部分的< code>boundary值。不设置< code>Content-Type头可确保requests将其设置为正确的值”(ref)。看看这个和这个。此外，请确保在< code>files中使用您在endpoint中给出的相同< code>key名称，即< code>file。因此，在Python请求中，它应该如下所示:

files = {('file', open('my_file.txt', 'rb'))}
response = requests.post(url='<your_url_here>', files=files, params=payload)

您还可以在此答案中找到有关如何发送其他数据以及文件的更多选项。