Commit 27f1b0b 1 parent 972f35b commit 27f1b0b Copy full SHA for 27f1b0b
File tree 2 files changed +4
-2
lines changed
2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 147
147
"list_downloaded_models" ,
148
148
"list_loaded_models" ,
149
149
"llm" ,
150
+ "prepare_file" ,
151
+ "prepare_image" ,
150
152
]
151
153
152
154
Original file line number Diff line number Diff line change @@ -51,15 +51,15 @@ def test_embedding_specific() -> None:
51
51
def test_prepare_file () -> None :
52
52
name = "example-file.txt"
53
53
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 )
55
55
assert file_handle .name == name
56
56
assert file_handle .size_bytes == len (raw_data )
57
57
assert file_handle .file_type == "text/plain"
58
58
59
59
60
60
@pytest .mark .lmstudio
61
61
def test_prepare_image () -> None :
62
- file_handle = lms .sync_api . prepare_image (IMAGE_FILEPATH )
62
+ file_handle = lms .prepare_image (IMAGE_FILEPATH )
63
63
assert file_handle .name == IMAGE_FILEPATH .name
64
64
assert file_handle .size_bytes == len (IMAGE_FILEPATH .read_bytes ())
65
65
assert file_handle .file_type == "image"
You can’t perform that action at this time.
0 commit comments