scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test: avoid global embedding setting mutation

Browse source 
Signed-off-by: Hasnaat Hussain <hasnaat.hussain.2@gmail.com>
This commit is contained in:
Hasnaat Hussain 2026-08-30 19:04:27 +05:00
parent 3fab69dd31
commit 8cf4b9f142
1 changed files with 6 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/litellm_core_utils/test_embedding_optional_params.py
View file

@ -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