Quellcode durchsuchen

fix: local/external models

Timothy Jaeryang Baek vor 4 Monaten
Ursprung
Commit
4c55ad71a7

+ 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