@stan-chen/simple-cli 0.2.3 → 0.2.4

package/dist/lib/agent.js

@@ -1,313 +0,0 @@
-/**
- * Agent Core - Main agent loop with reflection and retry
- * Implements Aider-style reasoning and error recovery
- */
-import { applyFileEdits, parseEditBlocks } from './editor.js';
-import { generateCommitMessage } from './git.js';
-import * as ui from './ui.js';
-/**
- * Parse LLM response into structured format
- */
-export function parseResponse(response) {
-    // Accept either a TypeLLMResponse object or a raw string (tests and some providers pass raw text)
-    const raw = typeof response === 'string' ? response : (response.raw || '');
-    // Extract edit blocks - Aider style blocks are within the raw text
-    const editBlocks = parseEditBlocks(raw);
-    // If we were given a structured TypeLLMResponse, prefer its typed fields
-    if (typeof response !== 'string') {
-        const thought = response.thought;
-        let tool = response.tool || 'none';
-        const args = response.args || {};
-        const message = response.message || '';
-        // normalize tool name to snake_case
-        if (tool && tool !== 'none') {
-            tool = tool.replace(/([a-z0-9])([A-Z])/g, '$1_$2').toLowerCase();
-        }
-        const action = tool !== 'none'
-            ? { tool, args }
-            : { tool: 'none', message: message || (tool === 'none' ? 'No action parsed' : '') };
-        return { thought, action, editBlocks };
-    }
-    // For raw string input, attempt to extract a <thought> block and a JSON action
-    let thought;
-    const thoughtMatch = raw.match(/<thought>[\s\S]*?<\/thought>/i);
-    if (thoughtMatch) {
-        thought = thoughtMatch[0].replace(/<\/?thought>/gi, '').trim();
-    }
-    // Extract first JSON-looking object from the raw text
-    let action = { tool: 'none', message: 'No action parsed' };
-    const jsonMatch = raw.match(/\{[\s\S]*\}/);
-    if (jsonMatch) {
-        try {
-            const parsed = JSON.parse(jsonMatch[0]);
-            let tool = parsed.tool || parsed.command || 'none';
-            const args = parsed.args || parsed.parameters || {};
-            if (tool && tool !== 'none') {
-                // normalize camelCase to snake_case
-                tool = String(tool).replace(/([a-z0-9])([A-Z])/g, '$1_$2').toLowerCase();
-                action = { tool, args };
-            }
-            else {
-                action = { tool: 'none', message: parsed.message || 'No action parsed' };
-            }
-        }
-        catch (e) {
-            // If JSON.parse fails, leave action as 'none'
-            action = { tool: 'none', message: 'No action parsed' };
-        }
-    }
-    return { thought, action, editBlocks };
-}
-/**
- * Build reflection prompt for retry
- */
-export function buildReflectionPrompt(context) {
-    let prompt = `## Reflection (Attempt ${context.attempt})
-
-Your previous response had errors that need to be fixed.
-
-### Previous Error
-${context.previousError}
-
-`;
-    if (context.failedEdits.length > 0) {
-        prompt += `### Failed Edits\n`;
-        for (const edit of context.failedEdits) {
-            prompt += `
-**File:** ${edit.file}
-**Error:** ${edit.error}
-${edit.suggestion ? `**Suggestion:** ${edit.suggestion}` : ''}
-`;
-        }
-    }
-    prompt += `
-### Instructions
-- Review the error carefully
-- The SEARCH block must EXACTLY match existing content
-- Include enough context to make the match unique
-- Try again with corrected SEARCH/REPLACE blocks
-`;
-    return prompt;
-}
-/**
- * Build lint error prompt
- */
-export function buildLintErrorPrompt(file, errors) {
-    return `## Lint Errors Detected
-
-The file \`${file}\` has syntax errors after your changes:
-
-\`\`\`
-${errors}
-\`\`\`
-
-Please fix these errors by providing corrected SEARCH/REPLACE blocks.
-Focus on the specific lines mentioned in the errors.
-`;
-}
-/**
- * Build test failure prompt
- */
-export function buildTestFailurePrompt(output) {
-    return `## Test Failure
-
-The tests failed after your changes:
-
-\`\`\`
-${output.slice(0, 2000)}${output.length > 2000 ? '\n... (truncated)' : ''}
-\`\`\`
-
-Please analyze the failure and fix the code.
-Provide SEARCH/REPLACE blocks for the necessary changes.
-`;
-}
-/**
- * Agent class - Main orchestration logic
- */
-export class Agent {
-    config;
-    git;
-    generateFn;
-    executeTool;
-    lintFn;
-    testFn;
-    constructor(options) {
-        this.config = options.config;
-        this.git = options.git;
-        this.generateFn = options.generateFn;
-        this.executeTool = options.executeTool;
-        this.lintFn = options.lintFn;
-        this.testFn = options.testFn;
-    }
-    /**
-     * Process a user message with reflection loop
-     */
-    async process(userMessage, history, systemPrompt) {
-        let messages = [
-            { role: 'system', content: systemPrompt },
-            ...history,
-            { role: 'user', content: userMessage },
-        ];
-        let attempt = 0;
-        let editResults = [];
-        let lintResults = [];
-        let testResult;
-        let commitResult;
-        while (attempt < this.config.maxReflections) {
-            attempt++;
-            // Generate response
-            const llmResponse = await ui.spin(attempt === 1 ? 'Thinking...' : `Reflecting (attempt ${attempt})...`, () => this.generateFn(messages));
-            // Parse response
-            const response = parseResponse(llmResponse);
-            // Show thought if present
-            if (response.thought) {
-                ui.showThought(response.thought);
-            }
-            // Apply edit blocks if present
-            if (response.editBlocks && response.editBlocks.length > 0) {
-                ui.step(`Applying ${response.editBlocks.length} edit(s)...`);
-                editResults = await applyFileEdits(response.editBlocks);
-                const failed = editResults.filter(r => !r.success);
-                const succeeded = editResults.filter(r => r.success);
-                // Show results
-                for (const result of succeeded) {
-                    ui.success(`✓ ${result.file}`);
-                    if (result.diff) {
-                        ui.showDiff(result.diff);
-                    }
-                }
-                for (const result of failed) {
-                    ui.error(`✗ ${result.file}: ${result.error}`);
-                    if (result.suggestion) {
-                        ui.note(result.suggestion, 'Suggestion');
-                    }
-                }
-                // If any edits failed, reflect and retry
-                if (failed.length > 0 && attempt < this.config.maxReflections) {
-                    const reflectionPrompt = buildReflectionPrompt({
-                        attempt,
-                        previousError: failed.map(f => f.error).join('\n'),
-                        previousResponse: llmResponse.raw || JSON.stringify(llmResponse),
-                        failedEdits: failed,
-                    });
-                    messages = [
-                        ...messages,
-                        { role: 'assistant', content: llmResponse.raw || JSON.stringify(llmResponse) },
-                        { role: 'user', content: reflectionPrompt },
-                    ];
-                    continue;
-                }
-                // Run linting if enabled and edits succeeded
-                if (this.config.autoLint && this.lintFn && succeeded.length > 0) {
-                    const filesToLint = [...new Set(succeeded.map(r => r.file))];
-                    for (const file of filesToLint) {
-                        const lintResult = await ui.spin(`Linting ${file}...`, () => this.lintFn(file));
-                        lintResults.push({ file, ...lintResult });
-                        if (!lintResult.passed && attempt < this.config.maxReflections) {
-                            ui.error(`Lint errors in ${file}`);
-                            const lintPrompt = buildLintErrorPrompt(file, lintResult.output);
-                            messages = [
-                                ...messages,
-                                { role: 'assistant', content: llmResponse.raw || JSON.stringify(llmResponse) },
-                                { role: 'user', content: lintPrompt },
-                            ];
-                            continue;
-                        }
-                    }
-                }
-                // Run tests if enabled
-                if (this.config.autoTest && this.testFn && succeeded.length > 0) {
-                    testResult = await ui.spin('Running tests...', () => this.testFn());
-                    if (!testResult.passed && attempt < this.config.maxReflections) {
-                        ui.error('Tests failed');
-                        const testPrompt = buildTestFailurePrompt(testResult.output);
-                        messages = [
-                            ...messages,
-                            { role: 'assistant', content: llmResponse.raw || JSON.stringify(llmResponse) },
-                            { role: 'user', content: testPrompt },
-                        ];
-                        continue;
-                    }
-                }
-                // Auto-commit if enabled and all checks passed
-                if (this.config.autoCommit && succeeded.length > 0) {
-                    const allLintsPassed = lintResults.every(r => r.passed);
-                    const testsPassed = !testResult || testResult.passed;
-                    if (allLintsPassed && testsPassed) {
-                        const diff = await this.git.diff();
-                        if (diff) {
-                            const commitMessage = await ui.spin('Generating commit message...', () => generateCommitMessage(diff, async (prompt) => {
-                                const res = await this.generateFn([
-                                    { role: 'user', content: prompt },
-                                ]);
-                                return res.message || res.thought || res.raw || '';
-                            }));
-                            commitResult = await this.git.commit({
-                                message: commitMessage,
-                                files: succeeded.map(r => r.file),
-                            }) || undefined;
-                            if (commitResult) {
-                                ui.success(`Committed: ${commitResult.hash} ${commitResult.message}`);
-                            }
-                        }
-                    }
-                }
-            }
-            // Execute tool action if not edit blocks
-            if (response.action && 'tool' in response.action && response.action.tool !== 'none' && !response.editBlocks?.length) {
-                const { tool, args } = response.action;
-                ui.showToolCall(tool, args || {});
-                try {
-                    const result = await this.executeTool(tool, args || {});
-                    const resultStr = typeof result === 'string' ? result : JSON.stringify(result, null, 2);
-                    ui.showToolResult(resultStr);
-                }
-                catch (error) {
-                    ui.error(`Tool error: ${error instanceof Error ? error.message : error}`);
-                }
-            }
-            return { response, editResults, lintResults, testResult, commitResult };
-        }
-        // Max reflections reached
-        ui.warning(`Max reflections (${this.config.maxReflections}) reached`);
-        return {
-            response: { action: { tool: 'none', message: 'Max reflections reached' } },
-            editResults,
-            lintResults,
-            testResult,
-            commitResult,
-        };
-    }
-}
-/**
- * Summarize conversation history to reduce tokens
- */
-export async function summarizeHistory(history, generateFn, maxMessages = 10) {
-    if (history.length <= maxMessages) {
-        return history;
-    }
-    // Keep first message (usually important context) and last few messages
-    const keepFirst = 1;
-    const keepLast = maxMessages - 2;
-    const toSummarize = history.slice(keepFirst, -keepLast);
-    if (toSummarize.length === 0) {
-        return history;
-    }
-    const summaryPrompt = `Summarize this conversation history concisely, preserving key information:
-
-${toSummarize.map(m => `${m.role}: ${m.content}`).join('\n\n')}
-
-Provide a brief summary that captures:
-1. Main topics discussed
-2. Key decisions made
-3. Important context for future messages`;
-    const summaryRes = await generateFn([
-        { role: 'user', content: summaryPrompt },
-    ]);
-    const summary = summaryRes.message || summaryRes.thought || summaryRes.raw || 'No summary generated';
-    return [
-        history[0], // Keep first
-        { role: 'system', content: `[Conversation Summary]\n${summary}` },
-        ...history.slice(-keepLast), // Keep last few
-    ];
-}

package/dist/lib/editor.d.ts

@@ -1,74 +0,0 @@
-/**
- * Intelligent Code Editor using ts-morph and fuzzy matching
- * Provides Aider-style SEARCH/REPLACE with smart matching
- */
-import { SourceFile, Node } from 'ts-morph';
-export interface EditBlock {
-    file: string;
-    search: string;
-    replace: string;
-}
-export interface EditResult {
-    file: string;
-    success: boolean;
-    applied: boolean;
-    error?: string;
-    diff?: string;
-    suggestion?: string;
-}
-/**
- * Apply a single edit to content
- */
-export declare function applyEdit(content: string, search: string, replace: string): {
-    success: boolean;
-    content: string;
-    method: string;
-    suggestion?: string;
-};
-/**
- * Apply edits to a file
- */
-export declare function applyFileEdits(edits: EditBlock[]): Promise<EditResult[]>;
-/**
- * Parse SEARCH/REPLACE blocks from LLM response
- */
-export declare function parseEditBlocks(response: string, validFiles?: string[]): EditBlock[];
-/**
- * TypeScript/JavaScript AST-aware editing using ts-morph
- */
-export declare class ASTEditor {
-    private project;
-    constructor();
-    /**
-     * Add a file to the project
-     */
-    addFile(path: string, content: string): SourceFile;
-    /**
-     * Get or create a source file
-     */
-    getSourceFile(path: string, content?: string): SourceFile | undefined;
-    /**
-     * Find a function by name
-     */
-    findFunction(sourceFile: SourceFile, name: string): Node | undefined;
-    /**
-     * Find a class by name
-     */
-    findClass(sourceFile: SourceFile, name: string): Node | undefined;
-    /**
-     * Rename a symbol across the project
-     */
-    renameSymbol(sourceFile: SourceFile, oldName: string, newName: string): boolean;
-    /**
-     * Get the modified content of a file
-     */
-    getContent(path: string): string | undefined;
-    /**
-     * Save all changes
-     */
-    saveAll(): Promise<void>;
-}
-/**
- * Create a code editor instance
- */
-export declare function createEditor(): ASTEditor;