fixed either in lineage req (#13088)

This commit is contained in:
Onkar Ravgan 2023-09-05 18:18:16 +05:30 committed by GitHub
parent 80c11ab4f5
commit 9af14a5a1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -606,19 +606,21 @@ class DbtSource(DbtServiceSource):
entity_list=from_es_result, fetch_multiple_entities=False entity_list=from_es_result, fetch_multiple_entities=False
) )
if from_entity and to_entity: if from_entity and to_entity:
yield AddLineageRequest( yield Either(
edge=EntitiesEdge( right=AddLineageRequest(
fromEntity=EntityReference( edge=EntitiesEdge(
id=from_entity.id.__root__, fromEntity=EntityReference(
type="table", id=from_entity.id.__root__,
), type="table",
toEntity=EntityReference( ),
id=to_entity.id.__root__, toEntity=EntityReference(
type="table", id=to_entity.id.__root__,
), type="table",
lineageDetails=LineageDetails( ),
source=LineageSource.DbtLineage lineageDetails=LineageDetails(
), source=LineageSource.DbtLineage
),
)
) )
) )