Skip to content

Commit cd34a4e

Browse files
kabirkabir
committed
feat!: Convert spec classes to records
This became apparent since in non-Quarkus environments you have to set a bunch of JVM flags, and also introduce default constructors, which in turn means making fields non-final. Additionally, added default constructors for some CDI beans to work in stricter CDI environments than Quarkus
1 parent 0b867d0 commit cd34a4e

51 files changed

Lines changed: 925 additions & 1198 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎compat-0.3/client/base/src/main/java/org/a2aproject/sdk/compat03/client/ClientTaskManager_v0_3.java‎

Lines changed: 18 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,10 @@ public Task_v0_3 saveTaskEvent(Task_v0_3 task) throws A2AClientInvalidArgsError_
5252

5353
public Task_v0_3 saveTaskEvent(TaskStatusUpdateEvent_v0_3 taskStatusUpdateEvent) throws A2AClientError_v0_3 {
5454
if (taskId == null) {
55-
taskId = taskStatusUpdateEvent.getTaskId();
55+
taskId = taskStatusUpdateEvent.taskId();
5656
}
5757
if (contextId == null) {
58-
contextId = taskStatusUpdateEvent.getContextId();
58+
contextId = taskStatusUpdateEvent.contextId();
5959
}
6060
Task_v0_3 task = currentTask;
6161
if (task == null) {
@@ -67,31 +67,27 @@ public Task_v0_3 saveTaskEvent(TaskStatusUpdateEvent_v0_3 taskStatusUpdateEvent)
6767
}
6868

6969
Task_v0_3.Builder taskBuilder = new Task_v0_3.Builder(task);
70-
if (taskStatusUpdateEvent.getStatus().message() != null) {
71-
if (task.getHistory() == null) {
72-
taskBuilder.history(taskStatusUpdateEvent.getStatus().message());
73-
} else {
74-
List<Message_v0_3> history = new ArrayList<>(task.getHistory());
75-
history.add(taskStatusUpdateEvent.getStatus().message());
76-
taskBuilder.history(history);
77-
}
70+
if (taskStatusUpdateEvent.status().message() != null) {
71+
List<Message_v0_3> history = new ArrayList<>(task.history());
72+
history.add(taskStatusUpdateEvent.status().message());
73+
taskBuilder.history(history);
7874
}
79-
if (taskStatusUpdateEvent.getMetadata() != null) {
80-
Map<String, Object> newMetadata = task.getMetadata() != null ? new HashMap<>(task.getMetadata()) : new HashMap<>();
81-
newMetadata.putAll(taskStatusUpdateEvent.getMetadata());
75+
if (taskStatusUpdateEvent.metadata() != null) {
76+
Map<String, Object> newMetadata = task.metadata() != null ? new HashMap<>(task.metadata()) : new HashMap<>();
77+
newMetadata.putAll(taskStatusUpdateEvent.metadata());
8278
taskBuilder.metadata(newMetadata);
8379
}
84-
taskBuilder.status(taskStatusUpdateEvent.getStatus());
80+
taskBuilder.status(taskStatusUpdateEvent.status());
8581
currentTask = taskBuilder.build();
8682
return currentTask;
8783
}
8884

8985
public Task_v0_3 saveTaskEvent(TaskArtifactUpdateEvent_v0_3 taskArtifactUpdateEvent) {
9086
if (taskId == null) {
91-
taskId = taskArtifactUpdateEvent.getTaskId();
87+
taskId = taskArtifactUpdateEvent.taskId();
9288
}
9389
if (contextId == null) {
94-
contextId = taskArtifactUpdateEvent.getContextId();
90+
contextId = taskArtifactUpdateEvent.contextId();
9591
}
9692
Task_v0_3 task = currentTask;
9793
if (task == null) {
@@ -115,13 +111,10 @@ public Task_v0_3 saveTaskEvent(TaskArtifactUpdateEvent_v0_3 taskArtifactUpdateEv
115111
*/
116112
public Task_v0_3 updateWithMessage(Message_v0_3 message, Task_v0_3 task) {
117113
Task_v0_3.Builder taskBuilder = new Task_v0_3.Builder(task);
118-
List<Message_v0_3> history = task.getHistory();
119-
if (history == null) {
120-
history = new ArrayList<>();
121-
}
122-
if (task.getStatus().message() != null) {
123-
history.add(task.getStatus().message());
124-
taskBuilder.status(new TaskStatus_v0_3(task.getStatus().state(), null, task.getStatus().timestamp()));
114+
List<Message_v0_3> history = new ArrayList<>(task.history());
115+
if (task.status().message() != null) {
116+
history.add(task.status().message());
117+
taskBuilder.status(new TaskStatus_v0_3(task.status().state(), null, task.status().timestamp()));
125118
}
126119
history.add(message);
127120
taskBuilder.history(history);
@@ -132,8 +125,8 @@ public Task_v0_3 updateWithMessage(Message_v0_3 message, Task_v0_3 task) {
132125
private void saveTask(Task_v0_3 task) {
133126
currentTask = task;
134127
if (taskId == null) {
135-
taskId = currentTask.getId();
136-
contextId = currentTask.getContextId();
128+
taskId = currentTask.id();
129+
contextId = currentTask.contextId();
137130
}
138131
}
139132
}

‎compat-0.3/client/base/src/test/java/org/a2aproject/sdk/compat03/A2A_v0_3_Test.java‎

Lines changed: 39 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,12 @@ public void testToUserMessage() {
2020
String text = "Hello, world!";
2121
Message_v0_3 message = A2A_v0_3.toUserMessage(text);
2222

23-
assertEquals(Message_v0_3.Role.USER, message.getRole());
24-
assertEquals(1, message.getParts().size());
25-
assertEquals(text, ((TextPart_v0_3) message.getParts().get(0)).getText());
26-
assertNotNull(message.getMessageId());
27-
assertNull(message.getContextId());
28-
assertNull(message.getTaskId());
23+
assertEquals(Message_v0_3.Role.USER, message.role());
24+
assertEquals(1, message.parts().size());
25+
assertEquals(text, ((TextPart_v0_3) message.parts().get(0)).text());
26+
assertNotNull(message.messageId());
27+
assertNull(message.contextId());
28+
assertNull(message.taskId());
2929
}
3030

3131
@Test
@@ -34,19 +34,19 @@ public void testToUserMessageWithId() {
3434
String messageId = "test-message-id";
3535
Message_v0_3 message = A2A_v0_3.toUserMessage(text, messageId);
3636

37-
assertEquals(Message_v0_3.Role.USER, message.getRole());
38-
assertEquals(messageId, message.getMessageId());
37+
assertEquals(Message_v0_3.Role.USER, message.role());
38+
assertEquals(messageId, message.messageId());
3939
}
4040

4141
@Test
4242
public void testToAgentMessage() {
4343
String text = "Hello, I'm an agent!";
4444
Message_v0_3 message = A2A_v0_3.toAgentMessage(text);
4545

46-
assertEquals(Message_v0_3.Role.AGENT, message.getRole());
47-
assertEquals(1, message.getParts().size());
48-
assertEquals(text, ((TextPart_v0_3) message.getParts().get(0)).getText());
49-
assertNotNull(message.getMessageId());
46+
assertEquals(Message_v0_3.Role.AGENT, message.role());
47+
assertEquals(1, message.parts().size());
48+
assertEquals(text, ((TextPart_v0_3) message.parts().get(0)).text());
49+
assertNotNull(message.messageId());
5050
}
5151

5252
@Test
@@ -55,8 +55,8 @@ public void testToAgentMessageWithId() {
5555
String messageId = "agent-message-id";
5656
Message_v0_3 message = A2A_v0_3.toAgentMessage(text, messageId);
5757

58-
assertEquals(Message_v0_3.Role.AGENT, message.getRole());
59-
assertEquals(messageId, message.getMessageId());
58+
assertEquals(Message_v0_3.Role.AGENT, message.role());
59+
assertEquals(messageId, message.messageId());
6060
}
6161

6262
@Test
@@ -67,14 +67,14 @@ public void testCreateUserTextMessage() {
6767

6868
Message_v0_3 message = A2A_v0_3.createUserTextMessage(text, contextId, taskId);
6969

70-
assertEquals(Message_v0_3.Role.USER, message.getRole());
71-
assertEquals(contextId, message.getContextId());
72-
assertEquals(taskId, message.getTaskId());
73-
assertEquals(1, message.getParts().size());
74-
assertEquals(text, ((TextPart_v0_3) message.getParts().get(0)).getText());
75-
assertNotNull(message.getMessageId());
76-
assertNull(message.getMetadata());
77-
assertNull(message.getReferenceTaskIds());
70+
assertEquals(Message_v0_3.Role.USER, message.role());
71+
assertEquals(contextId, message.contextId());
72+
assertEquals(taskId, message.taskId());
73+
assertEquals(1, message.parts().size());
74+
assertEquals(text, ((TextPart_v0_3) message.parts().get(0)).text());
75+
assertNotNull(message.messageId());
76+
assertNull(message.metadata());
77+
assertNull(message.referenceTaskIds());
7878
}
7979

8080
@Test
@@ -83,11 +83,11 @@ public void testCreateUserTextMessageWithNullParams() {
8383

8484
Message_v0_3 message = A2A_v0_3.createUserTextMessage(text, null, null);
8585

86-
assertEquals(Message_v0_3.Role.USER, message.getRole());
87-
assertNull(message.getContextId());
88-
assertNull(message.getTaskId());
89-
assertEquals(1, message.getParts().size());
90-
assertEquals(text, ((TextPart_v0_3) message.getParts().get(0)).getText());
86+
assertEquals(Message_v0_3.Role.USER, message.role());
87+
assertNull(message.contextId());
88+
assertNull(message.taskId());
89+
assertEquals(1, message.parts().size());
90+
assertEquals(text, ((TextPart_v0_3) message.parts().get(0)).text());
9191
}
9292

9393
@Test
@@ -98,12 +98,12 @@ public void testCreateAgentTextMessage() {
9898

9999
Message_v0_3 message = A2A_v0_3.createAgentTextMessage(text, contextId, taskId);
100100

101-
assertEquals(Message_v0_3.Role.AGENT, message.getRole());
102-
assertEquals(contextId, message.getContextId());
103-
assertEquals(taskId, message.getTaskId());
104-
assertEquals(1, message.getParts().size());
105-
assertEquals(text, ((TextPart_v0_3) message.getParts().get(0)).getText());
106-
assertNotNull(message.getMessageId());
101+
assertEquals(Message_v0_3.Role.AGENT, message.role());
102+
assertEquals(contextId, message.contextId());
103+
assertEquals(taskId, message.taskId());
104+
assertEquals(1, message.parts().size());
105+
assertEquals(text, ((TextPart_v0_3) message.parts().get(0)).text());
106+
assertNotNull(message.messageId());
107107
}
108108

109109
@Test
@@ -117,12 +117,12 @@ public void testCreateAgentPartsMessage() {
117117

118118
Message_v0_3 message = A2A_v0_3.createAgentPartsMessage(parts, contextId, taskId);
119119

120-
assertEquals(Message_v0_3.Role.AGENT, message.getRole());
121-
assertEquals(contextId, message.getContextId());
122-
assertEquals(taskId, message.getTaskId());
123-
assertEquals(2, message.getParts().size());
124-
assertEquals("Part 1", ((TextPart_v0_3) message.getParts().get(0)).getText());
125-
assertEquals("Part 2", ((TextPart_v0_3) message.getParts().get(1)).getText());
120+
assertEquals(Message_v0_3.Role.AGENT, message.role());
121+
assertEquals(contextId, message.contextId());
122+
assertEquals(taskId, message.taskId());
123+
assertEquals(2, message.parts().size());
124+
assertEquals("Part 1", ((TextPart_v0_3) message.parts().get(0)).text());
125+
assertEquals("Part 2", ((TextPart_v0_3) message.parts().get(1)).text());
126126
}
127127

128128
@Test

‎compat-0.3/client/transport/jsonrpc/src/main/java/org/a2aproject/sdk/compat03/client/transport/jsonrpc/sse/SSEEventListener_v0_3.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ private void handleMessage(JsonObject jsonObject, Future<Void> future) throws Js
7777
String resultJson = jsonObject.get("result").toString();
7878
StreamingEventKind_v0_3 event = JsonUtil_v0_3.fromJson(resultJson, StreamingEventKind_v0_3.class);
7979
eventHandler.accept(event);
80-
if (event instanceof TaskStatusUpdateEvent_v0_3 && ((TaskStatusUpdateEvent_v0_3) event).isFinal()) {
80+
if (event instanceof TaskStatusUpdateEvent_v0_3 tsue && tsue.isFinal()) {
8181
future.cancel(true); // close SSE channel
8282
}
8383
} else {

‎compat-0.3/client/transport/jsonrpc/src/test/java/org/a2aproject/sdk/compat03/client/transport/jsonrpc/JSONRPCTransportStreaming_v0_3_Test.java‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,8 @@ public void testSendStreamingMessageParams() {
7474
assertNotNull(params);
7575
assertEquals(message, params.message());
7676
assertEquals(configuration, params.configuration());
77-
assertEquals(Message_v0_3.Role.USER, params.message().getRole());
78-
assertEquals("test message", ((TextPart_v0_3) params.message().getParts().get(0)).getText());
77+
assertEquals(Message_v0_3.Role.USER, params.message().role());
78+
assertEquals("test message", ((TextPart_v0_3) params.message().parts().get(0)).text());
7979
}
8080

8181
@Test
@@ -159,16 +159,16 @@ public void testA2AClientResubscribeToTask() throws Exception {
159159
assertNotNull(eventKind);
160160
assertInstanceOf(Task_v0_3.class, eventKind);
161161
Task_v0_3 task = (Task_v0_3) eventKind;
162-
assertEquals("2", task.getId());
163-
assertEquals("context-1234", task.getContextId());
164-
assertEquals(TaskState_v0_3.COMPLETED, task.getStatus().state());
165-
List<Artifact_v0_3> artifacts = task.getArtifacts();
162+
assertEquals("2", task.id());
163+
assertEquals("context-1234", task.contextId());
164+
assertEquals(TaskState_v0_3.COMPLETED, task.status().state());
165+
List<Artifact_v0_3> artifacts = task.artifacts();
166166
assertEquals(1, artifacts.size());
167167
Artifact_v0_3 artifact = artifacts.get(0);
168168
assertEquals("artifact-1", artifact.artifactId());
169169
assertEquals("joke", artifact.name());
170170
Part_v0_3<?> part = artifact.parts().get(0);
171-
assertEquals(Part_v0_3.Kind.TEXT, part.getKind());
172-
assertEquals("Why did the chicken cross the road? To get to the other side!", ((TextPart_v0_3) part).getText());
171+
assertEquals(Part_v0_3.Kind.TEXT, ((TextPart_v0_3) part).kind());
172+
assertEquals("Why did the chicken cross the road? To get to the other side!", ((TextPart_v0_3) part).text());
173173
}
174174
}

0 commit comments

Comments
 (0)