@ashx-j/lunr 0.2.11 → 0.2.13

package/dist/core/agent-session.js

@@ -28,6 +28,7 @@ import { exportSessionToHtml } from "./export-html/index.js";
 import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
 import { ExtensionRunner, wrapRegisteredTools, } from "./extensions/index.js";
 import { emitSessionShutdownEvent } from "./extensions/runner.js";
+import { MEMORY_TOOL_NAMES } from "./memory-cap.js";
 import { ModelRegistry } from "./model-registry.js";
 import { gateToolCall } from "./permissions.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
@@ -150,6 +151,7 @@ export class AgentSession {
         this._customTools = config.customTools ?? [];
         this._cwd = config.cwd;
         this._modelRuntime = config.modelRuntime;
+        this._syncOpenAIServiceTier();
         this._extensionRunnerRef = config.extensionRunnerRef;
         this._initialActiveToolNames = config.initialActiveToolNames;
         this._allowedToolNames = config.allowedToolNames ? new Set(config.allowedToolNames) : undefined;
@@ -169,6 +171,19 @@ export class AgentSession {
     get modelRuntime() {
         return this._modelRuntime;
     }
+    get openaiFastMode() {
+        return this.settingsManager.getOpenAIFastMode();
+    }
+    setOpenAIFastMode(enabled) {
+        this.settingsManager.setOpenAIFastMode(enabled);
+        this._syncOpenAIServiceTier();
+    }
+    _syncOpenAIServiceTier() {
+        this.agent.serviceTier =
+            this.agent.state.model.provider === "openai-codex" && this.settingsManager.getOpenAIFastMode()
+                ? "fast"
+                : undefined;
+    }
     async _getRequiredRequestAuth(model) {
         let result;
         try {
@@ -239,9 +254,6 @@ export class AgentSession {
                     if (argPath)
                         pathsToSnapshot.push(argPath);
                 }
-                else if (toolName === "behavior_add" || toolName === "behavior_remove") {
-                    pathsToSnapshot.push(join(getAgentDir(), "behavior.md"));
-                }
                 else if (toolName === "memory_add" || toolName === "memory_remove") {
                     // lunr: simple-memory lives next to the lunr agent dir (~/.lunr/simple-memory)
                     pathsToSnapshot.push(join(dirname(getAgentDir()), "simple-memory", "memory.md"));
@@ -731,6 +743,10 @@ export class AgentSession {
         this._baseSystemPrompt = this._rebuildSystemPrompt(validToolNames);
         this.agent.state.systemPrompt = this._withSystemPromptAppend(this._systemPromptOverride ?? this._baseSystemPrompt);
     }
+    /** Rebuild the registry after a settings-backed feature changes availability. */
+    refreshToolRegistry() {
+        this._refreshToolRegistry();
+    }
     /** Whether compaction or branch summarization is currently running */
     get isCompacting() {
         return (this._autoCompactionAbortController !== undefined ||
@@ -816,6 +832,7 @@ export class AgentSession {
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
+            modelSlug: this.model ? `${this.model.provider}/${this.model.id}` : undefined,
             skills: loadedSkills,
             contextFiles: loadedContextFiles,
             customPrompt: loaderSystemPrompt,
@@ -826,6 +843,10 @@ export class AgentSession {
         };
         return buildSystemPrompt(this._baseSystemPromptOptions);
     }
+    _refreshSystemPromptForCurrentModel() {
+        this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
+        this.agent.state.systemPrompt = this._withSystemPromptAppend(this._systemPromptOverride ?? this._baseSystemPrompt);
+    }
     // =========================================================================
     // Prompting
     // =========================================================================
@@ -1295,6 +1316,8 @@ export class AgentSession {
         const previousModel = this.model;
         const thinkingLevel = this._getThinkingLevelForModelSwitch();
         this.agent.state.model = model;
+        this._syncOpenAIServiceTier();
+        this._refreshSystemPromptForCurrentModel();
         this.sessionManager.appendModelChange(model.provider, model.id);
         this.settingsManager.setDefaultModelAndProvider(model.provider, model.id);
         // Re-clamp thinking level for new model's capabilities
@@ -1331,6 +1354,8 @@ export class AgentSession {
         const thinkingLevel = this._getThinkingLevelForModelSwitch(next.thinkingLevel);
         // Apply model
         this.agent.state.model = next.model;
+        this._syncOpenAIServiceTier();
+        this._refreshSystemPromptForCurrentModel();
         this.sessionManager.appendModelChange(next.model.provider, next.model.id);
         this.settingsManager.setDefaultModelAndProvider(next.model.provider, next.model.id);
         // Apply thinking level.
@@ -1354,6 +1379,8 @@ export class AgentSession {
         const nextModel = availableModels[nextIndex];
         const thinkingLevel = this._getThinkingLevelForModelSwitch();
         this.agent.state.model = nextModel;
+        this._syncOpenAIServiceTier();
+        this._refreshSystemPromptForCurrentModel();
         this.sessionManager.appendModelChange(nextModel.provider, nextModel.id);
         this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
         // Re-clamp thinking level for new model's capabilities
@@ -2009,6 +2036,7 @@ export class AgentSession {
             return;
         }
         this.agent.state.model = refreshedModel;
+        this._syncOpenAIServiceTier();
     }
     _bindExtensionCore(runner) {
         const getCommands = () => {
@@ -2127,7 +2155,9 @@ export class AgentSession {
         const previousActiveToolNames = this.getActiveToolNames();
         const allowedToolNames = this._allowedToolNames;
         const excludedToolNames = this._excludedToolNames;
-        const isAllowedTool = (name) => (!allowedToolNames || allowedToolNames.has(name)) && !excludedToolNames?.has(name);
+        const isAllowedTool = (name) => (!allowedToolNames || allowedToolNames.has(name)) &&
+            !excludedToolNames?.has(name) &&
+            (this.settingsManager.getMemoryEnabled() || !MEMORY_TOOL_NAMES.has(name));
         const registeredTools = this._extensionRunner.getAllRegisteredTools();
         const allCustomTools = [
             ...registeredTools,