@google/adk 1.2.0 → 1.4.0

package/dist/cjs/plugins/base_plugin.js

@@ -45,6 +45,7 @@ var import_experimental = require("../utils/experimental.js");
 var ContextCompactionTrigger = /* @__PURE__ */ ((ContextCompactionTrigger2) => {
   ContextCompactionTrigger2["Manual"] = "Manual";
   ContextCompactionTrigger2["Auto"] = "Auto";
+  ContextCompactionTrigger2["AgentControlled"] = "AgentControlled";
   return ContextCompactionTrigger2;
 })(ContextCompactionTrigger || {});
 class BasePlugin {

package/dist/cjs/plugins/security_plugin.js

@@ -47,6 +47,11 @@ var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
   return PolicyOutcome2;
 })(PolicyOutcome || {});
 class InMemoryPolicyEngine {
+  /**
+   * Always returns {@link PolicyOutcome.ALLOW} for every tool call.
+   *
+   * @returns A promise resolving to an ALLOW result.
+   */
   async evaluate() {
     return Promise.resolve({
       outcome: "ALLOW" /* ALLOW */,
@@ -55,11 +60,22 @@ class InMemoryPolicyEngine {
   }
 }
 class SecurityPlugin extends import_base_plugin.BasePlugin {
+  /**
+   * @param params - Optional configuration. Defaults to {@link InMemoryPolicyEngine}
+   *   when no policy engine is provided.
+   */
   constructor(params) {
     var _a;
     super("security_plugin");
     this.policyEngine = (_a = params == null ? void 0 : params.policyEngine) != null ? _a : new InMemoryPolicyEngine();
   }
+  /**
+   * Intercepts tool calls, evaluating them against the policy engine before
+   * execution and handling confirmation flows for calls that require it.
+   *
+   * @returns A partial or error response object if the call is blocked or
+   *   awaiting confirmation, or `undefined` to allow execution to proceed.
+   */
   async beforeToolCallback({
     tool,
     toolArgs,

package/dist/cjs/runner/in_memory_runner.js

@@ -41,19 +41,30 @@ class InMemoryRunner extends import_runner.Runner {
    * Creates a new InMemoryRunner instance.
    *
    * @param params The configuration for the runner.
+   * @param params.app An optional application instance to run.
    * @param params.agent The root agent to run.
    * @param params.appName The application name. Defaults to `'InMemoryRunner'`.
    * @param params.plugins An optional list of plugins.
+   * @param params.resumabilityConfig An optional resumability configuration.
    */
   constructor(params) {
-    const { agent, appName = "InMemoryRunner", plugins = [] } = params;
+    var _a;
+    const {
+      agent,
+      appName = "InMemoryRunner",
+      plugins = [],
+      app,
+      resumabilityConfig
+    } = params;
     super({
+      app,
       appName,
       agent,
       plugins,
       artifactService: new import_in_memory_artifact_service.InMemoryArtifactService(),
       sessionService: new import_in_memory_session_service.InMemorySessionService(),
-      memoryService: new import_in_memory_memory_service.InMemoryMemoryService()
+      memoryService: new import_in_memory_memory_service.InMemoryMemoryService(),
+      resumabilityConfig: (_a = app == null ? void 0 : app.resumabilityConfig) != null ? _a : resumabilityConfig
     });
   }
 }

package/dist/cjs/runner/runner.js

@@ -25,14 +25,19 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var runner_exports = {};
 __export(runner_exports, {
   Runner: () => Runner,
+  determineAgentForResumption: () => determineAgentForResumption,
+  findEventByLastFunctionResponseId: () => findEventByLastFunctionResponseId,
+  isRoutableLlmAgent: () => isRoutableLlmAgent,
   isRunner: () => isRunner
 });
 module.exports = __toCommonJS(runner_exports);
 var import_genai = require("@google/genai");
 var import_api = require("@opentelemetry/api");
+var import_functions = require("../agents/functions.js");
 var import_invocation_context = require("../agents/invocation_context.js");
 var import_llm_agent = require("../agents/llm_agent.js");
 var import_run_config = require("../agents/run_config.js");
+var import_scoped_artifact_service = require("../artifacts/scoped_artifact_service.js");
 var import_built_in_code_executor = require("../code_executors/built_in_code_executor.js");
 var import_event = require("../events/event.js");
 var import_event_actions = require("../events/event_actions.js");
@@ -60,14 +65,24 @@ class Runner {
    */
   constructor(input) {
     this[_a] = true;
-    var _a2;
-    this.appName = input.appName;
-    this.agent = input.agent;
-    this.pluginManager = new import_plugin_manager.PluginManager((_a2 = input.plugins) != null ? _a2 : []);
+    var _a2, _b, _c, _d, _e, _f, _g, _h, _i;
+    const appName = (_b = (_a2 = input.app) == null ? void 0 : _a2.name) != null ? _b : input.appName;
+    const agent = (_d = (_c = input.app) == null ? void 0 : _c.rootAgent) != null ? _d : input.agent;
+    if (!agent) {
+      throw new Error(
+        "agent must be provided in runner constructor (or via app.rootAgent)"
+      );
+    }
+    this.appName = appName;
+    this.agent = agent;
+    const appPlugins = (_f = (_e = input.app) == null ? void 0 : _e.plugins) != null ? _f : [];
+    const configPlugins = (_g = input.plugins) != null ? _g : [];
+    this.pluginManager = new import_plugin_manager.PluginManager([...appPlugins, ...configPlugins]);
     this.artifactService = input.artifactService;
     this.sessionService = input.sessionService;
     this.memoryService = input.memoryService;
     this.credentialService = input.credentialService;
+    this.resumabilityConfig = (_i = (_h = input.app) == null ? void 0 : _h.resumabilityConfig) != null ? _i : input.resumabilityConfig;
   }
   /**
    * Runs the agent with a new, ephemeral session.
@@ -117,6 +132,9 @@ class Runner {
     const { userId, sessionId, stateDelta } = params;
     const runConfig = (0, import_run_config.createRunConfig)(params.runConfig);
     let newMessage = params.newMessage;
+    if (newMessage && !newMessage.role) {
+      newMessage.role = "user";
+    }
     const span = import_tracing.tracer.startSpan("invocation");
     const ctx = import_api.trace.setSpan(import_api.context.active(), span);
     try {
@@ -136,7 +154,7 @@ class Runner {
           if (!session) {
             if (!this.appName) {
               throw new Error(
-                `Session lookup failed: appName must be provided in runner constructor`
+                `Session lookup failed: appName must be provided in runner constructor (or via app.name)`
               );
             }
             throw new Error(`Session not found: ${sessionId}`);
@@ -153,7 +171,12 @@ class Runner {
             }
           }
           const invocationContext = new import_invocation_context.InvocationContext({
-            artifactService: this.artifactService,
+            artifactService: this.artifactService ? new import_scoped_artifact_service.ScopedArtifactService(
+              this.artifactService,
+              this.appName,
+              userId,
+              sessionId
+            ) : void 0,
             sessionService: this.sessionService,
             memoryService: this.memoryService,
             credentialService: this.credentialService,
@@ -280,6 +303,11 @@ class Runner {
     if (!this.artifactService || !((_a2 = message.parts) == null ? void 0 : _a2.length)) {
       return;
     }
+    const sessionKey = {
+      appName: this.appName,
+      userId,
+      sessionId
+    };
     for (let i = 0; i < message.parts.length; i++) {
       const part = message.parts[i];
       if (!part.inlineData) {
@@ -287,9 +315,7 @@ class Runner {
       }
       const fileName = `artifact_${invocationId}_${i}`;
       await this.artifactService.saveArtifact({
-        appName: this.appName,
-        userId,
-        sessionId,
+        ...sessionKey,
         filename: fileName,
         artifact: part
       });
@@ -302,86 +328,78 @@ class Runner {
    * Determines the next agent to run to continue the session. This is primarily
    * used for session resumption.
    */
-  // TODO - b/425992518: This is where LRO integration should happen.
-  // Needs clean up before we can generalize it.
+  /**
+   * Determines the next agent to run to continue the session. This is primarily
+   * used for session resumption across tool and LRO boundaries.
+   */
   determineAgentForResumption(session, rootAgent) {
-    const event = findEventByLastFunctionResponseId(session.events);
-    if (event && event.author) {
-      return rootAgent.findAgent(event.author) || rootAgent;
-    }
-    for (let i = session.events.length - 1; i >= 0; i--) {
-      import_logger.logger.info("event:", JSON.stringify(session.events[i]));
-      const event2 = session.events[i];
-      if (event2.author === "user" || !event2.author) {
-        continue;
-      }
-      if (event2.author === rootAgent.name) {
-        return rootAgent;
-      }
-      const agent = rootAgent.findSubAgent(event2.author);
-      if (!agent) {
-        import_logger.logger.warn(
-          `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
-        );
-        continue;
-      }
-      if (this.isRoutableLlmAgent(agent)) {
-        return agent;
-      }
-    }
-    return rootAgent;
+    return determineAgentForResumption(
+      session,
+      rootAgent,
+      this.resumabilityConfig
+    );
   }
   /**
    * Whether the agent to run can transfer to any other agent in the agent tree.
    *
-   * An agent is transferable if:
-   *  - It is an instance of `LlmAgent`.
-   *  - All its ancestors are also transferable (i.e., they have
-   *    `disallowTransferToParent` set to false).
-   *
    * @param agentToRun The agent to check for transferability.
    * @returns True if the agent can transfer, False otherwise.
    */
   isRoutableLlmAgent(agentToRun) {
-    let agent = agentToRun;
-    while (agent) {
-      if (!(0, import_llm_agent.isLlmAgent)(agent)) {
-        return false;
-      }
-      if (agent.disallowTransferToParent) {
-        return false;
-      }
-      agent = agent.parentAgent;
-    }
-    return true;
+    return isRoutableLlmAgent(agentToRun);
   }
   // TODO - b/425992518: Implement runLive and related methods.
 }
-function findEventByLastFunctionResponseId(events) {
-  var _a2, _b, _c, _d;
-  if (!events.length) {
-    return null;
-  }
-  const lastEvent = events[events.length - 1];
-  const functionCallId = (_d = (_c = (_b = (_a2 = lastEvent.content) == null ? void 0 : _a2.parts) == null ? void 0 : _b.find(
-    (part) => part.functionResponse
-  )) == null ? void 0 : _c.functionResponse) == null ? void 0 : _d.id;
-  if (!functionCallId) {
-    return null;
+function determineAgentForResumption(session, rootAgent, resumabilityConfig) {
+  const event = findEventByLastFunctionResponseId(session.events);
+  const isResumable = Boolean(resumabilityConfig == null ? void 0 : resumabilityConfig.isResumable);
+  if (event && event.author && isResumable) {
+    const resumedAgent = rootAgent.findAgent(event.author);
+    if (resumedAgent) {
+      return resumedAgent;
+    }
+    import_logger.logger.warn(
+      `Function response from an unknown agent: ${event.author}, event id: ${event.id}`
+    );
   }
-  for (let i = events.length - 2; i >= 0; i--) {
-    const event = events[i];
-    const functionCalls = (0, import_event.getFunctionCalls)(event);
-    if (!functionCalls) {
+  for (let i = session.events.length - 1; i >= 0; i--) {
+    import_logger.logger.debug("event:", JSON.stringify(session.events[i]));
+    const event2 = session.events[i];
+    if (event2.author === "user" || !event2.author) {
       continue;
     }
-    for (const functionCall of functionCalls) {
-      if (functionCall.id === functionCallId) {
-        return event;
-      }
+    if (event2.author === rootAgent.name) {
+      return rootAgent;
+    }
+    const agent = rootAgent.findSubAgent(event2.author);
+    if (!agent) {
+      import_logger.logger.warn(
+        `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
+      );
+      continue;
+    }
+    if (isRoutableLlmAgent(agent)) {
+      return agent;
     }
   }
-  return null;
+  return rootAgent;
+}
+function isRoutableLlmAgent(agentToRun) {
+  let agent = agentToRun;
+  while (agent) {
+    if (!(0, import_llm_agent.isLlmAgent)(agent)) {
+      return false;
+    }
+    if (agent.disallowTransferToParent) {
+      return false;
+    }
+    agent = agent.parentAgent;
+  }
+  return true;
+}
+function findEventByLastFunctionResponseId(events) {
+  var _a2;
+  return (_a2 = (0, import_functions.findMatchingFunctionCall)(events)) != null ? _a2 : null;
 }
 function getAllToolsets(agent) {
   const toolsets = [];
@@ -406,5 +424,8 @@ function getAllToolsets(agent) {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Runner,
+  determineAgentForResumption,
+  findEventByLastFunctionResponseId,
+  isRoutableLlmAgent,
   isRunner
 });

package/dist/cjs/sessions/base_session_service.js

@@ -26,7 +26,8 @@ var base_session_service_exports = {};
 __export(base_session_service_exports, {
   BaseSessionService: () => BaseSessionService,
   mergeStates: () => mergeStates,
-  trimTempDeltaState: () => trimTempDeltaState
+  trimTempDeltaState: () => trimTempDeltaState,
+  trimTempState: () => trimTempState
 });
 module.exports = __toCommonJS(base_session_service_exports);
 var import_lodash_es = require("lodash-es");
@@ -69,7 +70,12 @@ class BaseSessionService {
     }
     event = trimTempDeltaState(event);
     this.updateSessionState({ session, event });
-    session.events.push(event);
+    const index = session.events.findIndex((e) => e.id === event.id);
+    if (index >= 0) {
+      session.events[index] = event;
+    } else {
+      session.events.push(event);
+    }
     return event;
   }
   /**
@@ -103,6 +109,15 @@ function trimTempDeltaState(event) {
   event.actions.stateDelta = filteredStateDelta;
   return event;
 }
+function trimTempState(state) {
+  const filteredState = {};
+  for (const [key, value] of Object.entries(state)) {
+    if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
+      filteredState[key] = value;
+    }
+  }
+  return filteredState;
+}
 function mergeStates(appState = {}, userState = {}, sessionState = {}) {
   const merged = (0, import_lodash_es.cloneDeep)(sessionState);
   for (const [k, v] of Object.entries(appState)) {
@@ -117,5 +132,6 @@ function mergeStates(appState = {}, userState = {}, sessionState = {}) {
 0 && (module.exports = {
   BaseSessionService,
   mergeStates,
-  trimTempDeltaState
+  trimTempDeltaState,
+  trimTempState
 });

package/dist/cjs/sessions/database_session_service.js

@@ -119,7 +119,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
           appStateDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
         } else if (key.startsWith(import_state.State.USER_PREFIX)) {
           userStateDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-        } else {
+        } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
           sessionState[key] = value;
         }
       }
@@ -368,7 +368,7 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
             appDelta[key.replace(import_state.State.APP_PREFIX, "")] = value;
           } else if (key.startsWith(import_state.State.USER_PREFIX)) {
             userDelta[key.replace(import_state.State.USER_PREFIX, "")] = value;
-          } else {
+          } else if (!key.startsWith(import_state.State.TEMP_PREFIX)) {
             sessionDelta[key] = value;
           }
         }
@@ -382,16 +382,28 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
           storageSession.state = { ...storageSession.state, ...sessionDelta };
         }
       }
-      const newStorageEvent = txEm.create(import_schema.StorageEvent, {
+      const existingStorageEvent = await txEm.findOne(import_schema.StorageEvent, {
         id: trimmedEvent.id,
         appName: session.appName,
         userId: session.userId,
-        sessionId: session.id,
-        invocationId: trimmedEvent.invocationId,
-        timestamp: new Date(trimmedEvent.timestamp),
-        eventData: trimmedEvent
+        sessionId: session.id
       });
-      txEm.persist(newStorageEvent);
+      if (existingStorageEvent) {
+        existingStorageEvent.eventData = trimmedEvent;
+        existingStorageEvent.timestamp = new Date(trimmedEvent.timestamp);
+        txEm.persist(existingStorageEvent);
+      } else {
+        const newStorageEvent = txEm.create(import_schema.StorageEvent, {
+          id: trimmedEvent.id,
+          appName: session.appName,
+          userId: session.userId,
+          sessionId: session.id,
+          invocationId: trimmedEvent.invocationId,
+          timestamp: new Date(trimmedEvent.timestamp),
+          eventData: trimmedEvent
+        });
+        txEm.persist(newStorageEvent);
+      }
       await txEm.commit();
       storageSession.updateTime = new Date(event.timestamp);
       const newMergedState = (0, import_base_session_service.mergeStates)(
@@ -400,7 +412,12 @@ class DatabaseSessionService extends import_base_session_service.BaseSessionServ
         storageSession.state
       );
       session.state = newMergedState;
-      session.events.push(event);
+      const index = session.events.findIndex((e) => e.id === event.id);
+      if (index >= 0) {
+        session.events[index] = event;
+      } else {
+        session.events.push(event);
+      }
       session.lastUpdateTime = storageSession.updateTime.getTime();
     });
     return event;

package/dist/cjs/sessions/db/schema.js

@@ -35,6 +35,7 @@ __export(schema_exports, {
   ENTITIES: () => ENTITIES,
   SCHEMA_VERSION_1_JSON: () => SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY: () => SCHEMA_VERSION_KEY,
+  STORAGE_KEY_COLUMN_LENGTH: () => STORAGE_KEY_COLUMN_LENGTH,
   StorageAppState: () => StorageAppState,
   StorageEvent: () => StorageEvent,
   StorageMetadata: () => StorageMetadata,
@@ -51,6 +52,7 @@ var import_event = require("../../events/event.js");
  */
 const SCHEMA_VERSION_KEY = "schema_version";
 const SCHEMA_VERSION_1_JSON = "1";
+const STORAGE_KEY_COLUMN_LENGTH = 191;
 class CamelCaseToSnakeCaseJsonType extends import_core.JsonType {
   convertToDatabaseValue(value) {
     return JSON.stringify((0, import_event.transformToSnakeCaseEvent)(value));
@@ -79,7 +81,11 @@ let StorageAppState = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageAppState.prototype, "appName", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -102,10 +108,18 @@ let StorageUserState = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageUserState.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageUserState.prototype, "userId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -129,13 +143,21 @@ let StorageSession = class {
   }
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string" })
+  (0, import_core.PrimaryKey)({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageSession.prototype, "id", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageSession.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageSession.prototype, "userId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "json" })
@@ -161,16 +183,28 @@ import_core.PrimaryKey.name;
 let StorageEvent = class {
 };
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string" })
+  (0, import_core.PrimaryKey)({ type: "string", length: STORAGE_KEY_COLUMN_LENGTH })
 ], StorageEvent.prototype, "id", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "app_name" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "app_name",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "appName", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "user_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "user_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "userId", 2);
 __decorateClass([
-  (0, import_core.PrimaryKey)({ type: "string", fieldName: "session_id" })
+  (0, import_core.PrimaryKey)({
+    type: "string",
+    fieldName: "session_id",
+    length: STORAGE_KEY_COLUMN_LENGTH
+  })
 ], StorageEvent.prototype, "sessionId", 2);
 __decorateClass([
   (0, import_core.Property)({ type: "string", fieldName: "invocation_id" })
@@ -196,6 +230,7 @@ const ENTITIES = [
   ENTITIES,
   SCHEMA_VERSION_1_JSON,
   SCHEMA_VERSION_KEY,
+  STORAGE_KEY_COLUMN_LENGTH,
   StorageAppState,
   StorageEvent,
   StorageMetadata,

package/dist/cjs/sessions/in_memory_session_service.js

@@ -66,11 +66,12 @@ class InMemorySessionService extends import_base_session_service.BaseSessionServ
     sessionId
   }) {
     var _a;
+    const filteredState = state ? (0, import_base_session_service.trimTempState)(state) : void 0;
     const session = (0, import_session.createSession)({
       id: sessionId || (0, import_env_aware_utils.randomUUID)(),
       appName,
       userId,
-      state,
+      state: filteredState,
       events: [],
       lastUpdateTime: Date.now()
     });

package/dist/cjs/sessions/vertex_ai_session_service.js

@@ -33,7 +33,8 @@ var __decorateClass = (decorators, target, key, kind) => {
 var vertex_ai_session_service_exports = {};
 __export(vertex_ai_session_service_exports, {
   VertexAiSessionService: () => VertexAiSessionService,
-  isVertexAiConnectionString: () => isVertexAiConnectionString
+  isVertexAiConnectionString: () => isVertexAiConnectionString,
+  quoteFilterLiteral: () => quoteFilterLiteral
 });
 module.exports = __toCommonJS(vertex_ai_session_service_exports);
 var import_client = require("@google-cloud/vertexai/build/src/genai/client.js");
@@ -55,6 +56,10 @@ const HTTP_NOT_FOUND = 404;
 function isVertexAiConnectionString(uri) {
   return (uri == null ? void 0 : uri.startsWith("vertexai://")) || false;
 }
+function quoteFilterLiteral(value) {
+  const escaped = value.replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+  return `"${escaped}"`;
+}
 let VertexAiSessionService = class extends import_base_session_service.BaseSessionService {
   constructor(options) {
     super();
@@ -107,11 +112,12 @@ let VertexAiSessionService = class extends import_base_session_service.BaseSessi
   }) {
     var _a;
     const reasoningEngineId = this.getReasoningEngineId(appName);
+    const filteredState = state ? (0, import_base_session_service.trimTempState)(state) : void 0;
     let apiResponse = await this.sessions.createInternal({
       name: `reasoningEngines/${reasoningEngineId}`,
       userId,
       config: {
-        ...state ? { sessionState: state } : {},
+        ...filteredState ? { sessionState: filteredState } : {},
         ...sessionId ? { sessionId } : {}
       }
     });
@@ -221,7 +227,7 @@ let VertexAiSessionService = class extends import_base_session_service.BaseSessi
       const response = await this.sessions.listInternal({
         name: `reasoningEngines/${reasoningEngineId}`,
         config: {
-          ...userId ? { filter: `user_id="${userId}"` } : {},
+          ...userId ? { filter: `user_id=${quoteFilterLiteral(userId)}` } : {},
           ...pageToken ? { pageToken } : {}
         }
       });
@@ -282,10 +288,19 @@ let VertexAiSessionService = class extends import_base_session_service.BaseSessi
   }
   async deleteSession({
     appName,
-    userId: _userId,
+    userId,
     sessionId
   }) {
     const reasoningEngineId = this.getReasoningEngineId(appName);
+    const session = await this.getSession({
+      appName,
+      userId,
+      sessionId,
+      config: { numRecentEvents: 0 }
+    });
+    if (!session) {
+      return;
+    }
     await this.sessions.delete({
       name: `reasoningEngines/${reasoningEngineId}/sessions/${sessionId}`
     });
@@ -424,5 +439,6 @@ function _fromApiEvent(apiEventObj) {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   VertexAiSessionService,
-  isVertexAiConnectionString
+  isVertexAiConnectionString,
+  quoteFilterLiteral
 });