Skip to content

Commit 44ffd7f

Browse files
committed
Fix mypy errors from error_handlers.py
1 parent bf21978 commit 44ffd7f

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

src/a2a/client/client_task_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ async def save_task_event(
9191
await self._save_task(event)
9292
return event
9393
task_id_from_event = (
94-
event.id if isinstance(event, Task) else event.taskId
94+
event.id if isinstance(event, Task) else event.task_id
9595
)
9696
if not self._task_id:
9797
self._task_id = task_id_from_event
@@ -114,7 +114,7 @@ async def save_task_event(
114114
if isinstance(event, TaskStatusUpdateEvent):
115115
logger.debug(
116116
'Updating task %s status to: %s',
117-
event.taskId,
117+
event.task_id,
118118
event.status.state,
119119
)
120120
if event.status.message:

src/a2a/utils/error_handlers.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,11 @@ async def wrapper(*args: Any, **kwargs: Any) -> Response:
6565
logger.log(
6666
log_level,
6767
'Request error: '
68-
f"Code={error.root.code}, Message='{error.root.message}'"
69-
f'{", Data=" + str(error.root.data) if error.root.data else ""}',
68+
f"Code={error.code}, Message='{error.message}'"
69+
f'{", Data=" + str(error.data) if error.data else ""}',
7070
)
7171
return JSONResponse(
72-
content={'message': error.root.message}, status_code=http_code
72+
content={'message': error.message}, status_code=http_code
7373
)
7474
except Exception as e:
7575
logger.log(logging.ERROR, f'Unknown error occurred {e}')
@@ -102,8 +102,8 @@ async def wrapper(*args: Any, **kwargs: Any) -> Any:
102102
logger.log(
103103
log_level,
104104
'Request error: '
105-
f"Code={error.root.code}, Message='{error.root.message}'"
106-
f'{", Data=" + str(error.root.data) if error.root.data else ""}',
105+
f"Code={error.code}, Message='{error.message}'"
106+
f'{", Data=" + str(error.data) if error.data else ""}',
107107
)
108108
# Since the stream has started, we can't return a JSONResponse.
109109
# Instead, we runt the error handling logic (provides logging)

0 commit comments

Comments
 (0)