如何引导 FastAPI 从 Union 中选择正确的类型类型

Question

我有以下型号：

from pydantic import BaseModel

class A(BaseModel):
   tag: str
   field1: str

class B(BaseModel):
   tag: str
   field1: str
   field2: str

我正在使用它们来定义请求正文，如下所示：

@app.post('/route')
def handle(request: typing.Union[A, B])

documentation 声明：

首先包含最具体的类型，然后是不太具体的类型

但我想知道是否有一种方法可以引导 FastAPI 根据标签选择正确的模型。目前，我所有的模型都有一个常量标签（即类型A 总是有tag = 'A'）。有没有办法可以更改我的模型定义以确保分配的模型始终与请求正文中收到的标签匹配？

目前，我通过不键入 handle 函数来解决这个问题，而是专门将 tag 与模型匹配，但理想情况下我会使用 typing.Union 作为类型并确信它会得到它正确。

Answer 1

来自@rv.kvetch 的评论，将tag 字段注释为字符串文字，例如Literal['A'] 表示每次都选择了正确的类型。