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)