mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-13 23:11:40 +00:00
refactor(bedrock): drop default encoding_format without mutating params dict
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
7bf50d5334
commit
fb092107f5
2 changed files with 3 additions and 3 deletions
|
|
@ -3545,14 +3545,14 @@ def get_optional_params_embeddings(
|
|||
else: # unmapped model
|
||||
supported_params = []
|
||||
if non_default_params.get("encoding_format") == "float":
|
||||
non_default_params.pop("encoding_format")
|
||||
non_default_params = {k: v for k, v in non_default_params.items() if k != "encoding_format"}
|
||||
_check_valid_arg(supported_params=supported_params)
|
||||
final_params = {**kwargs}
|
||||
return final_params
|
||||
|
||||
supported_params = object.get_supported_openai_params()
|
||||
if non_default_params.get("encoding_format") == "float" and "encoding_format" not in supported_params:
|
||||
non_default_params.pop("encoding_format")
|
||||
non_default_params = {k: v for k, v in non_default_params.items() if k != "encoding_format"}
|
||||
_check_valid_arg(supported_params=supported_params)
|
||||
optional_params = object.map_openai_params(non_default_params=non_default_params, optional_params={})
|
||||
elif custom_llm_provider == "mistral":
|
||||
|
|
|
|||
|
|
@ -4426,7 +4426,7 @@ class TestBedrockEmbeddingDefaultEncodingFormat:
|
|||
"some-unmapped-bedrock-model",
|
||||
],
|
||||
)
|
||||
def test_encoding_format_float_is_dropped(self, model):
|
||||
def test_encoding_format_float_is_dropped(self, model: str):
|
||||
optional_params = litellm.utils.get_optional_params_embeddings(
|
||||
model=model,
|
||||
encoding_format="float",
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue