fix proxy connection pool creation

databricks · Jun 23, 2023 · 1b74087 · 1b74087
1 parent 54e3769
commit 1b74087
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/databricks/sql/auth/thrift_http_client.py b/src/databricks/sql/auth/thrift_http_client.py
@@ -109,7 +109,7 @@ def open(self):
                 headers={"Proxy-Authorization": self.proxy_auth},
             )
             self.__pool = proxy_manager.connection_from_host(
-                self.host, self.port, pool_kwargs=_pool_kwargs
+                self.realhost, self.realport, pool_kwargs=_pool_kwargs, scheme=self.scheme
             )
         else:
             self.__pool = pool_class(self.host, self.port, **_pool_kwargs)