Skip to content

Commit 615f608

Browse files
committed
refactor: use model_validate for legacy task conversion to improve type handling and Pyright compatibility.
1 parent 42aada9 commit 615f608

1 file changed

Lines changed: 10 additions & 7 deletions

File tree

src/a2a/server/tasks/database_task_store.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -168,13 +168,16 @@ def _from_orm(self, task_model: TaskModel) -> Task:
168168
return task
169169

170170
# Legacy conversion
171-
legacy_task = types_v03.Task(
172-
id=task_model.id,
173-
context_id=task_model.context_id,
174-
status=task_model.status,
175-
artifacts=task_model.artifacts or [],
176-
history=task_model.history or [],
177-
metadata=task_model.task_metadata or {},
171+
# Reconstruct legacy task using model_validate to handle dicts and resolve Pyright issues
172+
legacy_task = types_v03.Task.model_validate(
173+
{
174+
'id': task_model.id,
175+
'context_id': task_model.context_id,
176+
'status': task_model.status,
177+
'artifacts': task_model.artifacts,
178+
'history': task_model.history,
179+
'metadata': task_model.task_metadata,
180+
}
178181
)
179182
return conversions.to_core_task(legacy_task)
180183

0 commit comments

Comments
 (0)