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
)
if from_entity and to_entity:
yield AddLineageRequest(
edge=EntitiesEdge(
fromEntity=EntityReference(
id=from_entity.id.__root__,
type="table",
),
toEntity=EntityReference(
id=to_entity.id.__root__,
type="table",
),
lineageDetails=LineageDetails(
source=LineageSource.DbtLineage
),
yield Either(
right=AddLineageRequest(
edge=EntitiesEdge(
fromEntity=EntityReference(
id=from_entity.id.__root__,
type="table",
),
toEntity=EntityReference(
id=to_entity.id.__root__,
type="table",
),
lineageDetails=LineageDetails(
source=LineageSource.DbtLineage
),
)
)
)