File tree Expand file tree Collapse file tree 2 files changed +30
-0
lines changed
main/java/io/a2a/server/agentexecution
test/java/io/a2a/server/agentexecution Expand file tree Collapse file tree 2 files changed +30
-0
lines changed Original file line number Diff line number Diff line change 33import java .util .ArrayList ;
44import java .util .Collections ;
55import java .util .List ;
6+ import java .util .Map ;
67import java .util .UUID ;
78import java .util .stream .Collectors ;
89
@@ -198,6 +199,15 @@ public List<Task> getRelatedTasks() {
198199 return params != null ? params .configuration () : null ;
199200 }
200201
202+ /**
203+ * Returns the request metadata.
204+ *
205+ * @return the metadata, or null if not available
206+ */
207+ public @ Nullable Map <String , Object > getMetadata () {
208+ return (params != null && params .metadata () != null ) ? Collections .unmodifiableMap (params .metadata ()) : null ;
209+ }
210+
201211 /**
202212 * Returns the server call context.
203213 * <p>
Original file line number Diff line number Diff line change 1010import static org .mockito .Mockito .mockStatic ;
1111
1212import java .util .ArrayList ;
13+ import java .util .HashMap ;
1314import java .util .List ;
1415import java .util .UUID ;
1516
@@ -38,6 +39,7 @@ public void testInitWithoutParams() {
3839 RequestContext context = new RequestContext .Builder ().build ();
3940
4041 assertNull (context .getMessage ());
42+ assertNull (context .getMetadata ());
4143 assertNotNull (context .getTaskId ()); // Generated UUID
4244 assertNotNull (context .getContextId ()); // Generated UUID
4345 assertNull (context .getTask ());
@@ -70,6 +72,24 @@ public void testInitWithParamsNoIds() {
7072 }
7173 }
7274
75+ @ Test
76+ public void testInitWithParamsMetadata () {
77+ var message = Message .builder ().role (Message .Role .ROLE_USER ).parts (List .of (new TextPart ("" ))).build ();
78+ var metadata = new HashMap <String , Object >();
79+ metadata .put ("key" , "value" );
80+
81+ var params = MessageSendParams .builder ()
82+ .message (message )
83+ .metadata (metadata )
84+ .build ();
85+
86+ RequestContext context = new RequestContext .Builder ()
87+ .setParams (params )
88+ .build ();
89+
90+ assertEquals (metadata , context .getMetadata ());
91+ }
92+
7393 @ Test
7494 public void testInitWithTaskId () {
7595 String taskId = "task-123" ;
You can’t perform that action at this time.
0 commit comments