Skip to content

Commit c7d5d4b

Browse files
Hopefully final fix to databricks volume
1 parent 1328f39 commit c7d5d4b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

CHANGELOG.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
## [1.2.21]
22

33
* **fix: Enforce minimum version of databricks-sdk (>=0.62.0) for databricks-volumes connector**
4-
* **fix: Update databricks-volumes connector to use correct API parameter (data instead of contents) for files.upload() method in databricks-sdk >=0.62.0**
4+
* **fix: Update databricks-volumes connector to wrap file contents in io.BytesIO for compatibility with files.upload() BinaryIO requirement in databricks-sdk >=0.62.0**
55
* **fix: Add constraints to prevent platform compatibility issues with tritonclient/perf-analyzer dependencies**
66

77
## [1.2.20]

unstructured_ingest/processes/connectors/databricks/volumes.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import io
12
import os
23
from abc import ABC
34
from dataclasses import dataclass
@@ -220,9 +221,12 @@ def run(self, path: Path, file_data: FileData, **kwargs: Any) -> None:
220221
output_path = self.get_output_path(file_data=file_data)
221222
with open(path, "rb") as elements_file:
222223
try:
224+
# Read file bytes and wrap in BytesIO to create BinaryIO object
225+
file_bytes = elements_file.read()
226+
binary_data = io.BytesIO(file_bytes)
223227
self.connection_config.get_client().files.upload(
224228
file_path=output_path,
225-
data=elements_file,
229+
contents=binary_data,
226230
overwrite=True,
227231
)
228232
except Exception as e:

0 commit comments

Comments
 (0)