Skip to content

Commit 647935e

Browse files
authored
fix: task state from/to proto missing reject
1 parent cb7cdb3 commit 647935e

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/a2a/utils/proto_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,8 @@ def task_state(cls, state: types.TaskState) -> a2a_pb2.TaskState:
230230
return a2a_pb2.TaskState.TASK_STATE_INPUT_REQUIRED
231231
case types.TaskState.auth_required:
232232
return a2a_pb2.TaskState.TASK_STATE_AUTH_REQUIRED
233+
case types.TaskState.rejected:
234+
return a2a_pb2.TaskState.TASK_STATE_REJECTED
233235
case _:
234236
return a2a_pb2.TaskState.TASK_STATE_UNSPECIFIED
235237

@@ -703,6 +705,8 @@ def task_state(cls, state: a2a_pb2.TaskState) -> types.TaskState:
703705
return types.TaskState.input_required
704706
case a2a_pb2.TaskState.TASK_STATE_AUTH_REQUIRED:
705707
return types.TaskState.auth_required
708+
case a2a_pb2.TaskState.TASK_STATE_REJECTED:
709+
return types.TaskState.rejected
706710
case _:
707711
return types.TaskState.unknown
708712

0 commit comments

Comments
 (0)