@google/adk 1.6.0 → 2.0.0

package/dist/esm/plugins/global_instruction_plugin.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -8,7 +10,7 @@ const require = topLevelCreateRequire(import.meta.url);
 import { injectSessionState } from "../agents/instructions.js";
 import { ReadonlyContext } from "../agents/readonly_context.js";
 import { BasePlugin } from "./base_plugin.js";
-class GlobalInstructionPlugin extends BasePlugin {
+const _GlobalInstructionPlugin = class _GlobalInstructionPlugin extends BasePlugin {
   /**
    * Initializes the GlobalInstructionPlugin.
    *
@@ -63,7 +65,9 @@ ${existingInstruction}`;
     }
     return await this.globalInstruction(readonlyContext);
   }
-}
+};
+__name(_GlobalInstructionPlugin, "GlobalInstructionPlugin");
+let GlobalInstructionPlugin = _GlobalInstructionPlugin;
 export {
   GlobalInstructionPlugin
 };

package/dist/esm/plugins/logging_plugin.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -12,7 +14,7 @@ import {
 } from "../events/event.js";
 import { logger } from "../utils/logger.js";
 import { BasePlugin } from "./base_plugin.js";
-class LoggingPlugin extends BasePlugin {
+const _LoggingPlugin = class _LoggingPlugin extends BasePlugin {
   /**
    * Initialize the logging plugin.
    *
@@ -25,13 +27,13 @@ class LoggingPlugin extends BasePlugin {
     invocationContext,
     userMessage
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u{1F680} USER MESSAGE RECEIVED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Session ID: ${invocationContext.session.id}`);
     this.log(`   User ID: ${invocationContext.userId}`);
     this.log(`   App Name: ${invocationContext.appName}`);
-    this.log(`   Root Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(`   Root Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`);
     this.log(`   User Content: ${this.formatContent(userMessage)}`);
     if (invocationContext.branch) {
       this.log(`   Branch: ${invocationContext.branch}`);
@@ -41,10 +43,12 @@ class LoggingPlugin extends BasePlugin {
   async beforeRunCallback({
     invocationContext
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u{1F3C3} INVOCATION STARTING");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
-    this.log(`   Starting Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(
+      `   Starting Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`
+    );
     return void 0;
   }
   async onEventCallback({
@@ -73,10 +77,10 @@ class LoggingPlugin extends BasePlugin {
   async afterRunCallback({
     invocationContext
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u2705 INVOCATION COMPLETED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
-    this.log(`   Final Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(`   Final Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`);
     return void 0;
   }
   async beforeAgentCallback({
@@ -229,7 +233,9 @@ class LoggingPlugin extends BasePlugin {
     }
     return formatted;
   }
-}
+};
+__name(_LoggingPlugin, "LoggingPlugin");
+let LoggingPlugin = _LoggingPlugin;
 export {
   LoggingPlugin
 };

package/dist/esm/plugins/plugin_manager.js

@@ -1,12 +1,14 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 import { logger } from "../utils/logger.js";
-class PluginManager {
+const _PluginManager = class _PluginManager {
   /**
    * Initializes the plugin service.
    *
@@ -21,6 +23,10 @@ class PluginManager {
       }
     }
   }
+  /** Whether any plugin is registered. */
+  get hasPlugins() {
+    return this.plugins.size > 0;
+  }
   /**
    * Registers a new plugin.
    *
@@ -153,6 +159,34 @@ class PluginManager {
       "afterAgentCallback"
     );
   }
+  /**
+   * Runs the `beforeNodeCallback` for all plugins.
+   */
+  async runBeforeNodeCallback({
+    node,
+    nodeContext,
+    input
+  }) {
+    return this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.beforeNodeCallback({ node, nodeContext, input }),
+      "beforeNodeCallback"
+    );
+  }
+  /**
+   * Runs the `afterNodeCallback` for all plugins.
+   */
+  async runAfterNodeCallback({
+    node,
+    nodeContext,
+    output
+  }) {
+    return this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.afterNodeCallback({ node, nodeContext, output }),
+      "afterNodeCallback"
+    );
+  }
   /**
    * Runs the `beforeToolSelection` for all plugins.
    */
@@ -276,7 +310,9 @@ class PluginManager {
       "onToolErrorCallback"
     );
   }
-}
+};
+__name(_PluginManager, "PluginManager");
+let PluginManager = _PluginManager;
 export {
   PluginManager
 };

package/dist/esm/plugins/security_plugin.js

@@ -1,12 +1,14 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { REQUEST_CONFIRMATION_FUNCTION_CALL_NAME } from "../agents/functions.js";
 import { BasePlugin } from "../plugins/base_plugin.js";
-const REQUEST_CONFIRMATION_FUNCTION_CALL_NAME = "adk_request_confirmation";
 const TOOL_CALL_SECURITY_CHECK_STATES = "orcas_tool_call_security_check_states";
 const INTERMEDIATE_REQUIRE_TOOL_CALL_CONFIRMATION_ERROR = "This tool call needs external confirmation before completion.";
 var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
@@ -15,20 +17,23 @@ var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
   PolicyOutcome2["ALLOW"] = "ALLOW";
   return PolicyOutcome2;
 })(PolicyOutcome || {});
-class InMemoryPolicyEngine {
+const _InMemoryPolicyEngine = class _InMemoryPolicyEngine {
   /**
    * Always returns {@link PolicyOutcome.ALLOW} for every tool call.
    *
+   * @param _context - The tool and its arguments; ignored.
    * @returns A promise resolving to an ALLOW result.
    */
-  async evaluate() {
+  async evaluate(_context) {
     return Promise.resolve({
       outcome: "ALLOW" /* ALLOW */,
       reason: "For prototyping purpose, all tool calls are allowed."
     });
   }
-}
-class SecurityPlugin extends BasePlugin {
+};
+__name(_InMemoryPolicyEngine, "InMemoryPolicyEngine");
+let InMemoryPolicyEngine = _InMemoryPolicyEngine;
+const _SecurityPlugin = class _SecurityPlugin extends BasePlugin {
   /**
    * @param params - Optional configuration. Defaults to {@link InMemoryPolicyEngine}
    *   when no policy engine is provided.
@@ -118,7 +123,9 @@ class SecurityPlugin extends BasePlugin {
         return;
     }
   }
-}
+};
+__name(_SecurityPlugin, "SecurityPlugin");
+let SecurityPlugin = _SecurityPlugin;
 function getAskUserConfirmationFunctionCalls(event) {
   if (!event.content || !event.content.parts) {
     return [];
@@ -131,6 +138,7 @@ function getAskUserConfirmationFunctionCalls(event) {
   }
   return results;
 }
+__name(getAskUserConfirmationFunctionCalls, "getAskUserConfirmationFunctionCalls");
 export {
   InMemoryPolicyEngine,
   PolicyOutcome,

package/dist/esm/runner/in_memory_runner.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -9,7 +11,7 @@ import { InMemoryArtifactService } from "../artifacts/in_memory_artifact_service
 import { InMemoryMemoryService } from "../memory/in_memory_memory_service.js";
 import { InMemorySessionService } from "../sessions/in_memory_session_service.js";
 import { Runner } from "./runner.js";
-class InMemoryRunner extends Runner {
+const _InMemoryRunner = class _InMemoryRunner extends Runner {
   /**
    * Creates a new InMemoryRunner instance.
    *
@@ -40,7 +42,9 @@ class InMemoryRunner extends Runner {
       resumabilityConfig: (_a = app == null ? void 0 : app.resumabilityConfig) != null ? _a : resumabilityConfig
     });
   }
-}
+};
+__name(_InMemoryRunner, "InMemoryRunner");
+let InMemoryRunner = _InMemoryRunner;
 export {
   InMemoryRunner
 };

package/dist/esm/runner/runner.js

@@ -1,17 +1,22 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 var _a;
-import { createPartFromText } from "@google/genai";
+import { createPartFromText, Modality } from "@google/genai";
 import { context, trace } from "@opentelemetry/api";
+import { isBaseAgent } from "../agents/base_agent.js";
+import { reservedFunctionCallName } from "../agents/framework_function_calls.js";
 import { findMatchingFunctionCall } from "../agents/functions.js";
 import {
   InvocationContext,
-  newInvocationContextId
+  newInvocationContextId,
+  requireAgent
 } from "../agents/invocation_context.js";
 import { isLlmAgent } from "../agents/llm_agent.js";
 import { createRunConfig } from "../agents/run_config.js";
@@ -30,12 +35,17 @@ import {
 import { isBaseToolset } from "../tools/base_toolset.js";
 import { logger } from "../utils/logger.js";
 import { isGemini2OrAbove } from "../utils/model_name.js";
+import {
+  asRunnableRoot,
+  runNodeAsInvocation
+} from "../workflow/run_node_as_invocation.js";
 const RUNNER_SIGNATURE_SYMBOL = Symbol.for("google.adk.runner");
 function isRunner(obj) {
   return typeof obj === "object" && obj !== null && RUNNER_SIGNATURE_SYMBOL in obj && obj[RUNNER_SIGNATURE_SYMBOL] === true;
 }
+__name(isRunner, "isRunner");
 _a = RUNNER_SIGNATURE_SYMBOL;
-class Runner {
+const _Runner = class _Runner {
   /**
    * Creates a new Runner instance.
    *
@@ -52,7 +62,7 @@ class Runner {
       );
     }
     this.appName = appName;
-    this.agent = agent;
+    this.agent = asRunnableRoot(agent);
     const appPlugins = (_f = (_e = input.app) == null ? void 0 : _e.plugins) != null ? _f : [];
     const configPlugins = (_g = input.plugins) != null ? _g : [];
     this.pluginManager = new PluginManager([...appPlugins, ...configPlugins]);
@@ -113,6 +123,7 @@ class Runner {
     if (newMessage && !newMessage.role) {
       newMessage.role = "user";
     }
+    rejectReservedFunctionCalls(newMessage);
     const span = tracer.startSpan("invocation");
     const ctx = trace.setSpan(context.active(), span);
     try {
@@ -159,7 +170,7 @@ class Runner {
             memoryService: this.memoryService,
             credentialService: this.credentialService,
             invocationId: newInvocationContextId(),
-            agent: this.agent,
+            agent: isBaseAgent(this.agent) ? this.agent : void 0,
             session,
             userContent: newMessage,
             runConfig,
@@ -205,10 +216,12 @@ class Runner {
               return;
             }
           }
-          invocationContext.agent = this.determineAgentForResumption(
-            session,
-            this.agent
-          );
+          if (isBaseAgent(this.agent)) {
+            invocationContext.agent = this.determineAgentForResumption(
+              session,
+              this.agent
+            );
+          }
           if (newMessage) {
             const beforeRunCallbackResponse = await this.pluginManager.runBeforeRunCallback({
               invocationContext
@@ -231,9 +244,7 @@ class Runner {
               }
               yield earlyExitEvent;
             } else {
-              for await (const event of invocationContext.agent.runAsync(
-                invocationContext
-              )) {
+              for await (const event of this.runRoot(invocationContext)) {
                 if ((_h = params.abortSignal) == null ? void 0 : _h.aborted) {
                   return;
                 }
@@ -270,10 +281,29 @@ class Runner {
       );
     } finally {
       span.end();
-      const toolsets = getAllToolsets(this.agent);
+      const toolsets = isBaseAgent(this.agent) ? getAllToolsets(this.agent) : [];
       await Promise.allSettled(toolsets.map((t) => t.close()));
     }
   }
+  /**
+   * Runs whatever this runner was given as its root.
+   *
+   * An agent is run through `runAsync`, as always. A bare node — a `Workflow`
+   * handed to the runner directly — is driven by {@link runNodeAsInvocation}.
+   *
+   * Only the execution differs. Everything around it (the run callbacks, event
+   * persistence, cancellation) is shared, so the node path cannot drift from
+   * the agent path on the things that are not about execution. adk-python has
+   * two separate loops here and a TODO noting its node one lacks tracing and
+   * plugins; there is nothing to lack if there is only one loop.
+   */
+  async *runRoot(invocationContext) {
+    if (isBaseAgent(this.agent)) {
+      yield* requireAgent(invocationContext).runAsync(invocationContext);
+      return;
+    }
+    yield* runNodeAsInvocation(this.agent, invocationContext);
+  }
   /**
    * Saves artifacts from the message parts and replaces the inline data with
    * a file name placeholder and optional file reference.
@@ -372,8 +402,150 @@ class Runner {
   isRoutableLlmAgent(agentToRun) {
     return isRoutableLlmAgent(agentToRun);
   }
-  // TODO - b/425992518: Implement runLive and related methods.
+  /**
+   * Runs the agent in the live (bidirectional streaming) mode.
+   *
+   * Model media events that carry raw inline bytes (audio, video, or image)
+   * are yielded but not appended to the session to avoid persisting large
+   * blobs; events with `fileData` references and most other live events
+   * (transcriptions, tool calls, usage) are persisted as in `runAsync`.
+   *
+   * This feature is **experimental** and its API may change.
+   *
+   * @param params.userId The user ID of the session.
+   * @param params.sessionId The session ID of the session.
+   * @param params.liveRequestQueue The queue used to feed the live model.
+   * @param params.runConfig The run config for the agent.
+   * @param params.abortSignal Optional signal to abort the live run.
+   * @param params.liveSessionResumptionHandle Optional session resumption
+   *     handle observed from a prior `runLive` cycle on the same conversation.
+   *     When set, the agent's live flow opens the connection with
+   *     `liveConnectConfig.sessionResumption.handle` so the server restores its
+   *     state instead of relying on client-side history replay.
+   * @yields The events generated by the agent.
+   */
+  async *runLive(params) {
+    var _a2, _b, _c, _d;
+    if (!params.liveRequestQueue) {
+      throw new Error("liveRequestQueue is required for runLive.");
+    }
+    if (!isBaseAgent(this.agent)) {
+      throw new Error("runLive is only supported for agents.");
+    }
+    const agent = this.agent;
+    const runConfig = createRunConfig(params.runConfig);
+    if (!((_a2 = runConfig.responseModalities) == null ? void 0 : _a2.length)) {
+      runConfig.responseModalities = [Modality.AUDIO];
+    }
+    if ((_b = agent.subAgents) == null ? void 0 : _b.length) {
+      if (runConfig.responseModalities.includes(Modality.AUDIO)) {
+        (_c = runConfig.outputAudioTranscription) != null ? _c : runConfig.outputAudioTranscription = {};
+      }
+      (_d = runConfig.inputAudioTranscription) != null ? _d : runConfig.inputAudioTranscription = {};
+    }
+    const span = tracer.startSpan("invocation");
+    const ctx = trace.setSpan(context.active(), span);
+    try {
+      yield* runAsyncGeneratorWithOtelContext(
+        ctx,
+        this,
+        async function* () {
+          var _a3, _b2, _c2, _d2;
+          const session = await this.sessionService.getOrCreateSession({
+            appName: this.appName,
+            userId: params.userId,
+            sessionId: params.sessionId
+          });
+          if ((_a3 = params.abortSignal) == null ? void 0 : _a3.aborted) {
+            return;
+          }
+          const invocationContext = new InvocationContext({
+            artifactService: this.artifactService ? new ScopedArtifactService(
+              this.artifactService,
+              this.appName,
+              params.userId,
+              params.sessionId
+            ) : void 0,
+            sessionService: this.sessionService,
+            memoryService: this.memoryService,
+            credentialService: this.credentialService,
+            invocationId: newInvocationContextId(),
+            agent,
+            session,
+            runConfig,
+            pluginManager: this.pluginManager,
+            liveRequestQueue: params.liveRequestQueue,
+            abortSignal: params.abortSignal,
+            liveSessionResumptionHandle: params.liveSessionResumptionHandle
+          });
+          invocationContext.agent = this.determineAgentForResumption(
+            session,
+            agent
+          );
+          const beforeRunCallbackResponse = await this.pluginManager.runBeforeRunCallback({
+            invocationContext
+          });
+          if ((_b2 = params.abortSignal) == null ? void 0 : _b2.aborted) {
+            return;
+          }
+          if (beforeRunCallbackResponse) {
+            const earlyExitEvent = createEvent({
+              invocationId: invocationContext.invocationId,
+              author: "model",
+              content: beforeRunCallbackResponse
+            });
+            await this.sessionService.appendEvent({
+              session,
+              event: earlyExitEvent
+            });
+            yield earlyExitEvent;
+            return;
+          }
+          for await (const event of requireAgent(invocationContext).runLive(
+            invocationContext
+          )) {
+            if ((_c2 = params.abortSignal) == null ? void 0 : _c2.aborted) {
+              return;
+            }
+            const modifiedEvent = await this.pluginManager.runOnEventCallback({
+              invocationContext,
+              event
+            });
+            if ((_d2 = params.abortSignal) == null ? void 0 : _d2.aborted) {
+              return;
+            }
+            const eventToProcess = modifiedEvent != null ? modifiedEvent : event;
+            if (!eventToProcess.partial && !isLiveModelMediaEventWithInlineData(eventToProcess)) {
+              await this.sessionService.appendEvent({
+                session,
+                event: eventToProcess
+              });
+            }
+            yield eventToProcess;
+          }
+          await this.pluginManager.runAfterRunCallback({ invocationContext });
+        }
+      );
+    } finally {
+      span.end();
+    }
+  }
+};
+__name(_Runner, "Runner");
+let Runner = _Runner;
+function isLiveModelMediaEventWithInlineData(event) {
+  var _a2;
+  const parts = (_a2 = event.content) == null ? void 0 : _a2.parts;
+  if (!(parts == null ? void 0 : parts.length)) {
+    return false;
+  }
+  return parts.some((part) => {
+    var _a3, _b;
+    const mimeType = (_b = (_a3 = part.inlineData) == null ? void 0 : _a3.mimeType) == null ? void 0 : _b.toLowerCase();
+    return mimeType !== void 0 && (mimeType.startsWith("audio/") || mimeType.startsWith("video/") || mimeType.startsWith("image/"));
+  });
 }
+__name(isLiveModelMediaEventWithInlineData, "isLiveModelMediaEventWithInlineData");
 function determineAgentForResumption(session, rootAgent, resumabilityConfig) {
   const event = findEventByLastFunctionResponseId(session.events);
   const isResumable = Boolean(resumabilityConfig == null ? void 0 : resumabilityConfig.isResumable);
@@ -382,9 +554,11 @@ function determineAgentForResumption(session, rootAgent, resumabilityConfig) {
     if (resumedAgent) {
       return resumedAgent;
     }
-    logger.warn(
-      `Function response from an unknown agent: ${event.author}, event id: ${event.id}`
-    );
+    if (!isWorkflowNodeEvent(event)) {
+      logger.warn(
+        `Function response from an unknown agent: ${event.author}, event id: ${event.id}`
+      );
+    }
   }
   for (let i = session.events.length - 1; i >= 0; i--) {
     logger.debug("event:", JSON.stringify(session.events[i]));
@@ -397,9 +571,11 @@ function determineAgentForResumption(session, rootAgent, resumabilityConfig) {
     }
     const agent = rootAgent.findSubAgent(event2.author);
     if (!agent) {
-      logger.warn(
-        `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
-      );
+      if (!isWorkflowNodeEvent(event2)) {
+        logger.warn(
+          `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
+        );
+      }
       continue;
     }
     if (isRoutableLlmAgent(agent)) {
@@ -408,6 +584,12 @@ function determineAgentForResumption(session, rootAgent, resumabilityConfig) {
   }
   return rootAgent;
 }
+__name(determineAgentForResumption, "determineAgentForResumption");
+function isWorkflowNodeEvent(event) {
+  var _a2;
+  return Boolean((_a2 = event.nodeInfo) == null ? void 0 : _a2.path);
+}
+__name(isWorkflowNodeEvent, "isWorkflowNodeEvent");
 function isRoutableLlmAgent(agentToRun) {
   let agent = agentToRun;
   while (agent) {
@@ -421,10 +603,12 @@ function isRoutableLlmAgent(agentToRun) {
   }
   return true;
 }
+__name(isRoutableLlmAgent, "isRoutableLlmAgent");
 function findEventByLastFunctionResponseId(events) {
   var _a2;
   return (_a2 = findMatchingFunctionCall(events)) != null ? _a2 : null;
 }
+__name(findEventByLastFunctionResponseId, "findEventByLastFunctionResponseId");
 function getAllToolsets(agent) {
   const toolsets = [];
   const visited = /* @__PURE__ */ new Set();
@@ -442,9 +626,20 @@ function getAllToolsets(agent) {
       traverse(sub);
     }
   }
+  __name(traverse, "traverse");
   traverse(agent);
   return toolsets;
 }
+__name(getAllToolsets, "getAllToolsets");
+function rejectReservedFunctionCalls(newMessage) {
+  const reserved = reservedFunctionCallName(newMessage);
+  if (reserved) {
+    throw new Error(
+      `A client message may not contain a '${reserved}' function call: it is raised by the framework, and can only be answered with a function response.`
+    );
+  }
+}
+__name(rejectReservedFunctionCalls, "rejectReservedFunctionCalls");
 export {
   Runner,
   determineAgentForResumption,

package/dist/esm/sessions/base_session_service.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -7,7 +9,8 @@ const require = topLevelCreateRequire(import.meta.url);
  */
 import { cloneDeep } from "lodash-es";
 import { State } from "./state.js";
-class BaseSessionService {
+import { carryDeltaStamps, shouldApplyDeltaWrite } from "./state_write_order.js";
+const _BaseSessionService = class _BaseSessionService {
   /**
    * Gets a session or creates one if it doesn't exist.
    *
@@ -61,6 +64,9 @@ class BaseSessionService {
       if (key.startsWith(State.TEMP_PREFIX)) {
         continue;
       }
+      if (!shouldApplyDeltaWrite(session.state, event.actions.stateDelta, key)) {
+        continue;
+      }
       Object.defineProperty(session.state, key, {
         value,
         writable: true,
@@ -69,7 +75,9 @@ class BaseSessionService {
       });
     }
   }
-}
+};
+__name(_BaseSessionService, "BaseSessionService");
+let BaseSessionService = _BaseSessionService;
 function trimTempDeltaState(event) {
   if (!event.actions || !event.actions.stateDelta) {
     return event;
@@ -81,9 +89,11 @@ function trimTempDeltaState(event) {
       filteredStateDelta[key] = value;
     }
   }
+  carryDeltaStamps(stateDelta, filteredStateDelta);
   event.actions.stateDelta = filteredStateDelta;
   return event;
 }
+__name(trimTempDeltaState, "trimTempDeltaState");
 function trimTempState(state) {
   const filteredState = /* @__PURE__ */ Object.create(null);
   for (const [key, value] of Object.entries(state)) {
@@ -93,6 +103,7 @@ function trimTempState(state) {
   }
   return filteredState;
 }
+__name(trimTempState, "trimTempState");
 function mergeStates(appState = {}, userState = {}, sessionState = {}) {
   const merged = cloneDeep(sessionState);
   for (const [k, v] of Object.entries(appState)) {
@@ -103,6 +114,7 @@ function mergeStates(appState = {}, userState = {}, sessionState = {}) {
   }
   return merged;
 }
+__name(mergeStates, "mergeStates");
 export {
   BaseSessionService,
   mergeStates,