diff --git a/ingestion/src/metadata/ingestion/lineage/models.py b/ingestion/src/metadata/ingestion/lineage/models.py index fce558fe19f..419c788b747 100644 --- a/ingestion/src/metadata/ingestion/lineage/models.py +++ b/ingestion/src/metadata/ingestion/lineage/models.py @@ -17,6 +17,9 @@ from typing import Dict from metadata.generated.schema.entity.services.connections.database.athenaConnection import ( AthenaType, ) +from metadata.generated.schema.entity.services.connections.database.azureSQLConnection import ( + AzureSQLType, +) from metadata.generated.schema.entity.services.connections.database.bigQueryConnection import ( BigqueryType, ) @@ -105,6 +108,7 @@ MAP_CONNECTION_TYPE_DIALECT: Dict[str, Dialect] = { str(DeltaLakeType.DeltaLake.value): Dialect.SPARKSQL, str(SQLiteType.SQLite.value): Dialect.SQLITE, str(MssqlType.Mssql.value): Dialect.TSQL, + str(AzureSQLType.AzureSQL.value): Dialect.TSQL, }