Skip to content

Commit 68657a3

Browse files
remove un-needed GetChunksResponse
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 030edf8 commit 68657a3

File tree

2 files changed

+0
-36
lines changed

2 files changed

+0
-36
lines changed

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

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
if TYPE_CHECKING:
1111
from databricks.sql.client import Cursor
1212
from databricks.sql.result_set import ResultSet
13-
from databricks.sql.backend.sea.models.responses import GetChunksResponse
1413

1514
from databricks.sql.backend.databricks_client import DatabricksClient
1615
from databricks.sql.backend.types import (

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

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -196,38 +196,3 @@ class CreateSessionResponse:
196196
def from_dict(cls, data: Dict[str, Any]) -> "CreateSessionResponse":
197197
"""Create a CreateSessionResponse from a dictionary."""
198198
return cls(session_id=data.get("session_id", ""))
199-
200-
201-
@dataclass
202-
class GetChunksResponse:
203-
"""Response from getting chunks for a statement."""
204-
205-
statement_id: str
206-
external_links: List[ExternalLink]
207-
208-
@classmethod
209-
def from_dict(cls, data: Dict[str, Any]) -> "GetChunksResponse":
210-
"""Create a GetChunksResponse from a dictionary."""
211-
external_links = []
212-
if "external_links" in data:
213-
for link_data in data["external_links"]:
214-
external_links.append(
215-
ExternalLink(
216-
external_link=link_data.get("external_link", ""),
217-
expiration=link_data.get("expiration", ""),
218-
chunk_index=link_data.get("chunk_index", 0),
219-
byte_count=link_data.get("byte_count", 0),
220-
row_count=link_data.get("row_count", 0),
221-
row_offset=link_data.get("row_offset", 0),
222-
next_chunk_index=link_data.get("next_chunk_index"),
223-
next_chunk_internal_link=link_data.get(
224-
"next_chunk_internal_link"
225-
),
226-
http_headers=link_data.get("http_headers"),
227-
)
228-
)
229-
230-
return cls(
231-
statement_id=data.get("statement_id", ""),
232-
external_links=external_links,
233-
)

0 commit comments

Comments
 (0)