codegrunt 0.1.0

package/dist/core/agent/loop.js

@@ -0,0 +1,353 @@
+import chalk from 'chalk';
+import ora from 'ora';
+import { ContextManager } from '../context/manager.js';
+import { loadProjectGuide } from '../context/project-guide.js';
+import { getToolDefinitions } from '../tools/registry.js';
+import { executeTool, resetYesAll } from '../tools/executor.js';
+import { printToolCall, printToolResult, printAssistantHeader, printThinkingCollapsed } from '../../utils/display.js';
+import { MarkdownRenderer } from '../../utils/markdown.js';
+import { isReasonerModel, CHAT_CONTEXT_BUDGET } from '../../config.js';
+const MAX_ITERATIONS = 30;
+// ── Cumulative usage tracking ──────────────────────────────────────────────
+const sessionUsage = { inputTokens: 0, outputTokens: 0, cacheHitTokens: 0, cacheMissTokens: 0 };
+export function addUsage(u) {
+    sessionUsage.inputTokens += u.inputTokens;
+    sessionUsage.outputTokens += u.outputTokens;
+    sessionUsage.cacheHitTokens += u.cacheHitTokens;
+    sessionUsage.cacheMissTokens += u.cacheMissTokens;
+}
+export function getSessionUsage() {
+    return { ...sessionUsage };
+}
+export function resetSessionUsage() {
+    sessionUsage.inputTokens = 0;
+    sessionUsage.outputTokens = 0;
+    sessionUsage.cacheHitTokens = 0;
+    sessionUsage.cacheMissTokens = 0;
+}
+// Detect the system language from environment variables / locale.
+// Returns 'zh' for Chinese systems, otherwise 'en'.
+function detectSystemLanguage() {
+    const locale = process.env.LC_ALL
+        || process.env.LC_MESSAGES
+        || process.env.LANG
+        || '';
+    if (locale.toLowerCase().startsWith('zh'))
+        return 'zh';
+    // Also check Windows system locale
+    if (process.platform === 'win32') {
+        try {
+            const resolved = Intl.DateTimeFormat().resolvedOptions().locale;
+            if (resolved.toLowerCase().startsWith('zh'))
+                return 'zh';
+        }
+        catch { /* ignore */ }
+    }
+    return 'en';
+}
+// Read-type tools: acquiring information without side effects.
+// Used by the anti-hallucination check to detect whether the model has
+// grounded itself in project context before making changes.
+const READ_TOOL_NAMES = new Set(['read_file', 'search_files', 'list_directory']);
+const WRITE_TOOL_NAMES = new Set(['write_file', 'edit_file']);
+// System prompt is kept stable (no dynamic content) to maximise prompt cache hits.
+// For reasoner (R1) models: system prompt is injected into the first user message
+// because R1 API converts system messages internally anyway.
+// DeepSeek-optimized: explicit tool-calling patterns, structured thinking, and
+// clear formatting rules that work well with DeepSeek's instruction-following style.
+function buildSystemPrompt(guide, language) {
+    const langInstruction = language === 'zh'
+        ? `## Language
+- The user's system language is Chinese (zh). You MUST respond in Chinese (Simplified Chinese, 简体中文) at all times.
+- All explanations, summaries, and conversation with the user should be in Chinese.
+- Code and technical identifiers (variable names, file paths, commands) remain in their original language.`
+        : `## Language
+- Respond in English only.`;
+    const base = `You are CodeGrunt, an expert AI coding assistant running in the terminal. You are powered by DeepSeek, optimized for software engineering tasks.
+
+You have access to tools that let you read files, write files, edit files, run shell commands, list directories, and search code. Use them to complete the user's task.
+
+${langInstruction}
+
+## Core Guidelines
+- Read files before editing them to understand the current content
+- Prefer edit_file over write_file for modifying existing files
+- Run tests after making changes to verify correctness
+- When a task is complete, summarize what you did concisely
+
+## Tool Usage Best Practices
+- Chain tool calls when possible: read search results, then read relevant files, then make edits
+- For search_files: use specific, unique patterns to narrow results
+- For execute_shell: combine commands with && when possible; avoid interactive commands
+- For edit_file: old_string must match exactly including whitespace — copy from read_file output
+- For list_directory: start shallow (depth 1-2) then drill deeper as needed
+- Large tool outputs are truncated — use search or targeted reads when outputs are cut off
+
+## Code Quality
+- Follow existing code conventions in the project
+- Write idiomatic code for the language/framework being used
+- Add minimal, targeted comments for non-obvious logic only
+- Handle errors gracefully in production code
+
+## Anti-Hallucination Rules
+- NEVER invent APIs, functions, types, imports, or dependencies that don't exist in the project. Before using any library or internal API, you MUST read its definition file or find existing usage in the codebase via search_files.
+- When generating new code, you MUST first find and read at least one existing file that demonstrates the pattern, style, and conventions you plan to follow. Copy-adapt is safer than inventing.
+- Every code change must be traceable to something you actually READ during this session — not your training data. If you haven't read a relevant file yet, read it before writing.
+- If you're unsure whether a function/type/import path exists, use search_files or read_file to verify BEFORE writing code that depends on it.
+- For any non-trivial edit, add a brief comment in the code referencing the file(s) that informed your change (e.g., "// Ref: src/utils/billing.ts L23-45" or "// Following pattern from src/cli/commands.ts").`;
+    return guide ? base + guide : base;
+}
+function buildFirstUserPrefix(cwd, model, systemPrompt) {
+    const parts = [];
+    parts.push(`[cwd: ${cwd}]`);
+    // For reasoner models: embed system prompt in the first user message
+    if (isReasonerModel(model) && systemPrompt) {
+        parts.push(`\n[System Instructions]\n${systemPrompt}`);
+    }
+    return parts.join('\n') + '\n\n';
+}
+export async function runAgentLoop(options) {
+    const { task, cwd, config, provider, onText, onToolCall, onToolResult, signal } = options;
+    const model = config.model;
+    const context = options.context ?? new ContextManager(CHAT_CONTEXT_BUDGET);
+    const toolDefs = getToolDefinitions();
+    const guide = options.systemPromptOverride ? null : await loadProjectGuide(cwd);
+    const lang = detectSystemLanguage();
+    // When a skill provides a system prompt override, use it instead of the
+    // default coding-assistant identity. This lets skills define completely
+    // different roles (e.g. "You are a BaZi fortune-telling master").
+    const systemPrompt = options.systemPromptOverride
+        ? options.systemPromptOverride
+        : (guide ? buildSystemPrompt(guide, lang) : buildSystemPrompt(null, lang));
+    const isReasoner = isReasonerModel(model);
+    // Only push system prompt once per session (keeps it stable for cache hits).
+    // Reasoner models skip system role — prompt goes into first user message.
+    if (context.getMessages().length === 0) {
+        if (!isReasoner) {
+            context.push({ role: 'system', content: systemPrompt });
+        }
+    }
+    // Reset per-turn state
+    resetYesAll();
+    let hasReadThisTurn = false;
+    // Warned-this-turn flag: only inject the anti-hallucination reminder once per
+    // turn to avoid spamming context with repeated warnings.
+    let warnedBlindWrite = false;
+    // Prepend cwd (and system prompt for reasoner) to the first message of each turn
+    const isFirstTurn = context.getMessages().length <= (isReasoner ? 0 : 1);
+    const userContent = isFirstTurn
+        ? buildFirstUserPrefix(cwd, model, isReasoner ? systemPrompt : undefined) + task
+        : task;
+    context.push({ role: 'user', content: userContent });
+    for (let iteration = 0; iteration < MAX_ITERATIONS; iteration++) {
+        if (signal?.aborted)
+            break;
+        let assistantText = '';
+        let reasoningText = '';
+        const toolCallAccumulator = new Map();
+        let finishReason = null;
+        let outputTokens = 0;
+        let thinkingStartTime = null;
+        // Animated spinner: "⠋ Thinking… (3s · ↑ 42 tokens · iter 2/30)"
+        const startTime = Date.now();
+        const thinkingSpinner = ora({
+            text: chalk.gray('Thinking…'),
+            color: 'gray',
+            stream: process.stdout,
+        });
+        const updateThinkingText = () => {
+            const elapsed = Math.floor((Date.now() - startTime) / 1000);
+            const iterInfo = iteration > 0 ? ` · iter ${iteration + 1}/${MAX_ITERATIONS}` : '';
+            thinkingSpinner.text = chalk.gray(`Thinking… (${elapsed}s · ↑ ${outputTokens} tokens${iterInfo}  Esc to cancel)`);
+        };
+        const showThinking = () => {
+            if (!thinkingSpinner.isSpinning) {
+                thinkingSpinner.start();
+                updateThinkingText();
+                // Tick every second so the elapsed time updates even between chunks
+                const ticker = setInterval(() => {
+                    if (thinkingSpinner.isSpinning)
+                        updateThinkingText();
+                    else
+                        clearInterval(ticker);
+                }, 1000);
+            }
+            else {
+                updateThinkingText();
+            }
+        };
+        const hideThinking = () => {
+            if (thinkingSpinner.isSpinning) {
+                thinkingSpinner.stop();
+            }
+        };
+        const stream = provider.stream(context.getMessages(), {
+            model: config.model,
+            maxTokens: config.maxTokens,
+            // Reasoner models do not support temperature — omit it
+            temperature: isReasoner ? undefined : config.temperature,
+            reasoningEffort: isReasoner ? config.reasoningEffort : undefined,
+            topP: config.topP,
+            frequencyPenalty: config.frequencyPenalty,
+            presencePenalty: config.presencePenalty,
+            tools: toolDefs,
+            signal,
+        });
+        const md = new MarkdownRenderer();
+        // Start spinner immediately so user sees instant feedback while waiting
+        // for the first streaming chunk (important for reasoner models which may
+        // take 30-60 seconds to emit their first reasoning delta).
+        showThinking();
+        try {
+            for await (const chunk of stream) {
+                if (signal?.aborted)
+                    break;
+                if (chunk.type === 'text_delta') {
+                    hideThinking();
+                    if (!assistantText) {
+                        // Print assistant header before first text chunk
+                        printAssistantHeader();
+                    }
+                    assistantText += chunk.text;
+                    outputTokens += Math.ceil(chunk.text.length / 4);
+                    onText?.(chunk.text);
+                    const formatted = md.feed(chunk.text);
+                    if (formatted)
+                        process.stdout.write(formatted);
+                }
+                else if (chunk.type === 'reasoning_delta') {
+                    if (thinkingStartTime === null)
+                        thinkingStartTime = Date.now();
+                    reasoningText += chunk.text;
+                    outputTokens += Math.ceil(chunk.text.length / 4);
+                    showThinking();
+                }
+                else if (chunk.type === 'tool_call_delta') {
+                    const existing = toolCallAccumulator.get(chunk.index) ?? { id: '', name: '', arguments: '' };
+                    if (chunk.id)
+                        existing.id = chunk.id;
+                    if (chunk.name)
+                        existing.name = chunk.name;
+                    existing.arguments += chunk.arguments_delta;
+                    toolCallAccumulator.set(chunk.index, existing);
+                }
+                else if (chunk.type === 'finish') {
+                    hideThinking();
+                    finishReason = chunk.finish_reason;
+                }
+            }
+            if (signal?.aborted)
+                break;
+        }
+        finally {
+            // Always stop the spinner, even on abort/error.
+            // Without this, the ora timer keeps writing to stdout and corrupts
+            // subsequent terminal output (e.g. the next readMultilineInput panel).
+            hideThinking();
+        }
+        // Flush any remaining markdown buffer
+        const flushOut = md.flush();
+        if (flushOut)
+            process.stdout.write(flushOut);
+        // Show collapsed reasoning block after stream completes
+        if (reasoningText && thinkingStartTime !== null) {
+            const elapsed = Date.now() - thinkingStartTime;
+            printThinkingCollapsed(reasoningText, elapsed);
+        }
+        // Warn if response was truncated due to token limit (especially common with
+        // reasoner models whose chain-of-thought consumes most of maxTokens budget).
+        if (finishReason === 'length') {
+            process.stdout.write('\n');
+            if (lang === 'zh') {
+                process.stdout.write(chalk.yellow('⚠ 响应被截断（已达 token 上限）。请用 /config maxtokens <数值> 调高上限后重试（建议 32768 或更高）。\n'));
+            }
+            else {
+                process.stdout.write(chalk.yellow('⚠ Response truncated (token limit reached). Increase with /config maxtokens <value> (try 32768 or higher).\n'));
+            }
+        }
+        if (finishReason === 'stop' || (finishReason === 'length' && toolCallAccumulator.size === 0)) {
+            if (assistantText) {
+                context.push({
+                    role: 'assistant',
+                    content: assistantText,
+                    ...(reasoningText ? { reasoning_content: reasoningText } : {}),
+                });
+                process.stdout.write('\n');
+            }
+            break;
+        }
+        if (finishReason === 'tool_calls') {
+            if (assistantText)
+                process.stdout.write('\n');
+            const toolCalls = Array.from(toolCallAccumulator.entries())
+                .sort(([a], [b]) => a - b)
+                .map(([, tc]) => ({
+                id: tc.id,
+                type: 'function',
+                function: { name: tc.name, arguments: tc.arguments },
+            }));
+            context.push({
+                role: 'assistant',
+                content: null,
+                tool_calls: toolCalls,
+                ...(reasoningText ? { reasoning_content: reasoningText } : {}),
+            });
+            // ── Anti-hallucination: detect "blind write" pattern ─────────────────
+            // If the model issues write/edit without any read/search/list in the
+            // same batch and hasn't read anything this turn, inject a one-time
+            // reminder. This catches the most common hallucination vector: jumping
+            // straight to generating code without grounding in the codebase.
+            const hasWriteInBatch = toolCalls.some(tc => WRITE_TOOL_NAMES.has(tc.function.name));
+            const hasReadInBatch = toolCalls.some(tc => READ_TOOL_NAMES.has(tc.function.name));
+            if (hasWriteInBatch && !hasReadInBatch && !hasReadThisTurn && !warnedBlindWrite) {
+                warnedBlindWrite = true;
+                const warning = lang === 'zh'
+                    ? '⚠️ 检测到直接写入操作：你尚未读取任何项目文件就尝试编辑代码。这极大增加了凭空编造不存在的 API/类型/模式的风险。建议在写入之前先用 read_file 或 search_files 了解现有代码风格和可用的接口。'
+                    : '⚠️ Blind write detected: you are attempting to edit code without having read any project files first. This greatly increases the risk of inventing non-existent APIs, types, or patterns. Consider using read_file or search_files to ground yourself before writing.';
+                context.push({ role: 'user', content: warning });
+            }
+            for (const tc of toolCalls) {
+                let parsedArgs = {};
+                try {
+                    parsedArgs = JSON.parse(tc.function.arguments);
+                }
+                catch {
+                    // leave empty
+                }
+                // Track reads so subsequent writes in the same turn are considered grounded
+                if (READ_TOOL_NAMES.has(tc.function.name)) {
+                    hasReadThisTurn = true;
+                }
+                onToolCall?.(tc.function.name, parsedArgs);
+                printToolCall(tc.function.name, parsedArgs);
+                const toolSpinner = ora({
+                    text: chalk.gray(`Running ${tc.function.name}…`),
+                    color: 'cyan',
+                    stream: process.stdout,
+                }).start();
+                // Stop spinner before executing so confirmation UIs (edit_file, write_file)
+                // can take over stdout/raw-mode without the spinner corrupting their output.
+                toolSpinner.stop();
+                const result = await executeTool(tc.function.name, tc.function.arguments, cwd);
+                onToolResult?.(tc.function.name, result);
+                printToolResult(tc.function.name, result);
+                context.push({
+                    role: 'tool',
+                    tool_call_id: tc.id,
+                    content: result.success ? result.output : (result.error ?? result.output),
+                });
+                if (result.userRejected) {
+                    return;
+                }
+            }
+            // continue loop — model will process tool results
+        }
+        else {
+            // length or unknown finish reason — stop
+            if (assistantText)
+                process.stdout.write('\n');
+            break;
+        }
+    }
+}
+//# sourceMappingURL=loop.js.map

package/dist/core/agent/loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loop.js","sourceRoot":"","sources":["../../../src/core/agent/loop.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AACtH,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEvE,MAAM,cAAc,GAAG,EAAE,CAAC;AAE1B,8EAA8E;AAC9E,MAAM,YAAY,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC;AAChG,MAAM,UAAU,QAAQ,CAAC,CAAiG;IACxH,YAAY,CAAC,WAAW,IAAI,CAAC,CAAC,WAAW,CAAC;IAC1C,YAAY,CAAC,YAAY,IAAI,CAAC,CAAC,YAAY,CAAC;IAC5C,YAAY,CAAC,cAAc,IAAI,CAAC,CAAC,cAAc,CAAC;IAChD,YAAY,CAAC,eAAe,IAAI,CAAC,CAAC,eAAe,CAAC;AACpD,CAAC;AACD,MAAM,UAAU,eAAe;IAC7B,OAAO,EAAE,GAAG,YAAY,EAAE,CAAC;AAC7B,CAAC;AACD,MAAM,UAAU,iBAAiB;IAC/B,YAAY,CAAC,WAAW,GAAG,CAAC,CAAC;IAC7B,YAAY,CAAC,YAAY,GAAG,CAAC,CAAC;IAC9B,YAAY,CAAC,cAAc,GAAG,CAAC,CAAC;IAChC,YAAY,CAAC,eAAe,GAAG,CAAC,CAAC;AACnC,CAAC;AAED,kEAAkE;AAClE,oDAAoD;AACpD,SAAS,oBAAoB;IAC3B,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM;WAC5B,OAAO,CAAC,GAAG,CAAC,WAAW;WACvB,OAAO,CAAC,GAAG,CAAC,IAAI;WAChB,EAAE,CAAC;IACR,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACvD,mCAAmC;IACnC,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC,eAAe,EAAE,CAAC,MAAM,CAAC;YAChE,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;QAC3D,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,+DAA+D;AAC/D,uEAAuE;AACvE,4DAA4D;AAC5D,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,WAAW,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC,CAAC;AACjF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;AAE9D,mFAAmF;AACnF,kFAAkF;AAClF,6DAA6D;AAC7D,+EAA+E;AAC/E,qFAAqF;AACrF,SAAS,iBAAiB,CAAC,KAAoB,EAAE,QAAqB;IACpE,MAAM,eAAe,GAAG,QAAQ,KAAK,IAAI;QACvC,CAAC,CAAC;;;2GAGqG;QACvG,CAAC,CAAC;2BACqB,CAAC;IAE1B,MAAM,IAAI,GAAG;;;;EAIb,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;gNA2B+L,CAAC;IAE/M,OAAO,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;AACrC,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAW,EAAE,KAAa,EAAE,YAAqB;IAC7E,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;IAE5B,qEAAqE;IACrE,IAAI,eAAe,CAAC,KAAK,CAAC,IAAI,YAAY,EAAE,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;AACnC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAAwB;IACzD,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAC1F,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAE3B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,cAAc,CAAC,mBAAmB,CAAC,CAAC;IAC3E,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;IAEtC,MAAM,KAAK,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,gBAAgB,CAAC,GAAG,CAAC,CAAC;IAChF,MAAM,IAAI,GAAG,oBAAoB,EAAE,CAAC;IAEpC,wEAAwE;IACxE,wEAAwE;IACxE,kEAAkE;IAClE,MAAM,YAAY,GAAG,OAAO,CAAC,oBAAoB;QAC/C,CAAC,CAAC,OAAO,CAAC,oBAAoB;QAC9B,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAE7E,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAE1C,6EAA6E;IAC7E,0EAA0E;IAC1E,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,uBAAuB;IACvB,WAAW,EAAE,CAAC;IACd,IAAI,eAAe,GAAG,KAAK,CAAC;IAE5B,8EAA8E;IAC9E,yDAAyD;IACzD,IAAI,gBAAgB,GAAG,KAAK,CAAC;IAE7B,iFAAiF;IACjF,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG,WAAW;QAC7B,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI;QAChF,CAAC,CAAC,IAAI,CAAC;IAET,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;IAErD,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,cAAc,EAAE,SAAS,EAAE,EAAE,CAAC;QAChE,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM;QAE3B,IAAI,aAAa,GAAG,EAAE,CAAC;QACvB,IAAI,aAAa,GAAG,EAAE,CAAC;QACvB,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA2D,CAAC;QAC/F,IAAI,YAAY,GAAkB,IAAI,CAAC;QACvC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,iBAAiB,GAAkB,IAAI,CAAC;QAE5C,iEAAiE;QACjE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,GAAG,CAAC;YAC1B,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC;YAC7B,KAAK,EAAE,MAAM;YACb,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,GAAS,EAAE;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,SAAS,GAAG,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnF,eAAe,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,cAAc,OAAO,SAAS,YAAY,UAAU,QAAQ,kBAAkB,CAAC,CAAC;QACpH,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,GAAS,EAAE;YAC9B,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;gBAChC,eAAe,CAAC,KAAK,EAAE,CAAC;gBACxB,kBAAkB,EAAE,CAAC;gBACrB,oEAAoE;gBACpE,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,EAAE;oBAC9B,IAAI,eAAe,CAAC,UAAU;wBAAE,kBAAkB,EAAE,CAAC;;wBAChD,aAAa,CAAC,MAAM,CAAC,CAAC;gBAC7B,CAAC,EAAE,IAAI,CAAC,CAAC;YACX,CAAC;iBAAM,CAAC;gBACN,kBAAkB,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,GAAS,EAAE;YAC9B,IAAI,eAAe,CAAC,UAAU,EAAE,CAAC;gBAC/B,eAAe,CAAC,IAAI,EAAE,CAAC;YACzB,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE;YACpD,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,uDAAuD;YACvD,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW;YACxD,eAAe,EAAE,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS;YAChE,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,KAAK,EAAE,QAAQ;YACf,MAAM;SACP,CAAC,CAAC;QAEH,MAAM,EAAE,GAAG,IAAI,gBAAgB,EAAE,CAAC;QAElC,wEAAwE;QACxE,yEAAyE;QACzE,2DAA2D;QAC3D,YAAY,EAAE,CAAC;QAEf,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IAAI,MAAM,EAAE,OAAO;oBAAE,MAAM;gBAE3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAChC,YAAY,EAAE,CAAC;oBACf,IAAI,CAAC,aAAa,EAAE,CAAC;wBACnB,iDAAiD;wBACjD,oBAAoB,EAAE,CAAC;oBACzB,CAAC;oBACD,aAAa,IAAI,KAAK,CAAC,IAAI,CAAC;oBAC5B,YAAY,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACjD,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACrB,MAAM,SAAS,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACtC,IAAI,SAAS;wBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACjD,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC5C,IAAI,iBAAiB,KAAK,IAAI;wBAAE,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC/D,aAAa,IAAI,KAAK,CAAC,IAAI,CAAC;oBAC5B,YAAY,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACjD,YAAY,EAAE,CAAC;gBACjB,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC5C,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;oBAC7F,IAAI,KAAK,CAAC,EAAE;wBAAE,QAAQ,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;oBACrC,IAAI,KAAK,CAAC,IAAI;wBAAE,QAAQ,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBAC3C,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,eAAe,CAAC;oBAC5C,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;gBACjD,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACnC,YAAY,EAAE,CAAC;oBACf,YAAY,GAAG,KAAK,CAAC,aAAa,CAAC;gBACrC,CAAC;YACH,CAAC;YAED,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM;QAC7B,CAAC;gBAAS,CAAC;YACT,gDAAgD;YAChD,mEAAmE;YACnE,uEAAuE;YACvE,YAAY,EAAE,CAAC;QACjB,CAAC;QAED,sCAAsC;QACtC,MAAM,QAAQ,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,QAAQ;YAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAE7C,wDAAwD;QACxD,IAAI,aAAa,IAAI,iBAAiB,KAAK,IAAI,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,iBAAiB,CAAC;YAC/C,sBAAsB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QAED,4EAA4E;QAC5E,6EAA6E;QAC7E,IAAI,YAAY,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBAClB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,yEAAyE,CAAC,CAAC,CAAC;YAChH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,8GAA8G,CAAC,CAAC,CAAC;YACrJ,CAAC;QACH,CAAC;QAED,IAAI,YAAY,KAAK,MAAM,IAAI,CAAC,YAAY,KAAK,QAAQ,IAAI,mBAAmB,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7F,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,aAAa;oBACtB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC/D,CAAC,CAAC;gBACH,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM;QACR,CAAC;QAED,IAAI,YAAY,KAAK,YAAY,EAAE,CAAC;YAClC,IAAI,aAAa;gBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAE9C,MAAM,SAAS,GAAe,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC;iBACpE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBACzB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE;aACrD,CAAC,CAAC,CAAC;YAEN,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,SAAS;gBACrB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5C,CAAC,CAAC;YAEtB,wEAAwE;YACxE,qEAAqE;YACrE,mEAAmE;YACnE,uEAAuE;YACvE,iEAAiE;YACjE,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACrF,MAAM,cAAc,GAAG,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACnF,IAAI,eAAe,IAAI,CAAC,cAAc,IAAI,CAAC,eAAe,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAChF,gBAAgB,GAAG,IAAI,CAAC;gBACxB,MAAM,OAAO,GAAG,IAAI,KAAK,IAAI;oBAC3B,CAAC,CAAC,gHAAgH;oBAClH,CAAC,CAAC,uQAAuQ,CAAC;gBAC5Q,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC3B,IAAI,UAAU,GAA4B,EAAE,CAAC;gBAC7C,IAAI,CAAC;oBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAA4B,CAAC;gBAC5E,CAAC;gBAAC,MAAM,CAAC;oBACP,cAAc;gBAChB,CAAC;gBAED,4EAA4E;gBAC5E,IAAI,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC1C,eAAe,GAAG,IAAI,CAAC;gBACzB,CAAC;gBAED,UAAU,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBAC3C,aAAa,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBAE5C,MAAM,WAAW,GAAG,GAAG,CAAC;oBACtB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC;oBAChD,KAAK,EAAE,MAAM;oBACb,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC,CAAC,KAAK,EAAE,CAAC;gBAEX,4EAA4E;gBAC5E,6EAA6E;gBAC7E,WAAW,CAAC,IAAI,EAAE,CAAC;gBAEnB,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBAE/E,YAAY,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBACzC,eAAe,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAE1C,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,EAAE,CAAC,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC;iBAC1E,CAAC,CAAC;gBAEH,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBACxB,OAAO;gBACT,CAAC;YACH,CAAC;YAED,kDAAkD;QACpD,CAAC;aAAM,CAAC;YACN,yCAAyC;YACzC,IAAI,aAAa;gBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM;QACR,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/core/context/manager.d.ts

@@ -0,0 +1,26 @@
+import type { Message } from '../../types.js';
+export declare class ContextManager {
+    private messages;
+    private tokenBudget;
+    /**
+     * @param tokenBudget Maximum estimated tokens for stored messages.
+     *                    For DeepSeek-V4 (128K context), default ~90K leaves room for output.
+     *                    For R1 reasoner (1M context), can go much higher.
+     */
+    constructor(tokenBudget?: number);
+    push(message: Message): void;
+    getMessages(): Message[];
+    clear(): void;
+    /** Replace messages wholesale (used for cache-warm restart) */
+    setMessages(msgs: Message[]): void;
+    estimatedTokenCount(): number;
+    /** Adjust the token budget dynamically (e.g. when switching models) */
+    setTokenBudget(budget: number): void;
+    private estimateTokens;
+    private trim;
+    /**
+     * Remove the oldest message group from the conversation, preserving
+     * assistant(tool_calls) ↔ tool message pairing required by the LLM API.
+     */
+    private removeOldestGroup;
+}

package/dist/core/context/manager.js

@@ -0,0 +1,98 @@
+const CHARS_PER_TOKEN = 4;
+export class ContextManager {
+    messages = [];
+    tokenBudget;
+    /**
+     * @param tokenBudget Maximum estimated tokens for stored messages.
+     *                    For DeepSeek-V4 (128K context), default ~90K leaves room for output.
+     *                    For R1 reasoner (1M context), can go much higher.
+     */
+    constructor(tokenBudget = 90_000) {
+        this.tokenBudget = tokenBudget;
+    }
+    push(message) {
+        this.messages.push(message);
+        this.trim();
+    }
+    getMessages() {
+        return this.messages;
+    }
+    clear() {
+        this.messages = [];
+    }
+    /** Replace messages wholesale (used for cache-warm restart) */
+    setMessages(msgs) {
+        this.messages = msgs;
+        this.trim();
+    }
+    estimatedTokenCount() {
+        return this.estimateTokens();
+    }
+    /** Adjust the token budget dynamically (e.g. when switching models) */
+    setTokenBudget(budget) {
+        this.tokenBudget = budget;
+        this.trim();
+    }
+    estimateTokens() {
+        let total = 0;
+        for (const msg of this.messages) {
+            if ('content' in msg && msg.content) {
+                total += Math.ceil(String(msg.content).length / CHARS_PER_TOKEN);
+            }
+            // Account for reasoning_content in token count
+            if ('reasoning_content' in msg && msg.reasoning_content) {
+                total += Math.ceil(String(msg.reasoning_content).length / CHARS_PER_TOKEN);
+            }
+            // Tool calls consume tokens
+            if ('tool_calls' in msg && msg.tool_calls) {
+                for (const tc of msg.tool_calls) {
+                    total += Math.ceil(tc.function.name.length / CHARS_PER_TOKEN);
+                    total += Math.ceil(tc.function.arguments.length / CHARS_PER_TOKEN);
+                }
+            }
+        }
+        return total;
+    }
+    trim() {
+        const hasSystem = this.messages[0]?.role === 'system';
+        const startIdx = hasSystem ? 1 : 0;
+        // Keep system + at least the last 4 messages (user/assistant/tool pairs)
+        const minMessages = hasSystem ? 5 : 4;
+        while (this.estimateTokens() > this.tokenBudget && this.messages.length > minMessages) {
+            this.removeOldestGroup(startIdx);
+        }
+    }
+    /**
+     * Remove the oldest message group from the conversation, preserving
+     * assistant(tool_calls) ↔ tool message pairing required by the LLM API.
+     */
+    removeOldestGroup(startIdx) {
+        const msg = this.messages[startIdx];
+        if (msg && msg.role === 'assistant' && 'tool_calls' in msg && msg.tool_calls) {
+            // This is an assistant message with tool calls — must remove it
+            // together with all subsequent tool response messages.
+            let removeCount = 1;
+            while (startIdx + removeCount < this.messages.length &&
+                this.messages[startIdx + removeCount].role === 'tool') {
+                removeCount++;
+            }
+            this.messages.splice(startIdx, removeCount);
+        }
+        else if (msg && msg.role === 'tool') {
+            // Standalone tool message without preceding assistant(tool_calls) —
+            // this can happen if the assistant was already removed. Skip it
+            // to avoid breaking the stream, and also skip any contiguous tool messages.
+            let removeCount = 1;
+            while (startIdx + removeCount < this.messages.length &&
+                this.messages[startIdx + removeCount].role === 'tool') {
+                removeCount++;
+            }
+            this.messages.splice(startIdx, removeCount);
+        }
+        else {
+            // Plain message (user, assistant without tool_calls, system) — safe to remove
+            this.messages.splice(startIdx, 1);
+        }
+    }
+}
+//# sourceMappingURL=manager.js.map

package/dist/core/context/manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../src/core/context/manager.ts"],"names":[],"mappings":"AAEA,MAAM,eAAe,GAAG,CAAC,CAAC;AAE1B,MAAM,OAAO,cAAc;IACjB,QAAQ,GAAc,EAAE,CAAC;IACzB,WAAW,CAAS;IAE5B;;;;OAIG;IACH,YAAY,WAAW,GAAG,MAAM;QAC9B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,IAAI,CAAC,OAAgB;QACnB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;IACrB,CAAC;IAED,+DAA+D;IAC/D,WAAW,CAAC,IAAe;QACzB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,mBAAmB;QACjB,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;IAC/B,CAAC;IAED,uEAAuE;IACvE,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAC1B,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,cAAc;QACpB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChC,IAAI,SAAS,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBACpC,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC;YACnE,CAAC;YACD,+CAA+C;YAC/C,IAAI,mBAAmB,IAAI,GAAG,IAAI,GAAG,CAAC,iBAAiB,EAAE,CAAC;gBACxD,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC;YAC7E,CAAC;YACD,4BAA4B;YAC5B,IAAI,YAAY,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBAC1C,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;oBAChC,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC;oBAC9D,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,IAAI;QACV,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,QAAQ,CAAC;QACtD,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,yEAAyE;QACzE,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;YACtF,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,QAAgB;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEpC,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,YAAY,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YAC7E,gEAAgE;YAChE,uDAAuD;YACvD,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,OACE,QAAQ,GAAG,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAC7C,IAAI,CAAC,QAAQ,CAAC,QAAQ,GAAG,WAAW,CAAC,CAAC,IAAI,KAAK,MAAM,EACrD,CAAC;gBACD,WAAW,EAAE,CAAC;YAChB,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtC,oEAAoE;YACpE,gEAAgE;YAChE,4EAA4E;YAC5E,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,OACE,QAAQ,GAAG,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAC7C,IAAI,CAAC,QAAQ,CAAC,QAAQ,GAAG,WAAW,CAAC,CAAC,IAAI,KAAK,MAAM,EACrD,CAAC;gBACD,WAAW,EAAE,CAAC;YAChB,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,8EAA8E;YAC9E,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;CACF"}

package/dist/core/context/project-guide.d.ts

@@ -0,0 +1 @@
+export declare function loadProjectGuide(cwd: string): Promise<string | null>;

package/dist/core/context/project-guide.js

@@ -0,0 +1,17 @@
+import { readFile } from 'fs/promises';
+import { join } from 'path';
+// Priority order: CODEGRUNT.md > CLAUDE.md
+const GUIDE_FILES = ['CODEGRUNT.md', 'CLAUDE.md'];
+export async function loadProjectGuide(cwd) {
+    for (const filename of GUIDE_FILES) {
+        try {
+            const content = await readFile(join(cwd, filename), 'utf-8');
+            return `\n\n---\n# Project Guide (${filename})\n\n${content.trim()}`;
+        }
+        catch {
+            // not found, try next
+        }
+    }
+    return null;
+}
+//# sourceMappingURL=project-guide.js.map

package/dist/core/context/project-guide.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project-guide.js","sourceRoot":"","sources":["../../../src/core/context/project-guide.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5B,2CAA2C;AAC3C,MAAM,WAAW,GAAG,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;AAElD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,GAAW;IAChD,KAAK,MAAM,QAAQ,IAAI,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;YAC7D,OAAO,6BAA6B,QAAQ,QAAQ,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACvE,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/core/tools/edit_file.d.ts

@@ -0,0 +1,2 @@
+import type { Tool } from '../../types.js';
+export declare const editFileTool: Tool;

package/dist/core/tools/edit_file.js

@@ -0,0 +1,54 @@
+import { readFile, writeFile } from 'fs/promises';
+import { resolve } from 'path';
+import { printDiff } from '../../utils/display.js';
+export const editFileTool = {
+    definition: {
+        type: 'function',
+        function: {
+            name: 'edit_file',
+            description: 'Replace an exact string in a file with new content. The old_string must match exactly (including whitespace). Fails clearly if old_string is not found.',
+            parameters: {
+                type: 'object',
+                properties: {
+                    path: {
+                        type: 'string',
+                        description: 'The path to the file to edit',
+                    },
+                    old_string: {
+                        type: 'string',
+                        description: 'The exact string to find and replace',
+                    },
+                    new_string: {
+                        type: 'string',
+                        description: 'The string to replace it with',
+                    },
+                },
+                required: ['path', 'old_string', 'new_string'],
+            },
+        },
+    },
+    async execute(args) {
+        const filePath = resolve(args.path);
+        const oldString = args.old_string;
+        const newString = args.new_string;
+        try {
+            const content = await readFile(filePath, 'utf-8');
+            if (!content.includes(oldString)) {
+                return {
+                    success: false,
+                    output: '',
+                    error: `old_string not found in ${filePath}. The string must match exactly including whitespace and indentation.`,
+                };
+            }
+            const updated = content.replace(oldString, newString);
+            await writeFile(filePath, updated, 'utf-8');
+            printDiff(filePath, content, updated);
+            return { success: true, output: `Edited ${filePath}` };
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            return { success: false, output: '', error: `Failed to edit ${filePath}: ${message}` };
+        }
+    },
+};
+//# sourceMappingURL=edit_file.js.map

package/dist/core/tools/edit_file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit_file.js","sourceRoot":"","sources":["../../../src/core/tools/edit_file.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAE/B,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,MAAM,CAAC,MAAM,YAAY,GAAS;IAChC,UAAU,EAAE;QACV,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,yJAAyJ;YACtK,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,8BAA8B;qBAC5C;oBACD,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,sCAAsC;qBACpD;oBACD,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,+BAA+B;qBAC7C;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,YAAY,EAAE,YAAY,CAAC;aAC/C;SACF;KACF;IAED,KAAK,CAAC,OAAO,CAAC,IAAI;QAChB,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAoB,CAAC;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAoB,CAAC;QAE5C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAElD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,EAAE;oBACV,KAAK,EAAE,2BAA2B,QAAQ,uEAAuE;iBAClH,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YACtD,MAAM,SAAS,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC5C,SAAS,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YACtC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,QAAQ,EAAE,EAAE,CAAC;QACzD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,kBAAkB,QAAQ,KAAK,OAAO,EAAE,EAAE,CAAC;QACzF,CAAC;IACH,CAAC;CACF,CAAC"}

package/dist/core/tools/execute_shell.d.ts

@@ -0,0 +1,2 @@
+import type { Tool } from '../../types.js';
+export declare const executeShellTool: Tool;