Error: UnionDocNotInited

[mg3146]

Hi,

I'm getting an UnionDocNotInited error trying to launch a connection to the db. My setup looks something like the below:

class TradeBase(Document):       
    # Document-Admin Shared Fields
    last_update_datetime : dt.datetime = Field(default_factory=dt.datetime.now)
    trade_id : Indexed(UUID, unique = True) = Field(default_factory=trade_id)
    document_version : int = 1
    trade_status : bool = True    
        
class Trade(UnionDoc):    
    class Settings:
        name = 'Blotter_Test'
    

class Swaps(TradeBase):            
    def __init__(self,  **kwargs):
        super().__init__(**kwargs)
              
    class Settings:
        union_doc = Trade

class Swaps_Two(TradeBase):            
    def __init__(self,  **kwargs):
        super().__init__(**kwargs)
              
    class Settings:
        union_doc = Trade



async def blotter_connection():
    # Beanie uses Motor async client under the hood 
    client = AsyncIOMotorClient(conn_string)

    # Initialize beanie with the Product document class
    await init_beanie(database=client.db_name, document_models=[Swaps, Swaps_Two])

In IPython:

await blotter_connection()

 raise UnionDocNotInited

UnionDocNotInited

Any help is appreciated!!

[Vitalium]

In this example UnionDoc is also need to be added to document_models :

await init_beanie(database=client.db_name, document_models=[Trade, Swaps, Swaps_Two])

TradeBase in this case seems should be BaseModel type.