scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-24 00:52:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

test(utils): cover embedding dimensions in core-utils shard

Browse source 
Signed-off-by: Hasnaat Hussain <hasnaat.hussain.2@gmail.com>
This commit is contained in:
Hasnaat Hussain 2026-08-13 01:17:50 +05:00
parent 09fcacbaf9
commit 17cdd165cb
2 changed files with 51 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py Normal file
View file

@ -0,0 +1,51 @@
import pytest
import litellm
@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
litellm.drop_params = True
model_supported = litellm.utils.get_optional_params_embeddings(
model=f"{provider}/text-embedding-3-small",
custom_llm_provider=provider,
dimensions=512,
)
assert model_supported["dimensions"] == 512
explicitly_allowed = litellm.utils.get_optional_params_embeddings(
model=f"{provider}/legacy-model",
custom_llm_provider=provider,
dimensions=512,
allowed_openai_params=["dimensions"],
)
assert explicitly_allowed["dimensions"] == 512
finally:
litellm.drop_params = previous_drop_params

48
tests/test_litellm/test_utils.py
View file

@ -733,54 +733,6 @@ 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
litellm.drop_params = True
model_supported = litellm.utils.get_optional_params_embeddings(
model=f"{provider}/text-embedding-3-small",
custom_llm_provider=provider,
dimensions=512,
)
assert model_supported["dimensions"] == 512
explicitly_allowed = litellm.utils.get_optional_params_embeddings(
model=f"{provider}/legacy-model",
custom_llm_provider=provider,
dimensions=512,
allowed_openai_params=["dimensions"],
)
assert explicitly_allowed["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.