@ashx-j/lunr 0.2.9 → 0.2.11

package/dist/core/agent-session.js

@@ -219,12 +219,12 @@ export class AgentSession {
      * happens here instead of in wrappers.
      */
     _installAgentToolHooks() {
-        this.agent.beforeToolCall = async ({ toolCall, args }) => {
+        this.agent.beforeToolCall = async ({ toolCall, args, assistantMessage }) => {
             // lunr: permission gate (async) runs before sync gates — may show an approval dialog.
             // Explicit /swarm turns are pre-approved for the agent-swarm gate: the last
             // user message carries the literal [SWARM MODE] prefix on both TUI and gateway.
             const explicitSwarmTurn = toolCall.name === "subagent" ? isExplicitSwarmTurn(this.sessionManager.getBranch()) : undefined;
-            const permBlock = await gateToolCall(toolCall.name, args, this._cwd, this.sessionId, { explicitSwarmTurn });
+            const permBlock = await gateToolCall(toolCall.name, args, this._cwd, this.sessionId, { explicitSwarmTurn, assistantMessage });
             if (permBlock) {
                 return { block: true, reason: permBlock.reason };
             }
@@ -2594,10 +2594,12 @@ export class AgentSession {
             // Determine the new leaf position based on target type
             let newLeafId;
             let editorText;
+            let editorImages;
             if (targetEntry.type === "message" && targetEntry.message.role === "user") {
                 // User message: leaf = parent (null if root), text goes to editor
                 newLeafId = targetEntry.parentId;
                 editorText = this._extractUserMessageText(targetEntry.message.content);
+                editorImages = this._extractUserMessageImages(targetEntry.message.content);
             }
             else if (targetEntry.type === "custom_message") {
                 // Custom message: leaf = parent (null if root), text goes to editor
@@ -2650,7 +2652,7 @@ export class AgentSession {
                 fromExtension: summaryText ? fromExtension : undefined,
             });
             // Emit to custom tools
-            return { editorText, cancelled: false, summaryEntry };
+            return { editorText, editorImages, cancelled: false, summaryEntry };
         }
         finally {
             this._branchSummaryAbortController = undefined;
@@ -2685,6 +2687,12 @@ export class AgentSession {
         }
         return "";
     }
+    _extractUserMessageImages(content) {
+        if (typeof content === "string" || !Array.isArray(content))
+            return undefined;
+        const images = content.filter((part) => part.type === "image" && typeof part.data === "string" && typeof part.mimeType === "string");
+        return images.length > 0 ? images : undefined;
+    }
     /**
      * Get session statistics. Aggregates over ALL session entries (including
      * history that was compacted away), so token/cost totals reflect what was