From d197164125e29b0452cbb8199ea0a609a3c7263a Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Wed, 3 Jan 2024 21:41:05 +0530
Subject: [PATCH] Fix #14176: Add AzureSQL dialect in lineage (#14563)

---
 ingestion/src/metadata/ingestion/lineage/models.py | 4 ++++
 1 file changed, 4 insertions(+)

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,
 }