Skip to content

Commit 27f1b0b

Browse files
authored
Publish file and image prep convenience APIs (#39)
Part of #34
1 parent 972f35b commit 27f1b0b

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/lmstudio/sync_api.py

+2
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,8 @@
147147
"list_downloaded_models",
148148
"list_loaded_models",
149149
"llm",
150+
"prepare_file",
151+
"prepare_image",
150152
]
151153

152154

tests/test_convenience_api.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -51,15 +51,15 @@ def test_embedding_specific() -> None:
5151
def test_prepare_file() -> None:
5252
name = "example-file.txt"
5353
raw_data = b"raw data"
54-
file_handle = lms.sync_api.prepare_file(raw_data, name)
54+
file_handle = lms.prepare_file(raw_data, name)
5555
assert file_handle.name == name
5656
assert file_handle.size_bytes == len(raw_data)
5757
assert file_handle.file_type == "text/plain"
5858

5959

6060
@pytest.mark.lmstudio
6161
def test_prepare_image() -> None:
62-
file_handle = lms.sync_api.prepare_image(IMAGE_FILEPATH)
62+
file_handle = lms.prepare_image(IMAGE_FILEPATH)
6363
assert file_handle.name == IMAGE_FILEPATH.name
6464
assert file_handle.size_bytes == len(IMAGE_FILEPATH.read_bytes())
6565
assert file_handle.file_type == "image"

0 commit comments

Comments
 (0)