From 8cf4b9f142fde49117eb3dca47e006921eb9e099 Mon Sep 17 00:00:00 2001 From: Hasnaat Hussain Date: Sun, 30 Aug 2026 19:04:27 +0500 Subject: [PATCH] test: avoid global embedding setting mutation Signed-off-by: Hasnaat Hussain --- .../test_embedding_optional_params.py | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py b/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py index a5381f67c71..100d72baa74 100644 --- a/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py +++ b/tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py @@ -61,15 +61,11 @@ def test_embedding_dimensions_drop_params_for_openai_compatible_provider(provide ], ) def test_embedding_dimensions_preserved_for_provider_mappings(provider, model): - previous_drop_params = litellm.drop_params - try: - litellm.drop_params = True - optional_params = litellm.utils.get_optional_params_embeddings( - model=model, - custom_llm_provider=provider, - dimensions=128, - ) - finally: - litellm.drop_params = previous_drop_params + optional_params = litellm.utils.get_optional_params_embeddings( + model=model, + custom_llm_provider=provider, + dimensions=128, + drop_params=True, + ) assert optional_params["dimensions"] == 128