From 32fe2cb73cc53cd64010b287da032d9ccbba9bd9 Mon Sep 17 00:00:00 2001
From: Hasnaat Hussain <hasnaat.hussain.2@gmail.com>
Date: Wed, 12 Aug 2026 23:58:52 +0500
Subject: [PATCH] test(utils): exercise embedding drop params in CI

Signed-off-by: Hasnaat Hussain <hasnaat.hussain.2@gmail.com>
---
 tests/llm_translation/test_optional_params.py | 32 -------------------
 tests/test_litellm/test_utils.py              | 32 +++++++++++++++++++
 2 files changed, 32 insertions(+), 32 deletions(-)

diff --git a/tests/llm_translation/test_optional_params.py b/tests/llm_translation/test_optional_params.py
index 017857f0f77..997f5b3b73f 100644
--- a/tests/llm_translation/test_optional_params.py
+++ b/tests/llm_translation/test_optional_params.py
@@ -1120,38 +1120,6 @@ def test_lm_studio_embedding_params():
     assert len(optional_params) == 0
 
 
-@pytest.mark.parametrize("provider", ["azure", "together_ai"])
-def test_embedding_dimensions_drop_params_for_openai_compatible_provider(provider):
-    previous_drop_params = litellm.drop_params
-    try:
-        litellm.drop_params = False
-        dropped = get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-            drop_params=True,
-        )
-        assert "dimensions" not in dropped
-
-        litellm.drop_params = True
-        dropped_globally = get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-        )
-        assert "dimensions" not in dropped_globally
-
-        litellm.drop_params = False
-        preserved = get_optional_params_embeddings(
-            model=f"{provider}/dummy-model",
-            custom_llm_provider=provider,
-            dimensions=512,
-        )
-        assert preserved["dimensions"] == 512
-    finally:
-        litellm.drop_params = previous_drop_params
-
-
 def test_ollama_pydantic_obj():
     from pydantic import BaseModel
 
diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 6524353aa48..aaccf28110b 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -733,6 +733,38 @@ def test_cohere_embedding_optional_params():
     assert optional_params is not None
 
 
+@pytest.mark.parametrize("provider", ["azure", "together_ai"])
+def test_embedding_dimensions_drop_params_for_openai_compatible_provider(provider):
+    previous_drop_params = litellm.drop_params
+    try:
+        litellm.drop_params = False
+        dropped = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+            drop_params=True,
+        )
+        assert "dimensions" not in dropped
+
+        litellm.drop_params = True
+        dropped_globally = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+        )
+        assert "dimensions" not in dropped_globally
+
+        litellm.drop_params = False
+        preserved = litellm.utils.get_optional_params_embeddings(
+            model=f"{provider}/dummy-model",
+            custom_llm_provider=provider,
+            dimensions=512,
+        )
+        assert preserved["dimensions"] == 512
+    finally:
+        litellm.drop_params = previous_drop_params
+
+
 def validate_model_cost_values(model_data, exceptions=None):
     """
     Validates that cost values in model data do not exceed 1.