@google/adk 0.6.1 → 1.0.0

package/dist/cjs/utils/streaming_utils.js

@@ -0,0 +1,333 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var streaming_utils_exports = {};
+__export(streaming_utils_exports, {
+  StreamingResponseAggregator: () => StreamingResponseAggregator
+});
+module.exports = __toCommonJS(streaming_utils_exports);
+var import_genai = require("@google/genai");
+var import_jsonpath_plus = require("jsonpath-plus");
+var import_functions = require("../agents/functions.js");
+var import_feature_registry = require("../features/feature_registry.js");
+var import_llm_response = require("../models/llm_response.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class StreamingResponseAggregator {
+  constructor(isProgressiveMode = (0, import_feature_registry.isFeatureEnabled)(
+    import_feature_registry.FeatureName.PROGRESSIVE_SSE_STREAMING
+  )) {
+    this.isProgressiveMode = isProgressiveMode;
+    // For non-progressive SSE streaming mode
+    this.text = "";
+    this.thoughtText = "";
+    // For progressive SSE streaming mode: accumulate parts in order
+    this.partsSequence = [];
+    this.currentTextBuffer = "";
+    this.currentFcArgs = {};
+  }
+  flushTextBufferToSequence() {
+    if (!this.currentTextBuffer) {
+      return;
+    }
+    if (this.currentTextIsThought) {
+      this.partsSequence.push({
+        text: this.currentTextBuffer,
+        thought: true
+      });
+    } else {
+      this.partsSequence.push({
+        text: this.currentTextBuffer
+      });
+    }
+    this.currentTextBuffer = "";
+    this.currentTextIsThought = void 0;
+  }
+  getValueFromPartialArg(partialArg, jsonPath) {
+    let value = null;
+    let hasValue = false;
+    const stringValue = partialArg.stringValue;
+    const numberValue = partialArg.numberValue;
+    const boolValue = partialArg.boolValue;
+    const nullValue = partialArg.nullValue;
+    if (stringValue !== void 0) {
+      const stringChunk = stringValue;
+      hasValue = true;
+      const pathParts = import_jsonpath_plus.JSONPath.toPathArray(jsonPath).filter(
+        (p) => p !== "$" && p !== "$["
+      );
+      let existingValue = this.currentFcArgs;
+      for (const part of pathParts) {
+        if (existingValue && typeof existingValue === "object" && part in existingValue) {
+          existingValue = existingValue[part];
+        } else {
+          existingValue = void 0;
+          break;
+        }
+      }
+      if (typeof existingValue === "string") {
+        value = existingValue + stringChunk;
+      } else {
+        value = stringChunk;
+      }
+    } else if (numberValue !== void 0) {
+      value = numberValue;
+      hasValue = true;
+    } else if (boolValue !== void 0) {
+      value = boolValue;
+      hasValue = true;
+    } else if (nullValue !== void 0) {
+      value = null;
+      hasValue = true;
+    }
+    return [value, hasValue];
+  }
+  setValueByJsonPath(jsonPath, value) {
+    const pathParts = import_jsonpath_plus.JSONPath.toPathArray(jsonPath).filter(
+      (p) => p !== "$" && p !== "$["
+    );
+    let current = this.currentFcArgs;
+    for (let i = 0; i < pathParts.length - 1; i++) {
+      const part = pathParts[i];
+      if (!(part in current)) {
+        current[part] = {};
+      }
+      current = current[part];
+    }
+    if (pathParts.length > 0) {
+      current[pathParts[pathParts.length - 1]] = value;
+    }
+  }
+  flushFunctionCallToSequence() {
+    var _a;
+    if (this.currentFcName) {
+      const fcPart = {
+        functionCall: {
+          name: this.currentFcName,
+          args: JSON.parse(JSON.stringify(this.currentFcArgs)),
+          id: (_a = this.currentFcId) != null ? _a : (0, import_functions.generateClientFunctionCallId)()
+        }
+      };
+      if (this.currentThoughtSignature) {
+        fcPart.thoughtSignature = this.currentThoughtSignature.toString();
+      }
+      this.partsSequence.push(fcPart);
+      this.currentFcName = void 0;
+      this.currentFcArgs = {};
+      this.currentFcId = void 0;
+      this.currentThoughtSignature = void 0;
+    }
+  }
+  processStreamingFunctionCall(fc) {
+    if (fc.name) {
+      this.currentFcName = fc.name;
+    }
+    if (fc.id) {
+      this.currentFcId = fc.id;
+    }
+    for (const partialArg of fc.partialArgs || []) {
+      const jsonPath = partialArg.jsonPath;
+      if (!jsonPath) {
+        continue;
+      }
+      const [value, hasValue] = this.getValueFromPartialArg(
+        partialArg,
+        jsonPath
+      );
+      if (hasValue) {
+        this.setValueByJsonPath(jsonPath, value);
+      }
+    }
+    if (!fc.willContinue) {
+      this.flushTextBufferToSequence();
+      this.flushFunctionCallToSequence();
+    }
+  }
+  processFunctionCallPart(part) {
+    const fc = part.functionCall;
+    if (!fc) {
+      return;
+    }
+    if (part.thoughtSignature) {
+      this.lastThoughtSignature = part.thoughtSignature;
+    } else if (this.lastThoughtSignature) {
+      part.thoughtSignature = this.lastThoughtSignature.toString();
+    }
+    if (fc.partialArgs || fc.willContinue) {
+      if (!fc.id && !this.currentFcId) {
+        fc.id = (0, import_functions.generateClientFunctionCallId)();
+      }
+      if (part.thoughtSignature && !this.currentThoughtSignature) {
+        this.currentThoughtSignature = part.thoughtSignature;
+      }
+      this.processStreamingFunctionCall(fc);
+    } else {
+      if (fc.name) {
+        if (!fc.id) {
+          fc.id = (0, import_functions.generateClientFunctionCallId)();
+        }
+        this.flushTextBufferToSequence();
+        this.partsSequence.push(part);
+      }
+    }
+  }
+  async *processResponse(response) {
+    var _a, _b, _c;
+    this.response = response;
+    const llmResponse = (0, import_llm_response.createLlmResponse)(response);
+    this.usageMetadata = llmResponse.usageMetadata;
+    if (llmResponse.groundingMetadata) {
+      this.groundingMetadata = llmResponse.groundingMetadata;
+    }
+    if (llmResponse.citationMetadata) {
+      this.citationMetadata = llmResponse.citationMetadata;
+    }
+    if (llmResponse.finishReason) {
+      this.finishReason = llmResponse.finishReason;
+    }
+    if (llmResponse.content && llmResponse.content.parts) {
+      for (const part of llmResponse.content.parts) {
+        if (part.thoughtSignature) {
+          this.lastThoughtSignature = part.thoughtSignature;
+        } else if (part.functionCall && this.lastThoughtSignature) {
+          part.thoughtSignature = this.lastThoughtSignature.toString();
+        }
+      }
+    }
+    if (this.isProgressiveMode) {
+      if (llmResponse.content && llmResponse.content.parts) {
+        for (const part of llmResponse.content.parts) {
+          if (part.text) {
+            const isThought = (_a = part.thought) != null ? _a : false;
+            if (this.currentTextBuffer && isThought !== this.currentTextIsThought) {
+              this.flushTextBufferToSequence();
+            }
+            if (!this.currentTextBuffer) {
+              this.currentTextIsThought = isThought;
+            }
+            this.currentTextBuffer += part.text;
+          } else if (part.functionCall) {
+            this.processFunctionCallPart(part);
+          } else {
+            this.flushTextBufferToSequence();
+            this.partsSequence.push(part);
+          }
+        }
+      }
+      llmResponse.partial = true;
+      yield llmResponse;
+      return;
+    }
+    if (llmResponse.content && llmResponse.content.parts && typeof ((_b = llmResponse.content.parts[0]) == null ? void 0 : _b.text) === "string") {
+      const part0 = llmResponse.content.parts[0];
+      const partText = part0.text || "";
+      if (part0.thought) {
+        this.thoughtText += partText;
+      } else {
+        this.text += partText;
+      }
+      llmResponse.partial = true;
+    } else if ((this.thoughtText || this.text) && (!llmResponse.content || !llmResponse.content.parts || !((_c = llmResponse.content.parts[0]) == null ? void 0 : _c.inlineData))) {
+      const parts = [];
+      if (this.thoughtText) {
+        parts.push({ text: this.thoughtText, thought: true });
+      }
+      if (this.text) {
+        parts.push({ text: this.text });
+      }
+      yield {
+        content: {
+          role: "model",
+          parts
+        },
+        usageMetadata: llmResponse.usageMetadata,
+        partial: false
+      };
+      this.thoughtText = "";
+      this.text = "";
+    }
+    yield llmResponse;
+  }
+  close() {
+    var _a, _b;
+    if (!((_a = this.response) == null ? void 0 : _a.candidates) || this.response.candidates.length === 0) {
+      return;
+    }
+    if (this.isProgressiveMode) {
+      this.flushTextBufferToSequence();
+      this.flushFunctionCallToSequence();
+      const finalParts = this.partsSequence;
+      if (finalParts.length === 0) {
+        return;
+      }
+      const candidate = this.response.candidates[0];
+      const finishReason = (_b = this.finishReason) != null ? _b : candidate.finishReason;
+      return {
+        content: {
+          role: "model",
+          parts: finalParts
+        },
+        groundingMetadata: this.groundingMetadata,
+        citationMetadata: this.citationMetadata,
+        errorCode: finishReason === import_genai.FinishReason.STOP ? void 0 : finishReason,
+        errorMessage: finishReason === import_genai.FinishReason.STOP ? void 0 : candidate.finishMessage,
+        usageMetadata: this.usageMetadata,
+        finishReason,
+        partial: false
+      };
+    }
+    if ((this.text || this.thoughtText) && this.response.candidates && this.response.candidates.length > 0) {
+      const parts = [];
+      if (this.thoughtText) {
+        parts.push({ text: this.thoughtText, thought: true });
+      }
+      if (this.text) {
+        parts.push({ text: this.text });
+      }
+      const candidate = this.response.candidates[0];
+      const finishReason = candidate.finishReason;
+      return {
+        content: {
+          role: "model",
+          parts
+        },
+        groundingMetadata: this.groundingMetadata,
+        citationMetadata: this.citationMetadata,
+        errorCode: finishReason === import_genai.FinishReason.STOP ? void 0 : finishReason,
+        errorMessage: finishReason === import_genai.FinishReason.STOP ? void 0 : candidate.finishMessage,
+        usageMetadata: this.usageMetadata,
+        finishReason,
+        partial: false
+      };
+    }
+    return void 0;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  StreamingResponseAggregator
+});

package/dist/cjs/version.js

@@ -32,7 +32,7 @@ module.exports = __toCommonJS(version_exports);
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "0.6.1";
+const version = "1.0.0";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   version

package/dist/esm/a2a/a2a_event.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/a2a_remote_agent.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/a2a_remote_agent_run_processor.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/a2a_remote_agent_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/agent_card.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/agent_executor.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/agent_to_a2a.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/event_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/event_processor_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/executor_context.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/metadata_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/a2a/part_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/esm/agents/active_streaming_tool.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/esm/agents/base_agent.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -56,19 +58,20 @@ class BaseAgent {
         ctx,
         this,
         async function* () {
+          var _a2, _b;
           const context2 = this.createInvocationContext(parentContext);
           const beforeAgentCallbackEvent = await this.handleBeforeAgentCallback(context2);
           if (beforeAgentCallbackEvent) {
             yield beforeAgentCallbackEvent;
           }
-          if (context2.endInvocation) {
+          if (context2.endInvocation || ((_a2 = parentContext.abortSignal) == null ? void 0 : _a2.aborted)) {
             return;
           }
           traceAgentInvocation({ agent: this, invocationContext: context2 });
           for await (const event of this.runAsyncImpl(context2)) {
             yield event;
           }
-          if (context2.endInvocation) {
+          if (context2.endInvocation || ((_b = parentContext.abortSignal) == null ? void 0 : _b.aborted)) {
             return;
           }
           const afterAgentCallbackEvent = await this.handleAfterAgentCallback(context2);
@@ -150,12 +153,16 @@ class BaseAgent {
    *     generated.
    */
   async handleBeforeAgentCallback(invocationContext) {
+    var _a2;
     if (this.beforeAgentCallback.length === 0) {
       return void 0;
     }
     const callbackContext = new Context({ invocationContext });
     for (const callback of this.beforeAgentCallback) {
       const content = await callback(callbackContext);
+      if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
+        return;
+      }
       if (content) {
         invocationContext.endInvocation = true;
         return createEvent({
@@ -185,12 +192,16 @@ class BaseAgent {
    *     generated.
    */
   async handleAfterAgentCallback(invocationContext) {
+    var _a2;
     if (this.afterAgentCallback.length === 0) {
       return void 0;
     }
     const callbackContext = new Context({ invocationContext });
     for (const callback of this.afterAgentCallback) {
       const content = await callback(callbackContext);
+      if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
+        return;
+      }
       if (content) {
         return createEvent({
           invocationId: invocationContext.invocationId,

package/dist/esm/agents/context.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -30,6 +32,7 @@ class Context extends ReadonlyContext {
     );
     this.functionCallId = options.functionCallId;
     this.toolConfirmation = options.toolConfirmation;
+    this.abortSignal = options.invocationContext.abortSignal;
   }
   /**
    * The delta-aware state of the current session.

package/dist/esm/agents/functions.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -379,6 +381,7 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
   );
   const mergedActions = mergeEventActions(actionsList);
   return createEvent({
+    invocationId: baseEvent.invocationId,
     author: baseEvent.author,
     branch: baseEvent.branch,
     content: { role: "user", parts: mergedParts },

package/dist/esm/agents/instructions.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/esm/agents/invocation_context.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -47,6 +49,7 @@ class InvocationContext {
     this.liveRequestQueue = params.liveRequestQueue;
     this.activeStreamingTools = params.activeStreamingTools;
     this.pluginManager = params.pluginManager;
+    this.abortSignal = params.abortSignal;
   }
   /**
    * The app name of the current session.

package/dist/esm/agents/live_request_queue.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC