fix(ingest/snowflake): fix error on stored procs in non-SQL languages (#13127)

This commit is contained in:
Harshal Sheth 2025-04-08 17:00:56 -07:00 committed by GitHub
parent 742d060722
commit 1fca9855ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 1 deletions

View File

@ -26,6 +26,7 @@ from datahub.metadata.schema_classes import (
DataPlatformInstanceClass,
DataTransformClass,
DataTransformLogicClass,
QueryLanguageClass,
QueryStatementClass,
SubTypesClass,
)
@ -176,7 +177,17 @@ def _generate_job_workunits(
DataTransformClass(
queryStatement=QueryStatementClass(
value=procedure.procedure_definition,
language=procedure.language,
language=(
QueryLanguageClass.SQL
if procedure.language == "SQL"
# The language field uses a pretty limited enum.
# The "UNKNOWN" enum value is pretty new, so we don't want to
# emit it until it has broader server-side support. As a
# short-term solution, we map all languages to "SQL".
# TODO: Once we've released server 1.1.0, we should change
# this to be "UNKNOWN" for all languages except "SQL".
else QueryLanguageClass.SQL
),
),
)
]

View File

@ -17,5 +17,10 @@ record QueryStatement {
* A SQL Query
*/
SQL
/**
* Unknown query language
*/
UNKNOWN
} = "SQL"
}