Adding support for nested types, array and dates!

back/app/entity/schemas.py

@@ -1,9 +1,9 @@
 import uuid
 
 from datetime import datetime
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
-from .models import Entity, EntityType
+from .models import Entity, EntityType, Individual, Corporation
 from ..core.schemas import Writer
 
 
@@ -12,11 +12,12 @@ class EntityRead(Entity):
 
 
 class EntityCreate(Writer):
-    type: EntityType
     name: str
     address: str
+    entity_data: Individual | Corporation = Field(..., discriminator='type')
 
 
 class EntityUpdate(BaseModel):
     name: str
     address: str
+    entity_data: Individual | Corporation = Field(..., discriminator='type')