@ashx-j/lunr 0.2.21 → 0.2.22

package/dist/core/agent-session.js

@@ -36,6 +36,7 @@ import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-man
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { measureStartup } from "./timings.js";
+import { TODO_TOOL_NAMES } from "./todo-settings.js";
 import { createLocalBashOperations } from "./tools/bash.js";
 import { createAllToolDefinitions } from "./tools/index.js";
 import { createToolDefinitionFromAgentTool } from "./tools/tool-definition-wrapper.js";
@@ -640,6 +641,15 @@ export class AgentSession {
      * Remove all listeners and disconnect from agent.
      * Call this when completely done with the session.
      */
+    transferring = false;
+    setTransferring(value) {
+        this.transferring = value;
+    }
+    assertCanStartWork() {
+        this.sessionManager.assertWritable();
+        if (this.transferring)
+            throw new Error("Session transfer is in progress. Wait or reclaim the session.");
+    }
     dispose() {
         try {
             this.abortRetry();
@@ -655,6 +665,7 @@ export class AgentSession {
         this._disconnectFromAgent();
         this._eventListeners = [];
         cleanupSessionResources(this.sessionId);
+        this.sessionManager.dispose();
     }
     // =========================================================================
     // Read-only State Access
@@ -932,6 +943,7 @@ export class AgentSession {
      * @throws Error if no model selected or no API key available (when not streaming)
      */
     async prompt(text, options) {
+        this.assertCanStartWork();
         const expandPromptTemplates = options?.expandPromptTemplates ?? true;
         const preflightResult = options?.preflightResult;
         let messages;
@@ -1123,6 +1135,7 @@ export class AgentSession {
      * @throws Error if text is an extension command
      */
     async steer(text, images) {
+        this.assertCanStartWork();
         // Check for extension commands (cannot be queued)
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
@@ -1140,6 +1153,7 @@ export class AgentSession {
      * @throws Error if text is an extension command
      */
     async followUp(text, images) {
+        this.assertCanStartWork();
         // Check for extension commands (cannot be queued)
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
@@ -1205,6 +1219,7 @@ export class AgentSession {
      * @param options.deliverAs Delivery mode: "steer", "followUp", or "nextTurn"
      */
     async sendCustomMessage(message, options) {
+        this.assertCanStartWork();
         const appMessage = {
             role: "custom",
             customType: message.customType,
@@ -2177,7 +2192,7 @@ export class AgentSession {
                 }
                 void this.abort();
             },
-            hasPendingMessages: () => this.pendingMessageCount > 0,
+            hasPendingMessages: () => this.pendingMessageCount > 0 || this.agent.hasQueuedMessages(),
             shutdown: () => {
                 this._extensionShutdownHandler?.();
             },
@@ -2214,7 +2229,9 @@ export class AgentSession {
         const excludedToolNames = this._excludedToolNames;
         const isAllowedTool = (name) => (!allowedToolNames || allowedToolNames.has(name)) &&
             !excludedToolNames?.has(name) &&
-            (this.settingsManager.getMemoryEnabled() || !MEMORY_TOOL_NAMES.has(name));
+            (this.settingsManager.getMemoryEnabled() || !MEMORY_TOOL_NAMES.has(name)) &&
+            (this.settingsManager.getTodosEnabled() || !TODO_TOOL_NAMES.has(name)) &&
+            (name !== "browser" || this.settingsManager.getBrowserEnabled());
         const registeredTools = this._extensionRunner.getAllRegisteredTools();
         const allCustomTools = [
             ...registeredTools,
@@ -2484,6 +2501,7 @@ export class AgentSession {
      * @param options.operations Custom BashOperations for remote execution
      */
     async executeBash(command, onChunk, options) {
+        this.assertCanStartWork();
         this._bashAbortController = new AbortController();
         // Apply command prefix if configured (e.g., "shopt -s expand_aliases" for alias support)
         const prefix = this.settingsManager.getShellCommandPrefix();