From d3cb62c4c4c0a7dc955a02504d36e506e48cf0e4 Mon Sep 17 00:00:00 2001
From: Jacky Hu <jacky.hu@databricks.com>
Date: Fri, 25 Oct 2024 09:41:40 -0700
Subject: [PATCH] [PECO-2051] Add custom auth headers into cloud fetch request
 (#460)

Signed-off-by: Jacky Hu <jacky.hu@databricks.com>
---
 src/databricks/sql/cloudfetch/downloader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/databricks/sql/cloudfetch/downloader.py b/src/databricks/sql/cloudfetch/downloader.py
index 03c70054..228e07d6 100644
--- a/src/databricks/sql/cloudfetch/downloader.py
+++ b/src/databricks/sql/cloudfetch/downloader.py
@@ -100,6 +100,7 @@ def run(self) -> DownloadedFile:
                 self.link.fileLink,
                 timeout=self.settings.download_timeout,
                 verify=self._ssl_options.tls_verify,
+                headers=self.link.httpHeaders
                 # TODO: Pass cert from `self._ssl_options`
             )
             response.raise_for_status()