@google/adk 0.6.0 → 0.6.1

package/dist/cjs/a2a/a2a_remote_agent.js

@@ -51,17 +51,24 @@ class RemoteA2AAgent extends import_base_agent.BaseAgent {
     super(a2aConfig);
     this.a2aConfig = a2aConfig;
     this.isInitialized = false;
-    if (!a2aConfig.agentCard) {
-      throw new Error("AgentCard must be provided");
+    if (!a2aConfig.agentCard && !a2aConfig.client) {
+      throw new Error("Either AgentCard or Client must be provided");
     }
   }
   async init() {
     if (this.isInitialized) {
       return;
     }
-    this.card = await (0, import_agent_card.resolveAgentCard)(this.a2aConfig.agentCard);
-    const factory = this.a2aConfig.clientFactory || new import_client.ClientFactory();
-    this.client = await factory.createFromAgentCard(this.card);
+    if (this.a2aConfig.client) {
+      this.client = this.a2aConfig.client;
+    }
+    if (this.a2aConfig.agentCard) {
+      this.card = await (0, import_agent_card.resolveAgentCard)(this.a2aConfig.agentCard);
+      if (!this.client) {
+        const factory = this.a2aConfig.clientFactory || new import_client.ClientFactory();
+        this.client = await factory.createFromAgentCard(this.card);
+      }
+    }
     this.isInitialized = true;
   }
   async *runAsyncImpl(context) {
@@ -97,13 +104,11 @@ class RemoteA2AAgent extends import_base_agent.BaseAgent {
         messageId: (0, import_env_aware_utils.randomUUID)(),
         role: import_a2a_event.MessageRole.USER,
         parts,
-        metadata: {
-          ...(0, import_metadata_converter_utils.getA2ASessionMetadata)({
-            appName: context.session.appName,
-            userId: context.session.userId,
-            sessionId: context.session.id
-          })
-        }
+        metadata: (0, import_metadata_converter_utils.getA2ASessionMetadata)({
+          appName: context.session.appName,
+          userId: context.session.userId,
+          sessionId: context.session.id
+        })
       };
       if (taskId) message.taskId = taskId;
       if (contextId) message.contextId = contextId;
@@ -117,7 +122,7 @@ class RemoteA2AAgent extends import_base_agent.BaseAgent {
           await callback(context, params);
         }
       }
-      const useStreaming = ((_b = this.card.capabilities) == null ? void 0 : _b.streaming) !== false;
+      const useStreaming = this.card ? ((_b = this.card.capabilities) == null ? void 0 : _b.streaming) !== false : true;
       if (useStreaming) {
         for await (const chunk of this.client.sendMessageStream(params)) {
           if (this.a2aConfig.afterRequestCallbacks) {

package/dist/cjs/a2a/a2a_remote_agent_run_processor.js

@@ -179,6 +179,13 @@ class A2ARemoteAgentRunProcessor {
     }
     if (response) {
       toAdd["response"] = response;
+      if ((0, import_a2a_event.isTask)(response)) {
+        if (response.id) toAdd["task_id"] = response.id;
+        if (response.contextId) toAdd["context_id"] = response.contextId;
+      } else if (response.taskId) {
+        toAdd["task_id"] = response.taskId;
+        if (response.contextId) toAdd["context_id"] = response.contextId;
+      }
     }
     if (Object.keys(toAdd).length === 0) {
       return;