@dyyz1993/pi-coding-agent 0.69.26 → 0.70.1

package/dist/core/agent-session.js

@@ -30,6 +30,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 { handleLargeInput } from "./large-input.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
@@ -737,11 +738,12 @@ export class AgentSession {
                 if (!options?.streamingBehavior) {
                     throw new Error("Agent is already processing. Specify streamingBehavior ('steer' or 'followUp') to queue the message.");
                 }
+                const { text: streamingText } = handleLargeInput(expandedText);
                 if (options.streamingBehavior === "followUp") {
-                    await this._queueFollowUp(expandedText, currentImages);
+                    await this._queueFollowUp(streamingText, currentImages);
                 }
                 else {
-                    await this._queueSteer(expandedText, currentImages);
+                    await this._queueSteer(streamingText, currentImages);
                 }
                 preflightResult?.(true);
                 return;
@@ -771,8 +773,10 @@ export class AgentSession {
             }
             // Build messages array (custom message if any, then user message)
             messages = [];
+            // Handle large input: save to temp file and replace with preview
+            const { text: finalText } = handleLargeInput(expandedText);
             // Add user message
-            const userContent = [{ type: "text", text: expandedText }];
+            const userContent = [{ type: "text", text: finalText }];
             if (currentImages) {
                 userContent.push(...currentImages);
             }
@@ -833,7 +837,7 @@ export class AgentSession {
         if (!command)
             return false;
         // Get command context from extension runner (includes session control methods)
-        const ctx = this._extensionRunner.createCommandContext();
+        const ctx = this._extensionRunner.createCommandContext(commandName);
         try {
             await command.handler(args, ctx);
             return true;
@@ -887,14 +891,13 @@ export class AgentSession {
      * @throws Error if text is an extension command
      */
     async steer(text, images) {
-        // Check for extension commands (cannot be queued)
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
         }
-        // Expand skill commands and prompt templates
         let expandedText = this._expandSkillCommand(text);
         expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
-        await this._queueSteer(expandedText, images);
+        const { text: finalText } = handleLargeInput(expandedText);
+        await this._queueSteer(finalText, images);
     }
     /**
      * Queue a follow-up message to be processed after the agent finishes.
@@ -904,14 +907,13 @@ export class AgentSession {
      * @throws Error if text is an extension command
      */
     async followUp(text, images) {
-        // Check for extension commands (cannot be queued)
         if (text.startsWith("/")) {
             this._throwIfExtensionCommand(text);
         }
-        // Expand skill commands and prompt templates
         let expandedText = this._expandSkillCommand(text);
         expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
-        await this._queueFollowUp(expandedText, images);
+        const { text: finalText } = handleLargeInput(expandedText);
+        await this._queueFollowUp(finalText, images);
     }
     /**
      * Internal: Queue a steering message (already expanded, no extension command check).
@@ -1759,6 +1761,9 @@ export class AgentSession {
                 const id = this.sessionManager.appendCustomEntry(customType, data, options);
                 this._emit({ type: "custom_entry", customType, data, id, display: options?.display });
             },
+            foldEntry: (entryId, summary, originalTokens) => {
+                this.sessionManager.appendFold(entryId, summary, originalTokens);
+            },
             setSessionName: (name) => {
                 const oldName = this.sessionManager.getSessionName();
                 const trimmed = name.trim();