@tuanhung303/opencode-acp 0.0.1

package/dist/lib/strategies/tools.js

@@ -0,0 +1,288 @@
+import { tool } from "@opencode-ai/plugin";
+import { sendUnifiedNotification } from "../ui/notification";
+import { formatPruningResultForTool } from "../ui/utils";
+import { ensureSessionInitialized } from "../state";
+import { saveSessionState } from "../state/persistence";
+import { loadPrompt } from "../prompts";
+import { calculateTokensSaved, getCurrentParams } from "./utils";
+import { getFilePathFromParameters, isProtectedFilePath } from "../protected-file-patterns";
+const DISCARD_TOOL_DESCRIPTION = loadPrompt("discard-tool-spec");
+const EXTRACT_TOOL_DESCRIPTION = loadPrompt("extract-tool-spec");
+const RESTORE_TOOL_DESCRIPTION = loadPrompt("restore-tool-spec");
+/**
+ * Execute a prune operation using call IDs directly.
+ * This is the hash-based version that works with resolved call IDs.
+ */
+async function executePruneOperationByCallIds(ctx, toolCtx, callIds, hashes, reason, toolName, distillation) {
+    const { client, state, logger, config, workingDirectory } = ctx;
+    const sessionId = toolCtx.sessionID;
+    logger.info(`${toolName} tool invoked`);
+    logger.info(JSON.stringify({ hashes, reason }));
+    if (!callIds || callIds.length === 0) {
+        logger.debug(`${toolName} tool called but no valid call IDs resolved`);
+        throw new Error(`No valid hashes provided. Use hashes from tool outputs (e.g., #r_a1b2c#).`);
+    }
+    // Fetch messages to calculate tokens
+    const messagesResponse = await client.session.messages({
+        path: { id: sessionId },
+    });
+    const messages = messagesResponse.data || messagesResponse;
+    await ensureSessionInitialized(ctx.client, state, sessionId, logger, messages);
+    const currentParams = getCurrentParams(state, messages, logger);
+    // Validate that all call IDs exist in cache and aren't protected
+    for (const callId of callIds) {
+        const metadata = state.toolParameters.get(callId);
+        if (!metadata) {
+            const hash = state.callIdToHash.get(callId) || "unknown";
+            logger.debug("Rejecting prune request - call ID not in cache", { callId, hash });
+            throw new Error(`Invalid hash provided. The tool may have already been discarded.`);
+        }
+        const allProtectedTools = config.tools.settings.protectedTools;
+        if (allProtectedTools.includes(metadata.tool)) {
+            const hash = state.callIdToHash.get(callId) || "unknown";
+            logger.debug("Rejecting prune request - protected tool", {
+                callId,
+                hash,
+                tool: metadata.tool,
+            });
+            throw new Error(`Cannot discard: '${metadata.tool}' is a protected tool.\n` +
+                `Protected tools: ${allProtectedTools.join(", ")}\n` +
+                `To modify protection, update 'tools.settings.protectedTools' in your ACP config.`);
+        }
+        const filePath = getFilePathFromParameters(metadata.parameters);
+        if (isProtectedFilePath(filePath, config.protectedFilePatterns)) {
+            const hash = state.callIdToHash.get(callId) || "unknown";
+            logger.debug("Rejecting prune request - protected file path", {
+                callId,
+                hash,
+                tool: metadata.tool,
+                filePath,
+            });
+            const protectedPatterns = config.protectedFilePatterns.length > 0
+                ? `\nProtected patterns: ${config.protectedFilePatterns.join(", ")}`
+                : "";
+            throw new Error(`Cannot discard: ${filePath} is a protected file path.${protectedPatterns}\n` +
+                `To modify protection, update 'protectedFilePatterns' in your ACP config.`);
+        }
+    }
+    // Add to prune list
+    state.prune.toolIds.push(...callIds);
+    // Collect metadata for notification
+    const toolMetadata = new Map();
+    for (const callId of callIds) {
+        const toolParameters = state.toolParameters.get(callId);
+        if (toolParameters) {
+            toolMetadata.set(callId, toolParameters);
+        }
+        else {
+            logger.debug("No metadata found for call ID", { callId });
+        }
+    }
+    // Calculate token savings
+    const tokensSaved = calculateTokensSaved(state, messages, callIds);
+    state.stats.pruneTokenCounter += tokensSaved;
+    state.stats.pruneMessageCounter += callIds.length;
+    // Track strategy effectiveness
+    if (toolName === "Discard") {
+        state.stats.strategyStats.manualDiscard.count += callIds.length;
+        state.stats.strategyStats.manualDiscard.tokens += tokensSaved;
+    }
+    else if (toolName === "Extract") {
+        state.stats.strategyStats.extraction.count += callIds.length;
+        state.stats.strategyStats.extraction.tokens += tokensSaved;
+    }
+    // Store stats for display
+    state.lastDiscardStats = {
+        itemCount: callIds.length,
+        tokensSaved: state.stats.pruneTokenCounter,
+    };
+    // Record in discard history
+    state.discardHistory.push({
+        timestamp: Date.now(),
+        hashes,
+        tokensSaved,
+        reason,
+    });
+    await sendUnifiedNotification(client, logger, config, state, sessionId, callIds, toolMetadata, reason, currentParams, workingDirectory, distillation);
+    state.stats.totalPruneTokens += state.stats.pruneTokenCounter;
+    state.stats.pruneTokenCounter = 0;
+    state.stats.totalPruneMessages += state.stats.pruneMessageCounter;
+    state.stats.pruneMessageCounter = 0;
+    saveSessionState(state, logger).catch((err) => logger.error("Failed to persist state", { error: err.message }));
+    return formatPruningResultForTool(callIds, toolMetadata, workingDirectory);
+}
+export function createDiscardTool(ctx) {
+    return tool({
+        description: DISCARD_TOOL_DESCRIPTION,
+        args: {
+            reason: tool.schema
+                .string()
+                .describe("Why you're discarding: 'noise' | 'completion' | 'superseded' | 'exploration' | 'duplicate'"),
+            hashes: tool.schema
+                .array(tool.schema.string())
+                .describe("Hash identifiers from tool outputs (e.g., ['#r_a1b2c#', '#g_d4e5f#'])"),
+        },
+        async execute(args, toolCtx) {
+            const { state, logger } = ctx;
+            // Validate reason
+            const validReasons = [
+                "completion",
+                "noise",
+                "superseded",
+                "exploration",
+                "duplicate",
+            ];
+            if (!args.reason || !validReasons.includes(args.reason)) {
+                logger.debug("Invalid discard reason provided: " + args.reason);
+                throw new Error(`Invalid reason '${args.reason}'. Use one of: ${validReasons.join(", ")}`);
+            }
+            // Validate hashes provided
+            if (!args.hashes || args.hashes.length === 0) {
+                throw new Error("No hashes provided. Use hash identifiers from tool outputs (e.g., #r_a1b2c#).");
+            }
+            // Resolve hashes to call IDs
+            const callIds = [];
+            const validHashes = [];
+            const unknownHashes = [];
+            for (const hash of args.hashes) {
+                const callId = state.hashToCallId.get(hash);
+                if (callId) {
+                    // Skip if already pruned
+                    if (!state.prune.toolIds.includes(callId)) {
+                        callIds.push(callId);
+                        validHashes.push(hash);
+                    }
+                    else {
+                        logger.debug(`Hash ${hash} already pruned, skipping`);
+                    }
+                }
+                else {
+                    unknownHashes.push(hash);
+                    logger.warn(`Unknown hash: ${hash}`);
+                }
+            }
+            if (unknownHashes.length > 0 && callIds.length === 0) {
+                throw new Error(`Unknown hashes: ${unknownHashes.join(", ")}. Use hash identifiers from tool outputs.`);
+            }
+            if (callIds.length === 0) {
+                throw new Error("No valid hashes to discard. Tools may have already been discarded.");
+            }
+            logger.info(`Discard: ${validHashes.length} hashes resolved to ${callIds.length} call IDs`);
+            return executePruneOperationByCallIds(ctx, toolCtx, callIds, validHashes, args.reason, "Discard");
+        },
+    });
+}
+export function createExtractTool(ctx) {
+    return tool({
+        description: EXTRACT_TOOL_DESCRIPTION,
+        args: {
+            hashes: tool.schema
+                .array(tool.schema.string())
+                .describe("Hash identifiers from tool outputs (e.g., ['#r_a1b2c#', '#g_d4e5f#'])"),
+            distillation: tool.schema
+                .array(tool.schema.string())
+                .describe("REQUIRED. Array of strings, one per hash (positional: distillation[0] is for hashes[0], etc.)"),
+            preserve: tool.schema
+                .boolean()
+                .optional()
+                .describe("Set to true to keep original tool output after extraction. Default: false"),
+        },
+        async execute(args, toolCtx) {
+            const { state, logger } = ctx;
+            if (!args.distillation || args.distillation.length === 0) {
+                logger.debug("Extract tool called without distillation: " + JSON.stringify(args));
+                throw new Error("Missing distillation. You must provide a distillation string for each hash.");
+            }
+            if (!args.hashes || args.hashes.length === 0) {
+                throw new Error("No hashes provided. Use hash identifiers from tool outputs (e.g., #r_a1b2c#).");
+            }
+            // Resolve hashes to call IDs
+            const callIds = [];
+            const validHashes = [];
+            for (const hash of args.hashes) {
+                const callId = state.hashToCallId.get(hash);
+                if (callId && !state.prune.toolIds.includes(callId)) {
+                    callIds.push(callId);
+                    validHashes.push(hash);
+                }
+                else {
+                    logger.warn(`Unknown or already pruned hash: ${hash}`);
+                }
+            }
+            if (callIds.length === 0) {
+                throw new Error("No valid hashes to extract. Tools may have already been discarded.");
+            }
+            // Log the distillation for debugging/analysis
+            logger.info("Distillation data received:");
+            logger.info(JSON.stringify(args.distillation, null, 2));
+            // If preserve is true, don't prune - just return success with distillation
+            if (args.preserve) {
+                logger.info(`Extract with preserve=true - not pruning ${callIds.length} tools`);
+                return `Extracted knowledge from ${callIds.length} tool(s). Original outputs preserved.\n\nDistilled knowledge:\n${args.distillation.map((d, i) => `${i + 1}. ${d}`).join("\n")}`;
+            }
+            return executePruneOperationByCallIds(ctx, toolCtx, callIds, validHashes, "extraction", "Extract", args.distillation);
+        },
+    });
+}
+export function createRestoreTool(ctx) {
+    return tool({
+        description: RESTORE_TOOL_DESCRIPTION,
+        args: {
+            hashes: tool.schema
+                .array(tool.schema.string())
+                .describe("Hash identifiers of pruned tools to restore (e.g., ['#r_a1b2c#', '#g_d4e5f#'])"),
+        },
+        async execute(args, toolCtx) {
+            const { state, logger } = ctx;
+            const sessionId = toolCtx.sessionID;
+            if (!args.hashes || args.hashes.length === 0) {
+                throw new Error("No hashes provided. Use hash identifiers from pruned tool outputs (e.g., #r_a1b2c#).");
+            }
+            // Resolve hashes to call IDs and restore
+            const restored = [];
+            const notFound = [];
+            const notPruned = [];
+            for (const hash of args.hashes) {
+                const callId = state.hashToCallId.get(hash);
+                if (!callId) {
+                    notFound.push(hash);
+                    continue;
+                }
+                // Check if it's in the prune list
+                const pruneIndex = state.prune.toolIds.indexOf(callId);
+                if (pruneIndex === -1) {
+                    notPruned.push(hash);
+                    continue;
+                }
+                // Remove from prune list (restore)
+                state.prune.toolIds.splice(pruneIndex, 1);
+                restored.push(hash);
+                logger.info(`Restored tool with hash ${hash} (callId: ${callId})`);
+            }
+            // Build result message
+            const lines = [];
+            if (restored.length > 0) {
+                lines.push(`Restored ${restored.length} tool(s):`);
+                for (const hash of restored) {
+                    lines.push(`  ✓ ${hash}`);
+                }
+            }
+            if (notFound.length > 0) {
+                lines.push(`\nNot found (unknown hash):`);
+                for (const hash of notFound) {
+                    lines.push(`  ✗ ${hash}`);
+                }
+            }
+            if (notPruned.length > 0) {
+                lines.push(`\nNot pruned (already active):`);
+                for (const hash of notPruned) {
+                    lines.push(`  ○ ${hash}`);
+                }
+            }
+            // Save state
+            saveSessionState(state, logger).catch((err) => logger.error("Failed to persist state", { error: err.message }));
+            return lines.join("\n");
+        },
+    });
+}
+//# sourceMappingURL=tools.js.map

package/dist/lib/strategies/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../lib/strategies/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAA;AAG1C,OAAO,EAAe,uBAAuB,EAAE,MAAM,oBAAoB,CAAA;AACzE,OAAO,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAA;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AAEvD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AACvC,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAA;AAChE,OAAO,EAAE,yBAAyB,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAA;AAE3F,MAAM,wBAAwB,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAA;AAChE,MAAM,wBAAwB,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAA;AAChE,MAAM,wBAAwB,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAA;AAUhE;;;GAGG;AACH,KAAK,UAAU,8BAA8B,CACzC,GAAqB,EACrB,OAA8B,EAC9B,OAAiB,EACjB,MAAgB,EAChB,MAAmB,EACnB,QAAgB,EAChB,YAAuB;IAEvB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,GAAG,CAAA;IAC/D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;IAEnC,MAAM,CAAC,IAAI,CAAC,GAAG,QAAQ,eAAe,CAAC,CAAA;IACvC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;IAE/C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,GAAG,QAAQ,6CAA6C,CAAC,CAAA;QACtE,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAA;IAChG,CAAC;IAED,qCAAqC;IACrC,MAAM,gBAAgB,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;QACnD,IAAI,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE;KAC1B,CAAC,CAAA;IACF,MAAM,QAAQ,GAAgB,gBAAgB,CAAC,IAAI,IAAI,gBAAgB,CAAA;IAEvE,MAAM,wBAAwB,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAA;IAE9E,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAA;IAE/D,iEAAiE;IACjE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QACjD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,SAAS,CAAA;YACxD,MAAM,CAAC,KAAK,CAAC,gDAAgD,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;YAChF,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,CAAA;QACvF,CAAC;QAED,MAAM,iBAAiB,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAA;QAC9D,IAAI,iBAAiB,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,SAAS,CAAA;YACxD,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;gBACrD,MAAM;gBACN,IAAI;gBACJ,IAAI,EAAE,QAAQ,CAAC,IAAI;aACtB,CAAC,CAAA;YACF,MAAM,IAAI,KAAK,CACX,oBAAoB,QAAQ,CAAC,IAAI,0BAA0B;gBACvD,oBAAoB,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;gBACpD,kFAAkF,CACzF,CAAA;QACL,CAAC;QAED,MAAM,QAAQ,GAAG,yBAAyB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;QAC/D,IAAI,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,qBAAqB,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,SAAS,CAAA;YACxD,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;gBAC1D,MAAM;gBACN,IAAI;gBACJ,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,QAAQ;aACX,CAAC,CAAA;YACF,MAAM,iBAAiB,GACnB,MAAM,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC;gBACnC,CAAC,CAAC,yBAAyB,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBACpE,CAAC,CAAC,EAAE,CAAA;YACZ,MAAM,IAAI,KAAK,CACX,mBAAmB,QAAQ,6BAA6B,iBAAiB,IAAI;gBACzE,0EAA0E,CACjF,CAAA;QACL,CAAC;IACL,CAAC;IAED,oBAAoB;IACpB,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAA;IAEpC,oCAAoC;IACpC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAA;IAC1D,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,cAAc,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QACvD,IAAI,cAAc,EAAE,CAAC;YACjB,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAC5C,CAAC;aAAM,CAAC;YACJ,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,EAAE,MAAM,EAAE,CAAC,CAAA;QAC7D,CAAC;IACL,CAAC;IAED,0BAA0B;IAC1B,MAAM,WAAW,GAAG,oBAAoB,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAA;IAClE,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,WAAW,CAAA;IAC5C,KAAK,CAAC,KAAK,CAAC,mBAAmB,IAAI,OAAO,CAAC,MAAM,CAAA;IAEjD,+BAA+B;IAC/B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QACzB,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAA;QAC/D,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,IAAI,WAAW,CAAA;IACjE,CAAC;SAAM,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAChC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAA;QAC5D,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,WAAW,CAAA;IAC9D,CAAC;IAED,0BAA0B;IAC1B,KAAK,CAAC,gBAAgB,GAAG;QACrB,SAAS,EAAE,OAAO,CAAC,MAAM;QACzB,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;KAC7C,CAAA;IAED,4BAA4B;IAC5B,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC;QACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,MAAM;QACN,WAAW;QACX,MAAM;KACT,CAAC,CAAA;IAEF,MAAM,uBAAuB,CACzB,MAAM,EACN,MAAM,EACN,MAAM,EACN,KAAK,EACL,SAAS,EACT,OAAO,EACP,YAAY,EACZ,MAAM,EACN,aAAa,EACb,gBAAgB,EAChB,YAAY,CACf,CAAA;IAED,KAAK,CAAC,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAA;IAC7D,KAAK,CAAC,KAAK,CAAC,iBAAiB,GAAG,CAAC,CAAA;IACjC,KAAK,CAAC,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAAA;IACjE,KAAK,CAAC,KAAK,CAAC,mBAAmB,GAAG,CAAC,CAAA;IAEnC,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAC1C,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAClE,CAAA;IAED,OAAO,0BAA0B,CAAC,OAAO,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAA;AAC9E,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAqB;IACnD,OAAO,IAAI,CAAC;QACR,WAAW,EAAE,wBAAwB;QACrC,IAAI,EAAE;YACF,MAAM,EAAE,IAAI,CAAC,MAAM;iBACd,MAAM,EAAE;iBACR,QAAQ,CACL,4FAA4F,CAC/F;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;iBACd,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;iBAC3B,QAAQ,CAAC,uEAAuE,CAAC;SACzF;QACD,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO;YACvB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;YAE7B,kBAAkB;YAClB,MAAM,YAAY,GAAG;gBACjB,YAAY;gBACZ,OAAO;gBACP,YAAY;gBACZ,aAAa;gBACb,WAAW;aACL,CAAA;YACV,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAa,CAAC,EAAE,CAAC;gBAC7D,MAAM,CAAC,KAAK,CAAC,mCAAmC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAA;gBAC/D,MAAM,IAAI,KAAK,CACX,mBAAmB,IAAI,CAAC,MAAM,kBAAkB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC5E,CAAA;YACL,CAAC;YAED,2BAA2B;YAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CACX,+EAA+E,CAClF,CAAA;YACL,CAAC;YAED,6BAA6B;YAC7B,MAAM,OAAO,GAAa,EAAE,CAAA;YAC5B,MAAM,WAAW,GAAa,EAAE,CAAA;YAChC,MAAM,aAAa,GAAa,EAAE,CAAA;YAElC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBAC3C,IAAI,MAAM,EAAE,CAAC;oBACT,yBAAyB;oBACzB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;wBACxC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;wBACpB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBAC1B,CAAC;yBAAM,CAAC;wBACJ,MAAM,CAAC,KAAK,CAAC,QAAQ,IAAI,2BAA2B,CAAC,CAAA;oBACzD,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACJ,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACxB,MAAM,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAA;gBACxC,CAAC;YACL,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnD,MAAM,IAAI,KAAK,CACX,mBAAmB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,2CAA2C,CACzF,CAAA;YACL,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,IAAI,KAAK,CACX,oEAAoE,CACvE,CAAA;YACL,CAAC;YAED,MAAM,CAAC,IAAI,CACP,YAAY,WAAW,CAAC,MAAM,uBAAuB,OAAO,CAAC,MAAM,WAAW,CACjF,CAAA;YAED,OAAO,8BAA8B,CACjC,GAAG,EACH,OAAO,EACP,OAAO,EACP,WAAW,EACX,IAAI,CAAC,MAAqB,EAC1B,SAAS,CACZ,CAAA;QACL,CAAC;KACJ,CAAC,CAAA;AACN,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAqB;IACnD,OAAO,IAAI,CAAC;QACR,WAAW,EAAE,wBAAwB;QACrC,IAAI,EAAE;YACF,MAAM,EAAE,IAAI,CAAC,MAAM;iBACd,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;iBAC3B,QAAQ,CAAC,uEAAuE,CAAC;YACtF,YAAY,EAAE,IAAI,CAAC,MAAM;iBACpB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;iBAC3B,QAAQ,CACL,+FAA+F,CAClG;YACL,QAAQ,EAAE,IAAI,CAAC,MAAM;iBAChB,OAAO,EAAE;iBACT,QAAQ,EAAE;iBACV,QAAQ,CACL,2EAA2E,CAC9E;SACR;QACD,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO;YACvB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;YAE7B,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvD,MAAM,CAAC,KAAK,CAAC,4CAA4C,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAA;gBACjF,MAAM,IAAI,KAAK,CACX,6EAA6E,CAChF,CAAA;YACL,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CACX,+EAA+E,CAClF,CAAA;YACL,CAAC;YAED,6BAA6B;YAC7B,MAAM,OAAO,GAAa,EAAE,CAAA;YAC5B,MAAM,WAAW,GAAa,EAAE,CAAA;YAEhC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBAC3C,IAAI,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBAClD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;oBACpB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBAC1B,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,IAAI,CAAC,mCAAmC,IAAI,EAAE,CAAC,CAAA;gBAC1D,CAAC;YACL,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,IAAI,KAAK,CACX,oEAAoE,CACvE,CAAA;YACL,CAAC;YAED,8CAA8C;YAC9C,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAA;YAC1C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;YAEvD,2EAA2E;YAC3E,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CAAC,4CAA4C,OAAO,CAAC,MAAM,QAAQ,CAAC,CAAA;gBAC/E,OAAO,4BAA4B,OAAO,CAAC,MAAM,kEAAkE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;YACrL,CAAC;YAED,OAAO,8BAA8B,CACjC,GAAG,EACH,OAAO,EACP,OAAO,EACP,WAAW,EACX,YAA2B,EAC3B,SAAS,EACT,IAAI,CAAC,YAAY,CACpB,CAAA;QACL,CAAC;KACJ,CAAC,CAAA;AACN,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAqB;IACnD,OAAO,IAAI,CAAC;QACR,WAAW,EAAE,wBAAwB;QACrC,IAAI,EAAE;YACF,MAAM,EAAE,IAAI,CAAC,MAAM;iBACd,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;iBAC3B,QAAQ,CACL,gFAAgF,CACnF;SACR;QACD,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO;YACvB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;YAC7B,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;YAEnC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CACX,sFAAsF,CACzF,CAAA;YACL,CAAC;YAED,yCAAyC;YACzC,MAAM,QAAQ,GAAa,EAAE,CAAA;YAC7B,MAAM,QAAQ,GAAa,EAAE,CAAA;YAC7B,MAAM,SAAS,GAAa,EAAE,CAAA;YAE9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;oBACV,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACnB,SAAQ;gBACZ,CAAC;gBAED,kCAAkC;gBAClC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;gBACtD,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;oBACpB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACpB,SAAQ;gBACZ,CAAC;gBAED,mCAAmC;gBACnC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;gBACzC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACnB,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,aAAa,MAAM,GAAG,CAAC,CAAA;YACtE,CAAC;YAED,uBAAuB;YACvB,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,YAAY,QAAQ,CAAC,MAAM,WAAW,CAAC,CAAA;gBAClD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;oBAC1B,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;gBAC7B,CAAC;YACL,CAAC;YACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAA;gBACzC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;oBAC1B,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;gBAC7B,CAAC;YACL,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAA;gBAC5C,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC3B,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;gBAC7B,CAAC;YACL,CAAC;YAED,aAAa;YACb,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAC1C,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAClE,CAAA;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3B,CAAC;KACJ,CAAC,CAAA;AACN,CAAC"}

package/dist/lib/strategies/utils.d.ts

@@ -0,0 +1,11 @@
+import { SessionState, WithParts } from "../state";
+import { Logger } from "../logger";
+export declare function getCurrentParams(state: SessionState, messages: WithParts[], logger: Logger): {
+    providerId: string | undefined;
+    modelId: string | undefined;
+    agent: string | undefined;
+    variant: string | undefined;
+};
+export declare function countTokens(text: string): number;
+export declare const calculateTokensSaved: (state: SessionState, messages: WithParts[], pruneToolIds: string[]) => number;
+//# sourceMappingURL=utils.d.ts.map

package/dist/lib/strategies/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../lib/strategies/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,UAAU,CAAA;AAElD,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAIlC,wBAAgB,gBAAgB,CAC5B,KAAK,EAAE,YAAY,EACnB,QAAQ,EAAE,SAAS,EAAE,EACrB,MAAM,EAAE,MAAM,GACf;IACC,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;IAC9B,OAAO,EAAE,MAAM,GAAG,SAAS,CAAA;IAC3B,KAAK,EAAE,MAAM,GAAG,SAAS,CAAA;IACzB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAA;CAC9B,CAkBA;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAOhD;AAMD,eAAO,MAAM,oBAAoB,GAC7B,OAAO,YAAY,EACnB,UAAU,SAAS,EAAE,EACrB,cAAc,MAAM,EAAE,KACvB,MAyCF,CAAA"}

package/dist/lib/strategies/utils.js

@@ -0,0 +1,75 @@
+import { countTokens as anthropicCountTokens } from "@anthropic-ai/tokenizer";
+import { getLastUserMessage, isMessageCompacted } from "../shared-utils";
+export function getCurrentParams(state, messages, logger) {
+    const userMsg = getLastUserMessage(messages);
+    if (!userMsg) {
+        logger.debug("No user message found when determining current params");
+        return {
+            providerId: undefined,
+            modelId: undefined,
+            agent: undefined,
+            variant: state.variant,
+        };
+    }
+    const userInfo = userMsg.info;
+    const agent = userInfo.agent;
+    const providerId = userInfo.model.providerID;
+    const modelId = userInfo.model.modelID;
+    const variant = state.variant ?? userInfo.variant;
+    return { providerId, modelId, agent, variant };
+}
+export function countTokens(text) {
+    if (!text)
+        return 0;
+    try {
+        return anthropicCountTokens(text);
+    }
+    catch {
+        return Math.round(text.length / 4);
+    }
+}
+function estimateTokensBatch(texts) {
+    return texts.map(countTokens);
+}
+export const calculateTokensSaved = (state, messages, pruneToolIds) => {
+    try {
+        const contents = [];
+        for (const msg of messages) {
+            if (isMessageCompacted(state, msg)) {
+                continue;
+            }
+            const parts = Array.isArray(msg.parts) ? msg.parts : [];
+            for (const part of parts) {
+                if (part.type !== "tool" || !pruneToolIds.includes(part.callID)) {
+                    continue;
+                }
+                if (part.tool === "question") {
+                    const questions = part.state.input?.questions;
+                    if (questions !== undefined) {
+                        const content = typeof questions === "string" ? questions : JSON.stringify(questions);
+                        contents.push(content);
+                    }
+                    continue;
+                }
+                if (part.state.status === "completed") {
+                    const content = typeof part.state.output === "string"
+                        ? part.state.output
+                        : JSON.stringify(part.state.output);
+                    contents.push(content);
+                }
+                else if (part.state.status === "error") {
+                    const content = typeof part.state.error === "string"
+                        ? part.state.error
+                        : JSON.stringify(part.state.error);
+                    contents.push(content);
+                }
+            }
+        }
+        const tokenCounts = estimateTokensBatch(contents);
+        return tokenCounts.reduce((sum, count) => sum + count, 0);
+    }
+    catch (error) {
+        return 0;
+    }
+};
+//# sourceMappingURL=utils.js.map

package/dist/lib/strategies/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../lib/strategies/utils.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,IAAI,oBAAoB,EAAE,MAAM,yBAAyB,CAAA;AAC7E,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAA;AAExE,MAAM,UAAU,gBAAgB,CAC5B,KAAmB,EACnB,QAAqB,EACrB,MAAc;IAOd,MAAM,OAAO,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,uDAAuD,CAAC,CAAA;QACrE,OAAO;YACH,UAAU,EAAE,SAAS;YACrB,OAAO,EAAE,SAAS;YAClB,KAAK,EAAE,SAAS;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;SACzB,CAAA;IACL,CAAC;IACD,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAmB,CAAA;IAC5C,MAAM,KAAK,GAAW,QAAQ,CAAC,KAAK,CAAA;IACpC,MAAM,UAAU,GAAuB,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAA;IAChE,MAAM,OAAO,GAAuB,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAA;IAC1D,MAAM,OAAO,GAAuB,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAA;IAErE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;AAClD,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,IAAY;IACpC,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAA;IACnB,IAAI,CAAC;QACD,OAAO,oBAAoB,CAAC,IAAI,CAAC,CAAA;IACrC,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;IACtC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAe;IACxC,OAAO,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;AACjC,CAAC;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAChC,KAAmB,EACnB,QAAqB,EACrB,YAAsB,EAChB,EAAE;IACR,IAAI,CAAC;QACD,MAAM,QAAQ,GAAa,EAAE,CAAA;QAC7B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,IAAI,kBAAkB,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;gBACjC,SAAQ;YACZ,CAAC;YACD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAA;YACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACvB,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC9D,SAAQ;gBACZ,CAAC;gBACD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAA;oBAC7C,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;wBAC1B,MAAM,OAAO,GACT,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;wBACzE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;oBAC1B,CAAC;oBACD,SAAQ;gBACZ,CAAC;gBACD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACpC,MAAM,OAAO,GACT,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,QAAQ;wBACjC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM;wBACnB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;oBAC3C,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;gBAC1B,CAAC;qBAAM,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBACvC,MAAM,OAAO,GACT,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,KAAK,QAAQ;wBAChC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK;wBAClB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;oBAC1C,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;gBAC1B,CAAC;YACL,CAAC;QACL,CAAC;QACD,MAAM,WAAW,GAAa,mBAAmB,CAAC,QAAQ,CAAC,CAAA;QAC3D,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,EAAE,CAAC,CAAC,CAAA;IAC7D,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,OAAO,CAAC,CAAA;IACZ,CAAC;AACL,CAAC,CAAA"}

package/dist/lib/ui/notification.d.ts

@@ -0,0 +1,12 @@
+import type { Logger } from "../logger";
+import type { SessionState } from "../state";
+import { ToolParameterEntry } from "../state";
+import { PluginConfig } from "../config";
+export type PruneReason = "completion" | "noise" | "superseded" | "exploration" | "duplicate" | "extraction";
+export declare const PRUNE_REASON_LABELS: Record<PruneReason, string>;
+export interface NotifyOptions {
+    simplified?: boolean;
+}
+export declare function sendUnifiedNotification(client: any, logger: Logger, config: PluginConfig, state: SessionState, sessionId: string, pruneToolIds: string[], toolMetadata: Map<string, ToolParameterEntry>, reason: PruneReason | undefined, params: any, workingDirectory: string, distillation?: string[], options?: NotifyOptions): Promise<boolean>;
+export declare function sendIgnoredMessage(client: any, sessionID: string, text: string, params: any, logger: Logger): Promise<void>;
+//# sourceMappingURL=notification.d.ts.map

package/dist/lib/ui/notification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"notification.d.ts","sourceRoot":"","sources":["../../../lib/ui/notification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AACvC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAA;AAQ5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAA;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAA;AAExC,MAAM,MAAM,WAAW,GACjB,YAAY,GACZ,OAAO,GACP,YAAY,GACZ,aAAa,GACb,WAAW,GACX,YAAY,CAAA;AAClB,eAAO,MAAM,mBAAmB,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAO3D,CAAA;AA+DD,MAAM,WAAW,aAAa;IAC1B,UAAU,CAAC,EAAE,OAAO,CAAA;CACvB;AAED,wBAAsB,uBAAuB,CACzC,MAAM,EAAE,GAAG,EACX,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,YAAY,EACpB,KAAK,EAAE,YAAY,EACnB,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,EAAE,EACtB,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,EAC7C,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,MAAM,EAAE,GAAG,EACX,gBAAgB,EAAE,MAAM,EACxB,YAAY,CAAC,EAAE,MAAM,EAAE,EACvB,OAAO,CAAC,EAAE,aAAa,GACxB,OAAO,CAAC,OAAO,CAAC,CA6BlB;AAED,wBAAsB,kBAAkB,CACpC,MAAM,EAAE,GAAG,EACX,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,GAAG,EACX,MAAM,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAiCf"}

package/dist/lib/ui/notification.js

@@ -0,0 +1,81 @@
+import { countDistillationTokens, formatExtracted, formatPrunedItemsList, formatStatsHeader, formatTokenCount, } from "./utils";
+export const PRUNE_REASON_LABELS = {
+    completion: "Task Complete",
+    noise: "Noise Removal",
+    superseded: "Superseded",
+    exploration: "Dead-end Exploration",
+    duplicate: "Duplicate Content",
+    extraction: "Extraction",
+};
+function buildMinimalMessage(state, reason, distillation, showDistillation) {
+    const extractedTokens = countDistillationTokens(distillation);
+    const extractedSuffix = extractedTokens > 0 ? ` (extracted ${formatTokenCount(extractedTokens)})` : "";
+    const reasonSuffix = reason && extractedTokens === 0 ? ` — ${PRUNE_REASON_LABELS[reason]}` : "";
+    let message = formatStatsHeader(state.stats.totalPruneTokens, state.stats.pruneTokenCounter, state.stats.totalPruneMessages, state.stats.pruneMessageCounter) +
+        reasonSuffix +
+        extractedSuffix;
+    return message + formatExtracted(showDistillation ? distillation : undefined);
+}
+function buildDetailedMessage(state, reason, pruneToolIds, toolMetadata, workingDirectory, distillation, showDistillation, simplified = false) {
+    let message = formatStatsHeader(state.stats.totalPruneTokens, state.stats.pruneTokenCounter, state.stats.totalPruneMessages, state.stats.pruneMessageCounter);
+    if (pruneToolIds.length > 0) {
+        const pruneTokenCounterStr = `~${formatTokenCount(state.stats.pruneTokenCounter)}`;
+        const extractedTokens = countDistillationTokens(distillation);
+        const extractedSuffix = extractedTokens > 0 ? `, extracted ${formatTokenCount(extractedTokens)}` : "";
+        const reasonLabel = reason && extractedTokens === 0 ? ` — ${PRUNE_REASON_LABELS[reason]}` : "";
+        message += `\n\n▣ Pruning (${pruneTokenCounterStr}${extractedSuffix})${reasonLabel}`;
+        const itemLines = formatPrunedItemsList(pruneToolIds, toolMetadata, workingDirectory, simplified);
+        message += "\n" + itemLines.join("\n");
+    }
+    return (message + formatExtracted(showDistillation ? distillation : undefined)).trim();
+}
+export async function sendUnifiedNotification(client, logger, config, state, sessionId, pruneToolIds, toolMetadata, reason, params, workingDirectory, distillation, options) {
+    const hasPruned = pruneToolIds.length > 0;
+    if (!hasPruned) {
+        return false;
+    }
+    if (config.pruneNotification === "off") {
+        return false;
+    }
+    const showDistillation = config.tools.extract.showDistillation;
+    const simplified = options?.simplified ?? false;
+    const message = config.pruneNotification === "minimal"
+        ? buildMinimalMessage(state, reason, distillation, showDistillation)
+        : buildDetailedMessage(state, reason, pruneToolIds, toolMetadata, workingDirectory, distillation, showDistillation, simplified);
+    await sendIgnoredMessage(client, sessionId, message, params, logger);
+    return true;
+}
+export async function sendIgnoredMessage(client, sessionID, text, params, logger) {
+    const agent = params.agent || undefined;
+    const variant = params.variant || undefined;
+    const model = params.providerId && params.modelId
+        ? {
+            providerID: params.providerId,
+            modelID: params.modelId,
+        }
+        : undefined;
+    try {
+        await client.session.prompt({
+            path: {
+                id: sessionID,
+            },
+            body: {
+                noReply: true,
+                agent: agent,
+                model: model,
+                variant: variant,
+                parts: [
+                    {
+                        type: "text",
+                        text: text,
+                        ignored: true,
+                    },
+                ],
+            },
+        });
+    }
+    catch (error) {
+        logger.error("Failed to send notification", { error: error.message });
+    }
+}
+//# sourceMappingURL=notification.js.map

package/dist/lib/ui/notification.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"notification.js","sourceRoot":"","sources":["../../../lib/ui/notification.ts"],"names":[],"mappings":"AAEA,OAAO,EACH,uBAAuB,EACvB,eAAe,EACf,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,GACnB,MAAM,SAAS,CAAA;AAWhB,MAAM,CAAC,MAAM,mBAAmB,GAAgC;IAC5D,UAAU,EAAE,eAAe;IAC3B,KAAK,EAAE,eAAe;IACtB,UAAU,EAAE,YAAY;IACxB,WAAW,EAAE,sBAAsB;IACnC,SAAS,EAAE,mBAAmB;IAC9B,UAAU,EAAE,YAAY;CAC3B,CAAA;AAED,SAAS,mBAAmB,CACxB,KAAmB,EACnB,MAA+B,EAC/B,YAAkC,EAClC,gBAAyB;IAEzB,MAAM,eAAe,GAAG,uBAAuB,CAAC,YAAY,CAAC,CAAA;IAC7D,MAAM,eAAe,GACjB,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAA;IAClF,MAAM,YAAY,GAAG,MAAM,IAAI,eAAe,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,mBAAmB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;IAC/F,IAAI,OAAO,GACP,iBAAiB,CACb,KAAK,CAAC,KAAK,CAAC,gBAAgB,EAC5B,KAAK,CAAC,KAAK,CAAC,iBAAiB,EAC7B,KAAK,CAAC,KAAK,CAAC,kBAAkB,EAC9B,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAClC;QACD,YAAY;QACZ,eAAe,CAAA;IAEnB,OAAO,OAAO,GAAG,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;AACjF,CAAC;AAED,SAAS,oBAAoB,CACzB,KAAmB,EACnB,MAA+B,EAC/B,YAAsB,EACtB,YAA6C,EAC7C,gBAAwB,EACxB,YAAkC,EAClC,gBAAyB,EACzB,aAAsB,KAAK;IAE3B,IAAI,OAAO,GAAG,iBAAiB,CAC3B,KAAK,CAAC,KAAK,CAAC,gBAAgB,EAC5B,KAAK,CAAC,KAAK,CAAC,iBAAiB,EAC7B,KAAK,CAAC,KAAK,CAAC,kBAAkB,EAC9B,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAClC,CAAA;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,oBAAoB,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAA;QAClF,MAAM,eAAe,GAAG,uBAAuB,CAAC,YAAY,CAAC,CAAA;QAC7D,MAAM,eAAe,GACjB,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,gBAAgB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;QACjF,MAAM,WAAW,GACb,MAAM,IAAI,eAAe,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,mBAAmB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;QAC9E,OAAO,IAAI,kBAAkB,oBAAoB,GAAG,eAAe,IAAI,WAAW,EAAE,CAAA;QAEpF,MAAM,SAAS,GAAG,qBAAqB,CACnC,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,UAAU,CACb,CAAA;QACD,OAAO,IAAI,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC1C,CAAC;IAED,OAAO,CAAC,OAAO,GAAG,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;AAC1F,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,uBAAuB,CACzC,MAAW,EACX,MAAc,EACd,MAAoB,EACpB,KAAmB,EACnB,SAAiB,EACjB,YAAsB,EACtB,YAA6C,EAC7C,MAA+B,EAC/B,MAAW,EACX,gBAAwB,EACxB,YAAuB,EACvB,OAAuB;IAEvB,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,CAAA;IACzC,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,IAAI,MAAM,CAAC,iBAAiB,KAAK,KAAK,EAAE,CAAC;QACrC,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAA;IAC9D,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,KAAK,CAAA;IAE/C,MAAM,OAAO,GACT,MAAM,CAAC,iBAAiB,KAAK,SAAS;QAClC,CAAC,CAAC,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,CAAC;QACpE,CAAC,CAAC,oBAAoB,CAChB,KAAK,EACL,MAAM,EACN,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,YAAY,EACZ,gBAAgB,EAChB,UAAU,CACb,CAAA;IAEX,MAAM,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;IACpE,OAAO,IAAI,CAAA;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACpC,MAAW,EACX,SAAiB,EACjB,IAAY,EACZ,MAAW,EACX,MAAc;IAEd,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,SAAS,CAAA;IACvC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,SAAS,CAAA;IAC3C,MAAM,KAAK,GACP,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,OAAO;QAC/B,CAAC,CAAC;YACI,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;SAC1B;QACH,CAAC,CAAC,SAAS,CAAA;IAEnB,IAAI,CAAC;QACD,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YACxB,IAAI,EAAE;gBACF,EAAE,EAAE,SAAS;aAChB;YACD,IAAI,EAAE;gBACF,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,KAAK;gBACZ,OAAO,EAAE,OAAO;gBAChB,KAAK,EAAE;oBACH;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI;wBACV,OAAO,EAAE,IAAI;qBAChB;iBACJ;aACJ;SACJ,CAAC,CAAA;IACN,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;IACzE,CAAC;AACL,CAAC"}

package/dist/lib/ui/utils.d.ts

@@ -0,0 +1,10 @@
+import { ToolParameterEntry } from "../state";
+export declare function countDistillationTokens(distillation?: string[]): number;
+export declare function formatExtracted(distillation?: string[]): string;
+export declare function formatStatsHeader(totalTokensSaved: number, pruneTokenCounter: number, totalMessagesPruned: number, messagesPruned: number): string;
+export declare function formatTokenCount(tokens: number): string;
+export declare function truncate(str: string, maxLen?: number): string;
+export declare function shortenPath(input: string, workingDirectory?: string): string;
+export declare function formatPrunedItemsList(pruneToolIds: string[], toolMetadata: Map<string, ToolParameterEntry>, workingDirectory?: string, simplified?: boolean): string[];
+export declare function formatPruningResultForTool(prunedIds: string[], toolMetadata: Map<string, ToolParameterEntry>, workingDirectory?: string, simplified?: boolean): string;
+//# sourceMappingURL=utils.d.ts.map

package/dist/lib/ui/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../lib/ui/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAA;AAI7C,wBAAgB,uBAAuB,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAGvE;AAED,wBAAgB,eAAe,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAS/D;AAED,wBAAgB,iBAAiB,CAC7B,gBAAgB,EAAE,MAAM,EACxB,iBAAiB,EAAE,MAAM,EACzB,mBAAmB,EAAE,MAAM,EAC3B,cAAc,EAAE,MAAM,GACvB,MAAM,CAgBR;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKvD;AAED,wBAAgB,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,GAAE,MAAW,GAAG,MAAM,CAGjE;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,MAAM,CAU5E;AAeD,wBAAgB,qBAAqB,CACjC,YAAY,EAAE,MAAM,EAAE,EACtB,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,EAC7C,gBAAgB,CAAC,EAAE,MAAM,EACzB,UAAU,GAAE,OAAe,GAC5B,MAAM,EAAE,CAwCV;AAED,wBAAgB,0BAA0B,CACtC,SAAS,EAAE,MAAM,EAAE,EACnB,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,EAC7C,gBAAgB,CAAC,EAAE,MAAM,EACzB,UAAU,GAAE,OAAe,GAC5B,MAAM,CAWR"}

package/dist/lib/ui/utils.js

@@ -0,0 +1,113 @@
+import { extractParameterKey } from "../messages/utils";
+import { countTokens } from "../strategies/utils";
+export function countDistillationTokens(distillation) {
+    if (!distillation || distillation.length === 0)
+        return 0;
+    return countTokens(distillation.join("\n"));
+}
+export function formatExtracted(distillation) {
+    if (!distillation || distillation.length === 0) {
+        return "";
+    }
+    let result = `\n\n▣ Extracted`;
+    for (const finding of distillation) {
+        result += `\n───\n${finding}`;
+    }
+    return result;
+}
+export function formatStatsHeader(totalTokensSaved, pruneTokenCounter, totalMessagesPruned, messagesPruned) {
+    const totalMessages = totalMessagesPruned + messagesPruned;
+    const totalTokens = totalTokensSaved + pruneTokenCounter;
+    const totalTokensStr = `~${formatTokenCount(totalTokens)}`;
+    // Show addition indicator for current operation
+    const additionParts = [];
+    if (messagesPruned > 0) {
+        additionParts.push(`+${messagesPruned} 💬`);
+    }
+    if (pruneTokenCounter > 0) {
+        additionParts.push(`+${formatTokenCount(pruneTokenCounter)} 🪙`);
+    }
+    const additionSuffix = additionParts.length > 0 ? ` (${additionParts.join(", ")})` : "";
+    return `acp - ${totalMessages} 💬 · ${totalTokensStr} 🪙${additionSuffix}`;
+}
+export function formatTokenCount(tokens) {
+    if (tokens >= 1000) {
+        return `${(tokens / 1000).toFixed(1)}K`.replace(".0K", "K");
+    }
+    return tokens.toString();
+}
+export function truncate(str, maxLen = 60) {
+    if (str.length <= maxLen)
+        return str;
+    return str.slice(0, maxLen - 3) + "...";
+}
+export function shortenPath(input, workingDirectory) {
+    const inPathMatch = input.match(/^(.+) in (.+)$/);
+    if (inPathMatch?.[1] && inPathMatch[2]) {
+        const prefix = inPathMatch[1];
+        const pathPart = inPathMatch[2];
+        const shortenedPath = shortenSinglePath(pathPart, workingDirectory);
+        return `${prefix} in ${shortenedPath}`;
+    }
+    return shortenSinglePath(input, workingDirectory);
+}
+function shortenSinglePath(path, workingDirectory) {
+    if (workingDirectory) {
+        if (path.startsWith(workingDirectory + "/")) {
+            return path.slice(workingDirectory.length + 1);
+        }
+        if (path === workingDirectory) {
+            return ".";
+        }
+    }
+    return path;
+}
+export function formatPrunedItemsList(pruneToolIds, toolMetadata, workingDirectory, simplified = false) {
+    const lines = [];
+    for (const id of pruneToolIds) {
+        const metadata = toolMetadata.get(id);
+        if (metadata) {
+            const paramKey = extractParameterKey(metadata.tool, metadata.parameters);
+            if (paramKey) {
+                // Use 60 char limit to match notification style
+                const displayKey = truncate(shortenPath(paramKey, workingDirectory), 60);
+                if (simplified) {
+                    lines.push(displayKey);
+                }
+                else {
+                    lines.push(`→ ${metadata.tool}: ${displayKey}`);
+                }
+            }
+            else {
+                if (simplified) {
+                    lines.push(metadata.tool);
+                }
+                else {
+                    lines.push(`→ ${metadata.tool}`);
+                }
+            }
+        }
+    }
+    const knownCount = pruneToolIds.filter((id) => toolMetadata.has(id)).length;
+    const unknownCount = pruneToolIds.length - knownCount;
+    if (unknownCount > 0) {
+        if (simplified) {
+            lines.push(`(${unknownCount} tool${unknownCount > 1 ? "s" : ""} with unknown metadata)`);
+        }
+        else {
+            lines.push(`→ (${unknownCount} tool${unknownCount > 1 ? "s" : ""} with unknown metadata)`);
+        }
+    }
+    return lines;
+}
+export function formatPruningResultForTool(prunedIds, toolMetadata, workingDirectory, simplified = false) {
+    const lines = [];
+    lines.push(`Context pruning complete. Pruned ${prunedIds.length} tool outputs.`);
+    lines.push("");
+    if (prunedIds.length > 0) {
+        lines.push(`Semantically pruned (${prunedIds.length}):`);
+        lines.push(...formatPrunedItemsList(prunedIds, toolMetadata, workingDirectory, simplified));
+    }
+    return lines.join("\n").trim();
+}
+//# sourceMappingURL=utils.js.map