浏览代码

Merge pull request #13522 from salbahra/bug/fix-missing-api-config

fix: Correct missing `api_config` for embedding Ollama proxy endpoints
Tim Jaeryang Baek 5 月之前
父节点
当前提交
0a8199a056
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      backend/open_webui/routers/ollama.py

+ 8 - 0
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)