nex-code 0.3.4 → 0.3.7

package/cli/compactor.js

@@ -1,85 +0,0 @@
-/**
- * cli/compactor.js — LLM-Based Conversation Compacting
- *
- * Replaces old messages with a semantic summary via callChat(),
- * preserving context while freeing tokens. Silent fallback on any error.
- */
-
-const { callChat } = require('./providers/registry');
-const { estimateTokens } = require('./context-engine');
-
-const COMPACTION_ENABLED = process.env.NEX_COMPACTION !== 'false';
-const COMPACTION_MIN_MESSAGES = 6;
-const COMPACTION_SUMMARY_BUDGET = 500;
-
-const COMPACT_PROMPT = `Summarize this conversation history concisely. Focus on:
-- What files were read, created, or modified
-- Key decisions made and their rationale
-- Current state of the task (what's done, what's pending)
-- Any errors encountered and how they were resolved
-Be factual and brief. Use bullet points. Max 300 words.`;
-
-/**
- * Compact old messages into a single summary message via LLM.
- * @param {Array} messages - Old (non-compacted) messages to summarize
- * @returns {Promise<{ message: object, tokensRemoved: number } | null>}
- */
-async function compactMessages(messages) {
-  if (!COMPACTION_ENABLED || messages.length < COMPACTION_MIN_MESSAGES) return null;
-
-  const summaryMessages = [
-    { role: 'system', content: COMPACT_PROMPT },
-    { role: 'user', content: formatMessagesForSummary(messages) },
-  ];
-
-  try {
-    const result = await callChat(summaryMessages, [], {
-      temperature: 0,
-      maxTokens: COMPACTION_SUMMARY_BUDGET,
-    });
-    const summary = (result.content || '').trim();
-    if (!summary) return null;
-
-    const originalTokens = messages.reduce((sum, m) =>
-      sum + estimateTokens(m.content || '') + (m.tool_calls ? estimateTokens(JSON.stringify(m.tool_calls)) : 0), 0);
-    const summaryTokens = estimateTokens(summary);
-
-    if (summaryTokens >= originalTokens * 0.8) return null;
-
-    return {
-      message: {
-        role: 'system',
-        content: `[Conversation Summary — ${messages.length} messages compacted]\n${summary}`,
-        _compacted: true,
-        _originalCount: messages.length,
-      },
-      tokensRemoved: originalTokens - summaryTokens,
-    };
-  } catch {
-    return null;
-  }
-}
-
-/**
- * Format messages for the summary prompt input.
- * Each message is truncated to 500 chars to control input budget.
- */
-function formatMessagesForSummary(messages) {
-  return messages.map(m => {
-    const role = m.role === 'tool' ? 'tool_result' : m.role;
-    const content = (m.content || '').substring(0, 500);
-    if (m.tool_calls) {
-      const tools = m.tool_calls.map(tc => tc.function?.name).join(', ');
-      return `[${role}] ${content}\n  tools: ${tools}`;
-    }
-    return `[${role}] ${content}`;
-  }).join('\n\n');
-}
-
-module.exports = {
-  compactMessages,
-  formatMessagesForSummary,
-  COMPACTION_ENABLED,
-  COMPACTION_MIN_MESSAGES,
-  COMPACTION_SUMMARY_BUDGET,
-};

package/cli/context-engine.js

@@ -1,507 +0,0 @@
-/**
- * cli/context-engine.js — Token Management + Context Compression
- *
- * Tracks token usage per model, auto-compresses conversation history
- * when approaching context window limits, and provides smart file truncation.
- */
-
-const { getActiveModel } = require('./providers/registry');
-
-// ─── Token Estimation ──────────────────────────────────────────
-
-// Chars-per-token ratios vary by provider/model
-const TOKEN_RATIOS = {
-  anthropic: 3.5,
-  openai: 4.0,
-  gemini: 4.0,
-  ollama: 4.0,
-  local: 4.0,
-};
-
-/**
- * Get chars-per-token ratio for current provider.
- */
-function getTokenRatio() {
-  try {
-    const model = getActiveModel();
-    const provider = model?.provider || 'ollama';
-    return TOKEN_RATIOS[provider] || 4.0;
-  } catch {
-    return 4.0;
-  }
-}
-
-/**
- * Estimate token count for a string.
- * Uses provider-specific chars/token ratio for better accuracy.
- */
-function estimateTokens(text) {
-  if (!text) return 0;
-  if (typeof text !== 'string') text = JSON.stringify(text);
-  return Math.ceil(text.length / getTokenRatio());
-}
-
-/**
- * Estimate token count for a single message (including role overhead).
- * Each message has ~4 tokens of overhead (role, formatting).
- */
-function estimateMessageTokens(msg) {
-  const MESSAGE_OVERHEAD = 4;
-  let tokens = MESSAGE_OVERHEAD;
-
-  if (msg.content) {
-    tokens += estimateTokens(msg.content);
-  }
-
-  if (msg.tool_calls) {
-    for (const tc of msg.tool_calls) {
-      tokens += 4; // tool call overhead
-      tokens += estimateTokens(tc.function?.name || '');
-      const args = tc.function?.arguments;
-      if (typeof args === 'string') {
-        tokens += estimateTokens(args);
-      } else if (args) {
-        tokens += estimateTokens(JSON.stringify(args));
-      }
-    }
-  }
-
-  return tokens;
-}
-
-/**
- * Estimate total tokens for a message array.
- */
-function estimateMessagesTokens(messages) {
-  let total = 0;
-  for (const msg of messages) {
-    total += estimateMessageTokens(msg);
-  }
-  return total;
-}
-
-/**
- * Estimate tokens for tool definitions.
- */
-function estimateToolsTokens(tools) {
-  if (!tools || tools.length === 0) return 0;
-  return estimateTokens(JSON.stringify(tools));
-}
-
-// ─── Context Window ────────────────────────────────────────────
-
-/**
- * Get the context window size for the active model.
- * Falls back to a conservative default if unknown.
- */
-function getContextWindow() {
-  const model = getActiveModel();
-  return model?.contextWindow || 32768;
-}
-
-/**
- * Get current token usage breakdown.
- * @param {Array} messages - Current conversation messages
- * @param {Array} tools - Tool definitions
- * @returns {{ used: number, limit: number, percentage: number, breakdown: object }}
- */
-function getUsage(messages, tools) {
-  const messageTokens = estimateMessagesTokens(messages);
-  const toolTokens = estimateToolsTokens(tools);
-  const used = messageTokens + toolTokens;
-  const limit = getContextWindow();
-  const percentage = limit > 0 ? (used / limit) * 100 : 0;
-
-  // Breakdown
-  let systemTokens = 0;
-  let conversationTokens = 0;
-  let toolResultTokens = 0;
-
-  for (const msg of messages) {
-    const t = estimateMessageTokens(msg);
-    if (msg.role === 'system') {
-      systemTokens += t;
-    } else if (msg.role === 'tool') {
-      toolResultTokens += t;
-    } else {
-      conversationTokens += t;
-    }
-  }
-
-  return {
-    used,
-    limit,
-    percentage: Math.round(percentage * 10) / 10,
-    breakdown: {
-      system: systemTokens,
-      conversation: conversationTokens,
-      toolResults: toolResultTokens,
-      toolDefinitions: toolTokens,
-    },
-    messageCount: messages.length,
-  };
-}
-
-// ─── Auto-Compression ──────────────────────────────────────────
-
-const COMPRESSION_THRESHOLD = 0.75; // Compress when >75% full
-const SAFETY_MARGIN = 0.10;         // 10% reserve for token estimation errors
-const KEEP_RECENT = 10; // Always keep last N messages intact
-const TRUNCATE_TOOL_RESULT = 200; // Truncate old tool results to N chars
-const TRUNCATE_ASSISTANT = 500; // Truncate old assistant content to N chars
-
-/**
- * Smart compression for tool result content.
- * Preserves error messages, test summaries, and error traces at end of output.
- *
- * @param {string} content - Tool result text
- * @param {number} maxChars - Character budget
- * @returns {string} compressed content
- */
-function compressToolResult(content, maxChars) {
-  if (!content || content.length <= maxChars) return content;
-
-  // Error/status messages get 3x budget — highest value for LLM recovery
-  const isError = /^(ERROR|EXIT|BLOCKED|CANCELLED)/i.test(content);
-  const budget = isError ? maxChars * 3 : maxChars;
-  if (content.length <= budget) return content;
-
-  const lines = content.split('\n');
-
-  // Short outputs (≤10 lines): character-based 60/40 head/tail split
-  if (lines.length <= 10) {
-    const headChars = Math.floor(budget * 0.6);
-    const tailChars = Math.floor(budget * 0.4);
-    const head = content.substring(0, headChars);
-    const tail = content.substring(content.length - tailChars);
-    return head + `\n...(${content.length} chars total)...\n` + tail;
-  }
-
-  // Long outputs (>10 lines): line-based 40/40 head/tail split
-  const headCount = Math.floor(lines.length * 0.4);
-  const tailCount = Math.floor(lines.length * 0.4);
-
-  // Build head and tail within budget
-  let headLines = [];
-  let headLen = 0;
-  const headBudget = Math.floor(budget * 0.4);
-  for (let i = 0; i < headCount && headLen < headBudget; i++) {
-    headLines.push(lines[i]);
-    headLen += lines[i].length + 1;
-  }
-
-  let tailLines = [];
-  let tailLen = 0;
-  const tailBudget = Math.floor(budget * 0.4);
-  for (let i = lines.length - 1; i >= lines.length - tailCount && tailLen < tailBudget; i--) {
-    tailLines.unshift(lines[i]);
-    tailLen += lines[i].length + 1;
-  }
-
-  const omitted = lines.length - headLines.length - tailLines.length;
-  return headLines.join('\n') + `\n...(${omitted} lines omitted, ${lines.length} total)...\n` + tailLines.join('\n');
-}
-
-/**
- * Compress a single message to reduce token usage.
- * @param {object} msg
- * @param {string} level - 'light', 'medium', or 'aggressive'
- * @returns {object} compressed message
- */
-function compressMessage(msg, level = 'light') {
-  const maxContent = level === 'aggressive' ? 100 : level === 'medium' ? 200 : TRUNCATE_ASSISTANT;
-  const maxTool = level === 'aggressive' ? 50 : level === 'medium' ? 100 : TRUNCATE_TOOL_RESULT;
-
-  if (msg.role === 'tool') {
-    const content = typeof msg.content === 'string' ? msg.content : JSON.stringify(msg.content);
-    if (content.length > maxTool) {
-      return {
-        ...msg,
-        content: compressToolResult(content, maxTool),
-      };
-    }
-    return msg;
-  }
-
-  if (msg.role === 'assistant') {
-    const compressed = { ...msg };
-
-    // Truncate long content
-    if (compressed.content && compressed.content.length > maxContent) {
-      compressed.content =
-        compressed.content.substring(0, maxContent) + `\n...(truncated)`;
-    }
-
-    // Simplify tool_calls in old messages
-    if (compressed.tool_calls && level === 'aggressive') {
-      compressed.tool_calls = compressed.tool_calls.map((tc) => ({
-        ...tc,
-        function: {
-          name: tc.function.name,
-          arguments: typeof tc.function.arguments === 'string'
-            ? tc.function.arguments.substring(0, 50)
-            : tc.function.arguments,
-        },
-      }));
-    }
-
-    return compressed;
-  }
-
-  // User and system messages: keep as-is (they're important context)
-  return msg;
-}
-
-/**
- * Fit messages into the context window.
- * Compresses older messages if the conversation exceeds the threshold.
- *
- * Strategy:
- * 1. Always keep system prompt (first message) intact
- * 2. Always keep the most recent KEEP_RECENT messages intact
- * 3. Compress middle messages (light first, then aggressive)
- * 4. If still too large, remove oldest compressed messages
- *
- * @param {Array} messages - Full message array
- * @param {Array} tools - Tool definitions
- * @param {object} [options] - { threshold, keepRecent }
- * @returns {{ messages: Array, compressed: boolean, tokensRemoved: number }}
- */
-async function fitToContext(messages, tools, options = {}) {
-  const threshold = options.threshold ?? COMPRESSION_THRESHOLD;
-  const safetyMargin = options.safetyMargin ?? SAFETY_MARGIN;
-  const keepRecent = options.keepRecent ?? KEEP_RECENT;
-
-  const limit = getContextWindow();
-  const toolTokens = estimateToolsTokens(tools);
-  const targetMax = Math.floor(limit * (threshold - safetyMargin));
-  const available = targetMax - toolTokens;
-
-  const currentTokens = estimateMessagesTokens(messages);
-  const totalUsed = currentTokens + toolTokens;
-
-  // Under threshold → no compression needed
-  if (totalUsed <= targetMax) {
-    return { messages, compressed: false, compacted: false, tokensRemoved: 0 };
-  }
-
-  const originalTokens = currentTokens;
-
-  // Split: system + old messages + recent messages
-  let system = null;
-  let startIdx = 0;
-  if (messages.length > 0 && messages[0].role === 'system') {
-    system = messages[0];
-    startIdx = 1;
-  }
-
-  const recentStart = Math.max(startIdx, messages.length - keepRecent);
-  let oldMessages = messages.slice(startIdx, recentStart);
-  const recentMessages = messages.slice(recentStart);
-
-  // Phase 0: LLM Compacting
-  const nonCompacted = oldMessages.filter(m => !m._compacted);
-  if (nonCompacted.length >= 6) {
-    try {
-      const { compactMessages } = require('./compactor');
-      const compactResult = await compactMessages(nonCompacted);
-      if (compactResult) {
-        const kept = oldMessages.filter(m => m._compacted);
-        const compressedOld = [...kept, compactResult.message];
-        const r = buildResult(system, compressedOld, recentMessages);
-        const t = estimateMessagesTokens(r);
-        if (t + toolTokens <= targetMax) {
-          return { messages: r, compressed: true, compacted: true,
-                   tokensRemoved: originalTokens - t };
-        }
-        // Compacted but still too large → continue with compacted messages as base
-        oldMessages = compressedOld;
-      }
-    } catch { /* silent fallback */ }
-  }
-
-  // Determine compression level based on how far over target we are
-  const overageRatio = (totalUsed - targetMax) / targetMax;
-
-  // Phase 1: Light compression (≤15% over target)
-  let compressed = oldMessages.map((msg) => compressMessage(msg, 'light'));
-  let result = buildResult(system, compressed, recentMessages);
-  let tokens = estimateMessagesTokens(result);
-
-  if (tokens + toolTokens <= targetMax) {
-    return {
-      messages: result,
-      compressed: true,
-      compacted: false,
-      tokensRemoved: originalTokens - tokens,
-    };
-  }
-
-  // Phase 2: Medium compression (≤30% over target)
-  compressed = oldMessages.map((msg) => compressMessage(msg, 'medium'));
-  result = buildResult(system, compressed, recentMessages);
-  tokens = estimateMessagesTokens(result);
-
-  if (tokens + toolTokens <= targetMax) {
-    return {
-      messages: result,
-      compressed: true,
-      compacted: false,
-      tokensRemoved: originalTokens - tokens,
-    };
-  }
-
-  // Phase 3: Aggressive compression (>30% over target)
-  compressed = oldMessages.map((msg) => compressMessage(msg, 'aggressive'));
-  result = buildResult(system, compressed, recentMessages);
-  tokens = estimateMessagesTokens(result);
-
-  if (tokens + toolTokens <= targetMax) {
-    return {
-      messages: result,
-      compressed: true,
-      compacted: false,
-      tokensRemoved: originalTokens - tokens,
-    };
-  }
-
-  // Phase 4: Remove oldest messages until we fit
-  while (compressed.length > 0 && tokens + toolTokens > available) {
-    const removed = compressed.shift();
-    tokens -= estimateMessageTokens(removed);
-  }
-
-  result = buildResult(system, compressed, recentMessages);
-  tokens = estimateMessagesTokens(result);
-
-  return {
-    messages: result,
-    compressed: true,
-    compacted: false,
-    tokensRemoved: originalTokens - tokens,
-  };
-}
-
-function buildResult(system, oldMessages, recentMessages) {
-  const result = [];
-  if (system) result.push(system);
-  result.push(...oldMessages, ...recentMessages);
-  return result;
-}
-
-// ─── Smart File Truncation ─────────────────────────────────────
-
-/**
- * Truncate file content to fit within a token budget.
- * Keeps the beginning and end of the file (most useful parts).
- *
- * @param {string} content - File content
- * @param {number} maxTokens - Maximum tokens to use
- * @returns {string} truncated content
- */
-function truncateFileContent(content, maxTokens) {
-  if (!content) return '';
-
-  const currentTokens = estimateTokens(content);
-  if (currentTokens <= maxTokens) return content;
-
-  const maxChars = maxTokens * 4; // Reverse the estimation
-  const lines = content.split('\n');
-
-  // Keep first 60% and last 40%
-  const headChars = Math.floor(maxChars * 0.6);
-  const tailChars = Math.floor(maxChars * 0.4);
-
-  let headContent = '';
-  let headLines = 0;
-  for (const line of lines) {
-    if (headContent.length + line.length + 1 > headChars) break;
-    headContent += (headContent ? '\n' : '') + line;
-    headLines++;
-  }
-
-  let tailContent = '';
-  let tailLines = 0;
-  for (let i = lines.length - 1; i >= headLines; i--) {
-    const candidate = lines[i] + (tailContent ? '\n' : '') + tailContent;
-    if (candidate.length > tailChars) break;
-    tailContent = candidate;
-    tailLines++;
-  }
-
-  const skipped = lines.length - headLines - tailLines;
-  const separator = `\n\n... (${skipped} lines omitted, ${lines.length} total) ...\n\n`;
-
-  return headContent + separator + tailContent;
-}
-
-// ─── Force Compression (Context-Too-Long Recovery) ─────────────
-
-const FORCE_COMPRESS_KEEP_RECENT = 6;
-
-/**
- * Emergency compression when the API rejects with "context too long".
- * More aggressive than fitToContext: targets 50% of context window
- * and keeps only 6 recent messages.
- *
- * @param {Array} messages - Full message array (including system prompt)
- * @param {Array} tools - Tool definitions
- * @returns {{ messages: Array, tokensRemoved: number }}
- */
-function forceCompress(messages, tools) {
-  const limit = getContextWindow();
-  const toolTokens = estimateToolsTokens(tools);
-  const targetMax = Math.floor(limit * 0.5) - toolTokens; // 50% of context window
-  const originalTokens = estimateMessagesTokens(messages);
-
-  // Split: system + old + recent
-  let system = null;
-  let startIdx = 0;
-  if (messages.length > 0 && messages[0].role === 'system') {
-    system = messages[0];
-    startIdx = 1;
-  }
-
-  const recentStart = Math.max(startIdx, messages.length - FORCE_COMPRESS_KEEP_RECENT);
-  let oldMessages = messages.slice(startIdx, recentStart);
-  const recentMessages = messages.slice(recentStart);
-
-  // Aggressive compression on all old messages
-  let compressed = oldMessages.map((msg) => compressMessage(msg, 'aggressive'));
-
-  // Remove oldest messages until we fit
-  let result = buildResult(system, compressed, recentMessages);
-  let tokens = estimateMessagesTokens(result);
-
-  while (compressed.length > 0 && tokens > targetMax) {
-    const removed = compressed.shift();
-    tokens -= estimateMessageTokens(removed);
-  }
-
-  result = buildResult(system, compressed, recentMessages);
-  tokens = estimateMessagesTokens(result);
-
-  return {
-    messages: result,
-    tokensRemoved: originalTokens - tokens,
-  };
-}
-
-// ─── Exports ───────────────────────────────────────────────────
-
-module.exports = {
-  estimateTokens,
-  estimateMessageTokens,
-  estimateMessagesTokens,
-  estimateToolsTokens,
-  getContextWindow,
-  getUsage,
-  compressMessage,
-  compressToolResult,
-  fitToContext,
-  forceCompress,
-  truncateFileContent,
-  COMPRESSION_THRESHOLD,
-  SAFETY_MARGIN,
-  KEEP_RECENT,
-};

package/cli/context.js

@@ -1,98 +0,0 @@
-/**
- * cli/context.js — Auto-Context: package.json, git, README
- */
-
-const fs = require('fs');
-const path = require('path');
-const { execSync } = require('child_process');
-const { C } = require('./ui');
-const { getMergeConflicts } = require('./git');
-
-function safe(fn) {
-  try {
-    return fn();
-  } catch {
-    return null;
-  }
-}
-
-function gatherProjectContext(cwd) {
-  const parts = [];
-
-  // package.json
-  const pkgPath = path.join(cwd, 'package.json');
-  if (fs.existsSync(pkgPath)) {
-    try {
-      const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-      const info = { name: pkg.name, version: pkg.version };
-      if (pkg.scripts) info.scripts = Object.keys(pkg.scripts).slice(0, 15);
-      if (pkg.dependencies) info.deps = Object.keys(pkg.dependencies).length;
-      if (pkg.devDependencies) info.devDeps = Object.keys(pkg.devDependencies).length;
-      parts.push(`PACKAGE: ${JSON.stringify(info)}`);
-    } catch { /* ignore corrupt package.json */ }
-  }
-
-  // README (first 50 lines)
-  const readmePath = path.join(cwd, 'README.md');
-  if (fs.existsSync(readmePath)) {
-    const lines = fs.readFileSync(readmePath, 'utf-8').split('\n').slice(0, 50);
-    parts.push(`README (first 50 lines):\n${lines.join('\n')}`);
-  }
-
-  // Git info
-  const branch = safe(() => execSync('git branch --show-current', { cwd, encoding: 'utf-8', stdio: 'pipe' }).trim());
-  if (branch) parts.push(`GIT BRANCH: ${branch}`);
-
-  const status = safe(() => execSync('git status --short', { cwd, encoding: 'utf-8', timeout: 5000, stdio: 'pipe' }).trim());
-  if (status) parts.push(`GIT STATUS:\n${status}`);
-
-  const log = safe(() =>
-    execSync('git log --oneline -5', { cwd, encoding: 'utf-8', timeout: 5000, stdio: 'pipe' }).trim()
-  );
-  if (log) parts.push(`RECENT COMMITS:\n${log}`);
-
-  // Merge conflicts
-  const conflicts = getMergeConflicts();
-  if (conflicts.length > 0) {
-    const conflictFiles = conflicts.map(c => `  ${c.file}`).join('\n');
-    parts.push(`MERGE CONFLICTS (resolve before editing these files):\n${conflictFiles}`);
-  }
-
-  // .gitignore
-  const giPath = path.join(cwd, '.gitignore');
-  if (fs.existsSync(giPath)) {
-    const content = fs.readFileSync(giPath, 'utf-8').trim();
-    parts.push(`GITIGNORE:\n${content}`);
-  }
-
-  return parts.join('\n\n');
-}
-
-function printContext(cwd) {
-  const pkgPath = path.join(cwd, 'package.json');
-  let project = '';
-  if (fs.existsSync(pkgPath)) {
-    try {
-      const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-      project = `${pkg.name || '?'} v${pkg.version || '?'}`;
-    } catch { /* ignore corrupt package.json */ }
-  }
-
-  const branch = safe(() => execSync('git branch --show-current', { cwd, encoding: 'utf-8', stdio: 'pipe' }).trim());
-
-  if (project) console.log(`${C.dim}  project: ${project}${C.reset}`);
-  if (branch) console.log(`${C.dim}  branch: ${branch}${C.reset}`);
-
-  const conflicts = getMergeConflicts();
-  if (conflicts.length > 0) {
-    console.log(`${C.red}  ⚠ ${conflicts.length} unresolved merge conflict(s):${C.reset}`);
-    for (const c of conflicts) {
-      console.log(`${C.red}    ${c.file}${C.reset}`);
-    }
-    console.log(`${C.yellow}  → Resolve conflicts before starting tasks${C.reset}`);
-  }
-
-  console.log();
-}
-
-module.exports = { gatherProjectContext, printContext };