diff --git a/ingestion/src/metadata/ingestion/source/mssql.py b/ingestion/src/metadata/ingestion/source/mssql.py index f777d62024c..98b086e26ea 100644 --- a/ingestion/src/metadata/ingestion/source/mssql.py +++ b/ingestion/src/metadata/ingestion/source/mssql.py @@ -16,6 +16,9 @@ from metadata.generated.schema.entity.services.connections.database.mssqlConnect from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import ( OpenMetadataConnection, ) +from metadata.generated.schema.metadataIngestion.databaseServiceMetadataPipeline import ( + DatabaseServiceMetadataPipeline, +) from metadata.generated.schema.metadataIngestion.workflow import ( Source as WorkflowSource, ) @@ -41,4 +44,8 @@ class MssqlSource(SQLSource): raise InvalidSourceException( f"Expected MssqlConnection, but got {connection}" ) + if config.sourceConfig.config.sampleDataQuery == "select * from {}.{} limit 50": + config.sourceConfig.config.sampleDataQuery = ( + "select top 50 * from [{}].[{}]" + ) return cls(config, metadata_config)