From f18754b6ed4e1c238fb1038029f717f3f2d4a8b5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 10 Jul 2024 15:42:15 -0700 Subject: [PATCH] test - delete file --- litellm/tests/test_openai_batches.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/litellm/tests/test_openai_batches.py b/litellm/tests/test_openai_batches.py index 91357f7631..05a81a50e6 100644 --- a/litellm/tests/test_openai_batches.py +++ b/litellm/tests/test_openai_batches.py @@ -155,6 +155,15 @@ async def test_async_create_batch(): print("file obj = ", file_obj) assert file_obj.id == batch_input_file_id + # delete file + delete_file_response = await litellm.afile_delete( + file_id=batch_input_file_id, custom_llm_provider="openai" + ) + + print("delete file response = ", delete_file_response) + + assert delete_file_response.id == batch_input_file_id + # # write this file content to a file # with open("file_content.json", "w") as f: # json.dump(file_content, f)