@codemieai/code 0.0.2 → 0.0.4

package/dist/agents/codemie-code/prompts.js

@@ -0,0 +1,160 @@
+/**
+ * System Prompts for CodeMie Native Agent
+ *
+ * Contains the system prompt and instructions for the LangGraph ReAct agent
+ */
+export const SYSTEM_PROMPT = `You are CodeMie, an advanced AI coding assistant designed to help developers with various programming tasks.
+
+CAPABILITIES:
+- Read, write, and modify files in the project directory
+- Execute shell commands for building, testing, and development tasks
+- Perform Git operations (status, diff, add, commit, log)
+- Analyze code structure and provide recommendations
+- Help with debugging, refactoring, and code optimization
+
+GUIDELINES:
+- Always explain what you're doing before taking actions
+- Ask for confirmation before making significant changes
+- Provide clear, concise explanations of your reasoning
+- Follow best practices for the programming language being used
+- Be security-conscious when executing commands or modifying files
+
+CURRENT WORKING DIRECTORY: {workingDirectory}
+
+You have access to the following tools:`;
+/**
+ * Planning mode suffix for structured todo-based execution
+ * Inspired by LangChain-Code's Deep Agent planning approach
+ */
+export const PLANNING_SUFFIX = `
+## Planning & TODOs
+- In your FIRST 1-2 tool calls, call \`write_todos([...])\` with 3-8 concrete steps
+- Before working a step, call \`update_todo_status(index, "in_progress")\`
+- After finishing it, call \`update_todo_status(index, "completed")\`
+- Keep only one item "in_progress" at a time and keep todos verb-first and specific
+- Use specific, actionable language: "Read config.ts file" not "Understand configuration"
+
+## Todo Guidelines
+- pending: Task not yet started
+- in_progress: Currently working on (ONLY ONE allowed)
+- completed: Task finished successfully
+
+## Best Practices for Todos
+- Start with action verbs (Read, Write, Create, Update, Fix, Test, etc.)
+- Be specific about files, functions, or components
+- Break complex tasks into 3-8 manageable steps
+- Each step should have clear completion criteria
+- Avoid vague language like "handle", "deal with", "work on"
+
+Example good todos:
+- "Read package.json to understand dependencies"
+- "Create new React component in src/components/TodoList.tsx"
+- "Update API endpoint to handle authentication"
+- "Run tests to verify functionality"
+- "Commit changes with descriptive message"
+
+Example poor todos:
+- "Handle configuration" (too vague)
+- "Work on the authentication system" (too broad)
+- "Fix stuff" (unclear what needs fixing)
+`;
+/**
+ * Task-focused planning prompt for feature implementation
+ */
+export const FEATURE_PLANNING_PROMPT = `You are implementing a feature end-to-end with structured planning.
+
+Your process:
+1. **Planning Phase**: Create a detailed plan using write_todos with specific steps
+2. **Discovery Phase**: Use tools to understand the current codebase structure
+3. **Implementation Phase**: Make targeted changes with progress tracking
+4. **Verification Phase**: Test and validate your changes
+5. **Completion Phase**: Summarize what was accomplished
+
+Planning guidelines:
+- Break the feature into logical, sequential steps
+- Include discovery, implementation, testing, and documentation steps
+- Be specific about files and components to modify
+- Consider dependencies between steps
+- Plan for error handling and edge cases`;
+/**
+ * Bug fix planning prompt
+ */
+export const BUGFIX_PLANNING_PROMPT = `You are fixing a bug with a systematic approach.
+
+Your process:
+1. **Analysis Phase**: Understand the problem and create a plan using write_todos
+2. **Investigation Phase**: Use tools to locate the root cause
+3. **Fix Phase**: Implement a minimal, targeted fix
+4. **Testing Phase**: Verify the fix works and doesn't break anything else
+5. **Documentation Phase**: Document the fix and any lessons learned
+
+Planning guidelines:
+- Start by reproducing and understanding the issue
+- Plan investigation steps to narrow down the root cause
+- Design a minimal fix that addresses the core problem
+- Include regression testing in your plan
+- Consider edge cases and potential side effects`;
+/**
+ * Analysis planning prompt for code exploration
+ */
+export const ANALYSIS_PLANNING_PROMPT = `You are analyzing a codebase with a structured approach.
+
+Your process:
+1. **Planning Phase**: Create a plan using write_todos for systematic analysis
+2. **Overview Phase**: Get high-level understanding of project structure
+3. **Deep Dive Phase**: Examine specific components, patterns, and relationships
+4. **Assessment Phase**: Identify strengths, weaknesses, and opportunities
+5. **Reporting Phase**: Summarize findings with actionable recommendations
+
+Planning guidelines:
+- Plan to explore from general to specific
+- Include architecture, patterns, code quality, and dependencies
+- Look for potential issues, security concerns, and improvement opportunities
+- Structure findings in a clear, actionable format`;
+/**
+ * Get the system prompt with working directory substitution
+ */
+export function getSystemPrompt(workingDirectory) {
+    return SYSTEM_PROMPT.replace('{workingDirectory}', workingDirectory);
+}
+/**
+ * Get system prompt with planning mode enabled
+ */
+export function getSystemPromptWithPlanning(workingDirectory) {
+    return getSystemPrompt(workingDirectory) + PLANNING_SUFFIX;
+}
+/**
+ * Get task-specific planning prompt
+ */
+export function getTaskPlanningPrompt(taskType) {
+    switch (taskType) {
+        case 'feature':
+            return FEATURE_PLANNING_PROMPT;
+        case 'bugfix':
+            return BUGFIX_PLANNING_PROMPT;
+        case 'analysis':
+            return ANALYSIS_PLANNING_PROMPT;
+        default:
+            return FEATURE_PLANNING_PROMPT;
+    }
+}
+/**
+ * Detect task type from user input
+ */
+export function detectTaskType(task) {
+    const lowerTask = task.toLowerCase();
+    // Bug fix indicators
+    if (lowerTask.includes('fix') || lowerTask.includes('bug') || lowerTask.includes('error') ||
+        lowerTask.includes('issue') || lowerTask.includes('broken') || lowerTask.includes('crash')) {
+        return 'bugfix';
+    }
+    // Analysis indicators
+    if (lowerTask.includes('analyze') || lowerTask.includes('review') || lowerTask.includes('understand') ||
+        lowerTask.includes('explain') || lowerTask.includes('explore') || lowerTask.includes('what') ||
+        lowerTask.includes('how') || lowerTask.includes('why')) {
+        return 'analysis';
+    }
+    // Default to feature implementation
+    return 'feature';
+}
+//# sourceMappingURL=prompts.js.map

package/dist/agents/codemie-code/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../../src/agents/codemie-code/prompts.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;wCAkBW,CAAC;AAEzC;;;GAGG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+B9B,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG;;;;;;;;;;;;;;yCAcE,CAAC;AAE1C;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG;;;;;;;;;;;;;;iDAcW,CAAC;AAElD;;GAEG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG;;;;;;;;;;;;;mDAaW,CAAC;AAEpD;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,gBAAwB;IACtD,OAAO,aAAa,CAAC,OAAO,CAAC,oBAAoB,EAAE,gBAAgB,CAAC,CAAC;AACvE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CAAC,gBAAwB;IAClE,OAAO,eAAe,CAAC,gBAAgB,CAAC,GAAG,eAAe,CAAC;AAC7D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAA2C;IAC/E,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,SAAS;YACZ,OAAO,uBAAuB,CAAC;QACjC,KAAK,QAAQ;YACX,OAAO,sBAAsB,CAAC;QAChC,KAAK,UAAU;YACb,OAAO,wBAAwB,CAAC;QAClC;YACE,OAAO,uBAAuB,CAAC;IACnC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAErC,qBAAqB;IACrB,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;QACrF,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/F,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,sBAAsB;IACtB,IAAI,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC;QACjG,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC;QAC5F,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3D,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,oCAAoC;IACpC,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/agents/codemie-code/storage/todoStorage.d.ts

@@ -0,0 +1,78 @@
+/**
+ * Todo File Storage System
+ *
+ * Provides persistent storage for todos with both project-level and global backup storage
+ */
+import type { Todo } from '../types.js';
+export interface TodoStorageConfig {
+    workingDirectory: string;
+    enableGlobalBackup?: boolean;
+    debug?: boolean;
+}
+export interface TodoStorageMetadata {
+    version: string;
+    projectPath: string;
+    projectHash: string;
+    lastModified: Date;
+    sessionId: string;
+}
+export interface TodoFile {
+    metadata: TodoStorageMetadata;
+    todos: Todo[];
+}
+export declare class TodoFileStorage {
+    private workingDirectory;
+    private projectTodoPath;
+    private globalBackupDir;
+    private projectHash;
+    private sessionId;
+    private enableGlobalBackup;
+    private debug;
+    constructor(config: TodoStorageConfig);
+    /**
+     * Generate a hash for the project path to use as filename
+     */
+    private generateProjectHash;
+    /**
+     * Generate a unique session ID
+     */
+    private generateSessionId;
+    /**
+     * Save todos to both project and global storage
+     */
+    saveTodos(todos: Todo[]): Promise<void>;
+    /**
+     * Load todos from project storage (with global backup fallback)
+     */
+    loadTodos(): Promise<Todo[]>;
+    /**
+     * Save to project-level storage (.codemie/todos.json)
+     */
+    private saveToProject;
+    /**
+     * Save to global backup storage (~/.codemie/todos/{hash}.json)
+     */
+    private saveToGlobalBackup;
+    /**
+     * Load from project-level storage
+     */
+    private loadFromProject;
+    /**
+     * Load from global backup storage
+     */
+    private loadFromGlobalBackup;
+    /**
+     * Clear todos from both storages
+     */
+    clearTodos(): Promise<void>;
+    /**
+     * Get storage information
+     */
+    getStorageInfo(): {
+        projectPath: string;
+        globalBackupPath: string;
+        projectHash: string;
+        sessionId: string;
+    };
+}
+//# sourceMappingURL=todoStorage.d.ts.map

package/dist/agents/codemie-code/storage/todoStorage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"todoStorage.d.ts","sourceRoot":"","sources":["../../../../src/agents/codemie-code/storage/todoStorage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAExC,MAAM,WAAW,iBAAiB;IAChC,gBAAgB,EAAE,MAAM,CAAC;IACzB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,IAAI,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,QAAQ;IACvB,QAAQ,EAAE,mBAAmB,CAAC;IAC9B,KAAK,EAAE,IAAI,EAAE,CAAC;CACf;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,kBAAkB,CAAU;IACpC,OAAO,CAAC,KAAK,CAAU;gBAEX,MAAM,EAAE,iBAAiB;IAwBrC;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAI3B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B7C;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;IAiClC;;OAEG;YACW,aAAa;IAM3B;;OAEG;YACW,kBAAkB;IAahC;;OAEG;YACW,eAAe;IAkB7B;;OAEG;YACW,oBAAoB;IAwBlC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAqCjC;;OAEG;IACH,cAAc,IAAI;QAChB,WAAW,EAAE,MAAM,CAAC;QACpB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;KACnB;CAQF"}

package/dist/agents/codemie-code/storage/todoStorage.js

@@ -0,0 +1,225 @@
+/**
+ * Todo File Storage System
+ *
+ * Provides persistent storage for todos with both project-level and global backup storage
+ */
+import { promises as fs } from 'fs';
+import path from 'path';
+import os from 'os';
+import crypto from 'crypto';
+export class TodoFileStorage {
+    workingDirectory;
+    projectTodoPath;
+    globalBackupDir;
+    projectHash;
+    sessionId;
+    enableGlobalBackup;
+    debug;
+    constructor(config) {
+        this.workingDirectory = config.workingDirectory;
+        this.enableGlobalBackup = config.enableGlobalBackup ?? true;
+        this.debug = config.debug ?? false;
+        // Project-level storage path
+        this.projectTodoPath = path.join(this.workingDirectory, '.codemie', 'todos.json');
+        // Generate project hash for global storage
+        this.projectHash = this.generateProjectHash(this.workingDirectory);
+        // Global backup storage path
+        this.globalBackupDir = path.join(os.homedir(), '.codemie', 'todos');
+        // Generate session ID
+        this.sessionId = this.generateSessionId();
+        if (this.debug) {
+            console.log(`[TodoStorage] Project path: ${this.projectTodoPath}`);
+            console.log(`[TodoStorage] Global backup: ${this.globalBackupDir}`);
+            console.log(`[TodoStorage] Project hash: ${this.projectHash}`);
+        }
+    }
+    /**
+     * Generate a hash for the project path to use as filename
+     */
+    generateProjectHash(projectPath) {
+        return crypto.createHash('md5').update(projectPath).digest('hex').substring(0, 12);
+    }
+    /**
+     * Generate a unique session ID
+     */
+    generateSessionId() {
+        return `${Date.now()}-${Math.random().toString(36).substring(2, 8)}`;
+    }
+    /**
+     * Save todos to both project and global storage
+     */
+    async saveTodos(todos) {
+        const todoFile = {
+            metadata: {
+                version: '1.0.0',
+                projectPath: this.workingDirectory,
+                projectHash: this.projectHash,
+                lastModified: new Date(),
+                sessionId: this.sessionId
+            },
+            todos: todos
+        };
+        try {
+            // Save to project-level storage
+            await this.saveToProject(todoFile);
+            // Save to global backup if enabled
+            if (this.enableGlobalBackup) {
+                await this.saveToGlobalBackup(todoFile);
+            }
+            if (this.debug) {
+                console.log(`[TodoStorage] Saved ${todos.length} todos successfully`);
+            }
+        }
+        catch (error) {
+            console.error('[TodoStorage] Failed to save todos:', error);
+            throw new Error(`Failed to save todos: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Load todos from project storage (with global backup fallback)
+     */
+    async loadTodos() {
+        try {
+            // Try project-level storage first
+            const projectTodos = await this.loadFromProject();
+            if (projectTodos.length > 0) {
+                if (this.debug) {
+                    console.log(`[TodoStorage] Loaded ${projectTodos.length} todos from project storage`);
+                }
+                return projectTodos;
+            }
+            // Fallback to global backup
+            if (this.enableGlobalBackup) {
+                const globalTodos = await this.loadFromGlobalBackup();
+                if (globalTodos.length > 0) {
+                    if (this.debug) {
+                        console.log(`[TodoStorage] Loaded ${globalTodos.length} todos from global backup`);
+                    }
+                    return globalTodos;
+                }
+            }
+            if (this.debug) {
+                console.log('[TodoStorage] No existing todos found, starting fresh');
+            }
+            return [];
+        }
+        catch (error) {
+            console.warn('[TodoStorage] Failed to load todos:', error);
+            return [];
+        }
+    }
+    /**
+     * Save to project-level storage (.codemie/todos.json)
+     */
+    async saveToProject(todoFile) {
+        const projectDir = path.dirname(this.projectTodoPath);
+        await fs.mkdir(projectDir, { recursive: true });
+        await fs.writeFile(this.projectTodoPath, JSON.stringify(todoFile, null, 2), 'utf-8');
+    }
+    /**
+     * Save to global backup storage (~/.codemie/todos/{hash}.json)
+     */
+    async saveToGlobalBackup(todoFile) {
+        await fs.mkdir(this.globalBackupDir, { recursive: true });
+        const globalTodoPath = path.join(this.globalBackupDir, `${this.projectHash}.json`);
+        await fs.writeFile(globalTodoPath, JSON.stringify(todoFile, null, 2), 'utf-8');
+        // Also save a timestamped backup
+        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+        const backupPath = path.join(this.globalBackupDir, 'history', `${this.projectHash}-${timestamp}.json`);
+        await fs.mkdir(path.dirname(backupPath), { recursive: true });
+        await fs.writeFile(backupPath, JSON.stringify(todoFile, null, 2), 'utf-8');
+    }
+    /**
+     * Load from project-level storage
+     */
+    async loadFromProject() {
+        try {
+            const content = await fs.readFile(this.projectTodoPath, 'utf-8');
+            const todoFile = JSON.parse(content);
+            // Validate the structure
+            if (!todoFile.todos || !Array.isArray(todoFile.todos)) {
+                console.warn('[TodoStorage] Invalid project todo file structure');
+                return [];
+            }
+            return todoFile.todos;
+        }
+        catch {
+            // File doesn't exist or is invalid
+            return [];
+        }
+    }
+    /**
+     * Load from global backup storage
+     */
+    async loadFromGlobalBackup() {
+        try {
+            const globalTodoPath = path.join(this.globalBackupDir, `${this.projectHash}.json`);
+            const content = await fs.readFile(globalTodoPath, 'utf-8');
+            const todoFile = JSON.parse(content);
+            // Validate the structure and project match
+            if (!todoFile.todos || !Array.isArray(todoFile.todos)) {
+                console.warn('[TodoStorage] Invalid global todo file structure');
+                return [];
+            }
+            if (todoFile.metadata.projectHash !== this.projectHash) {
+                console.warn('[TodoStorage] Project hash mismatch in global backup');
+                return [];
+            }
+            return todoFile.todos;
+        }
+        catch {
+            // File doesn't exist or is invalid
+            return [];
+        }
+    }
+    /**
+     * Clear todos from both storages
+     */
+    async clearTodos() {
+        try {
+            // Clear project storage
+            await this.saveToProject({
+                metadata: {
+                    version: '1.0.0',
+                    projectPath: this.workingDirectory,
+                    projectHash: this.projectHash,
+                    lastModified: new Date(),
+                    sessionId: this.sessionId
+                },
+                todos: []
+            });
+            // Clear global backup
+            if (this.enableGlobalBackup) {
+                await this.saveToGlobalBackup({
+                    metadata: {
+                        version: '1.0.0',
+                        projectPath: this.workingDirectory,
+                        projectHash: this.projectHash,
+                        lastModified: new Date(),
+                        sessionId: this.sessionId
+                    },
+                    todos: []
+                });
+            }
+            if (this.debug) {
+                console.log('[TodoStorage] Cleared todos successfully');
+            }
+        }
+        catch (error) {
+            console.error('[TodoStorage] Failed to clear todos:', error);
+            throw new Error(`Failed to clear todos: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Get storage information
+     */
+    getStorageInfo() {
+        return {
+            projectPath: this.projectTodoPath,
+            globalBackupPath: path.join(this.globalBackupDir, `${this.projectHash}.json`),
+            projectHash: this.projectHash,
+            sessionId: this.sessionId
+        };
+    }
+}
+//# sourceMappingURL=todoStorage.js.map

package/dist/agents/codemie-code/storage/todoStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"todoStorage.js","sourceRoot":"","sources":["../../../../src/agents/codemie-code/storage/todoStorage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,MAAM,MAAM,QAAQ,CAAC;AAsB5B,MAAM,OAAO,eAAe;IAClB,gBAAgB,CAAS;IACzB,eAAe,CAAS;IACxB,eAAe,CAAS;IACxB,WAAW,CAAS;IACpB,SAAS,CAAS;IAClB,kBAAkB,CAAU;IAC5B,KAAK,CAAU;IAEvB,YAAY,MAAyB;QACnC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,kBAAkB,IAAI,IAAI,CAAC;QAC5D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC;QAEnC,6BAA6B;QAC7B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QAElF,2CAA2C;QAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAEnE,6BAA6B;QAC7B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAEpE,sBAAsB;QACtB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE1C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,+BAA+B,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;YACnE,OAAO,CAAC,GAAG,CAAC,gCAAgC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;YACpE,OAAO,CAAC,GAAG,CAAC,+BAA+B,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,WAAmB;QAC7C,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACrF,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACvE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,KAAa;QAC3B,MAAM,QAAQ,GAAa;YACzB,QAAQ,EAAE;gBACR,OAAO,EAAE,OAAO;gBAChB,WAAW,EAAE,IAAI,CAAC,gBAAgB;gBAClC,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,YAAY,EAAE,IAAI,IAAI,EAAE;gBACxB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B;YACD,KAAK,EAAE,KAAK;SACb,CAAC;QAEF,IAAI,CAAC;YACH,gCAAgC;YAChC,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YAEnC,mCAAmC;YACnC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,CAAC,MAAM,qBAAqB,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,yBAAyB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS;QACb,IAAI,CAAC;YACH,kCAAkC;YAClC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,GAAG,CAAC,wBAAwB,YAAY,CAAC,MAAM,6BAA6B,CAAC,CAAC;gBACxF,CAAC;gBACD,OAAO,YAAY,CAAC;YACtB,CAAC;YAED,4BAA4B;YAC5B,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;gBACtD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,GAAG,CAAC,wBAAwB,WAAW,CAAC,MAAM,2BAA2B,CAAC,CAAC;oBACrF,CAAC;oBACD,OAAO,WAAW,CAAC;gBACrB,CAAC;YACH,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;YACvE,CAAC;YACD,OAAO,EAAE,CAAC;QAEZ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;YAC3D,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,QAAkB;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACtD,MAAM,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACvF,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,QAAkB;QACjD,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC,WAAW,OAAO,CAAC,CAAC;QACnF,MAAM,EAAE,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAE/E,iCAAiC;QACjC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,WAAW,IAAI,SAAS,OAAO,CAAC,CAAC;QACvG,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;YACjE,MAAM,QAAQ,GAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAE/C,yBAAyB;YACzB,IAAI,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;gBAClE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,OAAO,QAAQ,CAAC,KAAK,CAAC;QACxB,CAAC;QAAC,MAAM,CAAC;YACP,mCAAmC;YACnC,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC,WAAW,OAAO,CAAC,CAAC;YACnF,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAE/C,2CAA2C;YAC3C,IAAI,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;gBACjE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,WAAW,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;gBACvD,OAAO,CAAC,IAAI,CAAC,sDAAsD,CAAC,CAAC;gBACrE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,OAAO,QAAQ,CAAC,KAAK,CAAC;QACxB,CAAC;QAAC,MAAM,CAAC;YACP,mCAAmC;YACnC,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC;YACH,wBAAwB;YACxB,MAAM,IAAI,CAAC,aAAa,CAAC;gBACvB,QAAQ,EAAE;oBACR,OAAO,EAAE,OAAO;oBAChB,WAAW,EAAE,IAAI,CAAC,gBAAgB;oBAClC,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,YAAY,EAAE,IAAI,IAAI,EAAE;oBACxB,SAAS,EAAE,IAAI,CAAC,SAAS;iBAC1B;gBACD,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;YAEH,sBAAsB;YACtB,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,MAAM,IAAI,CAAC,kBAAkB,CAAC;oBAC5B,QAAQ,EAAE;wBACR,OAAO,EAAE,OAAO;wBAChB,WAAW,EAAE,IAAI,CAAC,gBAAgB;wBAClC,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,YAAY,EAAE,IAAI,IAAI,EAAE;wBACxB,SAAS,EAAE,IAAI,CAAC,SAAS;qBAC1B;oBACD,KAAK,EAAE,EAAE;iBACV,CAAC,CAAC;YACL,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QAMZ,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,eAAe;YACjC,gBAAgB,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC,WAAW,OAAO,CAAC;YAC7E,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC;IACJ,CAAC;CACF"}

package/dist/agents/codemie-code/tokenUtils.d.ts

@@ -0,0 +1,108 @@
+/**
+ * Token Usage and Cost Calculation Utilities
+ *
+ * Provides utilities for extracting token usage from LLM responses
+ * and calculating costs for different providers and models.
+ */
+import type { TokenUsage } from './types.js';
+/**
+ * Provider pricing information (per 1M tokens)
+ * Updated as of November 2024 - check provider websites for latest pricing
+ */
+export declare const MODEL_PRICING: {
+    readonly 'gpt-4o': {
+        readonly input: 2.5;
+        readonly output: 10;
+    };
+    readonly 'gpt-4o-mini': {
+        readonly input: 0.15;
+        readonly output: 0.6;
+    };
+    readonly 'gpt-4-turbo': {
+        readonly input: 10;
+        readonly output: 30;
+    };
+    readonly 'gpt-4': {
+        readonly input: 30;
+        readonly output: 60;
+    };
+    readonly 'gpt-3.5-turbo': {
+        readonly input: 0.5;
+        readonly output: 1.5;
+    };
+    readonly 'claude-3-5-sonnet-20241022': {
+        readonly input: 3;
+        readonly output: 15;
+    };
+    readonly 'claude-3-5-sonnet-20240620': {
+        readonly input: 3;
+        readonly output: 15;
+    };
+    readonly 'claude-3-5-haiku-20241022': {
+        readonly input: 0.25;
+        readonly output: 1.25;
+    };
+    readonly 'claude-3-opus-20240229': {
+        readonly input: 15;
+        readonly output: 75;
+    };
+    readonly 'claude-3-sonnet-20240229': {
+        readonly input: 3;
+        readonly output: 15;
+    };
+    readonly 'claude-3-haiku-20240307': {
+        readonly input: 0.25;
+        readonly output: 1.25;
+    };
+    readonly 'gpt-4-*': {
+        readonly input: 10;
+        readonly output: 30;
+    };
+    readonly 'gpt-3.5-*': {
+        readonly input: 0.5;
+        readonly output: 1.5;
+    };
+    readonly 'claude-3-*': {
+        readonly input: 3;
+        readonly output: 15;
+    };
+};
+/**
+ * Extract token usage from LangChain AIMessage or response
+ * Based on LangChain v1.0+ usage_metadata format
+ */
+export declare function extractTokenUsage(response: any, model: string, _provider: string): TokenUsage | null;
+/**
+ * Calculate estimated cost for a model and token usage
+ */
+export declare function calculateCost(model: string, inputTokens: number, outputTokens: number): number;
+/**
+ * Get pricing information for a model
+ */
+export declare function getModelPricing(model: string): {
+    input: number;
+    output: number;
+} | null;
+/**
+ * Format cost as a human-readable string
+ */
+export declare function formatCost(cost: number): string;
+/**
+ * Format token count as a human-readable string
+ */
+export declare function formatTokens(tokens: number): string;
+/**
+ * Create a formatted summary of token usage
+ */
+export declare function formatTokenUsageSummary(usage: TokenUsage): string;
+/**
+ * Extract token usage from LangGraph stream chunk
+ * Based on LangChain documentation about "messages" stream mode
+ */
+export declare function extractTokenUsageFromStreamChunk(chunk: any, model: string, provider: string): TokenUsage | null;
+/**
+ * Extract token usage from final state after LangGraph completion
+ * This captures any usage information that wasn't caught during streaming
+ */
+export declare function extractTokenUsageFromFinalState(finalState: any, model: string, provider: string): TokenUsage | null;
+//# sourceMappingURL=tokenUtils.d.ts.map

package/dist/agents/codemie-code/tokenUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenUtils.d.ts","sourceRoot":"","sources":["../../../src/agents/codemie-code/tokenUtils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C;;;GAGG;AACH,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAoBhB,CAAC;AAEX;;;GAGG;AACH,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,GAAG,EACb,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,GAChB,UAAU,GAAG,IAAI,CAoEnB;AAED;;GAEG;AACH,wBAAgB,aAAa,CAC3B,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,GACnB,MAAM,CASR;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAkBvF;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK/C;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAInD;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,CAcjE;AAED;;;GAGG;AACH,wBAAgB,gCAAgC,CAC9C,KAAK,EAAE,GAAG,EACV,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,GACf,UAAU,GAAG,IAAI,CAgCnB;AAED;;;GAGG;AACH,wBAAgB,+BAA+B,CAC7C,UAAU,EAAE,GAAG,EACf,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,GACf,UAAU,GAAG,IAAI,CAkBnB"}