@google/adk 0.1.0

package/dist/cjs/agents/instructions.js

@@ -0,0 +1,110 @@
+/**
+  * @license
+  * Copyright 2025 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 instructions_exports = {};
+__export(instructions_exports, {
+  injectSessionState: () => injectSessionState
+});
+module.exports = __toCommonJS(instructions_exports);
+var import_state = require("../sessions/state.js");
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+async function injectSessionState(template, readonlyContext) {
+  const invocationContext = readonlyContext.invocationContext;
+  async function replaceMatchedKeyWithItsValue(match) {
+    let key = match[0].replace(/^\{+/, "").replace(/\}+$/, "").trim();
+    const isOptional = key.endsWith("?");
+    if (isOptional) {
+      key = key.slice(0, -1);
+    }
+    if (key.startsWith("artifact.")) {
+      const fileName = key.substring("artifact.".length);
+      if (invocationContext.artifactService === void 0) {
+        throw new Error("Artifact service is not initialized.");
+      }
+      const artifact = await invocationContext.artifactService.loadArtifact({
+        appName: invocationContext.session.appName,
+        userId: invocationContext.session.userId,
+        sessionId: invocationContext.session.id,
+        filename: fileName
+      });
+      if (!artifact) {
+        throw new Error(`Artifact ${fileName} not found.`);
+      }
+      return String(artifact);
+    }
+    if (!isValidStateName(key)) {
+      return match[0];
+    }
+    if (key in invocationContext.session.state) {
+      return String(invocationContext.session.state[key]);
+    }
+    if (isOptional) {
+      return "";
+    }
+    throw new Error(`Context variable not found: \`${key}\`.`);
+  }
+  const pattern = /\{+[^{}]*}+/g;
+  const result = [];
+  let lastEnd = 0;
+  const matches = template.matchAll(pattern);
+  for (const match of matches) {
+    result.push(template.slice(lastEnd, match.index));
+    const replacement = await replaceMatchedKeyWithItsValue(match);
+    result.push(replacement);
+    lastEnd = match.index + match[0].length;
+  }
+  result.push(template.slice(lastEnd));
+  return result.join("");
+}
+const isIdentifierPattern = /* @__PURE__ */ (() => {
+  return /^[a-zA-Z_][a-zA-Z0-9_]*$/;
+})();
+function isIdentifier(s) {
+  if (s === "" || s === void 0) {
+    return false;
+  }
+  return isIdentifierPattern.test(s);
+}
+const VALID_PREFIXES = [import_state.State.APP_PREFIX, import_state.State.USER_PREFIX, import_state.State.TEMP_PREFIX];
+function isValidStateName(variableName) {
+  const parts = variableName.split(":");
+  if (parts.length === 0 || parts.length > 2) {
+    return false;
+  }
+  if (parts.length === 1) {
+    return isIdentifier(variableName);
+  }
+  if (VALID_PREFIXES.includes(parts[0] + ":")) {
+    return isIdentifier(parts[1]);
+  }
+  return false;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  injectSessionState
+});

package/dist/cjs/agents/invocation_context.js

@@ -0,0 +1,109 @@
+/**
+  * @license
+  * Copyright 2025 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 invocation_context_exports = {};
+__export(invocation_context_exports, {
+  InvocationContext: () => InvocationContext,
+  newInvocationContextId: () => newInvocationContextId
+});
+module.exports = __toCommonJS(invocation_context_exports);
+var import_env_aware_utils = require("../utils/env_aware_utils.js");
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class InvocationCostManager {
+  constructor() {
+    this.numberOfLlmCalls = 0;
+  }
+  /**
+   * Increments the number of llm calls and enforces the limit.
+   *
+   * @param runConfig the run config of the invocation.
+   * @throws If number of llm calls made exceed the set threshold.
+   */
+  incrementAndEnforceLlmCallsLimit(runConfig) {
+    this.numberOfLlmCalls++;
+    if (runConfig && runConfig.maxLlmCalls > 0 && this.numberOfLlmCalls > runConfig.maxLlmCalls) {
+      throw new Error(
+        `Max number of llm calls limit of ${runConfig.maxLlmCalls} exceeded`
+      );
+    }
+  }
+}
+class InvocationContext {
+  /**
+   * @param params The parameters for creating an invocation context.
+   */
+  constructor(params) {
+    /**
+     * A container to keep track of different kinds of costs incurred as a part of
+     * this invocation.
+     */
+    this.invocationCostManager = new InvocationCostManager();
+    this.artifactService = params.artifactService;
+    this.sessionService = params.sessionService;
+    this.memoryService = params.memoryService;
+    this.invocationId = params.invocationId;
+    this.branch = params.branch;
+    this.agent = params.agent;
+    this.userContent = params.userContent;
+    this.session = params.session;
+    this.endInvocation = params.endInvocation || false;
+    this.transcriptionCache = params.transcriptionCache;
+    this.runConfig = params.runConfig;
+    this.liveRequestQueue = params.liveRequestQueue;
+    this.activeStreamingTools = params.activeStreamingTools;
+    this.pluginManager = params.pluginManager;
+  }
+  /**
+   * The app name of the current session.
+   */
+  get appName() {
+    return this.session.appName;
+  }
+  /**
+   * The user ID of the current session.
+   */
+  get userId() {
+    return this.session.userId;
+  }
+  /**
+   * Tracks number of llm calls made.
+   *
+   * @throws If number of llm calls made exceed the set threshold.
+   */
+  incrementLlmCallCount() {
+    this.invocationCostManager.incrementAndEnforceLlmCallsLimit(this.runConfig);
+  }
+}
+function newInvocationContextId() {
+  return `e-${(0, import_env_aware_utils.randomUUID)()}`;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  InvocationContext,
+  newInvocationContextId
+});

package/dist/cjs/agents/live_request_queue.js

@@ -0,0 +1,136 @@
+/**
+  * @license
+  * Copyright 2025 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 live_request_queue_exports = {};
+__export(live_request_queue_exports, {
+  LiveRequestQueue: () => LiveRequestQueue
+});
+module.exports = __toCommonJS(live_request_queue_exports);
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class LiveRequestQueue {
+  constructor() {
+    // Keeps track of the data that are waiting to be sent.
+    this.queue = [];
+    // Keeps track of the promises that are waiting for data.
+    this.resolveFnFifoQueue = [];
+    this.isClosed = false;
+  }
+  /**
+   * Adds a request to the queue. If there is a pending `get()` call, it
+   * will be resolved with the given request.
+   * @param req The request to send.
+   */
+  send(req) {
+    if (this.isClosed) {
+      throw new Error("Cannot send to a closed queue.");
+    }
+    if (this.resolveFnFifoQueue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      resolve(req);
+    } else {
+      this.queue.push(req);
+    }
+  }
+  /**
+   * Retrieves a request from the queue. If the queue is empty, it will
+   * wait until a request is available.
+   * @returns A promise that resolves with the next available request.
+   */
+  async get() {
+    if (this.queue.length > 0) {
+      return this.queue.shift();
+    }
+    if (this.isClosed) {
+      return { close: true };
+    }
+    return new Promise((resolve) => {
+      this.resolveFnFifoQueue.push(resolve);
+    });
+  }
+  /**
+   * Sends a close signal to the queue.
+   */
+  close() {
+    if (this.isClosed) {
+      return;
+    }
+    this.isClosed = true;
+    while (this.resolveFnFifoQueue.length > 0 && this.queue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      const req = this.queue.shift();
+      resolve(req);
+    }
+    const closeRequest = { close: true };
+    while (this.resolveFnFifoQueue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      resolve(closeRequest);
+    }
+  }
+  /**
+   * Sends a content object to the queue.
+   * @param content The content to send.
+   */
+  sendContent(content) {
+    this.send({ content });
+  }
+  /**
+   * Sends a blob to the model in realtime mode.
+   * @param blob The blob to send.
+   */
+  sendRealtime(blob) {
+    this.send({ blob });
+  }
+  /**
+   * Sends an activity start signal to mark the beginning of user input.
+   */
+  sendActivityStart() {
+    this.send({ activityStart: {} });
+  }
+  /**
+   * Sends an activity end signal to mark the end of user input.
+   */
+  sendActivityEnd() {
+    this.send({ activityEnd: {} });
+  }
+  /**
+   * Implements the async iterator protocol.
+   */
+  async *[Symbol.asyncIterator]() {
+    while (true) {
+      const request = await this.get();
+      yield request;
+      if (request.close) {
+        break;
+      }
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LiveRequestQueue
+});