FastAPI and Pydantic AnyOf in an Array - fastapi

Question

Is there a way to declare an AnyOf in Pydantic with FastAPI like this?
class RedPill(BaseModel):
kind: str = Literal['red']
name: str
size: int
class BluePill(BaseModel):
kind: str = Literal['blue']
serialnumber: str
class Matrix(BaseModel):
year: int
pills: List[Union[RedPill, BluePill]] = []
#api.post('/')
def sample(data: Matrix):
return
How do I represent a valid array of different objects.

Answer 1

so you were really close :)
you just need to change the Literal into the type and it should work
you can read about literal here
from pprint import pprint
from typing import List, Union
from pydantic.main import BaseModel
from pydantic.typing import Literal
class RedPill(BaseModel):
kind: Literal['red']
name: str
size: int
class BluePill(BaseModel):
kind: Literal['blue']
serialnumber: str
class Matrix(BaseModel):
year: int
pills: List[Union[RedPill, BluePill]] = []
if __name__ == "__main__":
pprint(Matrix.parse_obj({"year": 2021, "pills": [ {"kind": "blue", "serialnumber": "123"}, {"kind": "red", "name": "foo", "size": 100} ] }).dict())
## {'pills': [{'kind': 'blue', 'serialnumber': '123'},
{'kind': 'red', 'name': 'foo', 'size': 100}],
'year': 2021}