@dyyz1993/pi-coding-agent 0.69.13 → 0.69.17

package/dist/core/agent-session.js

@@ -1731,9 +1731,9 @@ export class AgentSession {
                     });
                 });
             },
-            appendEntry: (customType, data) => {
-                const id = this.sessionManager.appendCustomEntry(customType, data);
-                this._emit({ type: "custom_entry", customType, data, id });
+            appendEntry: (customType, data, options) => {
+                const id = this.sessionManager.appendCustomEntry(customType, data, options);
+                this._emit({ type: "custom_entry", customType, data, id, display: options?.display });
             },
             setSessionName: (name) => {
                 const oldName = this.sessionManager.getSessionName();
@@ -2423,11 +2423,39 @@ export class AgentSession {
             oldLeafId,
             summaryEntry,
             fromExtension: summaryText ? fromExtension : undefined,
+            skipFiles: options.skipFiles,
         });
         // Emit to custom tools
         this._branchSummaryAbortController = undefined;
         return { editorText, cancelled: false, summaryEntry };
     }
+    async previewRollback(targetId) {
+        const oldLeafId = this.sessionManager.getLeafId();
+        let newLeafId;
+        const targetEntry = this.sessionManager.getEntry(targetId);
+        if (!targetEntry) {
+            throw new Error(`Entry ${targetId} not found`);
+        }
+        if (targetEntry.type === "message" && targetEntry.message.role === "user") {
+            newLeafId = targetEntry.parentId;
+        }
+        else if (targetEntry.type === "custom_message") {
+            newLeafId = targetEntry.parentId;
+        }
+        else {
+            newLeafId = targetId;
+        }
+        const result = await this._extensionRunner.emit({
+            type: "session_tree",
+            newLeafId,
+            oldLeafId,
+            preview: true,
+        });
+        return (result ?? {
+            restored: [],
+            deleted: [],
+        });
+    }
     /**
      * Get all user messages from session for fork selector.
      */