openvibe 0.58.2 → 0.59.0

package/dist/core/agent-session.js

@@ -14,7 +14,7 @@
  */
 import { readFileSync } from "node:fs";
 import { basename, dirname } from "node:path";
-import { isContextOverflow, resetApiProviders, supportsXhigh } from "@mariozechner/pi-ai";
+import { isContextOverflow, modelsAreEqual, resetApiProviders, supportsXhigh } from "@mariozechner/pi-ai";
 import { theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
 import { sleep } from "../utils/sleep.js";
@@ -29,6 +29,7 @@ import { getLatestCompactionEntry } from "./session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "./slash-commands.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createAllTools } from "./tools/index.js";
+import { AGENT_MODES } from "./agent-modes.js";
 /**
  * Parse a skill block from message text.
  * Returns null if the text doesn't contain a skill block.
@@ -106,6 +107,8 @@ export class AgentSession {
     _toolPromptGuidelines = new Map();
     // Base system prompt (without extension appends) - used to apply fresh appends each turn
     _baseSystemPrompt = "";
+    // Current agent mode
+    _mode = "default";
     constructor(config) {
         this.agent = config.agent;
         this.sessionManager = config.sessionManager;
@@ -118,6 +121,7 @@ export class AgentSession {
         this._extensionRunnerRef = config.extensionRunnerRef;
         this._initialActiveToolNames = config.initialActiveToolNames;
         this._baseToolsOverride = config.baseToolsOverride;
+        this._mode = config.initialMode ?? "default";
         // Always subscribe to agent events for internal handling
         // (session persistence, extensions, auto-compaction, retry logic)
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
@@ -418,6 +422,84 @@ export class AgentSession {
     get thinkingLevel() {
         return this.agent.state.thinkingLevel;
     }
+    /** Current agent mode */
+    get mode() {
+        return this._mode;
+    }
+    /**
+     * Set agent mode.
+     * Updates tools, thinking level, and system prompt based on the mode.
+     */
+    setMode(mode) {
+        if (this._mode === mode)
+            return;
+        this._mode = mode;
+        const modeConfig = AGENT_MODES[mode];
+        const modeThinkingLevel = modeConfig.thinkingLevel;
+        const modeTools = modeConfig.tools;
+        this.agent.setThinkingLevel(modeThinkingLevel);
+        this.setActiveToolsByName(modeTools);
+        this._baseSystemPrompt = this._rebuildSystemPrompt(modeTools, modeThinkingLevel);
+        this.agent.setSystemPrompt(this._baseSystemPrompt);
+    }
+    /**
+     * Cycle to next agent mode.
+     * @returns New mode name
+     */
+    cycleMode() {
+        const modes = ["default", "plan", "spec"];
+        const currentIndex = modes.indexOf(this._mode);
+        const nextIndex = (currentIndex + 1) % modes.length;
+        const nextMode = modes[nextIndex];
+        this.setMode(nextMode);
+        return nextMode;
+    }
+    /**
+     * Set the model for the agent.
+     * Updates thinking level based on model capabilities.
+     */
+    async setModel(model) {
+        const previousModel = this.agent.state.model;
+        this.agent.setModel(model);
+        let effectiveThinking = this.thinkingLevel;
+        if (!model.reasoning) {
+            effectiveThinking = "off";
+        }
+        else if (effectiveThinking === "xhigh" && !supportsXhigh(model)) {
+            effectiveThinking = "high";
+        }
+        if (effectiveThinking !== this.thinkingLevel) {
+            this.agent.setThinkingLevel(effectiveThinking);
+            this.sessionManager.appendThinkingLevelChange(effectiveThinking);
+        }
+        this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames(), effectiveThinking);
+        this.agent.setSystemPrompt(this._baseSystemPrompt);
+    }
+    /**
+     * Cycle to next model in scoped models list.
+     * @returns Model cycle result or undefined if no scoped models
+     */
+    async cycleModel() {
+        if (this._scopedModels.length === 0) {
+            return undefined;
+        }
+        const currentModel = this.agent.state.model;
+        let currentIndex = -1;
+        for (let i = 0; i < this._scopedModels.length; i++) {
+            if (modelsAreEqual(this._scopedModels[i].model, currentModel)) {
+                currentIndex = i;
+                break;
+            }
+        }
+        const nextIndex = (currentIndex + 1) % this._scopedModels.length;
+        const nextModelConfig = this._scopedModels[nextIndex];
+        await this.setModel(nextModelConfig.model);
+        return {
+            model: nextModelConfig.model,
+            thinkingLevel: this.thinkingLevel,
+            isScoped: true,
+        };
+    }
     /** Whether agent is currently streaming a response */
     get isStreaming() {
         return this.agent.state.isStreaming;
@@ -561,6 +643,7 @@ export class AgentSession {
             toolSnippets,
             promptGuidelines,
             thinkingLevel: thinkingLevel ?? this.thinkingLevel,
+            mode: this._mode,
         });
     }
     // =========================================================================