diff --git a/src/a2a/server/tasks/database_task_store.py b/src/a2a/server/tasks/database_task_store.py index c677b8561..44887aa35 100644 --- a/src/a2a/server/tasks/database_task_store.py +++ b/src/a2a/server/tasks/database_task_store.py @@ -1,7 +1,6 @@ import logging from datetime import datetime, timezone -from typing import Any, cast try: @@ -146,25 +145,17 @@ def _from_orm(self, task_model: TaskModel) -> Task: context_id=task_model.context_id, ) if task_model.status: - ParseDict( - cast('dict[str, Any]', task_model.status), task.status - ) + ParseDict(task_model.status, task.status) if task_model.artifacts: - for art_dict in cast( - 'list[dict[str, Any]]', task_model.artifacts - ): + for art_dict in task_model.artifacts: art = task.artifacts.add() ParseDict(art_dict, art) if task_model.history: - for msg_dict in cast( - 'list[dict[str, Any]]', task_model.history - ): + for msg_dict in task_model.history: msg = task.history.add() ParseDict(msg_dict, msg) if task_model.task_metadata: - task.metadata.update( - cast('dict[str, Any]', task_model.task_metadata) - ) + task.metadata.update(task_model.task_metadata) return task # Legacy conversion