@timetotest/cli 0.2.4 → 0.3.1

package/dist/src/lib/context-compactor.js

@@ -0,0 +1,310 @@
+/**
+ * Context Compactor - Industry-standard conversation history management
+ *
+ * Features:
+ * - Accurate token counting using tiktoken
+ * - Conversation turn preservation (keeps user-assistant pairs intact)
+ * - LLM-based summarization for old context
+ * - Importance scoring to preserve critical messages
+ * - Tool result preservation for recent tools
+ */
+import { encoding_for_model } from "tiktoken";
+export class ContextCompactor {
+    config;
+    encoder = null;
+    constructor(config) {
+        this.config = {
+            maxTokens: config?.maxTokens ?? 200000,
+            targetTokens: config?.targetTokens ?? 150000,
+            preserveRecentTurns: config?.preserveRecentTurns ?? 10,
+            preserveRecentTools: config?.preserveRecentTools ?? 5,
+            enableSummarization: config?.enableSummarization ?? true,
+            model: config?.model ?? "gpt-4",
+        };
+    }
+    /**
+     * Get or initialize tokenizer
+     */
+    getEncoder() {
+        if (!this.encoder) {
+            try {
+                this.encoder = encoding_for_model(this.config.model);
+            }
+            catch {
+                // Fallback to cl100k_base (GPT-4/GPT-3.5-turbo)
+                this.encoder = encoding_for_model("gpt-4");
+            }
+        }
+        return this.encoder;
+    }
+    /**
+     * Accurate token counting using tiktoken
+     */
+    countTokens(messages) {
+        const encoder = this.getEncoder();
+        let totalTokens = 0;
+        for (const msg of messages) {
+            // Count role tokens
+            totalTokens += 4; // Every message has role overhead
+            // Count content tokens
+            if (typeof msg.content === "string") {
+                totalTokens += encoder.encode(msg.content).length;
+            }
+            else if (Array.isArray(msg.content)) {
+                totalTokens += encoder.encode(JSON.stringify(msg.content)).length;
+            }
+            // Count tool call tokens
+            if (msg.toolCalls) {
+                totalTokens += encoder.encode(JSON.stringify(msg.toolCalls)).length;
+            }
+            // Count metadata tokens
+            if (msg.metadata) {
+                totalTokens += encoder.encode(JSON.stringify(msg.metadata)).length;
+            }
+            // Count name/tool_call_id
+            if (msg.toolCall) {
+                totalTokens += encoder.encode(JSON.stringify(msg.toolCall)).length;
+            }
+        }
+        // Add overhead for message formatting
+        totalTokens += 3; // Every reply is primed with assistant
+        return totalTokens;
+    }
+    /**
+     * Check if compaction is needed
+     */
+    needsCompaction(messages) {
+        const tokens = this.countTokens(messages);
+        return tokens >= this.config.maxTokens;
+    }
+    /**
+     * Group messages into conversation turns
+     * A turn = user message + assistant response + any tool calls
+     */
+    groupIntoTurns(messages) {
+        const turns = [];
+        let i = 0;
+        while (i < messages.length) {
+            const msg = messages[i];
+            // Skip system messages (they're not part of turns)
+            if (msg.role === "system") {
+                i++;
+                continue;
+            }
+            // Start of a turn (user message)
+            if (msg.role === "user") {
+                const turn = {
+                    userMessage: msg,
+                    toolMessages: [],
+                    startIndex: i,
+                    endIndex: i,
+                    tokens: 0,
+                    importance: 0,
+                };
+                i++;
+                // Collect assistant response
+                if (i < messages.length && messages[i].role === "assistant") {
+                    turn.assistantMessage = messages[i];
+                    turn.endIndex = i;
+                    i++;
+                }
+                // Collect tool messages
+                while (i < messages.length && messages[i].role === "tool") {
+                    turn.toolMessages.push(messages[i]);
+                    turn.endIndex = i;
+                    i++;
+                }
+                // Calculate tokens for this turn
+                const turnMessages = messages.slice(turn.startIndex, turn.endIndex + 1);
+                turn.tokens = this.countTokens(turnMessages);
+                // Calculate importance score
+                turn.importance = this.calculateImportance(turn);
+                turns.push(turn);
+            }
+            else {
+                // Orphaned message (shouldn't happen but handle gracefully)
+                i++;
+            }
+        }
+        return turns;
+    }
+    /**
+     * Calculate importance score for a conversation turn
+     * Higher score = more important to preserve
+     */
+    calculateImportance(turn) {
+        let score = 1.0; // Base score
+        // Boost if contains errors
+        const hasError = turn.assistantMessage?.content
+            ?.toString()
+            .toLowerCase()
+            .includes("error") ||
+            turn.toolMessages.some((t) => t.content?.toString().toLowerCase().includes("error"));
+        if (hasError)
+            score += 2.0;
+        // Boost if user corrected the agent
+        const hasCorrection = turn.userMessage.content
+            ?.toString()
+            .toLowerCase()
+            .match(/no|wrong|incorrect|actually|fix|correct/);
+        if (hasCorrection)
+            score += 1.5;
+        // Boost if contains important tool calls
+        const importantTools = ["run_command", "write_file", "delete_file"];
+        const hasImportantTool = turn.toolMessages.some((t) => importantTools.includes(t.toolCall?.name || ""));
+        if (hasImportantTool)
+            score += 1.0;
+        // Penalize very long tool results (likely verbose output)
+        const hasLongToolResult = turn.toolMessages.some((t) => (t.content?.toString().length || 0) > 5000);
+        if (hasLongToolResult)
+            score -= 0.5;
+        return Math.max(0, score);
+    }
+    /**
+     * Main compaction method
+     */
+    async compact(messages, httpClient) {
+        const tokensBeforeCompaction = this.countTokens(messages);
+        if (tokensBeforeCompaction < this.config.maxTokens) {
+            return {
+                compactedMessages: messages,
+                removedCount: 0,
+                tokensBeforeCompaction,
+                tokensAfterCompaction: tokensBeforeCompaction,
+                strategy: "none",
+                summaryGenerated: false,
+            };
+        }
+        // Separate system messages from conversation
+        const systemMessages = messages.filter((m) => m.role === "system");
+        const conversationMessages = messages.filter((m) => m.role !== "system");
+        // Group into turns
+        const turns = this.groupIntoTurns(conversationMessages);
+        // Keep recent turns
+        const recentTurns = turns.slice(-this.config.preserveRecentTurns);
+        const oldTurns = turns.slice(0, -this.config.preserveRecentTurns);
+        // Extract recent messages
+        const recentMessages = [];
+        for (const turn of recentTurns) {
+            recentMessages.push(...conversationMessages.slice(turn.startIndex, turn.endIndex + 1));
+        }
+        // Preserve recent tool results
+        const recentToolMessages = recentMessages.filter((m) => m.role === "tool");
+        const toolsToPreserve = recentToolMessages.slice(-this.config.preserveRecentTools);
+        let compactedMessages;
+        let summaryGenerated = false;
+        if (this.config.enableSummarization && oldTurns.length > 0 && httpClient) {
+            // Generate LLM summary of old context
+            const summary = await this.generateSummary(oldTurns, httpClient);
+            summaryGenerated = true;
+            compactedMessages = [
+                ...systemMessages,
+                {
+                    role: "system",
+                    content: summary,
+                    timestamp: Date.now(),
+                    metadata: {
+                        type: "conversation_summary",
+                        turns_summarized: oldTurns.length,
+                        tokens_before: this.countTokens(conversationMessages.slice(0, oldTurns[oldTurns.length - 1].endIndex + 1)),
+                    },
+                },
+                ...recentMessages,
+            ];
+        }
+        else {
+            // Fallback: Keep high-importance turns from old context
+            const importantOldTurns = oldTurns
+                .sort((a, b) => b.importance - a.importance)
+                .slice(0, 3); // Keep top 3 important turns
+            const importantMessages = [];
+            for (const turn of importantOldTurns) {
+                importantMessages.push(...conversationMessages.slice(turn.startIndex, turn.endIndex + 1));
+            }
+            compactedMessages = [
+                ...systemMessages,
+                {
+                    role: "system",
+                    content: `[Earlier conversation: ${oldTurns.length} turns compacted. Preserved ${importantOldTurns.length} important turns.]`,
+                    timestamp: Date.now(),
+                    metadata: {
+                        type: "compaction_marker",
+                        turns_removed: oldTurns.length - importantOldTurns.length,
+                    },
+                },
+                ...importantMessages,
+                ...recentMessages,
+            ];
+        }
+        const tokensAfterCompaction = this.countTokens(compactedMessages);
+        return {
+            compactedMessages,
+            removedCount: messages.length - compactedMessages.length,
+            tokensBeforeCompaction,
+            tokensAfterCompaction,
+            strategy: summaryGenerated ? "llm-summarization" : "importance-based",
+            summaryGenerated,
+        };
+    }
+    /**
+     * Generate LLM summary of old conversation turns
+     */
+    async generateSummary(turns, httpClient) {
+        // Build a concise representation of the turns
+        const turnSummaries = turns.map((turn, idx) => {
+            const userContent = typeof turn.userMessage.content === "string"
+                ? turn.userMessage.content
+                : JSON.stringify(turn.userMessage.content);
+            const assistantContent = turn.assistantMessage
+                ? typeof turn.assistantMessage.content === "string"
+                    ? turn.assistantMessage.content
+                    : JSON.stringify(turn.assistantMessage.content)
+                : "[No response]";
+            const toolsUsed = turn.toolMessages
+                .map((t) => t.toolCall?.name)
+                .filter(Boolean)
+                .join(", ");
+            return `Turn ${idx + 1}:\nUser: ${userContent.substring(0, 200)}${userContent.length > 200 ? "..." : ""}\nAssistant: ${assistantContent.substring(0, 200)}${assistantContent.length > 200 ? "..." : ""}${toolsUsed ? `\nTools: ${toolsUsed}` : ""}`;
+        });
+        const conversationText = turnSummaries.join("\n\n");
+        try {
+            // Call backend to generate summary
+            const response = await httpClient.post("/api/v1/ai/agent/summarize", {
+                conversation: conversationText,
+                max_tokens: 500,
+            });
+            return (response.data?.summary ||
+                `[Summary of ${turns.length} conversation turns]`);
+        }
+        catch (error) {
+            // Fallback to simple summary if API fails
+            const userMessages = turns.length;
+            const toolCalls = turns.reduce((sum, t) => sum + t.toolMessages.length, 0);
+            return `[Earlier conversation: ${userMessages} user messages, ${toolCalls} tool executions. Context preserved for continuity.]`;
+        }
+    }
+    /**
+     * Get compaction statistics
+     */
+    getStats(messages) {
+        const tokens = this.countTokens(messages);
+        const turns = this.groupIntoTurns(messages.filter((m) => m.role !== "system"));
+        return {
+            totalMessages: messages.length,
+            tokens,
+            needsCompaction: tokens >= this.config.maxTokens,
+            utilizationPercent: (tokens / this.config.maxTokens) * 100,
+            conversationTurns: turns.length,
+        };
+    }
+    /**
+     * Clean up tokenizer
+     */
+    dispose() {
+        if (this.encoder) {
+            this.encoder.free();
+            this.encoder = null;
+        }
+    }
+}
+//# sourceMappingURL=context-compactor.js.map

package/dist/src/lib/context-compactor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-compactor.js","sourceRoot":"","sources":["../../../src/lib/context-compactor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAC,kBAAkB,EAAgB,MAAM,UAAU,CAAC;AA8B3D,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAmB;IACzB,OAAO,GAAoB,IAAI,CAAC;IAExC,YAAY,MAAkC;QAC5C,IAAI,CAAC,MAAM,GAAG;YACZ,SAAS,EAAE,MAAM,EAAE,SAAS,IAAI,MAAM;YACtC,YAAY,EAAE,MAAM,EAAE,YAAY,IAAI,MAAM;YAC5C,mBAAmB,EAAE,MAAM,EAAE,mBAAmB,IAAI,EAAE;YACtD,mBAAmB,EAAE,MAAM,EAAE,mBAAmB,IAAI,CAAC;YACrD,mBAAmB,EAAE,MAAM,EAAE,mBAAmB,IAAI,IAAI;YACxD,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,OAAO;SAChC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,KAAY,CAAC,CAAC;YAC9D,CAAC;YAAC,MAAM,CAAC;gBACP,gDAAgD;gBAChD,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAmB;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,oBAAoB;YACpB,WAAW,IAAI,CAAC,CAAC,CAAC,kCAAkC;YAEpD,uBAAuB;YACvB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACpC,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;YACpD,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtC,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;YACpE,CAAC;YAED,yBAAyB;YACzB,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;gBAClB,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC;YACtE,CAAC;YAED,wBAAwB;YACxB,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACjB,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;YACrE,CAAC;YAED,0BAA0B;YAC1B,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACjB,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;YACrE,CAAC;QACH,CAAC;QAED,sCAAsC;QACtC,WAAW,IAAI,CAAC,CAAC,CAAC,uCAAuC;QAEzD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAAmB;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC1C,OAAO,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;IACzC,CAAC;IAED;;;OAGG;IACK,cAAc,CAAC,QAAmB;QACxC,MAAM,KAAK,GAAuB,EAAE,CAAC;QACrC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEV,OAAO,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAExB,mDAAmD;YACnD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,CAAC,EAAE,CAAC;gBACJ,SAAS;YACX,CAAC;YAED,iCAAiC;YACjC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAqB;oBAC7B,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,EAAE;oBAChB,UAAU,EAAE,CAAC;oBACb,QAAQ,EAAE,CAAC;oBACX,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd,CAAC;gBAEF,CAAC,EAAE,CAAC;gBAEJ,6BAA6B;gBAC7B,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC5D,IAAI,CAAC,gBAAgB,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;oBACpC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;oBAClB,CAAC,EAAE,CAAC;gBACN,CAAC;gBAED,wBAAwB;gBACxB,OAAO,CAAC,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;oBACpC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;oBAClB,CAAC,EAAE,CAAC;gBACN,CAAC;gBAED,iCAAiC;gBACjC,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;gBACxE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;gBAE7C,6BAA6B;gBAC7B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAEjD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,4DAA4D;gBAC5D,CAAC,EAAE,CAAC;YACN,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;OAGG;IACK,mBAAmB,CAAC,IAAsB;QAChD,IAAI,KAAK,GAAG,GAAG,CAAC,CAAC,aAAa;QAE9B,2BAA2B;QAC3B,MAAM,QAAQ,GACZ,IAAI,CAAC,gBAAgB,EAAE,OAAO;YAC5B,EAAE,QAAQ,EAAE;aACX,WAAW,EAAE;aACb,QAAQ,CAAC,OAAO,CAAC;YACpB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC3B,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CACtD,CAAC;QACJ,IAAI,QAAQ;YAAE,KAAK,IAAI,GAAG,CAAC;QAE3B,oCAAoC;QACpC,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;YAC5C,EAAE,QAAQ,EAAE;aACX,WAAW,EAAE;aACb,KAAK,CAAC,yCAAyC,CAAC,CAAC;QACpD,IAAI,aAAa;YAAE,KAAK,IAAI,GAAG,CAAC;QAEhC,yCAAyC;QACzC,MAAM,cAAc,GAAG,CAAC,aAAa,EAAE,YAAY,EAAE,aAAa,CAAC,CAAC;QACpE,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACpD,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,CAChD,CAAC;QACF,IAAI,gBAAgB;YAAE,KAAK,IAAI,GAAG,CAAC;QAEnC,0DAA0D;QAC1D,MAAM,iBAAiB,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,IAAI,CAClD,CAAC;QACF,IAAI,iBAAiB;YAAE,KAAK,IAAI,GAAG,CAAC;QAEpC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAmB,EACnB,UAAgB;QAEhB,MAAM,sBAAsB,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE1D,IAAI,sBAAsB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YACnD,OAAO;gBACL,iBAAiB,EAAE,QAAQ;gBAC3B,YAAY,EAAE,CAAC;gBACf,sBAAsB;gBACtB,qBAAqB,EAAE,sBAAsB;gBAC7C,QAAQ,EAAE,MAAM;gBAChB,gBAAgB,EAAE,KAAK;aACxB,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACnE,MAAM,oBAAoB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEzE,mBAAmB;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QAExD,oBAAoB;QACpB,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAClE,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAElE,0BAA0B;QAC1B,MAAM,cAAc,GAAc,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,cAAc,CAAC,IAAI,CACjB,GAAG,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAClE,CAAC;QACJ,CAAC;QAED,+BAA+B;QAC/B,MAAM,kBAAkB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC3E,MAAM,eAAe,GAAG,kBAAkB,CAAC,KAAK,CAC9C,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CACjC,CAAC;QAEF,IAAI,iBAA4B,CAAC;QACjC,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAE7B,IAAI,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,EAAE,CAAC;YACzE,sCAAsC;YACtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACjE,gBAAgB,GAAG,IAAI,CAAC;YAExB,iBAAiB,GAAG;gBAClB,GAAG,cAAc;gBACjB;oBACE,IAAI,EAAE,QAAiB;oBACvB,OAAO,EAAE,OAAO;oBAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,QAAQ,EAAE;wBACR,IAAI,EAAE,sBAAsB;wBAC5B,gBAAgB,EAAE,QAAQ,CAAC,MAAM;wBACjC,aAAa,EAAE,IAAI,CAAC,WAAW,CAC7B,oBAAoB,CAAC,KAAK,CACxB,CAAC,EACD,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAC3C,CACF;qBACF;iBACF;gBACD,GAAG,cAAc;aAClB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,wDAAwD;YACxD,MAAM,iBAAiB,GAAG,QAAQ;iBAC/B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;iBAC3C,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,6BAA6B;YAE7C,MAAM,iBAAiB,GAAc,EAAE,CAAC;YACxC,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;gBACrC,iBAAiB,CAAC,IAAI,CACpB,GAAG,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAClE,CAAC;YACJ,CAAC;YAED,iBAAiB,GAAG;gBAClB,GAAG,cAAc;gBACjB;oBACE,IAAI,EAAE,QAAiB;oBACvB,OAAO,EAAE,0BAA0B,QAAQ,CAAC,MAAM,+BAA+B,iBAAiB,CAAC,MAAM,oBAAoB;oBAC7H,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,QAAQ,EAAE;wBACR,IAAI,EAAE,mBAAmB;wBACzB,aAAa,EAAE,QAAQ,CAAC,MAAM,GAAG,iBAAiB,CAAC,MAAM;qBAC1D;iBACF;gBACD,GAAG,iBAAiB;gBACpB,GAAG,cAAc;aAClB,CAAC;QACJ,CAAC;QAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;QAElE,OAAO;YACL,iBAAiB;YACjB,YAAY,EAAE,QAAQ,CAAC,MAAM,GAAG,iBAAiB,CAAC,MAAM;YACxD,sBAAsB;YACtB,qBAAqB;YACrB,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,kBAAkB;YACrE,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe,CAC3B,KAAyB,EACzB,UAAe;QAEf,8CAA8C;QAC9C,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5C,MAAM,WAAW,GACf,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,KAAK,QAAQ;gBAC1C,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO;gBAC1B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAE/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,gBAAgB;gBAC5C,CAAC,CAAC,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,KAAK,QAAQ;oBACjD,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO;oBAC/B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;gBACjD,CAAC,CAAC,eAAe,CAAC;YAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC;iBAC5B,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,OAAO,QAAQ,GAAG,GAAG,CAAC,YAAY,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,WAAW,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,gBAAgB,gBAAgB,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,YAAY,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACtP,CAAC,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpD,IAAI,CAAC;YACH,mCAAmC;YACnC,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;gBACnE,YAAY,EAAE,gBAAgB;gBAC9B,UAAU,EAAE,GAAG;aAChB,CAAC,CAAC;YAEH,OAAO,CACL,QAAQ,CAAC,IAAI,EAAE,OAAO;gBACtB,eAAe,KAAK,CAAC,MAAM,sBAAsB,CAClD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,0CAA0C;YAC1C,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC;YAClC,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAC5B,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,EACvC,CAAC,CACF,CAAC;YACF,OAAO,0BAA0B,YAAY,mBAAmB,SAAS,sDAAsD,CAAC;QAClI,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,QAAmB;QAO1B,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAC/B,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAC5C,CAAC;QAEF,OAAO;YACL,aAAa,EAAE,QAAQ,CAAC,MAAM;YAC9B,MAAM;YACN,eAAe,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YAChD,kBAAkB,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,GAAG;YAC1D,iBAAiB,EAAE,KAAK,CAAC,MAAM;SAChC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,CAAC;IACH,CAAC;CACF"}

package/dist/src/lib/http.js

@@ -36,6 +36,14 @@ export function createAgentHttpClient(options) {
             const response = await instance.post("/api/v1/ai/agent/reason", payload, { signal });
             return response.data;
         },
+        async getAvailableModels() {
+            const response = await instance.get("/api/v1/ai/models");
+            return response.data;
+        },
+        async setPreferredModel(modelId) {
+            const response = await instance.put("/api/v1/ai/preferred-model", { model_id: modelId });
+            return response.data;
+        },
         async startTest(payload, params) {
             const response = await instance.post("/api/v1/start-test", payload, { params });
             return response.data;

package/dist/src/lib/http.js.map

@@ -1 +1 @@
-{"version":3,"file":"http.js","sourceRoot":"","sources":["../../../src/lib/http.ts"],"names":[],"mappings":"AAAA,OAAO,KAAsB,MAAM,OAAO,CAAC;AAC3C,OAAO,EAAC,YAAY,EAAE,aAAa,EAAC,MAAM,aAAa,CAAC;AAGxD,MAAM,UAAU,gBAAgB,CAAC,OAAgB;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,YAAY,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,EAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;IAClE,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC3C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;QACtD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAwDD,MAAM,UAAU,qBAAqB,CAAC,OAAyB;IAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/C,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,OAAO;QACP,OAAO,EAAE,OAAO,CAAC,SAAS,IAAI,MAAM;KACrC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,EAAE,CAAC;QAC9C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;QACtD,CAAC;QACD,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAAC;QACtD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,KAAK,CAAC,eAAe,CACnB,OAA2B,EAC3B,MAAoB;YAEpB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAClC,yBAAyB,EACzB,OAAO,EACP,EAAC,MAAM,EAAC,CACT,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAID,KAAK,CAAC,SAAS,CACb,OAAyB,EACzB,MAA2D;YAE3D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAClC,oBAAoB,EACpB,OAAO,EACP,EAAC,MAAM,EAAC,CACT,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,MAAY;YACjC,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../../src/lib/http.ts"],"names":[],"mappings":"AAAA,OAAO,KAAsB,MAAM,OAAO,CAAC;AAC3C,OAAO,EAAC,YAAY,EAAE,aAAa,EAAC,MAAM,aAAa,CAAC;AAGxD,MAAM,UAAU,gBAAgB,CAAC,OAAgB;IAC/C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,YAAY,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,EAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;IAClE,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC3C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;QACtD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,OAAO,QAAQ,CAAC;AAClB,CAAC;AA8FD,MAAM,UAAU,qBAAqB,CAAC,OAAyB;IAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/C,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,OAAO;QACP,OAAO,EAAE,OAAO,CAAC,SAAS,IAAI,MAAM;KACrC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,EAAE,CAAC;QAC9C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;QACtD,CAAC;QACD,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAAC;QACtD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,KAAK,CAAC,eAAe,CACnB,OAA2B,EAC3B,MAAoB;YAEpB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAClC,yBAAyB,EACzB,OAAO,EACP,EAAC,MAAM,EAAC,CACT,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,CAAC,kBAAkB;YACtB,MAAM,QAAQ,GACZ,MAAM,QAAQ,CAAC,GAAG,CAA0B,mBAAmB,CAAC,CAAC;YACnE,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,CAAC,iBAAiB,CACrB,OAAe;YAEf,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,GAAG,CACjC,4BAA4B,EAC5B,EAAC,QAAQ,EAAE,OAAO,EAAC,CACpB,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,CAAC,SAAS,CACb,OAAyB,EACzB,MAA2D;YAE3D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAClC,oBAAoB,EACpB,OAAO,EACP,EAAC,MAAM,EAAC,CACT,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,MAAY;YACjC,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/lib/local-tools/code/__tests__/grep-search.test.js

@@ -0,0 +1,146 @@
+/**
+ * Unit tests for grep_search tool.
+ * Tests text search, regex search, case sensitivity, and context lines.
+ */
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import * as fs from "fs/promises";
+import * as path from "path";
+import * as os from "os";
+import { grepSearch } from "../grep-search.js";
+describe("grepSearch", () => {
+    let tempDir;
+    let context;
+    beforeEach(async () => {
+        tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "grep-search-test-"));
+        context = { workingDirectory: tempDir };
+    });
+    afterEach(async () => {
+        await fs.rm(tempDir, { recursive: true, force: true });
+    });
+    describe("text search", () => {
+        it("should find text matches in files", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "hello world\nfoo bar\nhello again");
+            const result = await grepSearch({ query: "hello" }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(2);
+            expect(result.matches[0].content).toContain("hello");
+        });
+        it("should return file path and line number", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "line1\nmatch here\nline3");
+            const result = await grepSearch({ query: "match" }, context);
+            expect(result.success).toBe(true);
+            expect(result.matches[0].file).toBe("test.txt");
+            expect(result.matches[0].line).toBe(2);
+        });
+        it("should return column position", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "prefix match suffix");
+            const result = await grepSearch({ query: "match" }, context);
+            expect(result.success).toBe(true);
+            expect(result.matches[0].column).toBe(8); // 1-indexed
+        });
+        it("should return error for empty query", async () => {
+            const result = await grepSearch({ query: "" }, context);
+            expect(result.success).toBe(false);
+            expect(result.error).toBe("Search query cannot be empty");
+        });
+        it("should search multiple files", async () => {
+            await fs.writeFile(path.join(tempDir, "file1.txt"), "match here");
+            await fs.writeFile(path.join(tempDir, "file2.txt"), "match there");
+            const result = await grepSearch({ query: "match" }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(2);
+            expect(result.files_searched).toBeGreaterThanOrEqual(2);
+        });
+    });
+    describe("regex search", () => {
+        it("should support regex patterns when is_regex is true", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "test123\ntest456\nother");
+            const result = await grepSearch({ query: "test\\d+", is_regex: true }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(2);
+        });
+        it("should return error for invalid regex", async () => {
+            const result = await grepSearch({ query: "[invalid", is_regex: true }, context);
+            expect(result.success).toBe(false);
+            expect(result.error).toBe("Invalid regex pattern");
+        });
+        it("should escape special characters in non-regex mode", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "test.file\ntest-file");
+            const result = await grepSearch({ query: "test.file", is_regex: false }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(1);
+        });
+    });
+    describe("case sensitivity", () => {
+        it("should be case-insensitive by default", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "Hello\nHELLO\nhello");
+            const result = await grepSearch({ query: "hello" }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(3);
+        });
+        it("should be case-sensitive when case_sensitive is true", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "Hello\nHELLO\nhello");
+            const result = await grepSearch({ query: "hello", case_sensitive: true }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(1);
+        });
+    });
+    describe("context lines", () => {
+        it("should include context lines when specified", async () => {
+            const content = "line1\nline2\nmatch\nline4\nline5";
+            await fs.writeFile(path.join(tempDir, "test.txt"), content);
+            const result = await grepSearch({ query: "match", context_lines: 1 }, context);
+            expect(result.success).toBe(true);
+            expect(result.matches[0].context_before).toEqual(["line2"]);
+            expect(result.matches[0].context_after).toEqual(["line4"]);
+        });
+        it("should handle context at file boundaries", async () => {
+            await fs.writeFile(path.join(tempDir, "test.txt"), "match\nline2");
+            const result = await grepSearch({ query: "match", context_lines: 2 }, context);
+            expect(result.success).toBe(true);
+            expect(result.matches[0].context_before).toBeUndefined();
+            expect(result.matches[0].context_after).toEqual(["line2"]);
+        });
+    });
+    describe("file pattern filtering", () => {
+        it("should filter by file pattern", async () => {
+            await fs.writeFile(path.join(tempDir, "test.ts"), "match");
+            await fs.writeFile(path.join(tempDir, "test.js"), "match");
+            const result = await grepSearch({ query: "match", pattern: "*.ts" }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(1);
+            expect(result.matches[0].file).toBe("test.ts");
+        });
+    });
+    describe("max_results limit", () => {
+        it("should limit results to max_results", async () => {
+            let content = "";
+            for (let i = 0; i < 20; i++) {
+                content += `match line ${i}\n`;
+            }
+            await fs.writeFile(path.join(tempDir, "test.txt"), content);
+            const result = await grepSearch({ query: "match", max_results: 5 }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(5);
+            expect(result.truncated).toBe(true);
+        });
+    });
+    describe("path validation (security)", () => {
+        it("should prevent directory traversal", async () => {
+            const result = await grepSearch({ query: "test", path: "../../../" }, context);
+            expect(result.success).toBe(false);
+            expect(result.error).toBe("Path validation failed: access denied");
+        });
+    });
+    describe("single file search", () => {
+        it("should search a single file when path is a file", async () => {
+            await fs.writeFile(path.join(tempDir, "target.txt"), "match here");
+            await fs.writeFile(path.join(tempDir, "other.txt"), "match there");
+            const result = await grepSearch({ query: "match", path: "target.txt" }, context);
+            expect(result.success).toBe(true);
+            expect(result.total_matches).toBe(1);
+            expect(result.files_searched).toBe(1);
+        });
+    });
+});
+//# sourceMappingURL=grep-search.test.js.map

package/dist/src/lib/local-tools/code/__tests__/grep-search.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"grep-search.test.js","sourceRoot":"","sources":["../../../../../../src/lib/local-tools/code/__tests__/grep-search.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAC,MAAM,QAAQ,CAAC;AACnE,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,EAAC,UAAU,EAAC,MAAM,mBAAmB,CAAC;AAG7C,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,OAAe,CAAC;IACpB,IAAI,OAAoB,CAAC;IAEzB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACxE,OAAO,GAAG,EAAC,gBAAgB,EAAE,OAAO,EAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAC,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAC9B,mCAAmC,CACpC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAC9B,0BAA0B,CAC3B,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,EAAE,EAAC,EAAE,OAAO,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,YAAY,CAAC,CAAC;YAClE,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,aAAa,CAAC,CAAC;YAEnE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAC9B,yBAAyB,CAC1B,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAC,EACnC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAC,EACnC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAC9B,sBAAsB,CACvB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAC,EACrC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAC,EACtC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,OAAO,GAAG,mCAAmC,CAAC;YACpD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAC,EAClC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAC,EAClC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,aAAa,EAAE,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAC,EACjC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC;YACjC,CAAC;YACD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,EAAC,EAChC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAC,EAClC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,YAAY,CAAC,CAAC;YACnE,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,aAAa,CAAC,CAAC;YAEnE,MAAM,MAAM,GAAG,MAAM,UAAU,CAC7B,EAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAC,EACpC,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}