From 32e8bdef6fa4db071177dabf824bee6d41b3ecf0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 28 Dec 2024 19:45:12 -0800 Subject: [PATCH] update clean up jobs --- docs/my-website/release_notes/v1.56.4/index.md | 13 +++++++++++++ .../batches_tests/test_openai_batches_and_files.py | 11 +++++------ 2 files changed, 18 insertions(+), 6 deletions(-) create mode 100644 docs/my-website/release_notes/v1.56.4/index.md diff --git a/docs/my-website/release_notes/v1.56.4/index.md b/docs/my-website/release_notes/v1.56.4/index.md new file mode 100644 index 0000000000..f5bc9d84c0 --- /dev/null +++ b/docs/my-website/release_notes/v1.56.4/index.md @@ -0,0 +1,13 @@ + +## New Features + + +## Dependency Upgrades + +- (Security fix) Upgrade to `fastapi==0.115.5` https://github.com/BerriAI/litellm/pull/7447 + +## Docs + + + + diff --git a/tests/batches_tests/test_openai_batches_and_files.py b/tests/batches_tests/test_openai_batches_and_files.py index e5c36468d5..447126f231 100644 --- a/tests/batches_tests/test_openai_batches_and_files.py +++ b/tests/batches_tests/test_openai_batches_and_files.py @@ -26,6 +26,7 @@ verbose_logger.setLevel(logging.DEBUG) from litellm.integrations.custom_logger import CustomLogger from litellm.types.utils import StandardLoggingPayload +import random def load_vertex_ai_credentials(): @@ -296,8 +297,10 @@ async def test_async_create_batch(provider): ) print("cancel_batch_response=", cancel_batch_response) - cleanup_azure_files() - cleanup_azure_ft_models() + if random.randint(1, 20) == 1: + print("Running random cleanup of Azure files and models...") + cleanup_azure_files() + cleanup_azure_ft_models() def cleanup_azure_files(): @@ -363,10 +366,6 @@ def cleanup_azure_ft_models(): print(f"Error on cleanup_azure_ft_models: {str(e)}") -# # cleanup_azure_files() -# cleanup_azure_ft_models() - - @pytest.mark.asyncio async def test_avertex_batch_prediction(): load_vertex_ai_credentials()