Skip to content

Commit 75d5bc1

Browse files
correct get chunk info endpoint
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent c372788 commit 75d5bc1

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

src/databricks/sql/backend/sea/backend.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -320,12 +320,9 @@ def get_chunk_links(
320320
"""
321321
from databricks.sql.backend.sea.models.responses import GetChunksResponse
322322

323-
params = {"chunk_index": chunk_index}
324-
325323
response_data = self.http_client._make_request(
326324
method="GET",
327-
path=self.CHUNKS_PATH_WITH_ID.format(statement_id),
328-
params=params,
325+
path=self.CHUNK_PATH_WITH_ID_AND_INDEX.format(statement_id, chunk_index),
329326
)
330327

331328
return GetChunksResponse.from_dict(response_data)

src/databricks/sql/cloud_fetch_queue.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def _fetch_chunk_link(self, chunk_index: int) -> Optional["ExternalLink"]:
360360

361361
# Use the SEA client to fetch the chunk links
362362
chunk_info = self._sea_client.get_chunk_links(self._statement_id, chunk_index)
363-
links = chunk_info.links
363+
links = chunk_info.external_links
364364

365365
if not links:
366366
logger.debug(

0 commit comments

Comments
 (0)