fix: Fix the error that occurs when the task model does not exist (#23169)

This commit is contained in:
Shirasawa 2026-04-01 20:01:25 +08:00 • committed by GitHub
parent 60676bfdcf
commit 3f7fc1a75a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -19,7 +19,7 @@ def get_task_model_id(default_model_id: str, task_model: str, task_model_externa
# 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('connection_type') == 'local':
if models.get(task_model_id, {}).get('connection_type') == 'local':
if task_model and task_model in models:
task_model_id = task_model
else: