[PECO-2051] Add custom auth headers into cloud fetch request

databricks · Oct 25, 2024 · 6d65bf7 · 6d65bf7
1 parent 55105fe
commit 6d65bf7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/examples/data_types.py b/examples/data_types.py
diff --git 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()