@@ -1321,7 +1321,7 @@ def creator(
13211321class EntityDef (TypeDef , kw_only = True ):
13221322 """Entity type definition."""
13231323
1324- attribute_defs : List [Dict [ str , Any ] ] = msgspec .field (default_factory = list )
1324+ attribute_defs : List [AttributeDef ] = msgspec .field (default_factory = list )
13251325 """Unused."""
13261326
13271327 business_attribute_defs : Union [Dict [str , List [Dict [str , Any ]]], None ] = (
@@ -1332,7 +1332,7 @@ class EntityDef(TypeDef, kw_only=True):
13321332 category : AtlanTypeCategory = AtlanTypeCategory .ENTITY
13331333 """Type category for entity definitions."""
13341334
1335- relationship_attribute_defs : List [Dict [ str , Any ] ] = msgspec .field (
1335+ relationship_attribute_defs : List [RelationshipAttributeDef ] = msgspec .field (
13361336 default_factory = list
13371337 )
13381338 """Unused."""
@@ -1360,7 +1360,7 @@ def reserved_type(self) -> bool:
13601360class RelationshipDef (TypeDef , kw_only = True ):
13611361 """Relationship type definition."""
13621362
1363- attribute_defs : List [Dict [ str , Any ] ] = msgspec .field (default_factory = list )
1363+ attribute_defs : List [AttributeDef ] = msgspec .field (default_factory = list )
13641364 """Unused."""
13651365
13661366 category : AtlanTypeCategory = AtlanTypeCategory .RELATIONSHIP
@@ -1372,13 +1372,13 @@ class RelationshipDef(TypeDef, kw_only=True):
13721372 end_def2 : Union [Dict [str , Any ], None ] = msgspec .field (default_factory = dict )
13731373 """Unused."""
13741374
1375- propagate_tags : str = "ONE_TO_TWO"
1375+ propagate_tags : Union [ str , None ] = None
13761376 """Unused."""
13771377
1378- relationship_category : str = "AGGREGATION"
1378+ relationship_category : Union [ str , None ] = None
13791379 """Unused."""
13801380
1381- relationship_label : str = "__SalesforceOrganization.reports"
1381+ relationship_label : Union [ str , None ] = None
13821382 """Unused."""
13831383
13841384 service_type : Union [str , None ] = None
0 commit comments