Skip to content

Commit f4ac109

Browse files
committed
fix: Add referenceTaskIds handling in ProtoUtils.java
1 parent d7e80de commit f4ac109

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,9 @@ public static io.a2a.grpc.Message message(Message message) {
164164
builder.addAllParts(message.getParts().stream().map(ToProto::part).collect(Collectors.toList()));
165165
}
166166
builder.setMetadata(struct(message.getMetadata()));
167+
if (message.getReferenceTaskIds() != null && !message.getReferenceTaskIds().isEmpty()) {
168+
builder.addAllReferenceTaskIds(message.getReferenceTaskIds());
169+
}
167170
return builder.build();
168171
}
169172

@@ -875,7 +878,7 @@ public static Message message(io.a2a.grpc.MessageOrBuilder message) {
875878
message.getMessageId().isEmpty() ? null : message.getMessageId(),
876879
message.getContextId().isEmpty() ? null : message.getContextId(),
877880
message.getTaskId().isEmpty() ? null : message.getTaskId(),
878-
null, // referenceTaskIds is not in grpc message
881+
message.getReferenceTaskIdsList().isEmpty() ? null : message.getReferenceTaskIdsList(),
879882
struct(message.getMetadata())
880883
);
881884
}

0 commit comments

Comments
 (0)