From 6dfef2003f9d782c7bb2bf917dd95c7dce6259bf Mon Sep 17 00:00:00 2001
From: Maxim <maxim@cube.dev>
Date: Fri, 9 Jan 2026 16:25:19 -0800
Subject: [PATCH] fix(athena-driver): accept AWS assume role parameters as part
 of the (#10300)

config
---
 packages/cubejs-athena-driver/src/AthenaDriver.ts | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/cubejs-athena-driver/src/AthenaDriver.ts b/packages/cubejs-athena-driver/src/AthenaDriver.ts
index 7589b6b92eb7b..fd8d5f76da755 100644
--- a/packages/cubejs-athena-driver/src/AthenaDriver.ts
+++ b/packages/cubejs-athena-driver/src/AthenaDriver.ts
@@ -51,6 +51,8 @@ interface AthenaDriverOptions extends AthenaClientConfig {
   exportBucket?: string
   pollTimeout?: number
   pollMaxInterval?: number
+  athenaAwsAssumeRoleArn?: string
+  athenaAwsAssumeRoleExternalId?: string
 
   /**
    * The export bucket CSV file escape symbol.
@@ -125,8 +127,13 @@ export class AthenaDriver extends BaseDriver implements DriverInterface {
       config.secretAccessKey ||
       getEnv('athenaAwsSecret', { dataSource });
 
-    const assumeRoleArn = getEnv('athenaAwsAssumeRoleArn', { dataSource });
-    const assumeRoleExternalId = getEnv('athenaAwsAssumeRoleExternalId', { dataSource });
+    const assumeRoleArn =
+      config.athenaAwsAssumeRoleArn ||
+      getEnv('athenaAwsAssumeRoleArn', { dataSource });
+
+    const assumeRoleExternalId =
+      config.athenaAwsAssumeRoleExternalId ||
+      getEnv('athenaAwsAssumeRoleExternalId', { dataSource });
 
     const { schema, ...restConfig } = config;