@google/adk 2.0.0 → 2.1.0

package/dist/cjs/models/chrome_prompt_llm.js

@@ -0,0 +1,304 @@
+/**
+  * @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 __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+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 chrome_prompt_llm_exports = {};
+__export(chrome_prompt_llm_exports, {
+  ChromeBuiltInLlm: () => ChromeBuiltInLlm,
+  ChromeModelUnavailableError: () => ChromeModelUnavailableError,
+  stripAdkIdentityPreamble: () => import_chrome_prompt_utils2.stripAdkIdentityPreamble
+});
+module.exports = __toCommonJS(chrome_prompt_llm_exports);
+var import_event = require("../events/event.js");
+var import_genai_schema_to_json = require("../utils/genai_schema_to_json.js");
+var import_base_llm = require("./base_llm.js");
+var import_chrome_prompt_utils = require("./chrome_prompt_utils.js");
+var import_chrome_prompt_utils2 = require("./chrome_prompt_utils.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const _ChromeModelUnavailableError = class _ChromeModelUnavailableError extends Error {
+  constructor(availability) {
+    super(
+      availability === "missing-api" ? "This browser does not expose the Prompt API (globalThis.LanguageModel is undefined)." : `Chrome's built-in model is not available (availability: "${availability}").`
+    );
+    this.availability = availability;
+    this.name = "ChromeModelUnavailableError";
+  }
+};
+__name(_ChromeModelUnavailableError, "ChromeModelUnavailableError");
+let ChromeModelUnavailableError = _ChromeModelUnavailableError;
+const _ChromeBuiltInLlm = class _ChromeBuiltInLlm extends import_base_llm.BaseLlm {
+  constructor(params = {}) {
+    var _a;
+    super({ model: (_a = params.model) != null ? _a : "chrome-on-device" });
+    this.params = params;
+  }
+  get api() {
+    var _a;
+    const api = (_a = this.params.languageModel) != null ? _a : globalThis.LanguageModel;
+    if (!api) throw new ChromeModelUnavailableError("missing-api");
+    return api;
+  }
+  /** Reports whether this browser can actually run the model. */
+  async availability() {
+    try {
+      return await this.api.availability({
+        expectedInputs: this.params.expectedInputs,
+        expectedOutputs: this.params.expectedOutputs
+      });
+    } catch {
+      return "unavailable";
+    }
+  }
+  diagnostic(diagnostic) {
+    var _a, _b;
+    (_b = (_a = this.params).onDiagnostic) == null ? void 0 : _b.call(_a, diagnostic);
+  }
+  /**
+   * Returns a session cloned from a warm base carrying `systemPrompt`, and
+   * rebuilds that base only when the system prompt or sampling changes.
+   */
+  async acquireSession(systemPrompt, signal) {
+    var _a, _b;
+    const key = JSON.stringify([
+      systemPrompt,
+      this.params.temperature,
+      this.params.topK,
+      this.params.expectedInputs
+    ]);
+    if (this.baseSessionKey !== key || !this.baseSessionPromise) {
+      (_b = (_a = this.baseSession) == null ? void 0 : _a.destroy) == null ? void 0 : _b.call(_a);
+      this.baseSession = void 0;
+      this.baseSessionKey = key;
+      this.baseSessionPromise = this.createBaseSession(systemPrompt);
+    }
+    let base;
+    try {
+      base = await this.baseSessionPromise;
+    } catch (error) {
+      this.baseSessionPromise = void 0;
+      this.baseSessionKey = void 0;
+      throw error;
+    }
+    return base.clone({ signal });
+  }
+  async createBaseSession(systemPrompt) {
+    var _a, _b;
+    const availability = await this.availability();
+    if (availability === "unavailable") {
+      throw new ChromeModelUnavailableError(availability);
+    }
+    const createOptions = {};
+    if (this.params.onDownloadProgress) {
+      createOptions.monitor = (monitor) => {
+        monitor.addEventListener("downloadprogress", (event) => {
+          this.params.onDownloadProgress(event.loaded);
+        });
+      };
+    }
+    if (systemPrompt) {
+      createOptions.initialPrompts = [{ role: "system", content: systemPrompt }];
+    }
+    if (this.params.expectedInputs) {
+      createOptions.expectedInputs = this.params.expectedInputs;
+    }
+    if (this.params.expectedOutputs) {
+      createOptions.expectedOutputs = this.params.expectedOutputs;
+    }
+    if (this.params.temperature !== void 0) {
+      createOptions.temperature = this.params.temperature;
+      createOptions.topK = (_a = this.params.topK) != null ? _a : 3;
+    }
+    const startedAt = performance.now();
+    let session;
+    try {
+      session = await this.api.create(createOptions);
+    } catch (error) {
+      if (this.params.temperature === void 0) throw error;
+      delete createOptions.temperature;
+      delete createOptions.topK;
+      session = await this.api.create(createOptions);
+    }
+    this.diagnostic({ phase: "create", ms: performance.now() - startedAt });
+    (_b = session.addEventListener) == null ? void 0 : _b.call(session, "contextoverflow", () => {
+      this.diagnostic({
+        phase: "context-overflow",
+        note: "history truncated by the browser",
+        contextUsage: session.contextUsage,
+        contextWindow: session.contextWindow
+      });
+    });
+    this.baseSession = session;
+    return session;
+  }
+  async *generateContentAsync(llmRequest, stream = false, abortSignal) {
+    var _a, _b, _c, _d;
+    const declarations = (0, import_chrome_prompt_utils.collectFunctionDeclarations)(llmRequest);
+    const useTools = declarations.length > 0;
+    const rawSystem = (0, import_chrome_prompt_utils.extractSystemInstruction)(llmRequest);
+    const systemParts = [
+      this.params.normalizeSystemPrompt ? this.params.normalizeSystemPrompt(rawSystem) : rawSystem
+    ];
+    if (useTools) systemParts.push((0, import_chrome_prompt_utils.renderToolInstructions)(declarations));
+    const systemPrompt = systemParts.filter(Boolean).join("\n\n");
+    this.maybeAppendUserContent(llmRequest);
+    const messages = (0, import_chrome_prompt_utils.contentsToMessages)((_a = llmRequest.contents) != null ? _a : []);
+    let session;
+    try {
+      session = await this.acquireSession(systemPrompt, abortSignal);
+      const responseJsonSchema = (_b = llmRequest.config) == null ? void 0 : _b.responseJsonSchema;
+      const responseSchema = (_c = llmRequest.config) == null ? void 0 : _c.responseSchema;
+      let responseConstraint;
+      if (useTools) {
+        responseConstraint = (0, import_chrome_prompt_utils.buildToolChoiceSchema)(declarations);
+      } else if ((0, import_chrome_prompt_utils.isRecord)(responseJsonSchema)) {
+        responseConstraint = responseJsonSchema;
+      } else if (responseSchema) {
+        responseConstraint = (0, import_genai_schema_to_json.genaiSchemaToJsonSchema)(responseSchema);
+      }
+      if (stream && !responseConstraint) {
+        yield* this.streamText(session, messages, abortSignal);
+        return;
+      }
+      const raw = await this.promptOnce(
+        session,
+        messages,
+        responseConstraint,
+        abortSignal
+      );
+      yield useTools ? this.parseToolChoice(raw, declarations) : (0, import_chrome_prompt_utils.finalText)(raw);
+    } catch (error) {
+      if ((0, import_chrome_prompt_utils.isAbortError)(error)) throw error;
+      yield (0, import_chrome_prompt_utils.errorResponse)(error);
+    } finally {
+      if (session && session !== this.baseSession) (_d = session.destroy) == null ? void 0 : _d.call(session);
+    }
+  }
+  async promptOnce(session, messages, responseConstraint, signal) {
+    const startedAt = performance.now();
+    const output = await session.prompt(messages, {
+      signal,
+      ...responseConstraint ? { responseConstraint, omitResponseConstraintInput: true } : {}
+    });
+    this.diagnostic({
+      phase: "prompt",
+      ms: performance.now() - startedAt,
+      contextUsage: session.contextUsage,
+      contextWindow: session.contextWindow
+    });
+    return output;
+  }
+  async *streamText(session, messages, signal) {
+    const reader = session.promptStreaming(messages, { signal }).getReader();
+    let accumulated = "";
+    try {
+      for (; ; ) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        accumulated += value;
+        yield {
+          content: { role: "model", parts: [{ text: value }] },
+          partial: true
+        };
+      }
+    } finally {
+      await reader.cancel();
+      reader.releaseLock();
+    }
+    yield {
+      content: { role: "model", parts: [{ text: accumulated }] },
+      turnComplete: true
+    };
+  }
+  /** Turns the constrained JSON reply into an ADK response. */
+  parseToolChoice(raw, declarations) {
+    let parsed;
+    try {
+      parsed = JSON.parse(raw);
+    } catch {
+      const match = raw.match(/\{[\s\S]*\}/);
+      if (match) {
+        try {
+          parsed = JSON.parse(match[0]);
+        } catch {
+        }
+      }
+    }
+    if (!(0, import_chrome_prompt_utils.isRecord)(parsed)) {
+      this.diagnostic({
+        phase: "parse-fallback",
+        note: "unparseable JSON; treated as text"
+      });
+      return (0, import_chrome_prompt_utils.finalText)(raw);
+    }
+    if (parsed["kind"] === "tool" && typeof parsed["name"] === "string") {
+      const name = parsed["name"];
+      if (!declarations.some((declaration) => declaration.name === name)) {
+        return (0, import_chrome_prompt_utils.finalText)(`The model requested an unknown tool "${name}".`);
+      }
+      const args = (0, import_chrome_prompt_utils.isRecord)(parsed["args"]) ? parsed["args"] : {};
+      return {
+        content: {
+          role: "model",
+          parts: [
+            { functionCall: { id: (0, import_event.generateClientFunctionCallId)(), name, args } }
+          ]
+        },
+        turnComplete: true
+      };
+    }
+    return (0, import_chrome_prompt_utils.finalText)(typeof parsed["text"] === "string" ? parsed["text"] : raw);
+  }
+  async connect(_llmRequest) {
+    throw new Error(
+      "Chrome's Prompt API has no bidirectional live mode, so connect() is not supported by ChromeBuiltInLlm."
+    );
+  }
+  /** Releases the warm base session. */
+  destroy() {
+    var _a, _b;
+    (_b = (_a = this.baseSession) == null ? void 0 : _a.destroy) == null ? void 0 : _b.call(_a);
+    this.baseSession = void 0;
+    this.baseSessionPromise = void 0;
+    this.baseSessionKey = void 0;
+  }
+};
+__name(_ChromeBuiltInLlm, "ChromeBuiltInLlm");
+// Left unanchored on purpose: LLMRegistry.resolve wraps each pattern in
+// `^...$` before matching, so anchoring here would double-anchor and never
+// match.
+_ChromeBuiltInLlm.supportedModels = [
+  /chrome-on-device/,
+  /chrome\/.*/
+];
+let ChromeBuiltInLlm = _ChromeBuiltInLlm;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ChromeBuiltInLlm,
+  ChromeModelUnavailableError,
+  stripAdkIdentityPreamble
+});

package/dist/cjs/models/chrome_prompt_utils.js

@@ -0,0 +1,229 @@
+/**
+  * @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 __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+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 chrome_prompt_utils_exports = {};
+__export(chrome_prompt_utils_exports, {
+  buildToolChoiceSchema: () => buildToolChoiceSchema,
+  collectFunctionDeclarations: () => collectFunctionDeclarations,
+  contentsToMessages: () => contentsToMessages,
+  errorResponse: () => errorResponse,
+  extractSystemInstruction: () => extractSystemInstruction,
+  finalText: () => finalText,
+  isAbortError: () => isAbortError,
+  isRecord: () => isRecord,
+  renderToolInstructions: () => renderToolInstructions,
+  renderToolResult: () => renderToolResult,
+  stripAdkIdentityPreamble: () => stripAdkIdentityPreamble
+});
+module.exports = __toCommonJS(chrome_prompt_utils_exports);
+var import_genai_schema_to_json = require("../utils/genai_schema_to_json.js");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+__name(isRecord, "isRecord");
+function argumentSchema(declaration) {
+  if (isRecord(declaration.parametersJsonSchema)) {
+    return declaration.parametersJsonSchema;
+  }
+  if (declaration.parameters) {
+    return (0, import_genai_schema_to_json.genaiSchemaToJsonSchema)(declaration.parameters);
+  }
+  return { type: "object", properties: {} };
+}
+__name(argumentSchema, "argumentSchema");
+function collectFunctionDeclarations(llmRequest) {
+  var _a, _b;
+  const declarations = [];
+  for (const tool of (_b = (_a = llmRequest.config) == null ? void 0 : _a.tools) != null ? _b : []) {
+    if (isRecord(tool) && Array.isArray(tool["functionDeclarations"])) {
+      declarations.push(
+        ...tool["functionDeclarations"]
+      );
+    }
+  }
+  return declarations;
+}
+__name(collectFunctionDeclarations, "collectFunctionDeclarations");
+function buildToolChoiceSchema(declarations) {
+  const branches = [
+    {
+      type: "object",
+      properties: {
+        kind: { type: "string", enum: ["final"] },
+        text: { type: "string" }
+      },
+      required: ["kind", "text"]
+    }
+  ];
+  for (const declaration of declarations) {
+    if (!declaration.name) continue;
+    branches.push({
+      type: "object",
+      properties: {
+        kind: { type: "string", enum: ["tool"] },
+        name: { type: "string", enum: [declaration.name] },
+        args: argumentSchema(declaration)
+      },
+      required: ["kind", "name", "args"]
+    });
+  }
+  return branches.length === 1 ? branches[0] : { anyOf: branches };
+}
+__name(buildToolChoiceSchema, "buildToolChoiceSchema");
+function renderToolInstructions(declarations) {
+  if (!declarations.length) return "";
+  const lines = declarations.map(
+    (declaration) => {
+      var _a;
+      return `- ${declaration.name}: ${(_a = declaration.description) != null ? _a : ""}
+  arguments: ${JSON.stringify(argumentSchema(declaration))}`;
+    }
+  );
+  return [
+    "You can call these tools:",
+    ...lines,
+    "",
+    'Reply with JSON only. To call a tool use {"kind":"tool","name":<tool>,"args":{...}}.',
+    'When you have the answer use {"kind":"final","text":<answer>}.'
+  ].join("\n");
+}
+__name(renderToolInstructions, "renderToolInstructions");
+function renderToolResult(response) {
+  if (response == null) return "{}";
+  let value = response;
+  if (isRecord(value) && Object.keys(value).length === 1 && "result" in value) {
+    value = value["result"];
+  }
+  if (typeof value === "string") {
+    const trimmed = value.trim();
+    if (trimmed.startsWith("{") || trimmed.startsWith("[")) {
+      try {
+        return JSON.stringify(JSON.parse(trimmed));
+      } catch {
+        return trimmed;
+      }
+    }
+    return trimmed;
+  }
+  try {
+    return JSON.stringify(value);
+  } catch {
+    return String(value);
+  }
+}
+__name(renderToolResult, "renderToolResult");
+function contentsToMessages(contents) {
+  var _a, _b;
+  const messages = [];
+  for (const content of contents != null ? contents : []) {
+    const role = content.role === "model" ? "assistant" : "user";
+    const chunks = [];
+    for (const part of (_a = content.parts) != null ? _a : []) {
+      if (part.text) chunks.push(part.text);
+      if (part.functionCall) {
+        const args = JSON.stringify((_b = part.functionCall.args) != null ? _b : {});
+        chunks.push(`[tool_call] ${part.functionCall.name}(${args})`);
+      }
+      if (part.functionResponse) {
+        const rendered = renderToolResult(part.functionResponse.response);
+        chunks.push(
+          `[tool_result] ${part.functionResponse.name} -> ${rendered}`
+        );
+      }
+    }
+    if (chunks.length) {
+      messages.push({ role, content: chunks.join("\n") });
+    }
+  }
+  return messages;
+}
+__name(contentsToMessages, "contentsToMessages");
+function extractSystemInstruction(llmRequest) {
+  var _a;
+  const instruction = (_a = llmRequest.config) == null ? void 0 : _a.systemInstruction;
+  if (!instruction) return "";
+  if (typeof instruction === "string") return instruction;
+  if (Array.isArray(instruction)) {
+    return instruction.map(
+      (entry) => {
+        var _a2;
+        return typeof entry === "string" ? entry : (_a2 = entry == null ? void 0 : entry.text) != null ? _a2 : "";
+      }
+    ).filter(Boolean).join("\n");
+  }
+  if (isRecord(instruction) && Array.isArray(instruction["parts"])) {
+    return instruction["parts"].map((part) => {
+      var _a2;
+      return (_a2 = part.text) != null ? _a2 : "";
+    }).filter(Boolean).join("\n");
+  }
+  return String(instruction);
+}
+__name(extractSystemInstruction, "extractSystemInstruction");
+function stripAdkIdentityPreamble(systemPrompt) {
+  return systemPrompt.replace(/^You are an agent\. Your internal name is "[^"]*"\.\s*/m, "").replace(/^The description about you is "[^"]*"\s*/m, "").trimStart();
+}
+__name(stripAdkIdentityPreamble, "stripAdkIdentityPreamble");
+function isAbortError(error) {
+  return isRecord(error) && error["name"] === "AbortError";
+}
+__name(isAbortError, "isAbortError");
+function finalText(text) {
+  return { content: { role: "model", parts: [{ text }] }, turnComplete: true };
+}
+__name(finalText, "finalText");
+function errorResponse(error) {
+  const details = isRecord(error) ? error : {};
+  const name = typeof details["name"] === "string" ? details["name"] : "UnknownError";
+  const message = typeof details["message"] === "string" ? details["message"] : String(error);
+  if (name === "QuotaExceededError") {
+    return {
+      errorCode: name,
+      errorMessage: `Prompt exceeded the context window (requested ${details["requested"]} of ${details["quota"]} tokens).`,
+      turnComplete: true
+    };
+  }
+  return { errorCode: name, errorMessage: message, turnComplete: true };
+}
+__name(errorResponse, "errorResponse");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  buildToolChoiceSchema,
+  collectFunctionDeclarations,
+  contentsToMessages,
+  errorResponse,
+  extractSystemInstruction,
+  finalText,
+  isAbortError,
+  isRecord,
+  renderToolInstructions,
+  renderToolResult,
+  stripAdkIdentityPreamble
+});

package/dist/cjs/models/google_llm.js

@@ -46,7 +46,7 @@ var import_llm_response = require("./llm_response.js");
  * SPDX-License-Identifier: Apache-2.0
  */
 var _a, _b;
-const GEMINI_MODEL_SYMBOL = Symbol.for("google.adk.geminiModel");
+const GEMINI_MODEL_SYMBOL = /* @__PURE__ */ Symbol.for("google.adk.geminiModel");
 function isGemini(obj) {
   return typeof obj === "object" && obj !== null && GEMINI_MODEL_SYMBOL in obj && obj[GEMINI_MODEL_SYMBOL] === true;
 }
@@ -91,9 +91,14 @@ const _Gemini = class _Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI
   /**
    * Sends a request to the Gemini model.
    *
-   * @param llmRequest LlmRequest, the request to send to the Gemini model.
-   * @param stream bool = false, whether to do streaming call.
-   * @yields LlmResponse: The model response.
+   * @param llmRequest The request to send to the Gemini model.
+   * @param stream Whether to make a streaming call. Defaults to `false`.
+   * @yields Each {@link LlmResponse} produced by the model. A non-streaming
+   *     call yields exactly one; a streaming call yields the partial responses
+   *     as they arrive, then a final aggregated response whenever the stream
+   *     carried either content or usage/grounding/citation metadata. A
+   *     metadata-only stream therefore still ends with one, holding the
+   *     metadata and no `content`; only a stream with neither ends silently.
    */
   async *generateContentAsync(llmRequest, stream = false, abortSignal) {
     var _a2, _b2, _c;

package/dist/cjs/models/interactions_utils.js

@@ -695,13 +695,12 @@ async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
     });
     const aggregatedParts = [];
     for await (const event of responses) {
-      const sseEvent = event;
-      const interactionId = extractStreamInteractionId(sseEvent);
+      const interactionId = extractStreamInteractionId(event);
       if (interactionId) {
         currentInteractionId = interactionId;
       }
       const llmResponse = convertInteractionEventToLlmResponse(
-        sseEvent,
+        event,
         aggregatedParts,
         currentInteractionId
       );

package/dist/cjs/models/registry.js

@@ -30,6 +30,7 @@ __export(registry_exports, {
 module.exports = __toCommonJS(registry_exports);
 var import_logger = require("../utils/logger.js");
 var import_apigee_llm = require("./apigee_llm.js");
+var import_chrome_prompt_llm = require("./chrome_prompt_llm.js");
 var import_google_llm = require("./google_llm.js");
 /**
  * @license
@@ -121,6 +122,7 @@ _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(import_google_llm.Gemini);
 LLMRegistry.register(import_apigee_llm.ApigeeLlm);
+LLMRegistry.register(import_chrome_prompt_llm.ChromeBuiltInLlm);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LLMRegistry