added a2ui support and display part

Author: vishalmysore

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>io.github.vishalmysore</groupId>
     <artifactId>a2ajava</artifactId>
-    <version>1.0.2</version>
+    <version>1.0.3</version>
     <name>A2A Protocol Implementation for Java</name>
     <description>
         Java implementation of the A2A protocol v1.0, which allows for the exchange of data between different AI systems.

src/main/java/io/github/vishalmysore/a2ui/A2UIAware.java

@@ -315,7 +315,7 @@ default Map<String, Object> buildA2UIMessageWithData(String surfaceId, String ro
 
         return messages;
     }
-    default boolean isUICallback(ThreadLocal<ActionCallback> callback) {
-        return callback != null && callback.get()!=null && callback.get().getType().equals(CallBackType.A2UI.name());
+    default boolean isUICallback(ActionCallback callback) {
+        return callback != null && callback.getType().equals(CallBackType.A2UI.name());
     }
 }