@entelligentsia/forgecli 1.0.25 → 1.0.40

package/node_modules/@earendil-works/pi-coding-agent/dist/core/agent-session.js

@@ -15,7 +15,7 @@
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { basename, dirname } from "node:path";
 import { clampThinkingLevel, cleanupSessionResources, getSupportedThinkingLevels, isContextOverflow, modelsAreEqual, resetApiProviders, streamSimple, } from "@earendil-works/pi-ai";
-import { theme } from "../modes/interactive/theme/theme.js";
+import { getThemeByName, theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
 import { resolvePath } from "../utils/paths.js";
 import { sleep } from "../utils/sleep.js";
@@ -97,6 +97,7 @@ export class AgentSession {
     _baseToolsOverride;
     _sessionStartEvent;
     _extensionUIContext;
+    _extensionMode = "print";
     _extensionCommandContextActions;
     _extensionAbortHandler;
     _extensionShutdownHandler;
@@ -1235,6 +1236,10 @@ export class AgentSession {
     // =========================================================================
     // Queue Mode Management
     // =========================================================================
+    syncQueueModesFromSettings() {
+        this.agent.steeringMode = this.settingsManager.getSteeringMode();
+        this.agent.followUpMode = this.settingsManager.getFollowUpMode();
+    }
     /**
      * Set steering message mode.
      * Saves to settings.
@@ -1629,6 +1634,9 @@ export class AgentSession {
         if (bindings.uiContext !== undefined) {
             this._extensionUIContext = bindings.uiContext;
         }
+        if (bindings.mode !== undefined) {
+            this._extensionMode = bindings.mode;
+        }
         if (bindings.commandContextActions !== undefined) {
             this._extensionCommandContextActions = bindings.commandContextActions;
         }
@@ -1686,7 +1694,7 @@ export class AgentSession {
         return `extension:${name}`;
     }
     _applyExtensionBindings(runner) {
-        runner.setUIContext(this._extensionUIContext);
+        runner.setUIContext(this._extensionUIContext, this._extensionMode);
         runner.bindCommandContext(this._extensionCommandContextActions);
         this._extensionErrorUnsubscriber?.();
         this._extensionErrorUnsubscriber = this._extensionErrorListener
@@ -1773,6 +1781,7 @@ export class AgentSession {
         }, {
             getModel: () => this.model,
             isIdle: () => !this.isStreaming,
+            isProjectTrusted: () => this.settingsManager.isProjectTrusted(),
             getSignal: () => this.agent.signal,
             abort: () => {
                 if (this._extensionAbortHandler) {
@@ -1799,6 +1808,7 @@ export class AgentSession {
                 })();
             },
             getSystemPrompt: () => this.systemPrompt,
+            getSystemPromptOptions: () => this._baseSystemPromptOptions,
         }, {
             registerProvider: (name, config) => {
                 this._modelRegistry.registerProvider(name, config);
@@ -1926,6 +1936,7 @@ export class AgentSession {
         const previousFlagValues = this._extensionRunner.getFlagValues();
         await emitSessionShutdownEvent(this._extensionRunner, { type: "session_shutdown", reason: "reload" });
         await this.settingsManager.reload();
+        this.syncQueueModesFromSettings();
         resetApiProviders();
         await this._resourceLoader.reload();
         this._buildRuntime({
@@ -2220,6 +2231,7 @@ export class AgentSession {
                     customInstructions,
                     replaceInstructions,
                     reserveTokens: branchSummarySettings.reserveTokens,
+                    streamFn: this.agent.streamFn,
                 });
                 if (result.aborted) {
                     return { cancelled: true, aborted: true };
@@ -2422,7 +2434,8 @@ export class AgentSession {
      * @returns Path to exported file
      */
     async exportToHtml(outputPath) {
-        const themeName = this.settingsManager.getTheme();
+        const configuredThemeName = this.settingsManager.getTheme();
+        const themeName = configuredThemeName && getThemeByName(configuredThemeName) ? configuredThemeName : undefined;
         // Create tool renderer if we have an extension runner (for custom tool HTML rendering)
         const toolRenderer = createToolHtmlRenderer({
             getToolDefinition: (name) => this.getToolDefinition(name),