phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add VertexAITextEmbeddingConfig

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 12:53:43 -07:00
parent ea12519b98
commit 9f9b6e81a7
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -2621,8 +2621,11 @@ def get_optional_params_embeddings(
request_type="embeddings",
)
_check_valid_arg(supported_params=supported_params)
optional_params = litellm.VertexAITextEmbeddingConfig().map_openai_params(
non_default_params=non_default_params, optional_params={}
(
optional_params,
kwargs,
) = litellm.VertexAITextEmbeddingConfig().map_openai_params(
non_default_params=non_default_params, optional_params={}, kwargs=kwargs
)
final_params = {**optional_params, **kwargs}
return final_params