Преглед изворни кода

Fix missing api_config for embedding Ollama proxy endpoints

Samer Albahra пре 5 месеци
родитељ
комит
d5e866d839
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)