diff --git a/integration/common/openlineage/common/provider/dbt/processor.py b/integration/common/openlineage/common/provider/dbt/processor.py index 6887930cfb..c4955e06d2 100644 --- a/integration/common/openlineage/common/provider/dbt/processor.py +++ b/integration/common/openlineage/common/provider/dbt/processor.py @@ -40,6 +40,7 @@ class Adapter(Enum): DATABRICKS = "databricks" SQLSERVER = "sqlserver" DREMIO = "dremio" + ATHENA = "athena" @staticmethod def adapters() -> str: @@ -583,6 +584,8 @@ def extract_namespace(self, profile: Dict) -> str: return f"mssql://{profile['server']}:{profile['port']}" elif self.adapter_type == Adapter.DREMIO: return f"dremio://{profile['software_host']}:{profile['port']}" + elif self.adapter_type == Adapter.ATHENA: + return f"awsathena://athena.{profile['region_name']}.amazonaws.com" elif self.adapter_type == Adapter.SPARK: port = ""