This commit is contained in:
lwcobo 2025-04-24 00:56:22 -07:00 committed by GitHub
commit 374083dcac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3278,7 +3278,9 @@ async def aembedding(*args, **kwargs) -> EmbeddingResponse:
func_with_context = partial(ctx.run, func)
_, custom_llm_provider, _, _ = get_llm_provider(
model=model, api_base=kwargs.get("api_base", None)
model=model,
custom_llm_provider=kwargs.get("custom_llm_provider", None),
api_base=kwargs.get("api_base", None),
)
# Await normally