Merge pull request #111 from fjuma/spec

Add support for extensions

Author: maeste

examples/src/main/java/io/a2a/examples/helloworld/server/AgentCardProducer.java

@@ -23,7 +23,11 @@ public AgentCard agentCard() {
                 .url("http://localhost:9999")
                 .version("1.0.0")
                 .documentationUrl("http://example.com/docs")
-                .capabilities(new AgentCapabilities(true, true, true))
+                .capabilities(new AgentCapabilities.Builder()
+                        .streaming(true)
+                        .pushNotifications(true)
+                        .stateTransitionHistory(true)
+                        .build())
                 .defaultInputModes(Collections.singletonList("text"))
                 .defaultOutputModes(Collections.singletonList("text"))
                 .skills(Collections.singletonList(new AgentSkill.Builder()

src/main/java/io/a2a/spec/AgentCapabilities.java

@@ -1,7 +1,42 @@
 package io.a2a.spec;
 
+import java.util.List;
+
 /**
  * An agent's capabilities.
  */
-public record AgentCapabilities(boolean streaming, boolean pushNotifications, boolean stateTransitionHistory) {
+public record AgentCapabilities(boolean streaming, boolean pushNotifications, boolean stateTransitionHistory,
+                                List<AgentExtension> extensions) {
+
+    public static class Builder {
+
+        private boolean streaming;
+        private boolean pushNotifications;
+        private boolean stateTransitionHistory;
+        private List<AgentExtension> extensions;
+
+        public Builder streaming(boolean streaming) {
+            this.streaming = streaming;
+            return this;
+        }
+
+        public Builder pushNotifications(boolean pushNotifications) {
+            this.pushNotifications = pushNotifications;
+            return this;
+        }
+
+        public Builder stateTransitionHistory(boolean stateTransitionHistory) {
+            this.stateTransitionHistory = stateTransitionHistory;
+            return this;
+        }
+
+        public Builder extensions(List<AgentExtension> extensions) {
+            this.extensions = extensions;
+            return this;
+        }
+
+        public AgentCapabilities build() {
+            return new AgentCapabilities(streaming, pushNotifications, stateTransitionHistory, extensions);
+        }
+    }
 }

src/main/java/io/a2a/spec/AgentExtension.java

@@ -0,0 +1,44 @@
+package io.a2a.spec;
+
+import java.util.Map;
+
+import io.a2a.util.Assert;
+
+public record AgentExtension (String description, Map<String, Object> params, boolean required, String uri) {
+
+    public AgentExtension {
+        Assert.checkNotNullParam("uri", uri);
+    }
+
+    public static class Builder {
+        String description;
+        Map<String, Object> params;
+        boolean required;
+        String uri;
+
+        public Builder description(String description) {
+            this.description = description;
+            return this;
+        }
+
+        public Builder params(Map<String, Object> params) {
+            this.params = params;
+            return this;
+        }
+
+        public Builder required(boolean required) {
+            this.required = required;
+            return this;
+        }
+
+        public Builder uri(String uri) {
+            this.uri = uri;
+            return this;
+        }
+
+        public AgentExtension build() {
+            return new AgentExtension(description, params, required, uri);
+        }
+    }
+
+}

src/test/java/io/a2a/server/apps/AgentCardProducer.java

@@ -22,7 +22,11 @@ public AgentCard agentCard() {
                 .url("http://localhost:8081")
                 .version("1.0")
                 .documentationUrl("http://example.com/docs")
-                .capabilities(new AgentCapabilities(true, true, true))
+                .capabilities(new AgentCapabilities.Builder()
+                        .streaming(true)
+                        .pushNotifications(true)
+                        .stateTransitionHistory(true)
+                        .build())
                 .defaultInputModes(Collections.singletonList("text"))
                 .defaultOutputModes(Collections.singletonList("text"))
                 .skills(new ArrayList<>())

src/test/java/io/a2a/server/requesthandlers/JSONRPCHandlerTest.java

@@ -1251,7 +1251,11 @@ private static AgentCard createAgentCard(boolean streaming, boolean pushNotifica
                 .url("http://example.com")
                 .version("1.0")
                 .documentationUrl("http://example.com/docs")
-                .capabilities(new AgentCapabilities(streaming, pushNotifications, stateTransitionHistory))
+                .capabilities(new AgentCapabilities.Builder()
+                        .streaming(streaming)
+                        .pushNotifications(pushNotifications)
+                        .stateTransitionHistory(stateTransitionHistory)
+                        .build())
                 .defaultInputModes(new ArrayList<>())
                 .defaultOutputModes(new ArrayList<>())
                 .skills(new ArrayList<>())

tck/src/main/java/io/a2a/tck/server/AgentCardProducer.java

@@ -23,7 +23,11 @@ public AgentCard agentCard() {
                 .url("http://localhost:9999")
                 .version("1.0.0")
                 .documentationUrl("http://example.com/docs")
-                .capabilities(new AgentCapabilities(true, true, true))
+                .capabilities(new AgentCapabilities.Builder()
+                        .streaming(true)
+                        .pushNotifications(true)
+                        .stateTransitionHistory(true)
+                        .build())
                 .defaultInputModes(Collections.singletonList("text"))
                 .defaultOutputModes(Collections.singletonList("text"))
                 .skills(Collections.singletonList(new AgentSkill.Builder()