From d5e866d839ed835078d2c412eb4c8f4a6312bf80 Mon Sep 17 00:00:00 2001 From: Samer Albahra Date: Mon, 5 May 2025 12:33:07 -0400 Subject: [PATCH] Fix missing api_config for embedding Ollama proxy endpoints --- backend/open_webui/routers/ollama.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/backend/open_webui/routers/ollama.py b/backend/open_webui/routers/ollama.py index 4e8c83e08..790f7dece 100644 --- a/backend/open_webui/routers/ollama.py +++ b/backend/open_webui/routers/ollama.py @@ -883,6 +883,10 @@ async def embed( ) url = request.app.state.config.OLLAMA_BASE_URLS[url_idx] + api_config = request.app.state.config.OLLAMA_API_CONFIGS.get( + str(url_idx), + request.app.state.config.OLLAMA_API_CONFIGS.get(url, {}), # Legacy support + ) key = get_api_key(url_idx, url, request.app.state.config.OLLAMA_API_CONFIGS) prefix_id = api_config.get("prefix_id", None) @@ -966,6 +970,10 @@ async def embeddings( ) url = request.app.state.config.OLLAMA_BASE_URLS[url_idx] + api_config = request.app.state.config.OLLAMA_API_CONFIGS.get( + str(url_idx), + request.app.state.config.OLLAMA_API_CONFIGS.get(url, {}), # Legacy support + ) key = get_api_key(url_idx, url, request.app.state.config.OLLAMA_API_CONFIGS) prefix_id = api_config.get("prefix_id", None)