scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

refactor(azure_ai): reuse the common embedding dispatch

Browse source
This commit is contained in:
Emerson Gomes 2026-11-03 21:35:38 +01:00
parent a9ea7aaae3
commit b04dc3b997
No known key found for this signature in database
GPG key ID: D3DF28AB5D1B5E17
1 changed files with 8 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
litellm/main.py
View file

@ -7280,7 +7280,13 @@ def embedding(
aembedding=aembedding,
drop_params=drop_unsupported_params,
)
elif custom_llm_provider == "oci":
elif custom_llm_provider == "oci" or (
custom_llm_provider == "azure_ai"
and ProviderConfigManager.get_provider_embedding_config(
model=model, provider=LlmProviders.AZURE_AI, api_base=api_base
)
is not None
):
response = base_llm_http_handler.embedding(
model=model,
input=input,
@ -7784,20 +7790,7 @@ def embedding(
or get_azure_ai_entra_token(litellm_params=litellm_params_dict)
)
azure_embedding_handler: Final = (
partial(
base_llm_http_handler.embedding,
custom_llm_provider=custom_llm_provider,
litellm_params=litellm_params_dict,
headers=headers,
)
if ProviderConfigManager.get_provider_embedding_config(
model=model, provider=LlmProviders.AZURE_AI, api_base=api_base
)
is not None
else azure_ai_embedding.embedding
)
response = azure_embedding_handler(
response = azure_ai_embedding.embedding(
model=model,
input=input,
api_base=api_base,