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: