From 486037ff8b0cb7dcf7a90363b999130ead94c8db Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 23 Nov 2023 21:16:35 -0800 Subject: [PATCH] (fix) proxy: /embeddings --- litellm/proxy/proxy_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 09956660b1..17f703fe69 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -625,8 +625,8 @@ async def embeddings(request: Request): router_model_names = [m["model_name"] for m in llm_model_list] if llm_model_list is not None else [] if llm_router is not None and data["model"] in router_model_names: # model in router model list response = await llm_router.aembedding(**data) - else: - response = litellm.aembedding(**data) + else: + response = await litellm.aembedding(**data) return response except Exception as e: traceback.print_exc()