Define Document from Pydantic BaseModel

[z0z0r4]

I import BaseModel A, and want to define a Document class B, which can inherit all attributes of A.

Can I directly define B like this example?

# A should be imported from another package
class A(BaseModel):
    name: str

class B(Document, A):
    pass

# do some db operation with B