瀏覽代碼

fix: local/external models

Timothy Jaeryang Baek 4 月之前
父節點
當前提交
4c55ad71a7
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      backend/open_webui/utils/task.py
  2. 1 1
      src/lib/components/chat/ModelSelector/Selector.svelte

+ 1 - 1
backend/open_webui/utils/task.py

@@ -22,7 +22,7 @@ def get_task_model_id(
     # Set the task model
     task_model_id = default_model_id
     # Check if the user has a custom task model and use that model
-    if models[task_model_id].get("owned_by") == "ollama":
+    if models[task_model_id].get("connection_type") == "local":
         if task_model and task_model in models:
             task_model_id = task_model
     else:

+ 1 - 1
src/lib/components/chat/ModelSelector/Selector.svelte

@@ -557,7 +557,7 @@
 											</svg>
 										</div>
 									</Tooltip>
-								{:else if item.model.owned_by === 'openai'}
+								{:else if item.model.connection_type === 'external'}
 									<Tooltip content={`${'External'}`}>
 										<div class="translate-y-[1px]">
 											<svg