fix: local/external models

This commit is contained in:
Timothy Jaeryang Baek 2025-05-23 02:48:31 +04:00
parent 1f632d3570
commit 4c55ad71a7
2 changed files with 2 additions and 2 deletions

View File

@ -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:

View File

@ -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