phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test - list batches

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-30 08:19:25 -07:00
parent 43a06f408c
commit 4206354ee7
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_openai_batches_and_files.py
View file

@ -72,6 +72,10 @@ def test_create_batch():
assert retrieved_batch.id == create_batch_response.id assert retrieved_batch.id == create_batch_response.id
# list all batches
list_batches = litellm.list_batches(custom_llm_provider="openai", limit=2)
print("list_batches=", list_batches)
file_content = litellm.file_content( file_content = litellm.file_content(
file_id=batch_input_file_id, custom_llm_provider="openai" file_id=batch_input_file_id, custom_llm_provider="openai"
) )
@ -140,6 +144,10 @@ async def test_async_create_batch():
assert retrieved_batch.id == create_batch_response.id assert retrieved_batch.id == create_batch_response.id
# list all batches
list_batches = await litellm.alist_batches(custom_llm_provider="openai", limit=2)
print("list_batches=", list_batches)
# try to get file content for our original file # try to get file content for our original file
file_content = await litellm.afile_content( file_content = await litellm.afile_content(