From a3a652a85f5cd78e31bc28687e2f7ef29a3e6494 Mon Sep 17 00:00:00 2001
From: Biplob Biswas <biplob.biswas@holidu.com>
Date: Thu, 21 Nov 2024 18:12:46 +0100
Subject: [PATCH] Add indication for retrieving schema metadata for a list of
 relations and not all

---
 dbt-athena/src/dbt/adapters/athena/impl.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/dbt-athena/src/dbt/adapters/athena/impl.py b/dbt-athena/src/dbt/adapters/athena/impl.py
index 5b94a4c9..c0211bb2 100755
--- a/dbt-athena/src/dbt/adapters/athena/impl.py
+++ b/dbt-athena/src/dbt/adapters/athena/impl.py
@@ -65,6 +65,7 @@
 from dbt.adapters.base import ConstraintSupport, PythonJobHelper, available
 from dbt.adapters.base.impl import AdapterConfig
 from dbt.adapters.base.relation import BaseRelation, InformationSchema
+from dbt.adapters.capability import Capability, CapabilityDict, CapabilitySupport, Support
 from dbt.adapters.contracts.connection import AdapterResponse
 from dbt.adapters.contracts.relation import RelationConfig
 from dbt.adapters.sql import SQLAdapter
@@ -146,6 +147,13 @@ class AthenaAdapter(SQLAdapter):
         ConstraintType.foreign_key: ConstraintSupport.NOT_SUPPORTED,
     }
 
+    _capabilities = CapabilityDict(
+        {
+            Capability.SchemaMetadataByRelations: CapabilitySupport(support=Support.Full),
+            Capability.TableLastModifiedMetadata: CapabilitySupport(support=Support.Unknown),
+        }
+    )
+
     @classmethod
     def date_function(cls) -> str:
         return "now()"