From 4206354ee79e4b880807ab2c857b119c07dbcb4d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 30 Jul 2024 08:19:25 -0700 Subject: [PATCH] test - list batches --- litellm/tests/test_openai_batches_and_files.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/tests/test_openai_batches_and_files.py b/litellm/tests/test_openai_batches_and_files.py index 14aee6e19..e8bde4d20 100644 --- a/litellm/tests/test_openai_batches_and_files.py +++ b/litellm/tests/test_openai_batches_and_files.py @@ -72,6 +72,10 @@ def test_create_batch(): 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_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 + # 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 file_content = await litellm.afile_content(