Skip to content

Commit 4154215

Browse files
committed
fix: issue 226
Checking if the TaskUpdate has an update message before trying to convert it. Signed-off-by: Emmanuel Hugonnet <[email protected]>
1 parent 77b0c7b commit 4154215

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

spec-grpc/src/main/java/io/a2a/grpc/utils/ProtoUtils.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -822,9 +822,12 @@ private static DataPart dataPart(io.a2a.grpc.DataPart dataPart) {
822822
}
823823

824824
private static TaskStatus taskStatus(io.a2a.grpc.TaskStatus taskStatus) {
825-
return new TaskStatus(
826-
taskState(taskStatus.getState()),
827-
message(taskStatus.getUpdate()),
825+
TaskState state = taskState(taskStatus.getState());
826+
if (state == null) {
827+
return null;
828+
}
829+
return new TaskStatus(state,
830+
taskStatus.hasUpdate() ? message(taskStatus.getUpdate()) : null,
828831
LocalDateTime.ofInstant(Instant.ofEpochSecond(taskStatus.getTimestamp().getSeconds(), taskStatus.getTimestamp().getNanos()), ZoneOffset.UTC)
829832
);
830833
}

0 commit comments

Comments
 (0)