Skip to content

Commit e81b35a

Browse files
authored
fix: Add additional methods in TaskUpdater (#378)
Small updates to TaskUpdater that I found useful while working on #338
1 parent c9fe654 commit e81b35a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

server-common/src/main/java/io/a2a/server/tasks/TaskUpdater.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,11 @@ private void updateStatus(TaskState taskState) {
3232
updateStatus(taskState, null, taskState.isFinal());
3333
}
3434

35-
private void updateStatus(TaskState taskState, Message message) {
35+
public void updateStatus(TaskState taskState, Message message) {
3636
updateStatus(taskState, message, taskState.isFinal());
3737
}
3838

39-
private void updateStatus(TaskState state, Message message, boolean isFinal) {
39+
public void updateStatus(TaskState state, Message message, boolean isFinal) {
4040
synchronized (stateLock) {
4141
// Check if we're already in a terminal state
4242
if (terminalStateReached.get()) {
@@ -62,6 +62,10 @@ public String getContextId() {
6262
return this.contextId;
6363
}
6464

65+
public String getTaskId() {
66+
return this.taskId;
67+
}
68+
6569
public void addArtifact(List<Part<?>> parts) {
6670
addArtifact(parts, null, null, null);
6771
}

0 commit comments

Comments
 (0)