瀏覽代碼

fix: ollama prefix id

Timothy Jaeryang Baek 3 月之前
父節點
當前提交
bb45d35a36
共有 1 個文件被更改,包括 10 次插入0 次删除
  1. 10 0
      backend/open_webui/routers/ollama.py

+ 10 - 0
backend/open_webui/routers/ollama.py

@@ -1030,6 +1030,16 @@ async def generate_chat_completion(
     if ":" not in payload["model"]:
         payload["model"] = f"{payload['model']}:latest"
 
+    await get_all_models(request)
+    model = request.app.state.OLLAMA_MODELS.get(model_id)
+    if model:
+        url_idx = model["urls"].pop()
+    else:
+        raise HTTPException(
+            status_code=404,
+            detail="Model not found",
+        )
+
     url = await get_ollama_url(request, payload["model"], url_idx)
     api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
         str(url_idx),