Skip to content

Commit 9802cb0

Browse files
committed
fix: convert auth_required state in proto utils
Signed-off-by: Tomas Pilar <[email protected]>
1 parent 9193208 commit 9802cb0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/a2a/utils/proto_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,8 @@ def task_state(cls, state: types.TaskState) -> a2a_pb2.TaskState:
122122
return a2a_pb2.TaskState.TASK_STATE_FAILED
123123
case types.TaskState.input_required:
124124
return a2a_pb2.TaskState.TASK_STATE_INPUT_REQUIRED
125+
case types.TaskState.auth_required:
126+
return a2a_pb2.TaskState.TASK_STATE_AUTH_REQUIRED
125127
case _:
126128
return a2a_pb2.TaskState.TASK_STATE_UNSPECIFIED
127129

@@ -568,6 +570,8 @@ def task_state(cls, state: a2a_pb2.TaskState) -> types.TaskState:
568570
return types.TaskState.failed
569571
case a2a_pb2.TaskState.TASK_STATE_INPUT_REQUIRED:
570572
return types.TaskState.input_required
573+
case a2a_pb2.TaskState.TASK_STATE_AUTH_REQUIRED:
574+
return types.TaskState.auth_required
571575
case _:
572576
return types.TaskState.unknown
573577

0 commit comments

Comments
 (0)