@stackmemoryai/stackmemory 0.5.2 → 0.5.4

package/dist/cli/commands/sweep.js

@@ -1,9 +1,19 @@
 import { Command } from "commander";
 import chalk from "chalk";
 import ora from "ora";
-import { existsSync, readFileSync } from "fs";
-import { join } from "path";
+import {
+  existsSync,
+  readFileSync,
+  writeFileSync,
+  mkdirSync,
+  copyFileSync,
+  chmodSync
+} from "fs";
+import { join, dirname } from "path";
+import { fileURLToPath } from "url";
 import { spawn, execSync } from "child_process";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
 function findPythonScript() {
   const locations = [
     join(
@@ -16,7 +26,7 @@ function findPythonScript() {
     join(
       process.cwd(),
       "node_modules",
-      "@stackmemory",
+      "@stackmemoryai",
       "sweep-addon",
       "python",
       "sweep_predict.py"
@@ -30,6 +40,32 @@ function findPythonScript() {
   }
   return null;
 }
+function findHookSource() {
+  const locations = [
+    join(process.cwd(), "templates", "claude-hooks", "post-edit-sweep.js"),
+    join(
+      process.cwd(),
+      "node_modules",
+      "@stackmemoryai",
+      "stackmemory",
+      "templates",
+      "claude-hooks",
+      "post-edit-sweep.js"
+    ),
+    join(
+      dirname(dirname(dirname(__dirname))),
+      "templates",
+      "claude-hooks",
+      "post-edit-sweep.js"
+    )
+  ];
+  for (const loc of locations) {
+    if (existsSync(loc)) {
+      return loc;
+    }
+  }
+  return null;
+}
 async function findPython() {
   const candidates = ["python3", "python"];
   for (const cmd of candidates) {
@@ -280,12 +316,144 @@ hf_hub_download(
       console.log(chalk.gray(`Tokens: ${result.tokens_generated}`));
     }
     if (options.output) {
-      const { writeFileSync } = await import("fs");
-      writeFileSync(options.output, result.predicted_content || "");
+      const { writeFileSync: writeFileSync2 } = await import("fs");
+      writeFileSync2(options.output, result.predicted_content || "");
       console.log(chalk.green(`
 Written to: ${options.output}`));
     }
   });
+  const hookCmd = cmd.command("hook").description("Manage Claude Code integration hook");
+  hookCmd.command("install").description("Install Sweep prediction hook for Claude Code").action(async () => {
+    const spinner = ora("Installing Sweep hook...").start();
+    const homeDir = process.env.HOME || "";
+    const hookDir = join(homeDir, ".claude", "hooks");
+    const sweepDir = join(homeDir, ".stackmemory", "sweep");
+    const hooksJsonPath = join(homeDir, ".claude", "hooks.json");
+    try {
+      mkdirSync(hookDir, { recursive: true });
+      mkdirSync(sweepDir, { recursive: true });
+      const hookSource = findHookSource();
+      if (!hookSource) {
+        spinner.fail(chalk.red("Hook template not found"));
+        console.log(
+          chalk.gray("Ensure stackmemory is installed from the repository")
+        );
+        process.exit(1);
+      }
+      const hookDest = join(hookDir, "post-edit-sweep.js");
+      copyFileSync(hookSource, hookDest);
+      chmodSync(hookDest, "755");
+      const pythonScriptSource = findPythonScript();
+      if (pythonScriptSource) {
+        const pythonDest = join(sweepDir, "sweep_predict.py");
+        copyFileSync(pythonScriptSource, pythonDest);
+      }
+      if (existsSync(hooksJsonPath)) {
+        const hooks = JSON.parse(readFileSync(hooksJsonPath, "utf-8"));
+        if (!hooks["post-tool-use"]) {
+          hooks["post-tool-use"] = hookDest;
+          writeFileSync(hooksJsonPath, JSON.stringify(hooks, null, 2));
+        } else if (!hooks["post-tool-use"].includes("sweep")) {
+          spinner.warn(chalk.yellow("post-tool-use hook already configured"));
+          console.log(chalk.gray(`Existing: ${hooks["post-tool-use"]}`));
+          console.log(chalk.gray(`Hook installed at: ${hookDest}`));
+          console.log(
+            chalk.gray("You may need to manually configure the hook chain")
+          );
+          return;
+        }
+      } else {
+        const hooks = { "post-tool-use": hookDest };
+        writeFileSync(hooksJsonPath, JSON.stringify(hooks, null, 2));
+      }
+      spinner.succeed(chalk.green("Sweep hook installed"));
+      console.log(chalk.gray(`Hook: ${hookDest}`));
+      console.log(chalk.gray(`Config: ${hooksJsonPath}`));
+      console.log("");
+      console.log(chalk.bold("Usage:"));
+      console.log("  Hook runs automatically after Edit/Write operations");
+      console.log("  Predictions appear after 2+ edits in session");
+      console.log("  Disable: export SWEEP_ENABLED=false");
+    } catch (error) {
+      spinner.fail(chalk.red("Installation failed"));
+      console.log(chalk.gray(error.message));
+      process.exit(1);
+    }
+  });
+  hookCmd.command("status").description("Check hook installation status").action(async () => {
+    const homeDir = process.env.HOME || "";
+    const hookPath = join(homeDir, ".claude", "hooks", "post-edit-sweep.js");
+    const hooksJsonPath = join(homeDir, ".claude", "hooks.json");
+    const statePath = join(homeDir, ".stackmemory", "sweep-state.json");
+    console.log(chalk.bold("\nSweep Hook Status\n"));
+    const hookInstalled = existsSync(hookPath);
+    console.log(
+      `Hook installed: ${hookInstalled ? chalk.green("Yes") : chalk.yellow("No")}`
+    );
+    if (existsSync(hooksJsonPath)) {
+      const hooks = JSON.parse(readFileSync(hooksJsonPath, "utf-8"));
+      const configured = hooks["post-tool-use"] && hooks["post-tool-use"].includes("sweep");
+      console.log(
+        `Hook configured: ${configured ? chalk.green("Yes") : chalk.yellow("No")}`
+      );
+    } else {
+      console.log(`Hook configured: ${chalk.yellow("No hooks.json")}`);
+    }
+    const enabled = process.env.SWEEP_ENABLED !== "false";
+    console.log(
+      `Enabled: ${enabled ? chalk.green("Yes") : chalk.yellow("Disabled (SWEEP_ENABLED=false)")}`
+    );
+    if (existsSync(statePath)) {
+      try {
+        const state = JSON.parse(readFileSync(statePath, "utf-8"));
+        console.log(
+          chalk.gray(
+            `
+Recent diffs tracked: ${state.recentDiffs?.length || 0}`
+          )
+        );
+        if (state.lastPrediction) {
+          const age = Date.now() - state.lastPrediction.timestamp;
+          const ageStr = age < 6e4 ? `${Math.round(age / 1e3)}s ago` : `${Math.round(age / 6e4)}m ago`;
+          console.log(chalk.gray(`Last prediction: ${ageStr}`));
+        }
+      } catch {
+      }
+    }
+    if (!hookInstalled) {
+      console.log(chalk.bold("\nTo install: stackmemory sweep hook install"));
+    }
+  });
+  hookCmd.command("disable").description("Disable the Sweep hook").action(() => {
+    console.log(chalk.bold("\nTo disable Sweep predictions:\n"));
+    console.log("  Temporarily: export SWEEP_ENABLED=false");
+    console.log("  Permanently: Add to ~/.zshrc or ~/.bashrc");
+    console.log("");
+    console.log("Or remove the hook:");
+    console.log("  rm ~/.claude/hooks/post-edit-sweep.js");
+  });
+  hookCmd.command("clear").description("Clear hook state (recent diffs and predictions)").action(() => {
+    const homeDir = process.env.HOME || "";
+    const statePath = join(homeDir, ".stackmemory", "sweep-state.json");
+    if (existsSync(statePath)) {
+      writeFileSync(
+        statePath,
+        JSON.stringify(
+          {
+            recentDiffs: [],
+            lastPrediction: null,
+            pendingPrediction: null,
+            fileContents: {}
+          },
+          null,
+          2
+        )
+      );
+      console.log(chalk.green("Sweep state cleared"));
+    } else {
+      console.log(chalk.gray("No state file found"));
+    }
+  });
   cmd.action(async () => {
     const status = await checkSweepStatus();
     console.log(chalk.bold("\nSweep 1.5B Addon Status\n"));

package/dist/cli/commands/sweep.js.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
   "sources": ["../../../src/cli/commands/sweep.ts"],
-  "sourcesContent": ["/**\n * Sweep command for StackMemory\n * Provides next-edit predictions using the Sweep 1.5B model\n *\n * Usage:\n *   stackmemory sweep setup          Install dependencies and optionally download model\n *   stackmemory sweep status         Check if Sweep addon is properly configured\n *   stackmemory sweep predict <file> Run prediction on a file\n */\n\nimport { Command } from 'commander';\nimport chalk from 'chalk';\nimport ora from 'ora';\nimport { existsSync, readFileSync } from 'fs';\nimport { join } from 'path';\nimport { spawn, execSync } from 'child_process';\n\ninterface SweepStatus {\n  installed: boolean;\n  model_downloaded: boolean;\n  python_path?: string;\n  model_path?: string;\n  error?: string;\n}\n\ninterface SweepPredictResult {\n  success: boolean;\n  predicted_content?: string;\n  file_path?: string;\n  latency_ms?: number;\n  tokens_generated?: number;\n  error?: string;\n  message?: string;\n}\n\nfunction findPythonScript(): string | null {\n  const locations = [\n    join(\n      process.cwd(),\n      'packages',\n      'sweep-addon',\n      'python',\n      'sweep_predict.py'\n    ),\n    join(\n      process.cwd(),\n      'node_modules',\n      '@stackmemory',\n      'sweep-addon',\n      'python',\n      'sweep_predict.py'\n    ),\n    join(process.env.HOME || '', '.stackmemory', 'sweep', 'sweep_predict.py'),\n  ];\n\n  for (const loc of locations) {\n    if (existsSync(loc)) {\n      return loc;\n    }\n  }\n  return null;\n}\n\nasync function findPython(): Promise<string | null> {\n  const candidates = ['python3', 'python'];\n\n  for (const cmd of candidates) {\n    try {\n      execSync(`${cmd} --version`, { stdio: 'pipe' });\n      return cmd;\n    } catch {\n      continue;\n    }\n  }\n  return null;\n}\n\nasync function checkSweepStatus(): Promise<SweepStatus> {\n  const pythonPath = await findPython();\n  if (!pythonPath) {\n    return {\n      installed: false,\n      model_downloaded: false,\n      error: 'Python not found. Install Python 3.10+',\n    };\n  }\n\n  const scriptPath = findPythonScript();\n  if (!scriptPath) {\n    return {\n      installed: false,\n      model_downloaded: false,\n      python_path: pythonPath,\n      error: 'Sweep addon not installed. Run: stackmemory sweep setup',\n    };\n  }\n\n  const homeDir = process.env.HOME || '';\n  const modelPath = join(\n    homeDir,\n    '.stackmemory',\n    'models',\n    'sweep',\n    'sweep-next-edit-1.5b.q8_0.v2.gguf'\n  );\n  const modelDownloaded = existsSync(modelPath);\n\n  return {\n    installed: true,\n    model_downloaded: modelDownloaded,\n    python_path: pythonPath,\n    model_path: modelDownloaded ? modelPath : undefined,\n  };\n}\n\nasync function runPrediction(\n  filePath: string,\n  pythonPath: string,\n  scriptPath: string\n): Promise<SweepPredictResult> {\n  if (!existsSync(filePath)) {\n    return {\n      success: false,\n      error: 'file_not_found',\n      message: `File not found: ${filePath}`,\n    };\n  }\n\n  const currentContent = readFileSync(filePath, 'utf-8');\n\n  const input = {\n    file_path: filePath,\n    current_content: currentContent,\n  };\n\n  return new Promise((resolve) => {\n    const proc = spawn(pythonPath, [scriptPath], {\n      stdio: ['pipe', 'pipe', 'pipe'],\n    });\n\n    let stdout = '';\n    let stderr = '';\n\n    proc.stdout.on('data', (data) => (stdout += data));\n    proc.stderr.on('data', (data) => (stderr += data));\n\n    proc.on('close', (code) => {\n      try {\n        if (stdout.trim()) {\n          const result = JSON.parse(stdout.trim());\n          resolve(result);\n        } else if (code !== 0) {\n          resolve({\n            success: false,\n            error: 'process_error',\n            message: stderr || `Process exited with code ${code}`,\n          });\n        } else {\n          resolve({\n            success: false,\n            error: 'no_output',\n            message: 'No output from prediction script',\n          });\n        }\n      } catch {\n        resolve({\n          success: false,\n          error: 'parse_error',\n          message: `Failed to parse output: ${stdout}`,\n        });\n      }\n    });\n\n    proc.on('error', (error) => {\n      resolve({\n        success: false,\n        error: 'spawn_error',\n        message: error.message,\n      });\n    });\n\n    proc.stdin.write(JSON.stringify(input));\n    proc.stdin.end();\n  });\n}\n\nexport function createSweepCommand(): Command {\n  const cmd = new Command('sweep')\n    .description(\n      'Next-edit predictions using Sweep 1.5B model (optional addon)'\n    )\n    .addHelpText(\n      'after',\n      `\nExamples:\n  stackmemory sweep setup              Install Python dependencies\n  stackmemory sweep setup --download   Also download the model (1.5GB)\n  stackmemory sweep status             Check addon status\n  stackmemory sweep predict src/app.ts Predict next edit for a file\n\nRequirements:\n  - Python 3.10+\n  - pip packages: huggingface_hub, llama-cpp-python\n\nThe Sweep 1.5B model predicts what code changes you'll make next based on:\n  - Current file content\n  - Recent changes (diffs)\n  - Context from other files\n\nModel is downloaded from HuggingFace on first prediction (~1.5GB).\n`\n    );\n\n  cmd\n    .command('setup')\n    .description('Install Python dependencies for Sweep addon')\n    .option('--download', 'Also download the model now')\n    .action(async (options) => {\n      const spinner = ora('Checking Python...').start();\n\n      const pythonPath = await findPython();\n      if (!pythonPath) {\n        spinner.fail(chalk.red('Python not found'));\n        console.log(chalk.gray('Please install Python 3.10+'));\n        process.exit(1);\n      }\n\n      spinner.text = 'Installing Python dependencies...';\n\n      try {\n        execSync(\n          `${pythonPath} -m pip install --quiet huggingface_hub llama-cpp-python`,\n          {\n            stdio: 'pipe',\n          }\n        );\n        spinner.succeed(chalk.green('Python dependencies installed'));\n      } catch {\n        spinner.fail(chalk.red('Failed to install dependencies'));\n        console.log(\n          chalk.gray(\n            `Run: ${pythonPath} -m pip install huggingface_hub llama-cpp-python`\n          )\n        );\n        process.exit(1);\n      }\n\n      if (options.download) {\n        const downloadSpinner = ora('Downloading Sweep 1.5B model...').start();\n        downloadSpinner.text = 'Downloading model from HuggingFace (~1.5GB)...';\n\n        try {\n          execSync(\n            `${pythonPath} -c \"\nfrom huggingface_hub import hf_hub_download\nimport os\nmodel_dir = os.path.expanduser('~/.stackmemory/models/sweep')\nos.makedirs(model_dir, exist_ok=True)\nhf_hub_download(\n    repo_id='sweepai/sweep-next-edit-1.5B',\n    filename='sweep-next-edit-1.5b.q8_0.v2.gguf',\n    repo_type='model',\n    local_dir=model_dir,\n    local_dir_use_symlinks=False\n)\n\"`,\n            { stdio: 'pipe', timeout: 600000 }\n          );\n          downloadSpinner.succeed(chalk.green('Model downloaded'));\n        } catch {\n          downloadSpinner.fail(chalk.red('Model download failed'));\n          console.log(chalk.gray('Model will be downloaded on first use'));\n        }\n      } else {\n        console.log(\n          chalk.gray('\\nModel will be downloaded on first prediction (~1.5GB)')\n        );\n        console.log(chalk.gray('Or run: stackmemory sweep setup --download'));\n      }\n\n      console.log(chalk.bold('\\nSetup complete!'));\n    });\n\n  cmd\n    .command('status')\n    .description('Check Sweep addon status')\n    .action(async () => {\n      console.log(chalk.bold('\\nSweep 1.5B Addon Status\\n'));\n\n      const status = await checkSweepStatus();\n\n      if (status.error) {\n        console.log(chalk.red(`Error: ${status.error}`));\n        console.log('');\n      }\n\n      console.log(\n        `Python: ${status.python_path ? chalk.green(status.python_path) : chalk.red('Not found')}`\n      );\n      console.log(\n        `Addon installed: ${status.installed ? chalk.green('Yes') : chalk.yellow('No')}`\n      );\n      console.log(\n        `Model downloaded: ${status.model_downloaded ? chalk.green('Yes') : chalk.yellow('No (will download on first use)')}`\n      );\n\n      if (status.model_path) {\n        console.log(chalk.gray(`Model path: ${status.model_path}`));\n      }\n\n      if (!status.installed) {\n        console.log(chalk.bold('\\nTo install:'));\n        console.log('  stackmemory sweep setup');\n      }\n    });\n\n  cmd\n    .command('predict <file>')\n    .description('Predict next edit for a file')\n    .option('-o, --output <path>', 'Write prediction to file instead of stdout')\n    .option('--json', 'Output raw JSON result')\n    .action(async (file, options) => {\n      const status = await checkSweepStatus();\n\n      if (!status.installed) {\n        console.error(chalk.red('Sweep addon not installed'));\n        console.log(chalk.gray('Run: stackmemory sweep setup'));\n        process.exit(1);\n      }\n\n      const scriptPath = findPythonScript();\n      if (!scriptPath || !status.python_path) {\n        console.error(chalk.red('Could not find Sweep prediction script'));\n        process.exit(1);\n      }\n\n      const spinner = ora('Running prediction...').start();\n\n      if (!status.model_downloaded) {\n        spinner.text = 'Downloading model (first time only, ~1.5GB)...';\n      }\n\n      const result = await runPrediction(file, status.python_path, scriptPath);\n\n      if (!result.success) {\n        spinner.fail(\n          chalk.red(`Prediction failed: ${result.message || result.error}`)\n        );\n        process.exit(1);\n      }\n\n      spinner.succeed(chalk.green('Prediction complete'));\n\n      if (options.json) {\n        console.log(JSON.stringify(result, null, 2));\n        return;\n      }\n\n      console.log(chalk.bold('\\nPredicted content:'));\n      console.log(chalk.gray('\u2500'.repeat(50)));\n      console.log(result.predicted_content);\n      console.log(chalk.gray('\u2500'.repeat(50)));\n\n      if (result.latency_ms) {\n        console.log(chalk.gray(`Latency: ${result.latency_ms}ms`));\n      }\n      if (result.tokens_generated) {\n        console.log(chalk.gray(`Tokens: ${result.tokens_generated}`));\n      }\n\n      if (options.output) {\n        const { writeFileSync } = await import('fs');\n        writeFileSync(options.output, result.predicted_content || '');\n        console.log(chalk.green(`\\nWritten to: ${options.output}`));\n      }\n    });\n\n  cmd.action(async () => {\n    const status = await checkSweepStatus();\n    console.log(chalk.bold('\\nSweep 1.5B Addon Status\\n'));\n\n    console.log(\n      `Installed: ${status.installed ? chalk.green('Yes') : chalk.yellow('No')}`\n    );\n    console.log(\n      `Model ready: ${status.model_downloaded ? chalk.green('Yes') : chalk.yellow('No')}`\n    );\n\n    if (!status.installed) {\n      console.log(chalk.bold('\\nRun: stackmemory sweep setup'));\n    } else {\n      console.log(chalk.bold('\\nUsage: stackmemory sweep predict <file>'));\n    }\n  });\n\n  return cmd;\n}\n\nexport default createSweepCommand();\n"],
-  "mappings": "AAUA,SAAS,eAAe;AACxB,OAAO,WAAW;AAClB,OAAO,SAAS;AAChB,SAAS,YAAY,oBAAoB;AACzC,SAAS,YAAY;AACrB,SAAS,OAAO,gBAAgB;AAoBhC,SAAS,mBAAkC;AACzC,QAAM,YAAY;AAAA,IAChB;AAAA,MACE,QAAQ,IAAI;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA;AAAA,MACE,QAAQ,IAAI;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,KAAK,QAAQ,IAAI,QAAQ,IAAI,gBAAgB,SAAS,kBAAkB;AAAA,EAC1E;AAEA,aAAW,OAAO,WAAW;AAC3B,QAAI,WAAW,GAAG,GAAG;AACnB,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAe,aAAqC;AAClD,QAAM,aAAa,CAAC,WAAW,QAAQ;AAEvC,aAAW,OAAO,YAAY;AAC5B,QAAI;AACF,eAAS,GAAG,GAAG,cAAc,EAAE,OAAO,OAAO,CAAC;AAC9C,aAAO;AAAA,IACT,QAAQ;AACN;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAe,mBAAyC;AACtD,QAAM,aAAa,MAAM,WAAW;AACpC,MAAI,CAAC,YAAY;AACf,WAAO;AAAA,MACL,WAAW;AAAA,MACX,kBAAkB;AAAA,MAClB,OAAO;AAAA,IACT;AAAA,EACF;AAEA,QAAM,aAAa,iBAAiB;AACpC,MAAI,CAAC,YAAY;AACf,WAAO;AAAA,MACL,WAAW;AAAA,MACX,kBAAkB;AAAA,MAClB,aAAa;AAAA,MACb,OAAO;AAAA,IACT;AAAA,EACF;AAEA,QAAM,UAAU,QAAQ,IAAI,QAAQ;AACpC,QAAM,YAAY;AAAA,IAChB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,QAAM,kBAAkB,WAAW,SAAS;AAE5C,SAAO;AAAA,IACL,WAAW;AAAA,IACX,kBAAkB;AAAA,IAClB,aAAa;AAAA,IACb,YAAY,kBAAkB,YAAY;AAAA,EAC5C;AACF;AAEA,eAAe,cACb,UACA,YACA,YAC6B;AAC7B,MAAI,CAAC,WAAW,QAAQ,GAAG;AACzB,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO;AAAA,MACP,SAAS,mBAAmB,QAAQ;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,iBAAiB,aAAa,UAAU,OAAO;AAErD,QAAM,QAAQ;AAAA,IACZ,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAEA,SAAO,IAAI,QAAQ,CAAC,YAAY;AAC9B,UAAM,OAAO,MAAM,YAAY,CAAC,UAAU,GAAG;AAAA,MAC3C,OAAO,CAAC,QAAQ,QAAQ,MAAM;AAAA,IAChC,CAAC;AAED,QAAI,SAAS;AACb,QAAI,SAAS;AAEb,SAAK,OAAO,GAAG,QAAQ,CAAC,SAAU,UAAU,IAAK;AACjD,SAAK,OAAO,GAAG,QAAQ,CAAC,SAAU,UAAU,IAAK;AAEjD,SAAK,GAAG,SAAS,CAAC,SAAS;AACzB,UAAI;AACF,YAAI,OAAO,KAAK,GAAG;AACjB,gBAAM,SAAS,KAAK,MAAM,OAAO,KAAK,CAAC;AACvC,kBAAQ,MAAM;AAAA,QAChB,WAAW,SAAS,GAAG;AACrB,kBAAQ;AAAA,YACN,SAAS;AAAA,YACT,OAAO;AAAA,YACP,SAAS,UAAU,4BAA4B,IAAI;AAAA,UACrD,CAAC;AAAA,QACH,OAAO;AACL,kBAAQ;AAAA,YACN,SAAS;AAAA,YACT,OAAO;AAAA,YACP,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF,QAAQ;AACN,gBAAQ;AAAA,UACN,SAAS;AAAA,UACT,OAAO;AAAA,UACP,SAAS,2BAA2B,MAAM;AAAA,QAC5C,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,SAAK,GAAG,SAAS,CAAC,UAAU;AAC1B,cAAQ;AAAA,QACN,SAAS;AAAA,QACT,OAAO;AAAA,QACP,SAAS,MAAM;AAAA,MACjB,CAAC;AAAA,IACH,CAAC;AAED,SAAK,MAAM,MAAM,KAAK,UAAU,KAAK,CAAC;AACtC,SAAK,MAAM,IAAI;AAAA,EACjB,CAAC;AACH;AAEO,SAAS,qBAA8B;AAC5C,QAAM,MAAM,IAAI,QAAQ,OAAO,EAC5B;AAAA,IACC;AAAA,EACF,EACC;AAAA,IACC;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBF;AAEF,MACG,QAAQ,OAAO,EACf,YAAY,6CAA6C,EACzD,OAAO,cAAc,6BAA6B,EAClD,OAAO,OAAO,YAAY;AACzB,UAAM,UAAU,IAAI,oBAAoB,EAAE,MAAM;AAEhD,UAAM,aAAa,MAAM,WAAW;AACpC,QAAI,CAAC,YAAY;AACf,cAAQ,KAAK,MAAM,IAAI,kBAAkB,CAAC;AAC1C,cAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AACrD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,YAAQ,OAAO;AAEf,QAAI;AACF;AAAA,QACE,GAAG,UAAU;AAAA,QACb;AAAA,UACE,OAAO;AAAA,QACT;AAAA,MACF;AACA,cAAQ,QAAQ,MAAM,MAAM,+BAA+B,CAAC;AAAA,IAC9D,QAAQ;AACN,cAAQ,KAAK,MAAM,IAAI,gCAAgC,CAAC;AACxD,cAAQ;AAAA,QACN,MAAM;AAAA,UACJ,QAAQ,UAAU;AAAA,QACpB;AAAA,MACF;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,QAAI,QAAQ,UAAU;AACpB,YAAM,kBAAkB,IAAI,iCAAiC,EAAE,MAAM;AACrE,sBAAgB,OAAO;AAEvB,UAAI;AACF;AAAA,UACE,GAAG,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,UAab,EAAE,OAAO,QAAQ,SAAS,IAAO;AAAA,QACnC;AACA,wBAAgB,QAAQ,MAAM,MAAM,kBAAkB,CAAC;AAAA,MACzD,QAAQ;AACN,wBAAgB,KAAK,MAAM,IAAI,uBAAuB,CAAC;AACvD,gBAAQ,IAAI,MAAM,KAAK,uCAAuC,CAAC;AAAA,MACjE;AAAA,IACF,OAAO;AACL,cAAQ;AAAA,QACN,MAAM,KAAK,yDAAyD;AAAA,MACtE;AACA,cAAQ,IAAI,MAAM,KAAK,4CAA4C,CAAC;AAAA,IACtE;AAEA,YAAQ,IAAI,MAAM,KAAK,mBAAmB,CAAC;AAAA,EAC7C,CAAC;AAEH,MACG,QAAQ,QAAQ,EAChB,YAAY,0BAA0B,EACtC,OAAO,YAAY;AAClB,YAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AAErD,UAAM,SAAS,MAAM,iBAAiB;AAEtC,QAAI,OAAO,OAAO;AAChB,cAAQ,IAAI,MAAM,IAAI,UAAU,OAAO,KAAK,EAAE,CAAC;AAC/C,cAAQ,IAAI,EAAE;AAAA,IAChB;AAEA,YAAQ;AAAA,MACN,WAAW,OAAO,cAAc,MAAM,MAAM,OAAO,WAAW,IAAI,MAAM,IAAI,WAAW,CAAC;AAAA,IAC1F;AACA,YAAQ;AAAA,MACN,oBAAoB,OAAO,YAAY,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IAChF;AACA,YAAQ;AAAA,MACN,qBAAqB,OAAO,mBAAmB,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,iCAAiC,CAAC;AAAA,IACrH;AAEA,QAAI,OAAO,YAAY;AACrB,cAAQ,IAAI,MAAM,KAAK,eAAe,OAAO,UAAU,EAAE,CAAC;AAAA,IAC5D;AAEA,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,IAAI,MAAM,KAAK,eAAe,CAAC;AACvC,cAAQ,IAAI,2BAA2B;AAAA,IACzC;AAAA,EACF,CAAC;AAEH,MACG,QAAQ,gBAAgB,EACxB,YAAY,8BAA8B,EAC1C,OAAO,uBAAuB,4CAA4C,EAC1E,OAAO,UAAU,wBAAwB,EACzC,OAAO,OAAO,MAAM,YAAY;AAC/B,UAAM,SAAS,MAAM,iBAAiB;AAEtC,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,MAAM,MAAM,IAAI,2BAA2B,CAAC;AACpD,cAAQ,IAAI,MAAM,KAAK,8BAA8B,CAAC;AACtD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,UAAM,aAAa,iBAAiB;AACpC,QAAI,CAAC,cAAc,CAAC,OAAO,aAAa;AACtC,cAAQ,MAAM,MAAM,IAAI,wCAAwC,CAAC;AACjE,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,UAAM,UAAU,IAAI,uBAAuB,EAAE,MAAM;AAEnD,QAAI,CAAC,OAAO,kBAAkB;AAC5B,cAAQ,OAAO;AAAA,IACjB;AAEA,UAAM,SAAS,MAAM,cAAc,MAAM,OAAO,aAAa,UAAU;AAEvE,QAAI,CAAC,OAAO,SAAS;AACnB,cAAQ;AAAA,QACN,MAAM,IAAI,sBAAsB,OAAO,WAAW,OAAO,KAAK,EAAE;AAAA,MAClE;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,YAAQ,QAAQ,MAAM,MAAM,qBAAqB,CAAC;AAElD,QAAI,QAAQ,MAAM;AAChB,cAAQ,IAAI,KAAK,UAAU,QAAQ,MAAM,CAAC,CAAC;AAC3C;AAAA,IACF;AAEA,YAAQ,IAAI,MAAM,KAAK,sBAAsB,CAAC;AAC9C,YAAQ,IAAI,MAAM,KAAK,SAAI,OAAO,EAAE,CAAC,CAAC;AACtC,YAAQ,IAAI,OAAO,iBAAiB;AACpC,YAAQ,IAAI,MAAM,KAAK,SAAI,OAAO,EAAE,CAAC,CAAC;AAEtC,QAAI,OAAO,YAAY;AACrB,cAAQ,IAAI,MAAM,KAAK,YAAY,OAAO,UAAU,IAAI,CAAC;AAAA,IAC3D;AACA,QAAI,OAAO,kBAAkB;AAC3B,cAAQ,IAAI,MAAM,KAAK,WAAW,OAAO,gBAAgB,EAAE,CAAC;AAAA,IAC9D;AAEA,QAAI,QAAQ,QAAQ;AAClB,YAAM,EAAE,cAAc,IAAI,MAAM,OAAO,IAAI;AAC3C,oBAAc,QAAQ,QAAQ,OAAO,qBAAqB,EAAE;AAC5D,cAAQ,IAAI,MAAM,MAAM;AAAA,cAAiB,QAAQ,MAAM,EAAE,CAAC;AAAA,IAC5D;AAAA,EACF,CAAC;AAEH,MAAI,OAAO,YAAY;AACrB,UAAM,SAAS,MAAM,iBAAiB;AACtC,YAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AAErD,YAAQ;AAAA,MACN,cAAc,OAAO,YAAY,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IAC1E;AACA,YAAQ;AAAA,MACN,gBAAgB,OAAO,mBAAmB,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IACnF;AAEA,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,IAAI,MAAM,KAAK,gCAAgC,CAAC;AAAA,IAC1D,OAAO;AACL,cAAQ,IAAI,MAAM,KAAK,2CAA2C,CAAC;AAAA,IACrE;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAEA,IAAO,gBAAQ,mBAAmB;",
-  "names": []
+  "sourcesContent": ["/**\n * Sweep command for StackMemory\n * Provides next-edit predictions using the Sweep 1.5B model\n *\n * Usage:\n *   stackmemory sweep setup          Install dependencies and optionally download model\n *   stackmemory sweep status         Check if Sweep addon is properly configured\n *   stackmemory sweep predict <file> Run prediction on a file\n */\n\nimport { Command } from 'commander';\nimport chalk from 'chalk';\nimport ora from 'ora';\nimport {\n  existsSync,\n  readFileSync,\n  writeFileSync,\n  mkdirSync,\n  copyFileSync,\n  chmodSync,\n} from 'fs';\nimport { join, dirname } from 'path';\nimport { fileURLToPath } from 'url';\nimport { spawn, execSync } from 'child_process';\n\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = dirname(__filename);\n\ninterface SweepStatus {\n  installed: boolean;\n  model_downloaded: boolean;\n  python_path?: string;\n  model_path?: string;\n  error?: string;\n}\n\ninterface SweepPredictResult {\n  success: boolean;\n  predicted_content?: string;\n  file_path?: string;\n  latency_ms?: number;\n  tokens_generated?: number;\n  error?: string;\n  message?: string;\n}\n\nfunction findPythonScript(): string | null {\n  const locations = [\n    join(\n      process.cwd(),\n      'packages',\n      'sweep-addon',\n      'python',\n      'sweep_predict.py'\n    ),\n    join(\n      process.cwd(),\n      'node_modules',\n      '@stackmemoryai',\n      'sweep-addon',\n      'python',\n      'sweep_predict.py'\n    ),\n    join(process.env.HOME || '', '.stackmemory', 'sweep', 'sweep_predict.py'),\n  ];\n\n  for (const loc of locations) {\n    if (existsSync(loc)) {\n      return loc;\n    }\n  }\n  return null;\n}\n\nfunction findHookSource(): string | null {\n  const locations = [\n    join(process.cwd(), 'templates', 'claude-hooks', 'post-edit-sweep.js'),\n    join(\n      process.cwd(),\n      'node_modules',\n      '@stackmemoryai',\n      'stackmemory',\n      'templates',\n      'claude-hooks',\n      'post-edit-sweep.js'\n    ),\n    join(\n      dirname(dirname(dirname(__dirname))),\n      'templates',\n      'claude-hooks',\n      'post-edit-sweep.js'\n    ),\n  ];\n\n  for (const loc of locations) {\n    if (existsSync(loc)) {\n      return loc;\n    }\n  }\n  return null;\n}\n\nasync function findPython(): Promise<string | null> {\n  const candidates = ['python3', 'python'];\n\n  for (const cmd of candidates) {\n    try {\n      execSync(`${cmd} --version`, { stdio: 'pipe' });\n      return cmd;\n    } catch {\n      continue;\n    }\n  }\n  return null;\n}\n\nasync function checkSweepStatus(): Promise<SweepStatus> {\n  const pythonPath = await findPython();\n  if (!pythonPath) {\n    return {\n      installed: false,\n      model_downloaded: false,\n      error: 'Python not found. Install Python 3.10+',\n    };\n  }\n\n  const scriptPath = findPythonScript();\n  if (!scriptPath) {\n    return {\n      installed: false,\n      model_downloaded: false,\n      python_path: pythonPath,\n      error: 'Sweep addon not installed. Run: stackmemory sweep setup',\n    };\n  }\n\n  const homeDir = process.env.HOME || '';\n  const modelPath = join(\n    homeDir,\n    '.stackmemory',\n    'models',\n    'sweep',\n    'sweep-next-edit-1.5b.q8_0.v2.gguf'\n  );\n  const modelDownloaded = existsSync(modelPath);\n\n  return {\n    installed: true,\n    model_downloaded: modelDownloaded,\n    python_path: pythonPath,\n    model_path: modelDownloaded ? modelPath : undefined,\n  };\n}\n\nasync function runPrediction(\n  filePath: string,\n  pythonPath: string,\n  scriptPath: string\n): Promise<SweepPredictResult> {\n  if (!existsSync(filePath)) {\n    return {\n      success: false,\n      error: 'file_not_found',\n      message: `File not found: ${filePath}`,\n    };\n  }\n\n  const currentContent = readFileSync(filePath, 'utf-8');\n\n  const input = {\n    file_path: filePath,\n    current_content: currentContent,\n  };\n\n  return new Promise((resolve) => {\n    const proc = spawn(pythonPath, [scriptPath], {\n      stdio: ['pipe', 'pipe', 'pipe'],\n    });\n\n    let stdout = '';\n    let stderr = '';\n\n    proc.stdout.on('data', (data) => (stdout += data));\n    proc.stderr.on('data', (data) => (stderr += data));\n\n    proc.on('close', (code) => {\n      try {\n        if (stdout.trim()) {\n          const result = JSON.parse(stdout.trim());\n          resolve(result);\n        } else if (code !== 0) {\n          resolve({\n            success: false,\n            error: 'process_error',\n            message: stderr || `Process exited with code ${code}`,\n          });\n        } else {\n          resolve({\n            success: false,\n            error: 'no_output',\n            message: 'No output from prediction script',\n          });\n        }\n      } catch {\n        resolve({\n          success: false,\n          error: 'parse_error',\n          message: `Failed to parse output: ${stdout}`,\n        });\n      }\n    });\n\n    proc.on('error', (error) => {\n      resolve({\n        success: false,\n        error: 'spawn_error',\n        message: error.message,\n      });\n    });\n\n    proc.stdin.write(JSON.stringify(input));\n    proc.stdin.end();\n  });\n}\n\nexport function createSweepCommand(): Command {\n  const cmd = new Command('sweep')\n    .description(\n      'Next-edit predictions using Sweep 1.5B model (optional addon)'\n    )\n    .addHelpText(\n      'after',\n      `\nExamples:\n  stackmemory sweep setup              Install Python dependencies\n  stackmemory sweep setup --download   Also download the model (1.5GB)\n  stackmemory sweep status             Check addon status\n  stackmemory sweep predict src/app.ts Predict next edit for a file\n\nRequirements:\n  - Python 3.10+\n  - pip packages: huggingface_hub, llama-cpp-python\n\nThe Sweep 1.5B model predicts what code changes you'll make next based on:\n  - Current file content\n  - Recent changes (diffs)\n  - Context from other files\n\nModel is downloaded from HuggingFace on first prediction (~1.5GB).\n`\n    );\n\n  cmd\n    .command('setup')\n    .description('Install Python dependencies for Sweep addon')\n    .option('--download', 'Also download the model now')\n    .action(async (options) => {\n      const spinner = ora('Checking Python...').start();\n\n      const pythonPath = await findPython();\n      if (!pythonPath) {\n        spinner.fail(chalk.red('Python not found'));\n        console.log(chalk.gray('Please install Python 3.10+'));\n        process.exit(1);\n      }\n\n      spinner.text = 'Installing Python dependencies...';\n\n      try {\n        execSync(\n          `${pythonPath} -m pip install --quiet huggingface_hub llama-cpp-python`,\n          {\n            stdio: 'pipe',\n          }\n        );\n        spinner.succeed(chalk.green('Python dependencies installed'));\n      } catch {\n        spinner.fail(chalk.red('Failed to install dependencies'));\n        console.log(\n          chalk.gray(\n            `Run: ${pythonPath} -m pip install huggingface_hub llama-cpp-python`\n          )\n        );\n        process.exit(1);\n      }\n\n      if (options.download) {\n        const downloadSpinner = ora('Downloading Sweep 1.5B model...').start();\n        downloadSpinner.text = 'Downloading model from HuggingFace (~1.5GB)...';\n\n        try {\n          execSync(\n            `${pythonPath} -c \"\nfrom huggingface_hub import hf_hub_download\nimport os\nmodel_dir = os.path.expanduser('~/.stackmemory/models/sweep')\nos.makedirs(model_dir, exist_ok=True)\nhf_hub_download(\n    repo_id='sweepai/sweep-next-edit-1.5B',\n    filename='sweep-next-edit-1.5b.q8_0.v2.gguf',\n    repo_type='model',\n    local_dir=model_dir,\n    local_dir_use_symlinks=False\n)\n\"`,\n            { stdio: 'pipe', timeout: 600000 }\n          );\n          downloadSpinner.succeed(chalk.green('Model downloaded'));\n        } catch {\n          downloadSpinner.fail(chalk.red('Model download failed'));\n          console.log(chalk.gray('Model will be downloaded on first use'));\n        }\n      } else {\n        console.log(\n          chalk.gray('\\nModel will be downloaded on first prediction (~1.5GB)')\n        );\n        console.log(chalk.gray('Or run: stackmemory sweep setup --download'));\n      }\n\n      console.log(chalk.bold('\\nSetup complete!'));\n    });\n\n  cmd\n    .command('status')\n    .description('Check Sweep addon status')\n    .action(async () => {\n      console.log(chalk.bold('\\nSweep 1.5B Addon Status\\n'));\n\n      const status = await checkSweepStatus();\n\n      if (status.error) {\n        console.log(chalk.red(`Error: ${status.error}`));\n        console.log('');\n      }\n\n      console.log(\n        `Python: ${status.python_path ? chalk.green(status.python_path) : chalk.red('Not found')}`\n      );\n      console.log(\n        `Addon installed: ${status.installed ? chalk.green('Yes') : chalk.yellow('No')}`\n      );\n      console.log(\n        `Model downloaded: ${status.model_downloaded ? chalk.green('Yes') : chalk.yellow('No (will download on first use)')}`\n      );\n\n      if (status.model_path) {\n        console.log(chalk.gray(`Model path: ${status.model_path}`));\n      }\n\n      if (!status.installed) {\n        console.log(chalk.bold('\\nTo install:'));\n        console.log('  stackmemory sweep setup');\n      }\n    });\n\n  cmd\n    .command('predict <file>')\n    .description('Predict next edit for a file')\n    .option('-o, --output <path>', 'Write prediction to file instead of stdout')\n    .option('--json', 'Output raw JSON result')\n    .action(async (file, options) => {\n      const status = await checkSweepStatus();\n\n      if (!status.installed) {\n        console.error(chalk.red('Sweep addon not installed'));\n        console.log(chalk.gray('Run: stackmemory sweep setup'));\n        process.exit(1);\n      }\n\n      const scriptPath = findPythonScript();\n      if (!scriptPath || !status.python_path) {\n        console.error(chalk.red('Could not find Sweep prediction script'));\n        process.exit(1);\n      }\n\n      const spinner = ora('Running prediction...').start();\n\n      if (!status.model_downloaded) {\n        spinner.text = 'Downloading model (first time only, ~1.5GB)...';\n      }\n\n      const result = await runPrediction(file, status.python_path, scriptPath);\n\n      if (!result.success) {\n        spinner.fail(\n          chalk.red(`Prediction failed: ${result.message || result.error}`)\n        );\n        process.exit(1);\n      }\n\n      spinner.succeed(chalk.green('Prediction complete'));\n\n      if (options.json) {\n        console.log(JSON.stringify(result, null, 2));\n        return;\n      }\n\n      console.log(chalk.bold('\\nPredicted content:'));\n      console.log(chalk.gray('\u2500'.repeat(50)));\n      console.log(result.predicted_content);\n      console.log(chalk.gray('\u2500'.repeat(50)));\n\n      if (result.latency_ms) {\n        console.log(chalk.gray(`Latency: ${result.latency_ms}ms`));\n      }\n      if (result.tokens_generated) {\n        console.log(chalk.gray(`Tokens: ${result.tokens_generated}`));\n      }\n\n      if (options.output) {\n        const { writeFileSync } = await import('fs');\n        writeFileSync(options.output, result.predicted_content || '');\n        console.log(chalk.green(`\\nWritten to: ${options.output}`));\n      }\n    });\n\n  const hookCmd = cmd\n    .command('hook')\n    .description('Manage Claude Code integration hook');\n\n  hookCmd\n    .command('install')\n    .description('Install Sweep prediction hook for Claude Code')\n    .action(async () => {\n      const spinner = ora('Installing Sweep hook...').start();\n\n      const homeDir = process.env.HOME || '';\n      const hookDir = join(homeDir, '.claude', 'hooks');\n      const sweepDir = join(homeDir, '.stackmemory', 'sweep');\n      const hooksJsonPath = join(homeDir, '.claude', 'hooks.json');\n\n      try {\n        mkdirSync(hookDir, { recursive: true });\n        mkdirSync(sweepDir, { recursive: true });\n\n        const hookSource = findHookSource();\n        if (!hookSource) {\n          spinner.fail(chalk.red('Hook template not found'));\n          console.log(\n            chalk.gray('Ensure stackmemory is installed from the repository')\n          );\n          process.exit(1);\n        }\n\n        const hookDest = join(hookDir, 'post-edit-sweep.js');\n        copyFileSync(hookSource, hookDest);\n        chmodSync(hookDest, '755');\n\n        const pythonScriptSource = findPythonScript();\n        if (pythonScriptSource) {\n          const pythonDest = join(sweepDir, 'sweep_predict.py');\n          copyFileSync(pythonScriptSource, pythonDest);\n        }\n\n        if (existsSync(hooksJsonPath)) {\n          const hooks = JSON.parse(readFileSync(hooksJsonPath, 'utf-8'));\n          if (!hooks['post-tool-use']) {\n            hooks['post-tool-use'] = hookDest;\n            writeFileSync(hooksJsonPath, JSON.stringify(hooks, null, 2));\n          } else if (!hooks['post-tool-use'].includes('sweep')) {\n            spinner.warn(chalk.yellow('post-tool-use hook already configured'));\n            console.log(chalk.gray(`Existing: ${hooks['post-tool-use']}`));\n            console.log(chalk.gray(`Hook installed at: ${hookDest}`));\n            console.log(\n              chalk.gray('You may need to manually configure the hook chain')\n            );\n            return;\n          }\n        } else {\n          const hooks = { 'post-tool-use': hookDest };\n          writeFileSync(hooksJsonPath, JSON.stringify(hooks, null, 2));\n        }\n\n        spinner.succeed(chalk.green('Sweep hook installed'));\n        console.log(chalk.gray(`Hook: ${hookDest}`));\n        console.log(chalk.gray(`Config: ${hooksJsonPath}`));\n        console.log('');\n        console.log(chalk.bold('Usage:'));\n        console.log('  Hook runs automatically after Edit/Write operations');\n        console.log('  Predictions appear after 2+ edits in session');\n        console.log('  Disable: export SWEEP_ENABLED=false');\n      } catch (error) {\n        spinner.fail(chalk.red('Installation failed'));\n        console.log(chalk.gray((error as Error).message));\n        process.exit(1);\n      }\n    });\n\n  hookCmd\n    .command('status')\n    .description('Check hook installation status')\n    .action(async () => {\n      const homeDir = process.env.HOME || '';\n      const hookPath = join(homeDir, '.claude', 'hooks', 'post-edit-sweep.js');\n      const hooksJsonPath = join(homeDir, '.claude', 'hooks.json');\n      const statePath = join(homeDir, '.stackmemory', 'sweep-state.json');\n\n      console.log(chalk.bold('\\nSweep Hook Status\\n'));\n\n      const hookInstalled = existsSync(hookPath);\n      console.log(\n        `Hook installed: ${hookInstalled ? chalk.green('Yes') : chalk.yellow('No')}`\n      );\n\n      if (existsSync(hooksJsonPath)) {\n        const hooks = JSON.parse(readFileSync(hooksJsonPath, 'utf-8'));\n        const configured =\n          hooks['post-tool-use'] && hooks['post-tool-use'].includes('sweep');\n        console.log(\n          `Hook configured: ${configured ? chalk.green('Yes') : chalk.yellow('No')}`\n        );\n      } else {\n        console.log(`Hook configured: ${chalk.yellow('No hooks.json')}`);\n      }\n\n      const enabled = process.env.SWEEP_ENABLED !== 'false';\n      console.log(\n        `Enabled: ${enabled ? chalk.green('Yes') : chalk.yellow('Disabled (SWEEP_ENABLED=false)')}`\n      );\n\n      if (existsSync(statePath)) {\n        try {\n          const state = JSON.parse(readFileSync(statePath, 'utf-8'));\n          console.log(\n            chalk.gray(\n              `\\nRecent diffs tracked: ${state.recentDiffs?.length || 0}`\n            )\n          );\n          if (state.lastPrediction) {\n            const age = Date.now() - state.lastPrediction.timestamp;\n            const ageStr =\n              age < 60000\n                ? `${Math.round(age / 1000)}s ago`\n                : `${Math.round(age / 60000)}m ago`;\n            console.log(chalk.gray(`Last prediction: ${ageStr}`));\n          }\n        } catch {\n          // Ignore parse errors\n        }\n      }\n\n      if (!hookInstalled) {\n        console.log(chalk.bold('\\nTo install: stackmemory sweep hook install'));\n      }\n    });\n\n  hookCmd\n    .command('disable')\n    .description('Disable the Sweep hook')\n    .action(() => {\n      console.log(chalk.bold('\\nTo disable Sweep predictions:\\n'));\n      console.log('  Temporarily: export SWEEP_ENABLED=false');\n      console.log('  Permanently: Add to ~/.zshrc or ~/.bashrc');\n      console.log('');\n      console.log('Or remove the hook:');\n      console.log('  rm ~/.claude/hooks/post-edit-sweep.js');\n    });\n\n  hookCmd\n    .command('clear')\n    .description('Clear hook state (recent diffs and predictions)')\n    .action(() => {\n      const homeDir = process.env.HOME || '';\n      const statePath = join(homeDir, '.stackmemory', 'sweep-state.json');\n\n      if (existsSync(statePath)) {\n        writeFileSync(\n          statePath,\n          JSON.stringify(\n            {\n              recentDiffs: [],\n              lastPrediction: null,\n              pendingPrediction: null,\n              fileContents: {},\n            },\n            null,\n            2\n          )\n        );\n        console.log(chalk.green('Sweep state cleared'));\n      } else {\n        console.log(chalk.gray('No state file found'));\n      }\n    });\n\n  cmd.action(async () => {\n    const status = await checkSweepStatus();\n    console.log(chalk.bold('\\nSweep 1.5B Addon Status\\n'));\n\n    console.log(\n      `Installed: ${status.installed ? chalk.green('Yes') : chalk.yellow('No')}`\n    );\n    console.log(\n      `Model ready: ${status.model_downloaded ? chalk.green('Yes') : chalk.yellow('No')}`\n    );\n\n    if (!status.installed) {\n      console.log(chalk.bold('\\nRun: stackmemory sweep setup'));\n    } else {\n      console.log(chalk.bold('\\nUsage: stackmemory sweep predict <file>'));\n    }\n  });\n\n  return cmd;\n}\n\nexport default createSweepCommand();\n"],
+  "mappings": "AAUA,SAAS,eAAe;AACxB,OAAO,WAAW;AAClB,OAAO,SAAS;AAChB;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,MAAM,eAAe;AAC9B,SAAS,qBAAqB;AAC9B,SAAS,OAAO,gBAAgB;AAEhC,MAAM,aAAa,cAAc,YAAY,GAAG;AAChD,MAAM,YAAY,QAAQ,UAAU;AAoBpC,SAAS,mBAAkC;AACzC,QAAM,YAAY;AAAA,IAChB;AAAA,MACE,QAAQ,IAAI;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA;AAAA,MACE,QAAQ,IAAI;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,KAAK,QAAQ,IAAI,QAAQ,IAAI,gBAAgB,SAAS,kBAAkB;AAAA,EAC1E;AAEA,aAAW,OAAO,WAAW;AAC3B,QAAI,WAAW,GAAG,GAAG;AACnB,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,iBAAgC;AACvC,QAAM,YAAY;AAAA,IAChB,KAAK,QAAQ,IAAI,GAAG,aAAa,gBAAgB,oBAAoB;AAAA,IACrE;AAAA,MACE,QAAQ,IAAI;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA;AAAA,MACE,QAAQ,QAAQ,QAAQ,SAAS,CAAC,CAAC;AAAA,MACnC;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,aAAW,OAAO,WAAW;AAC3B,QAAI,WAAW,GAAG,GAAG;AACnB,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAe,aAAqC;AAClD,QAAM,aAAa,CAAC,WAAW,QAAQ;AAEvC,aAAW,OAAO,YAAY;AAC5B,QAAI;AACF,eAAS,GAAG,GAAG,cAAc,EAAE,OAAO,OAAO,CAAC;AAC9C,aAAO;AAAA,IACT,QAAQ;AACN;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAe,mBAAyC;AACtD,QAAM,aAAa,MAAM,WAAW;AACpC,MAAI,CAAC,YAAY;AACf,WAAO;AAAA,MACL,WAAW;AAAA,MACX,kBAAkB;AAAA,MAClB,OAAO;AAAA,IACT;AAAA,EACF;AAEA,QAAM,aAAa,iBAAiB;AACpC,MAAI,CAAC,YAAY;AACf,WAAO;AAAA,MACL,WAAW;AAAA,MACX,kBAAkB;AAAA,MAClB,aAAa;AAAA,MACb,OAAO;AAAA,IACT;AAAA,EACF;AAEA,QAAM,UAAU,QAAQ,IAAI,QAAQ;AACpC,QAAM,YAAY;AAAA,IAChB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,QAAM,kBAAkB,WAAW,SAAS;AAE5C,SAAO;AAAA,IACL,WAAW;AAAA,IACX,kBAAkB;AAAA,IAClB,aAAa;AAAA,IACb,YAAY,kBAAkB,YAAY;AAAA,EAC5C;AACF;AAEA,eAAe,cACb,UACA,YACA,YAC6B;AAC7B,MAAI,CAAC,WAAW,QAAQ,GAAG;AACzB,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO;AAAA,MACP,SAAS,mBAAmB,QAAQ;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,iBAAiB,aAAa,UAAU,OAAO;AAErD,QAAM,QAAQ;AAAA,IACZ,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAEA,SAAO,IAAI,QAAQ,CAAC,YAAY;AAC9B,UAAM,OAAO,MAAM,YAAY,CAAC,UAAU,GAAG;AAAA,MAC3C,OAAO,CAAC,QAAQ,QAAQ,MAAM;AAAA,IAChC,CAAC;AAED,QAAI,SAAS;AACb,QAAI,SAAS;AAEb,SAAK,OAAO,GAAG,QAAQ,CAAC,SAAU,UAAU,IAAK;AACjD,SAAK,OAAO,GAAG,QAAQ,CAAC,SAAU,UAAU,IAAK;AAEjD,SAAK,GAAG,SAAS,CAAC,SAAS;AACzB,UAAI;AACF,YAAI,OAAO,KAAK,GAAG;AACjB,gBAAM,SAAS,KAAK,MAAM,OAAO,KAAK,CAAC;AACvC,kBAAQ,MAAM;AAAA,QAChB,WAAW,SAAS,GAAG;AACrB,kBAAQ;AAAA,YACN,SAAS;AAAA,YACT,OAAO;AAAA,YACP,SAAS,UAAU,4BAA4B,IAAI;AAAA,UACrD,CAAC;AAAA,QACH,OAAO;AACL,kBAAQ;AAAA,YACN,SAAS;AAAA,YACT,OAAO;AAAA,YACP,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF,QAAQ;AACN,gBAAQ;AAAA,UACN,SAAS;AAAA,UACT,OAAO;AAAA,UACP,SAAS,2BAA2B,MAAM;AAAA,QAC5C,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,SAAK,GAAG,SAAS,CAAC,UAAU;AAC1B,cAAQ;AAAA,QACN,SAAS;AAAA,QACT,OAAO;AAAA,QACP,SAAS,MAAM;AAAA,MACjB,CAAC;AAAA,IACH,CAAC;AAED,SAAK,MAAM,MAAM,KAAK,UAAU,KAAK,CAAC;AACtC,SAAK,MAAM,IAAI;AAAA,EACjB,CAAC;AACH;AAEO,SAAS,qBAA8B;AAC5C,QAAM,MAAM,IAAI,QAAQ,OAAO,EAC5B;AAAA,IACC;AAAA,EACF,EACC;AAAA,IACC;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBF;AAEF,MACG,QAAQ,OAAO,EACf,YAAY,6CAA6C,EACzD,OAAO,cAAc,6BAA6B,EAClD,OAAO,OAAO,YAAY;AACzB,UAAM,UAAU,IAAI,oBAAoB,EAAE,MAAM;AAEhD,UAAM,aAAa,MAAM,WAAW;AACpC,QAAI,CAAC,YAAY;AACf,cAAQ,KAAK,MAAM,IAAI,kBAAkB,CAAC;AAC1C,cAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AACrD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,YAAQ,OAAO;AAEf,QAAI;AACF;AAAA,QACE,GAAG,UAAU;AAAA,QACb;AAAA,UACE,OAAO;AAAA,QACT;AAAA,MACF;AACA,cAAQ,QAAQ,MAAM,MAAM,+BAA+B,CAAC;AAAA,IAC9D,QAAQ;AACN,cAAQ,KAAK,MAAM,IAAI,gCAAgC,CAAC;AACxD,cAAQ;AAAA,QACN,MAAM;AAAA,UACJ,QAAQ,UAAU;AAAA,QACpB;AAAA,MACF;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,QAAI,QAAQ,UAAU;AACpB,YAAM,kBAAkB,IAAI,iCAAiC,EAAE,MAAM;AACrE,sBAAgB,OAAO;AAEvB,UAAI;AACF;AAAA,UACE,GAAG,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,UAab,EAAE,OAAO,QAAQ,SAAS,IAAO;AAAA,QACnC;AACA,wBAAgB,QAAQ,MAAM,MAAM,kBAAkB,CAAC;AAAA,MACzD,QAAQ;AACN,wBAAgB,KAAK,MAAM,IAAI,uBAAuB,CAAC;AACvD,gBAAQ,IAAI,MAAM,KAAK,uCAAuC,CAAC;AAAA,MACjE;AAAA,IACF,OAAO;AACL,cAAQ;AAAA,QACN,MAAM,KAAK,yDAAyD;AAAA,MACtE;AACA,cAAQ,IAAI,MAAM,KAAK,4CAA4C,CAAC;AAAA,IACtE;AAEA,YAAQ,IAAI,MAAM,KAAK,mBAAmB,CAAC;AAAA,EAC7C,CAAC;AAEH,MACG,QAAQ,QAAQ,EAChB,YAAY,0BAA0B,EACtC,OAAO,YAAY;AAClB,YAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AAErD,UAAM,SAAS,MAAM,iBAAiB;AAEtC,QAAI,OAAO,OAAO;AAChB,cAAQ,IAAI,MAAM,IAAI,UAAU,OAAO,KAAK,EAAE,CAAC;AAC/C,cAAQ,IAAI,EAAE;AAAA,IAChB;AAEA,YAAQ;AAAA,MACN,WAAW,OAAO,cAAc,MAAM,MAAM,OAAO,WAAW,IAAI,MAAM,IAAI,WAAW,CAAC;AAAA,IAC1F;AACA,YAAQ;AAAA,MACN,oBAAoB,OAAO,YAAY,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IAChF;AACA,YAAQ;AAAA,MACN,qBAAqB,OAAO,mBAAmB,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,iCAAiC,CAAC;AAAA,IACrH;AAEA,QAAI,OAAO,YAAY;AACrB,cAAQ,IAAI,MAAM,KAAK,eAAe,OAAO,UAAU,EAAE,CAAC;AAAA,IAC5D;AAEA,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,IAAI,MAAM,KAAK,eAAe,CAAC;AACvC,cAAQ,IAAI,2BAA2B;AAAA,IACzC;AAAA,EACF,CAAC;AAEH,MACG,QAAQ,gBAAgB,EACxB,YAAY,8BAA8B,EAC1C,OAAO,uBAAuB,4CAA4C,EAC1E,OAAO,UAAU,wBAAwB,EACzC,OAAO,OAAO,MAAM,YAAY;AAC/B,UAAM,SAAS,MAAM,iBAAiB;AAEtC,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,MAAM,MAAM,IAAI,2BAA2B,CAAC;AACpD,cAAQ,IAAI,MAAM,KAAK,8BAA8B,CAAC;AACtD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,UAAM,aAAa,iBAAiB;AACpC,QAAI,CAAC,cAAc,CAAC,OAAO,aAAa;AACtC,cAAQ,MAAM,MAAM,IAAI,wCAAwC,CAAC;AACjE,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,UAAM,UAAU,IAAI,uBAAuB,EAAE,MAAM;AAEnD,QAAI,CAAC,OAAO,kBAAkB;AAC5B,cAAQ,OAAO;AAAA,IACjB;AAEA,UAAM,SAAS,MAAM,cAAc,MAAM,OAAO,aAAa,UAAU;AAEvE,QAAI,CAAC,OAAO,SAAS;AACnB,cAAQ;AAAA,QACN,MAAM,IAAI,sBAAsB,OAAO,WAAW,OAAO,KAAK,EAAE;AAAA,MAClE;AACA,cAAQ,KAAK,CAAC;AAAA,IAChB;AAEA,YAAQ,QAAQ,MAAM,MAAM,qBAAqB,CAAC;AAElD,QAAI,QAAQ,MAAM;AAChB,cAAQ,IAAI,KAAK,UAAU,QAAQ,MAAM,CAAC,CAAC;AAC3C;AAAA,IACF;AAEA,YAAQ,IAAI,MAAM,KAAK,sBAAsB,CAAC;AAC9C,YAAQ,IAAI,MAAM,KAAK,SAAI,OAAO,EAAE,CAAC,CAAC;AACtC,YAAQ,IAAI,OAAO,iBAAiB;AACpC,YAAQ,IAAI,MAAM,KAAK,SAAI,OAAO,EAAE,CAAC,CAAC;AAEtC,QAAI,OAAO,YAAY;AACrB,cAAQ,IAAI,MAAM,KAAK,YAAY,OAAO,UAAU,IAAI,CAAC;AAAA,IAC3D;AACA,QAAI,OAAO,kBAAkB;AAC3B,cAAQ,IAAI,MAAM,KAAK,WAAW,OAAO,gBAAgB,EAAE,CAAC;AAAA,IAC9D;AAEA,QAAI,QAAQ,QAAQ;AAClB,YAAM,EAAE,eAAAA,eAAc,IAAI,MAAM,OAAO,IAAI;AAC3C,MAAAA,eAAc,QAAQ,QAAQ,OAAO,qBAAqB,EAAE;AAC5D,cAAQ,IAAI,MAAM,MAAM;AAAA,cAAiB,QAAQ,MAAM,EAAE,CAAC;AAAA,IAC5D;AAAA,EACF,CAAC;AAEH,QAAM,UAAU,IACb,QAAQ,MAAM,EACd,YAAY,qCAAqC;AAEpD,UACG,QAAQ,SAAS,EACjB,YAAY,+CAA+C,EAC3D,OAAO,YAAY;AAClB,UAAM,UAAU,IAAI,0BAA0B,EAAE,MAAM;AAEtD,UAAM,UAAU,QAAQ,IAAI,QAAQ;AACpC,UAAM,UAAU,KAAK,SAAS,WAAW,OAAO;AAChD,UAAM,WAAW,KAAK,SAAS,gBAAgB,OAAO;AACtD,UAAM,gBAAgB,KAAK,SAAS,WAAW,YAAY;AAE3D,QAAI;AACF,gBAAU,SAAS,EAAE,WAAW,KAAK,CAAC;AACtC,gBAAU,UAAU,EAAE,WAAW,KAAK,CAAC;AAEvC,YAAM,aAAa,eAAe;AAClC,UAAI,CAAC,YAAY;AACf,gBAAQ,KAAK,MAAM,IAAI,yBAAyB,CAAC;AACjD,gBAAQ;AAAA,UACN,MAAM,KAAK,qDAAqD;AAAA,QAClE;AACA,gBAAQ,KAAK,CAAC;AAAA,MAChB;AAEA,YAAM,WAAW,KAAK,SAAS,oBAAoB;AACnD,mBAAa,YAAY,QAAQ;AACjC,gBAAU,UAAU,KAAK;AAEzB,YAAM,qBAAqB,iBAAiB;AAC5C,UAAI,oBAAoB;AACtB,cAAM,aAAa,KAAK,UAAU,kBAAkB;AACpD,qBAAa,oBAAoB,UAAU;AAAA,MAC7C;AAEA,UAAI,WAAW,aAAa,GAAG;AAC7B,cAAM,QAAQ,KAAK,MAAM,aAAa,eAAe,OAAO,CAAC;AAC7D,YAAI,CAAC,MAAM,eAAe,GAAG;AAC3B,gBAAM,eAAe,IAAI;AACzB,wBAAc,eAAe,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAAA,QAC7D,WAAW,CAAC,MAAM,eAAe,EAAE,SAAS,OAAO,GAAG;AACpD,kBAAQ,KAAK,MAAM,OAAO,uCAAuC,CAAC;AAClE,kBAAQ,IAAI,MAAM,KAAK,aAAa,MAAM,eAAe,CAAC,EAAE,CAAC;AAC7D,kBAAQ,IAAI,MAAM,KAAK,sBAAsB,QAAQ,EAAE,CAAC;AACxD,kBAAQ;AAAA,YACN,MAAM,KAAK,mDAAmD;AAAA,UAChE;AACA;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,QAAQ,EAAE,iBAAiB,SAAS;AAC1C,sBAAc,eAAe,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAAA,MAC7D;AAEA,cAAQ,QAAQ,MAAM,MAAM,sBAAsB,CAAC;AACnD,cAAQ,IAAI,MAAM,KAAK,SAAS,QAAQ,EAAE,CAAC;AAC3C,cAAQ,IAAI,MAAM,KAAK,WAAW,aAAa,EAAE,CAAC;AAClD,cAAQ,IAAI,EAAE;AACd,cAAQ,IAAI,MAAM,KAAK,QAAQ,CAAC;AAChC,cAAQ,IAAI,uDAAuD;AACnE,cAAQ,IAAI,gDAAgD;AAC5D,cAAQ,IAAI,uCAAuC;AAAA,IACrD,SAAS,OAAO;AACd,cAAQ,KAAK,MAAM,IAAI,qBAAqB,CAAC;AAC7C,cAAQ,IAAI,MAAM,KAAM,MAAgB,OAAO,CAAC;AAChD,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF,CAAC;AAEH,UACG,QAAQ,QAAQ,EAChB,YAAY,gCAAgC,EAC5C,OAAO,YAAY;AAClB,UAAM,UAAU,QAAQ,IAAI,QAAQ;AACpC,UAAM,WAAW,KAAK,SAAS,WAAW,SAAS,oBAAoB;AACvE,UAAM,gBAAgB,KAAK,SAAS,WAAW,YAAY;AAC3D,UAAM,YAAY,KAAK,SAAS,gBAAgB,kBAAkB;AAElE,YAAQ,IAAI,MAAM,KAAK,uBAAuB,CAAC;AAE/C,UAAM,gBAAgB,WAAW,QAAQ;AACzC,YAAQ;AAAA,MACN,mBAAmB,gBAAgB,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IAC5E;AAEA,QAAI,WAAW,aAAa,GAAG;AAC7B,YAAM,QAAQ,KAAK,MAAM,aAAa,eAAe,OAAO,CAAC;AAC7D,YAAM,aACJ,MAAM,eAAe,KAAK,MAAM,eAAe,EAAE,SAAS,OAAO;AACnE,cAAQ;AAAA,QACN,oBAAoB,aAAa,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,MAC1E;AAAA,IACF,OAAO;AACL,cAAQ,IAAI,oBAAoB,MAAM,OAAO,eAAe,CAAC,EAAE;AAAA,IACjE;AAEA,UAAM,UAAU,QAAQ,IAAI,kBAAkB;AAC9C,YAAQ;AAAA,MACN,YAAY,UAAU,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,gCAAgC,CAAC;AAAA,IAC3F;AAEA,QAAI,WAAW,SAAS,GAAG;AACzB,UAAI;AACF,cAAM,QAAQ,KAAK,MAAM,aAAa,WAAW,OAAO,CAAC;AACzD,gBAAQ;AAAA,UACN,MAAM;AAAA,YACJ;AAAA,wBAA2B,MAAM,aAAa,UAAU,CAAC;AAAA,UAC3D;AAAA,QACF;AACA,YAAI,MAAM,gBAAgB;AACxB,gBAAM,MAAM,KAAK,IAAI,IAAI,MAAM,eAAe;AAC9C,gBAAM,SACJ,MAAM,MACF,GAAG,KAAK,MAAM,MAAM,GAAI,CAAC,UACzB,GAAG,KAAK,MAAM,MAAM,GAAK,CAAC;AAChC,kBAAQ,IAAI,MAAM,KAAK,oBAAoB,MAAM,EAAE,CAAC;AAAA,QACtD;AAAA,MACF,QAAQ;AAAA,MAER;AAAA,IACF;AAEA,QAAI,CAAC,eAAe;AAClB,cAAQ,IAAI,MAAM,KAAK,8CAA8C,CAAC;AAAA,IACxE;AAAA,EACF,CAAC;AAEH,UACG,QAAQ,SAAS,EACjB,YAAY,wBAAwB,EACpC,OAAO,MAAM;AACZ,YAAQ,IAAI,MAAM,KAAK,mCAAmC,CAAC;AAC3D,YAAQ,IAAI,2CAA2C;AACvD,YAAQ,IAAI,6CAA6C;AACzD,YAAQ,IAAI,EAAE;AACd,YAAQ,IAAI,qBAAqB;AACjC,YAAQ,IAAI,yCAAyC;AAAA,EACvD,CAAC;AAEH,UACG,QAAQ,OAAO,EACf,YAAY,iDAAiD,EAC7D,OAAO,MAAM;AACZ,UAAM,UAAU,QAAQ,IAAI,QAAQ;AACpC,UAAM,YAAY,KAAK,SAAS,gBAAgB,kBAAkB;AAElE,QAAI,WAAW,SAAS,GAAG;AACzB;AAAA,QACE;AAAA,QACA,KAAK;AAAA,UACH;AAAA,YACE,aAAa,CAAC;AAAA,YACd,gBAAgB;AAAA,YAChB,mBAAmB;AAAA,YACnB,cAAc,CAAC;AAAA,UACjB;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF;AACA,cAAQ,IAAI,MAAM,MAAM,qBAAqB,CAAC;AAAA,IAChD,OAAO;AACL,cAAQ,IAAI,MAAM,KAAK,qBAAqB,CAAC;AAAA,IAC/C;AAAA,EACF,CAAC;AAEH,MAAI,OAAO,YAAY;AACrB,UAAM,SAAS,MAAM,iBAAiB;AACtC,YAAQ,IAAI,MAAM,KAAK,6BAA6B,CAAC;AAErD,YAAQ;AAAA,MACN,cAAc,OAAO,YAAY,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IAC1E;AACA,YAAQ;AAAA,MACN,gBAAgB,OAAO,mBAAmB,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO,IAAI,CAAC;AAAA,IACnF;AAEA,QAAI,CAAC,OAAO,WAAW;AACrB,cAAQ,IAAI,MAAM,KAAK,gCAAgC,CAAC;AAAA,IAC1D,OAAO;AACL,cAAQ,IAAI,MAAM,KAAK,2CAA2C,CAAC;AAAA,IACrE;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAEA,IAAO,gBAAQ,mBAAmB;",
+  "names": ["writeFileSync"]
 }

package/dist/cli/index.js

@@ -49,7 +49,7 @@ import {
   enableChromaDB,
   getStorageModeDescription
 } from "../core/config/storage-config.js";
-const VERSION = "0.5.2";
+const VERSION = "0.5.3";
 UpdateChecker.checkForUpdates(VERSION, true).catch(() => {
 });
 program.name("stackmemory").description(

package/dist/cli/index.js.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
   "sources": ["../../src/cli/index.ts"],
-  "sourcesContent": ["#!/usr/bin/env node\n/**\n * StackMemory CLI\n * Command-line interface for StackMemory operations\n */\n\n// Set environment flag for CLI usage to skip async context bridge\nprocess.env['STACKMEMORY_CLI'] = 'true';\n\n// Load environment variables\nimport 'dotenv/config';\n\n// Initialize tracing system early\nimport { initializeTracing, trace } from '../core/trace/index.js';\ninitializeTracing();\n\nimport { program } from 'commander';\nimport { logger } from '../core/monitoring/logger.js';\nimport { FrameManager } from '../core/context/frame-manager.js';\nimport { sessionManager, FrameQueryMode } from '../core/session/index.js';\nimport { sharedContextLayer } from '../core/context/shared-context-layer.js';\nimport { UpdateChecker } from '../core/utils/update-checker.js';\nimport { ProgressTracker } from '../core/monitoring/progress-tracker.js';\nimport { registerProjectCommands } from './commands/projects.js';\nimport { registerLinearCommands } from './commands/linear.js';\nimport { createSessionCommands } from './commands/session.js';\nimport { registerWorktreeCommands } from './commands/worktree.js';\nimport { registerOnboardingCommand } from './commands/onboard.js';\nimport { createTaskCommands } from './commands/tasks.js';\nimport { createSearchCommand } from './commands/search.js';\nimport { createLogCommand } from './commands/log.js';\nimport { createContextCommands } from './commands/context.js';\nimport { createConfigCommand } from './commands/config.js';\nimport { createHandoffCommand } from './commands/handoff.js';\nimport {\n  createDecisionCommand,\n  createMemoryCommand,\n} from './commands/decision.js';\nimport { createStorageCommand } from './commands/storage.js';\nimport { createSkillsCommand } from './commands/skills.js';\nimport { createTestCommand } from './commands/test.js';\nimport clearCommand from './commands/clear.js';\nimport createWorkflowCommand from './commands/workflow.js';\nimport monitorCommand from './commands/monitor.js';\nimport qualityCommand from './commands/quality.js';\nimport createRalphCommand from './commands/ralph.js';\nimport serviceCommand from './commands/service.js';\nimport { registerLoginCommand } from './commands/login.js';\nimport { registerSignupCommand } from './commands/signup.js';\nimport { registerLogoutCommand, registerDbCommands } from './commands/db.js';\nimport { createSweepCommand } from './commands/sweep.js';\nimport { ProjectManager } from '../core/projects/project-manager.js';\nimport Database from 'better-sqlite3';\nimport { join } from 'path';\nimport { existsSync, mkdirSync } from 'fs';\nimport inquirer from 'inquirer';\nimport chalk from 'chalk';\nimport {\n  loadStorageConfig,\n  enableChromaDB,\n  getStorageModeDescription,\n} from '../core/config/storage-config.js';\n\nconst VERSION = '0.5.2';\n\n// Check for updates on CLI startup\nUpdateChecker.checkForUpdates(VERSION, true).catch(() => {\n  // Silently ignore errors\n});\n\nprogram\n  .name('stackmemory')\n  .description(\n    'Lossless memory runtime for AI coding tools - organizes context as a call stack instead of linear chat logs, with team collaboration and infinite retention'\n  )\n  .version(VERSION);\n\nprogram\n  .command('init')\n  .description(\n    `Initialize StackMemory in current project\n\nStorage Modes:\n  SQLite (default): Local only, fast, no setup required\n  ChromaDB (hybrid): Adds semantic search and cloud backup, requires API key`\n  )\n  .option('--sqlite', 'Use SQLite-only storage (default, skip prompts)')\n  .option(\n    '--chromadb',\n    'Enable ChromaDB for semantic search (prompts for API key)'\n  )\n  .option('--skip-storage-prompt', 'Skip storage configuration prompt')\n  .action(async (options) => {\n    try {\n      const projectRoot = process.cwd();\n      const dbDir = join(projectRoot, '.stackmemory');\n\n      if (!existsSync(dbDir)) {\n        mkdirSync(dbDir, { recursive: true });\n      }\n\n      // Handle storage configuration\n      let storageConfig = loadStorageConfig();\n      const isFirstTimeSetup =\n        !storageConfig.chromadb.enabled && storageConfig.mode === 'sqlite';\n\n      // Skip prompts if --sqlite flag or --skip-storage-prompt\n      if (options.sqlite || options.skipStoragePrompt) {\n        // Use SQLite-only (default)\n        console.log(chalk.gray('Using SQLite-only storage mode.'));\n      } else if (options.chromadb) {\n        // User explicitly requested ChromaDB, prompt for API key\n        await promptAndEnableChromaDB();\n      } else if (isFirstTimeSetup && process.stdin.isTTY) {\n        // Interactive mode - ask user about ChromaDB\n        console.log(chalk.cyan('\\nStorage Configuration'));\n        console.log(chalk.gray('StackMemory supports two storage modes:\\n'));\n        console.log(chalk.white('  SQLite (default):'));\n        console.log(chalk.gray('    - Local storage only'));\n        console.log(chalk.gray('    - Fast and simple'));\n        console.log(chalk.gray('    - No external dependencies\\n'));\n        console.log(chalk.white('  ChromaDB (hybrid):'));\n        console.log(chalk.gray('    - Semantic search across your context'));\n        console.log(chalk.gray('    - Cloud backup capability'));\n        console.log(chalk.gray('    - Requires ChromaDB API key\\n'));\n\n        const { enableChroma } = await inquirer.prompt([\n          {\n            type: 'confirm',\n            name: 'enableChroma',\n            message: 'Enable ChromaDB for semantic search? (requires API key)',\n            default: false,\n          },\n        ]);\n\n        if (enableChroma) {\n          await promptAndEnableChromaDB();\n        } else {\n          console.log(chalk.gray('Using SQLite-only storage mode.'));\n        }\n      }\n\n      // Initialize SQLite database\n      const dbPath = join(dbDir, 'context.db');\n      const db = new Database(dbPath);\n      new FrameManager(db, 'cli-project');\n\n      logger.info('StackMemory initialized successfully', { projectRoot });\n      console.log(\n        chalk.green('\\n[OK] StackMemory initialized in'),\n        projectRoot\n      );\n\n      // Show current storage mode\n      storageConfig = loadStorageConfig();\n      console.log(chalk.gray(`Storage mode: ${getStorageModeDescription()}`));\n\n      db.close();\n    } catch (error: unknown) {\n      logger.error('Failed to initialize StackMemory', error as Error);\n      console.error(\n        chalk.red('[ERROR] Initialization failed:'),\n        (error as Error).message\n      );\n      process.exit(1);\n    }\n  });\n\n/**\n * Prompt user for ChromaDB configuration and enable it\n */\nasync function promptAndEnableChromaDB(): Promise<void> {\n  const answers = await inquirer.prompt([\n    {\n      type: 'password',\n      name: 'apiKey',\n      message: 'Enter your ChromaDB API key:',\n      validate: (input: string) => {\n        if (!input || input.trim().length === 0) {\n          return 'API key is required for ChromaDB';\n        }\n        return true;\n      },\n    },\n    {\n      type: 'input',\n      name: 'apiUrl',\n      message: 'ChromaDB API URL (press Enter for default):',\n      default: 'https://api.trychroma.com',\n    },\n  ]);\n\n  enableChromaDB({\n    apiKey: answers.apiKey,\n    apiUrl: answers.apiUrl,\n  });\n\n  console.log(chalk.green('[OK] ChromaDB enabled for semantic search.'));\n  console.log(\n    chalk.gray('API key saved to ~/.stackmemory/storage-config.json')\n  );\n}\n\nprogram\n  .command('status')\n  .description('Show current StackMemory status')\n  .option('--all', 'Show all active frames across sessions')\n  .option('--project', 'Show all active frames in current project')\n  .option('--session <id>', 'Show frames for specific session')\n  .action(async (options) => {\n    return trace.command('stackmemory-status', options, async () => {\n      try {\n        const projectRoot = process.cwd();\n        const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n        if (!existsSync(dbPath)) {\n          console.log(\n            '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n          );\n          return;\n        }\n\n        // Check for updates and display if available\n        await UpdateChecker.checkForUpdates(VERSION);\n\n        // Initialize session manager and shared context\n        await sessionManager.initialize();\n        await sharedContextLayer.initialize();\n\n        const session = await sessionManager.getOrCreateSession({\n          projectPath: projectRoot,\n          sessionId: options.session,\n        });\n\n        // Auto-discover shared context on startup\n        const contextDiscovery = await sharedContextLayer.autoDiscoverContext();\n\n        // Show context hints if available\n        if (\n          contextDiscovery.hasSharedContext &&\n          contextDiscovery.sessionCount > 1\n        ) {\n          console.log(`\\n\uD83D\uDCA1 Shared Context Available:`);\n          console.log(\n            `   ${contextDiscovery.sessionCount} sessions with shared context`\n          );\n\n          if (contextDiscovery.recentPatterns.length > 0) {\n            console.log(`   Recent patterns:`);\n            contextDiscovery.recentPatterns.slice(0, 3).forEach((p) => {\n              console.log(\n                `     \u2022 ${p.type}: ${p.pattern.slice(0, 50)} (${p.frequency}x)`\n              );\n            });\n          }\n\n          if (contextDiscovery.lastDecisions.length > 0) {\n            console.log(\n              `   Last decision: ${contextDiscovery.lastDecisions[0].decision.slice(0, 60)}`\n            );\n          }\n        }\n\n        const db = new Database(dbPath);\n        const frameManager = new FrameManager(db, session.projectId);\n\n        // Set query mode based on options\n        if (options.all) {\n          frameManager.setQueryMode(FrameQueryMode.ALL_ACTIVE);\n        } else if (options.project) {\n          frameManager.setQueryMode(FrameQueryMode.PROJECT_ACTIVE);\n        }\n\n        const activeFrames = frameManager.getActiveFramePath();\n        const stackDepth = frameManager.getStackDepth();\n\n        // Always get total counts across all sessions\n        const totalStats = db\n          .prepare(\n            `\n        SELECT \n          COUNT(*) as total_frames,\n          SUM(CASE WHEN state = 'active' THEN 1 ELSE 0 END) as active_frames,\n          SUM(CASE WHEN state = 'closed' THEN 1 ELSE 0 END) as closed_frames,\n          COUNT(DISTINCT run_id) as total_sessions\n        FROM frames\n        WHERE project_id = ?\n      `\n          )\n          .get(session.projectId) as {\n          total_frames: number;\n          active_frames: number;\n          closed_frames: number;\n          total_sessions: number;\n        };\n\n        const contextCount = db\n          .prepare(\n            `\n        SELECT COUNT(*) as count FROM contexts\n      `\n          )\n          .get() as { count: number };\n\n        const eventCount = db\n          .prepare(\n            `\n        SELECT COUNT(*) as count FROM events e\n        JOIN frames f ON e.frame_id = f.frame_id\n        WHERE f.project_id = ?\n      `\n          )\n          .get(session.projectId) as { count: number };\n\n        console.log('\uD83D\uDCCA StackMemory Status:');\n        console.log(\n          `   Session: ${session.sessionId.slice(0, 8)} (${session.state}, ${Math.round((Date.now() - session.startedAt) / 1000 / 60)}min old)`\n        );\n        console.log(`   Project: ${session.projectId}`);\n        if (session.branch) {\n          console.log(`   Branch: ${session.branch}`);\n        }\n\n        // Show total database statistics\n        console.log(`\\n   Database Statistics (this project):`);\n        console.log(\n          `     Frames: ${totalStats.total_frames || 0} (${totalStats.active_frames || 0} active, ${totalStats.closed_frames || 0} closed)`\n        );\n        console.log(`     Events: ${eventCount.count || 0}`);\n        console.log(`     Sessions: ${totalStats.total_sessions || 0}`);\n        console.log(\n          `     Cached contexts: ${contextCount.count || 0} (global)`\n        );\n\n        // Show recent activity\n        const recentFrames = db\n          .prepare(\n            `\n        SELECT name, type, state, datetime(created_at, 'unixepoch') as created\n        FROM frames\n        WHERE project_id = ?\n        ORDER BY created_at DESC\n        LIMIT 3\n      `\n          )\n          .all(session.projectId) as Array<{\n          name: string;\n          type: string;\n          state: string;\n          created: string;\n        }>;\n\n        if (recentFrames.length > 0) {\n          console.log(`\\n   Recent Activity:`);\n          recentFrames.forEach((f) => {\n            const stateIcon = f.state === 'active' ? '\uD83D\uDFE2' : '\u26AB';\n            console.log(\n              `     ${stateIcon} ${f.name} [${f.type}] - ${f.created}`\n            );\n          });\n        }\n\n        console.log(`\\n   Current Session:`);\n        console.log(`     Stack depth: ${stackDepth}`);\n        console.log(`     Active frames: ${activeFrames.length}`);\n\n        if (activeFrames.length > 0) {\n          activeFrames.forEach((frame, i) => {\n            const indent = '     ' + '  '.repeat(frame.depth || i);\n            const prefix = i === 0 ? '\u2514\u2500' : '  \u2514\u2500';\n            console.log(`${indent}${prefix} ${frame.name} [${frame.type}]`);\n          });\n        }\n\n        // Show other sessions if in default mode\n        if (!options.all && !options.project) {\n          const otherSessions = await sessionManager.listSessions({\n            projectId: session.projectId,\n            state: 'active',\n          });\n\n          const otherActive = otherSessions.filter(\n            (s) => s.sessionId !== session.sessionId\n          );\n          if (otherActive.length > 0) {\n            console.log(`\\n   Other Active Sessions (same project):`);\n            otherActive.forEach((s) => {\n              const age = Math.round(\n                (Date.now() - s.lastActiveAt) / 1000 / 60 / 60\n              );\n              console.log(\n                `     - ${s.sessionId.slice(0, 8)}: ${s.branch || 'main'}, ${age}h old`\n              );\n            });\n            console.log(`\\n   Tip: Use --all to see frames across sessions`);\n          }\n        }\n\n        db.close();\n      } catch (error: unknown) {\n        logger.error('Failed to get status', error as Error);\n        console.error('\u274C Status check failed:', (error as Error).message);\n        process.exit(1);\n      }\n    });\n  });\n\nprogram\n  .command('update-check')\n  .description('Check for StackMemory updates')\n  .action(async () => {\n    try {\n      console.log('\uD83D\uDD0D Checking for updates...');\n      await UpdateChecker.forceCheck(VERSION);\n    } catch (error: unknown) {\n      logger.error('Update check failed', error as Error);\n      console.error('\u274C Update check failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('progress')\n  .description('Show current progress and recent changes')\n  .action(async () => {\n    try {\n      const projectRoot = process.cwd();\n      const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n      if (!existsSync(dbPath)) {\n        console.log(\n          '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n        );\n        return;\n      }\n\n      const progress = new ProgressTracker(projectRoot);\n      console.log(progress.getSummary());\n    } catch (error: unknown) {\n      logger.error('Failed to show progress', error as Error);\n      console.error('\u274C Failed to show progress:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('mcp-server')\n  .description('Start StackMemory MCP server for Claude Desktop')\n  .option('-p, --project <path>', 'Project root directory', process.cwd())\n  .action(async (options) => {\n    try {\n      const { runMCPServer } = await import('../integrations/mcp/server.js');\n\n      // Set project root\n      process.env['PROJECT_ROOT'] = options.project;\n\n      console.log('\uD83D\uDE80 Starting StackMemory MCP Server...');\n      console.log(`   Project: ${options.project}`);\n      console.log(`   Version: ${VERSION}`);\n\n      // Check for updates silently\n      UpdateChecker.checkForUpdates(VERSION, true).catch(() => {});\n\n      // Start the MCP server\n      await runMCPServer();\n    } catch (error: unknown) {\n      logger.error('Failed to start MCP server', error as Error);\n      console.error('\u274C MCP server failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\n// Add test context command\nprogram\n  .command('context:test')\n  .description('Test context persistence by creating sample frames')\n  .action(async () => {\n    try {\n      const projectRoot = process.cwd();\n      const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n      if (!existsSync(dbPath)) {\n        console.log(\n          '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n        );\n        return;\n      }\n\n      const db = new Database(dbPath);\n      const frameManager = new FrameManager(db, 'cli-project');\n\n      // Create test frames\n      console.log('\uD83D\uDCDD Creating test context frames...');\n\n      const rootFrame = frameManager.createFrame({\n        type: 'task',\n        name: 'Test Session',\n        inputs: { test: true, timestamp: new Date().toISOString() },\n      });\n\n      const taskFrame = frameManager.createFrame({\n        type: 'subtask',\n        name: 'Sample Task',\n        inputs: { description: 'Testing context persistence' },\n        parentFrameId: rootFrame,\n      });\n\n      const commandFrame = frameManager.createFrame({\n        type: 'tool_scope',\n        name: 'test-command',\n        inputs: { args: ['--test'] },\n        parentFrameId: taskFrame,\n      });\n\n      // Add some events\n      frameManager.addEvent(\n        'observation',\n        {\n          message: 'Test event recorded',\n        },\n        commandFrame\n      );\n\n      console.log('\u2705 Test frames created!');\n      console.log(`\uD83D\uDCCA Stack depth: ${frameManager.getStackDepth()}`);\n      console.log(\n        `\uD83D\uDD04 Active frames: ${frameManager.getActiveFramePath().length}`\n      );\n\n      // Close one frame to test state changes\n      frameManager.closeFrame(commandFrame);\n      console.log(\n        `\uD83D\uDCCA After closing command frame: depth = ${frameManager.getStackDepth()}`\n      );\n\n      db.close();\n    } catch (error: unknown) {\n      logger.error('Test context failed', error as Error);\n      console.error('\u274C Test failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\n// Register project management commands\n// Register command modules\nregisterOnboardingCommand(program);\nregisterSignupCommand(program);\nregisterLoginCommand(program);\nregisterLogoutCommand(program);\nregisterDbCommands(program);\nregisterProjectCommands(program);\nregisterWorktreeCommands(program);\n\n// Register Linear integration commands\nregisterLinearCommands(program);\n\n// Register session management commands\nprogram.addCommand(createSessionCommands());\n\n// Register enhanced CLI commands\nprogram.addCommand(createTaskCommands());\nprogram.addCommand(createSearchCommand());\nprogram.addCommand(createLogCommand());\nprogram.addCommand(createContextCommands());\nprogram.addCommand(createConfigCommand());\nprogram.addCommand(createHandoffCommand());\nprogram.addCommand(createDecisionCommand());\nprogram.addCommand(createMemoryCommand());\nprogram.addCommand(createStorageCommand());\nprogram.addCommand(createSkillsCommand());\nprogram.addCommand(createTestCommand());\nprogram.addCommand(clearCommand);\nprogram.addCommand(createWorkflowCommand());\nprogram.addCommand(monitorCommand);\nprogram.addCommand(qualityCommand);\nprogram.addCommand(createRalphCommand());\nprogram.addCommand(serviceCommand);\nprogram.addCommand(createSweepCommand());\n\n// Register dashboard command\nprogram\n  .command('dashboard')\n  .description('Display monitoring dashboard in terminal')\n  .option('-w, --watch', 'Auto-refresh dashboard')\n  .option('-i, --interval <seconds>', 'Refresh interval in seconds', '5')\n  .action(async (options) => {\n    const { dashboardCommand } = await import('./commands/dashboard.js');\n    await dashboardCommand.handler(options);\n  });\n\n// Auto-detect current project on startup\nif (process.argv.length > 2) {\n  const manager = ProjectManager.getInstance();\n  manager.detectProject().catch(() => {\n    // Silently fail if not in a project directory\n  });\n}\n\n// Only parse when running as main module (not when imported for testing)\nconst isMainModule =\n  import.meta.url === `file://${process.argv[1]}` ||\n  process.argv[1]?.endsWith('/stackmemory') ||\n  process.argv[1]?.endsWith('index.ts') ||\n  process.argv[1]?.includes('tsx');\n\nif (isMainModule) {\n  program.parse();\n}\n\nexport { program };\n"],
+  "sourcesContent": ["#!/usr/bin/env node\n/**\n * StackMemory CLI\n * Command-line interface for StackMemory operations\n */\n\n// Set environment flag for CLI usage to skip async context bridge\nprocess.env['STACKMEMORY_CLI'] = 'true';\n\n// Load environment variables\nimport 'dotenv/config';\n\n// Initialize tracing system early\nimport { initializeTracing, trace } from '../core/trace/index.js';\ninitializeTracing();\n\nimport { program } from 'commander';\nimport { logger } from '../core/monitoring/logger.js';\nimport { FrameManager } from '../core/context/frame-manager.js';\nimport { sessionManager, FrameQueryMode } from '../core/session/index.js';\nimport { sharedContextLayer } from '../core/context/shared-context-layer.js';\nimport { UpdateChecker } from '../core/utils/update-checker.js';\nimport { ProgressTracker } from '../core/monitoring/progress-tracker.js';\nimport { registerProjectCommands } from './commands/projects.js';\nimport { registerLinearCommands } from './commands/linear.js';\nimport { createSessionCommands } from './commands/session.js';\nimport { registerWorktreeCommands } from './commands/worktree.js';\nimport { registerOnboardingCommand } from './commands/onboard.js';\nimport { createTaskCommands } from './commands/tasks.js';\nimport { createSearchCommand } from './commands/search.js';\nimport { createLogCommand } from './commands/log.js';\nimport { createContextCommands } from './commands/context.js';\nimport { createConfigCommand } from './commands/config.js';\nimport { createHandoffCommand } from './commands/handoff.js';\nimport {\n  createDecisionCommand,\n  createMemoryCommand,\n} from './commands/decision.js';\nimport { createStorageCommand } from './commands/storage.js';\nimport { createSkillsCommand } from './commands/skills.js';\nimport { createTestCommand } from './commands/test.js';\nimport clearCommand from './commands/clear.js';\nimport createWorkflowCommand from './commands/workflow.js';\nimport monitorCommand from './commands/monitor.js';\nimport qualityCommand from './commands/quality.js';\nimport createRalphCommand from './commands/ralph.js';\nimport serviceCommand from './commands/service.js';\nimport { registerLoginCommand } from './commands/login.js';\nimport { registerSignupCommand } from './commands/signup.js';\nimport { registerLogoutCommand, registerDbCommands } from './commands/db.js';\nimport { createSweepCommand } from './commands/sweep.js';\nimport { ProjectManager } from '../core/projects/project-manager.js';\nimport Database from 'better-sqlite3';\nimport { join } from 'path';\nimport { existsSync, mkdirSync } from 'fs';\nimport inquirer from 'inquirer';\nimport chalk from 'chalk';\nimport {\n  loadStorageConfig,\n  enableChromaDB,\n  getStorageModeDescription,\n} from '../core/config/storage-config.js';\n\nconst VERSION = '0.5.3';\n\n// Check for updates on CLI startup\nUpdateChecker.checkForUpdates(VERSION, true).catch(() => {\n  // Silently ignore errors\n});\n\nprogram\n  .name('stackmemory')\n  .description(\n    'Lossless memory runtime for AI coding tools - organizes context as a call stack instead of linear chat logs, with team collaboration and infinite retention'\n  )\n  .version(VERSION);\n\nprogram\n  .command('init')\n  .description(\n    `Initialize StackMemory in current project\n\nStorage Modes:\n  SQLite (default): Local only, fast, no setup required\n  ChromaDB (hybrid): Adds semantic search and cloud backup, requires API key`\n  )\n  .option('--sqlite', 'Use SQLite-only storage (default, skip prompts)')\n  .option(\n    '--chromadb',\n    'Enable ChromaDB for semantic search (prompts for API key)'\n  )\n  .option('--skip-storage-prompt', 'Skip storage configuration prompt')\n  .action(async (options) => {\n    try {\n      const projectRoot = process.cwd();\n      const dbDir = join(projectRoot, '.stackmemory');\n\n      if (!existsSync(dbDir)) {\n        mkdirSync(dbDir, { recursive: true });\n      }\n\n      // Handle storage configuration\n      let storageConfig = loadStorageConfig();\n      const isFirstTimeSetup =\n        !storageConfig.chromadb.enabled && storageConfig.mode === 'sqlite';\n\n      // Skip prompts if --sqlite flag or --skip-storage-prompt\n      if (options.sqlite || options.skipStoragePrompt) {\n        // Use SQLite-only (default)\n        console.log(chalk.gray('Using SQLite-only storage mode.'));\n      } else if (options.chromadb) {\n        // User explicitly requested ChromaDB, prompt for API key\n        await promptAndEnableChromaDB();\n      } else if (isFirstTimeSetup && process.stdin.isTTY) {\n        // Interactive mode - ask user about ChromaDB\n        console.log(chalk.cyan('\\nStorage Configuration'));\n        console.log(chalk.gray('StackMemory supports two storage modes:\\n'));\n        console.log(chalk.white('  SQLite (default):'));\n        console.log(chalk.gray('    - Local storage only'));\n        console.log(chalk.gray('    - Fast and simple'));\n        console.log(chalk.gray('    - No external dependencies\\n'));\n        console.log(chalk.white('  ChromaDB (hybrid):'));\n        console.log(chalk.gray('    - Semantic search across your context'));\n        console.log(chalk.gray('    - Cloud backup capability'));\n        console.log(chalk.gray('    - Requires ChromaDB API key\\n'));\n\n        const { enableChroma } = await inquirer.prompt([\n          {\n            type: 'confirm',\n            name: 'enableChroma',\n            message: 'Enable ChromaDB for semantic search? (requires API key)',\n            default: false,\n          },\n        ]);\n\n        if (enableChroma) {\n          await promptAndEnableChromaDB();\n        } else {\n          console.log(chalk.gray('Using SQLite-only storage mode.'));\n        }\n      }\n\n      // Initialize SQLite database\n      const dbPath = join(dbDir, 'context.db');\n      const db = new Database(dbPath);\n      new FrameManager(db, 'cli-project');\n\n      logger.info('StackMemory initialized successfully', { projectRoot });\n      console.log(\n        chalk.green('\\n[OK] StackMemory initialized in'),\n        projectRoot\n      );\n\n      // Show current storage mode\n      storageConfig = loadStorageConfig();\n      console.log(chalk.gray(`Storage mode: ${getStorageModeDescription()}`));\n\n      db.close();\n    } catch (error: unknown) {\n      logger.error('Failed to initialize StackMemory', error as Error);\n      console.error(\n        chalk.red('[ERROR] Initialization failed:'),\n        (error as Error).message\n      );\n      process.exit(1);\n    }\n  });\n\n/**\n * Prompt user for ChromaDB configuration and enable it\n */\nasync function promptAndEnableChromaDB(): Promise<void> {\n  const answers = await inquirer.prompt([\n    {\n      type: 'password',\n      name: 'apiKey',\n      message: 'Enter your ChromaDB API key:',\n      validate: (input: string) => {\n        if (!input || input.trim().length === 0) {\n          return 'API key is required for ChromaDB';\n        }\n        return true;\n      },\n    },\n    {\n      type: 'input',\n      name: 'apiUrl',\n      message: 'ChromaDB API URL (press Enter for default):',\n      default: 'https://api.trychroma.com',\n    },\n  ]);\n\n  enableChromaDB({\n    apiKey: answers.apiKey,\n    apiUrl: answers.apiUrl,\n  });\n\n  console.log(chalk.green('[OK] ChromaDB enabled for semantic search.'));\n  console.log(\n    chalk.gray('API key saved to ~/.stackmemory/storage-config.json')\n  );\n}\n\nprogram\n  .command('status')\n  .description('Show current StackMemory status')\n  .option('--all', 'Show all active frames across sessions')\n  .option('--project', 'Show all active frames in current project')\n  .option('--session <id>', 'Show frames for specific session')\n  .action(async (options) => {\n    return trace.command('stackmemory-status', options, async () => {\n      try {\n        const projectRoot = process.cwd();\n        const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n        if (!existsSync(dbPath)) {\n          console.log(\n            '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n          );\n          return;\n        }\n\n        // Check for updates and display if available\n        await UpdateChecker.checkForUpdates(VERSION);\n\n        // Initialize session manager and shared context\n        await sessionManager.initialize();\n        await sharedContextLayer.initialize();\n\n        const session = await sessionManager.getOrCreateSession({\n          projectPath: projectRoot,\n          sessionId: options.session,\n        });\n\n        // Auto-discover shared context on startup\n        const contextDiscovery = await sharedContextLayer.autoDiscoverContext();\n\n        // Show context hints if available\n        if (\n          contextDiscovery.hasSharedContext &&\n          contextDiscovery.sessionCount > 1\n        ) {\n          console.log(`\\n\uD83D\uDCA1 Shared Context Available:`);\n          console.log(\n            `   ${contextDiscovery.sessionCount} sessions with shared context`\n          );\n\n          if (contextDiscovery.recentPatterns.length > 0) {\n            console.log(`   Recent patterns:`);\n            contextDiscovery.recentPatterns.slice(0, 3).forEach((p) => {\n              console.log(\n                `     \u2022 ${p.type}: ${p.pattern.slice(0, 50)} (${p.frequency}x)`\n              );\n            });\n          }\n\n          if (contextDiscovery.lastDecisions.length > 0) {\n            console.log(\n              `   Last decision: ${contextDiscovery.lastDecisions[0].decision.slice(0, 60)}`\n            );\n          }\n        }\n\n        const db = new Database(dbPath);\n        const frameManager = new FrameManager(db, session.projectId);\n\n        // Set query mode based on options\n        if (options.all) {\n          frameManager.setQueryMode(FrameQueryMode.ALL_ACTIVE);\n        } else if (options.project) {\n          frameManager.setQueryMode(FrameQueryMode.PROJECT_ACTIVE);\n        }\n\n        const activeFrames = frameManager.getActiveFramePath();\n        const stackDepth = frameManager.getStackDepth();\n\n        // Always get total counts across all sessions\n        const totalStats = db\n          .prepare(\n            `\n        SELECT \n          COUNT(*) as total_frames,\n          SUM(CASE WHEN state = 'active' THEN 1 ELSE 0 END) as active_frames,\n          SUM(CASE WHEN state = 'closed' THEN 1 ELSE 0 END) as closed_frames,\n          COUNT(DISTINCT run_id) as total_sessions\n        FROM frames\n        WHERE project_id = ?\n      `\n          )\n          .get(session.projectId) as {\n          total_frames: number;\n          active_frames: number;\n          closed_frames: number;\n          total_sessions: number;\n        };\n\n        const contextCount = db\n          .prepare(\n            `\n        SELECT COUNT(*) as count FROM contexts\n      `\n          )\n          .get() as { count: number };\n\n        const eventCount = db\n          .prepare(\n            `\n        SELECT COUNT(*) as count FROM events e\n        JOIN frames f ON e.frame_id = f.frame_id\n        WHERE f.project_id = ?\n      `\n          )\n          .get(session.projectId) as { count: number };\n\n        console.log('\uD83D\uDCCA StackMemory Status:');\n        console.log(\n          `   Session: ${session.sessionId.slice(0, 8)} (${session.state}, ${Math.round((Date.now() - session.startedAt) / 1000 / 60)}min old)`\n        );\n        console.log(`   Project: ${session.projectId}`);\n        if (session.branch) {\n          console.log(`   Branch: ${session.branch}`);\n        }\n\n        // Show total database statistics\n        console.log(`\\n   Database Statistics (this project):`);\n        console.log(\n          `     Frames: ${totalStats.total_frames || 0} (${totalStats.active_frames || 0} active, ${totalStats.closed_frames || 0} closed)`\n        );\n        console.log(`     Events: ${eventCount.count || 0}`);\n        console.log(`     Sessions: ${totalStats.total_sessions || 0}`);\n        console.log(\n          `     Cached contexts: ${contextCount.count || 0} (global)`\n        );\n\n        // Show recent activity\n        const recentFrames = db\n          .prepare(\n            `\n        SELECT name, type, state, datetime(created_at, 'unixepoch') as created\n        FROM frames\n        WHERE project_id = ?\n        ORDER BY created_at DESC\n        LIMIT 3\n      `\n          )\n          .all(session.projectId) as Array<{\n          name: string;\n          type: string;\n          state: string;\n          created: string;\n        }>;\n\n        if (recentFrames.length > 0) {\n          console.log(`\\n   Recent Activity:`);\n          recentFrames.forEach((f) => {\n            const stateIcon = f.state === 'active' ? '\uD83D\uDFE2' : '\u26AB';\n            console.log(\n              `     ${stateIcon} ${f.name} [${f.type}] - ${f.created}`\n            );\n          });\n        }\n\n        console.log(`\\n   Current Session:`);\n        console.log(`     Stack depth: ${stackDepth}`);\n        console.log(`     Active frames: ${activeFrames.length}`);\n\n        if (activeFrames.length > 0) {\n          activeFrames.forEach((frame, i) => {\n            const indent = '     ' + '  '.repeat(frame.depth || i);\n            const prefix = i === 0 ? '\u2514\u2500' : '  \u2514\u2500';\n            console.log(`${indent}${prefix} ${frame.name} [${frame.type}]`);\n          });\n        }\n\n        // Show other sessions if in default mode\n        if (!options.all && !options.project) {\n          const otherSessions = await sessionManager.listSessions({\n            projectId: session.projectId,\n            state: 'active',\n          });\n\n          const otherActive = otherSessions.filter(\n            (s) => s.sessionId !== session.sessionId\n          );\n          if (otherActive.length > 0) {\n            console.log(`\\n   Other Active Sessions (same project):`);\n            otherActive.forEach((s) => {\n              const age = Math.round(\n                (Date.now() - s.lastActiveAt) / 1000 / 60 / 60\n              );\n              console.log(\n                `     - ${s.sessionId.slice(0, 8)}: ${s.branch || 'main'}, ${age}h old`\n              );\n            });\n            console.log(`\\n   Tip: Use --all to see frames across sessions`);\n          }\n        }\n\n        db.close();\n      } catch (error: unknown) {\n        logger.error('Failed to get status', error as Error);\n        console.error('\u274C Status check failed:', (error as Error).message);\n        process.exit(1);\n      }\n    });\n  });\n\nprogram\n  .command('update-check')\n  .description('Check for StackMemory updates')\n  .action(async () => {\n    try {\n      console.log('\uD83D\uDD0D Checking for updates...');\n      await UpdateChecker.forceCheck(VERSION);\n    } catch (error: unknown) {\n      logger.error('Update check failed', error as Error);\n      console.error('\u274C Update check failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('progress')\n  .description('Show current progress and recent changes')\n  .action(async () => {\n    try {\n      const projectRoot = process.cwd();\n      const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n      if (!existsSync(dbPath)) {\n        console.log(\n          '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n        );\n        return;\n      }\n\n      const progress = new ProgressTracker(projectRoot);\n      console.log(progress.getSummary());\n    } catch (error: unknown) {\n      logger.error('Failed to show progress', error as Error);\n      console.error('\u274C Failed to show progress:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('mcp-server')\n  .description('Start StackMemory MCP server for Claude Desktop')\n  .option('-p, --project <path>', 'Project root directory', process.cwd())\n  .action(async (options) => {\n    try {\n      const { runMCPServer } = await import('../integrations/mcp/server.js');\n\n      // Set project root\n      process.env['PROJECT_ROOT'] = options.project;\n\n      console.log('\uD83D\uDE80 Starting StackMemory MCP Server...');\n      console.log(`   Project: ${options.project}`);\n      console.log(`   Version: ${VERSION}`);\n\n      // Check for updates silently\n      UpdateChecker.checkForUpdates(VERSION, true).catch(() => {});\n\n      // Start the MCP server\n      await runMCPServer();\n    } catch (error: unknown) {\n      logger.error('Failed to start MCP server', error as Error);\n      console.error('\u274C MCP server failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\n// Add test context command\nprogram\n  .command('context:test')\n  .description('Test context persistence by creating sample frames')\n  .action(async () => {\n    try {\n      const projectRoot = process.cwd();\n      const dbPath = join(projectRoot, '.stackmemory', 'context.db');\n\n      if (!existsSync(dbPath)) {\n        console.log(\n          '\u274C StackMemory not initialized. Run \"stackmemory init\" first.'\n        );\n        return;\n      }\n\n      const db = new Database(dbPath);\n      const frameManager = new FrameManager(db, 'cli-project');\n\n      // Create test frames\n      console.log('\uD83D\uDCDD Creating test context frames...');\n\n      const rootFrame = frameManager.createFrame({\n        type: 'task',\n        name: 'Test Session',\n        inputs: { test: true, timestamp: new Date().toISOString() },\n      });\n\n      const taskFrame = frameManager.createFrame({\n        type: 'subtask',\n        name: 'Sample Task',\n        inputs: { description: 'Testing context persistence' },\n        parentFrameId: rootFrame,\n      });\n\n      const commandFrame = frameManager.createFrame({\n        type: 'tool_scope',\n        name: 'test-command',\n        inputs: { args: ['--test'] },\n        parentFrameId: taskFrame,\n      });\n\n      // Add some events\n      frameManager.addEvent(\n        'observation',\n        {\n          message: 'Test event recorded',\n        },\n        commandFrame\n      );\n\n      console.log('\u2705 Test frames created!');\n      console.log(`\uD83D\uDCCA Stack depth: ${frameManager.getStackDepth()}`);\n      console.log(\n        `\uD83D\uDD04 Active frames: ${frameManager.getActiveFramePath().length}`\n      );\n\n      // Close one frame to test state changes\n      frameManager.closeFrame(commandFrame);\n      console.log(\n        `\uD83D\uDCCA After closing command frame: depth = ${frameManager.getStackDepth()}`\n      );\n\n      db.close();\n    } catch (error: unknown) {\n      logger.error('Test context failed', error as Error);\n      console.error('\u274C Test failed:', (error as Error).message);\n      process.exit(1);\n    }\n  });\n\n// Register project management commands\n// Register command modules\nregisterOnboardingCommand(program);\nregisterSignupCommand(program);\nregisterLoginCommand(program);\nregisterLogoutCommand(program);\nregisterDbCommands(program);\nregisterProjectCommands(program);\nregisterWorktreeCommands(program);\n\n// Register Linear integration commands\nregisterLinearCommands(program);\n\n// Register session management commands\nprogram.addCommand(createSessionCommands());\n\n// Register enhanced CLI commands\nprogram.addCommand(createTaskCommands());\nprogram.addCommand(createSearchCommand());\nprogram.addCommand(createLogCommand());\nprogram.addCommand(createContextCommands());\nprogram.addCommand(createConfigCommand());\nprogram.addCommand(createHandoffCommand());\nprogram.addCommand(createDecisionCommand());\nprogram.addCommand(createMemoryCommand());\nprogram.addCommand(createStorageCommand());\nprogram.addCommand(createSkillsCommand());\nprogram.addCommand(createTestCommand());\nprogram.addCommand(clearCommand);\nprogram.addCommand(createWorkflowCommand());\nprogram.addCommand(monitorCommand);\nprogram.addCommand(qualityCommand);\nprogram.addCommand(createRalphCommand());\nprogram.addCommand(serviceCommand);\nprogram.addCommand(createSweepCommand());\n\n// Register dashboard command\nprogram\n  .command('dashboard')\n  .description('Display monitoring dashboard in terminal')\n  .option('-w, --watch', 'Auto-refresh dashboard')\n  .option('-i, --interval <seconds>', 'Refresh interval in seconds', '5')\n  .action(async (options) => {\n    const { dashboardCommand } = await import('./commands/dashboard.js');\n    await dashboardCommand.handler(options);\n  });\n\n// Auto-detect current project on startup\nif (process.argv.length > 2) {\n  const manager = ProjectManager.getInstance();\n  manager.detectProject().catch(() => {\n    // Silently fail if not in a project directory\n  });\n}\n\n// Only parse when running as main module (not when imported for testing)\nconst isMainModule =\n  import.meta.url === `file://${process.argv[1]}` ||\n  process.argv[1]?.endsWith('/stackmemory') ||\n  process.argv[1]?.endsWith('index.ts') ||\n  process.argv[1]?.includes('tsx');\n\nif (isMainModule) {\n  program.parse();\n}\n\nexport { program };\n"],
   "mappings": ";AAOA,QAAQ,IAAI,iBAAiB,IAAI;AAGjC,OAAO;AAGP,SAAS,mBAAmB,aAAa;AACzC,kBAAkB;AAElB,SAAS,eAAe;AACxB,SAAS,cAAc;AACvB,SAAS,oBAAoB;AAC7B,SAAS,gBAAgB,sBAAsB;AAC/C,SAAS,0BAA0B;AACnC,SAAS,qBAAqB;AAC9B,SAAS,uBAAuB;AAChC,SAAS,+BAA+B;AACxC,SAAS,8BAA8B;AACvC,SAAS,6BAA6B;AACtC,SAAS,gCAAgC;AACzC,SAAS,iCAAiC;AAC1C,SAAS,0BAA0B;AACnC,SAAS,2BAA2B;AACpC,SAAS,wBAAwB;AACjC,SAAS,6BAA6B;AACtC,SAAS,2BAA2B;AACpC,SAAS,4BAA4B;AACrC;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,4BAA4B;AACrC,SAAS,2BAA2B;AACpC,SAAS,yBAAyB;AAClC,OAAO,kBAAkB;AACzB,OAAO,2BAA2B;AAClC,OAAO,oBAAoB;AAC3B,OAAO,oBAAoB;AAC3B,OAAO,wBAAwB;AAC/B,OAAO,oBAAoB;AAC3B,SAAS,4BAA4B;AACrC,SAAS,6BAA6B;AACtC,SAAS,uBAAuB,0BAA0B;AAC1D,SAAS,0BAA0B;AACnC,SAAS,sBAAsB;AAC/B,OAAO,cAAc;AACrB,SAAS,YAAY;AACrB,SAAS,YAAY,iBAAiB;AACtC,OAAO,cAAc;AACrB,OAAO,WAAW;AAClB;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,MAAM,UAAU;AAGhB,cAAc,gBAAgB,SAAS,IAAI,EAAE,MAAM,MAAM;AAEzD,CAAC;AAED,QACG,KAAK,aAAa,EAClB;AAAA,EACC;AACF,EACC,QAAQ,OAAO;AAElB,QACG,QAAQ,MAAM,EACd;AAAA,EACC;AAAA;AAAA;AAAA;AAAA;AAKF,EACC,OAAO,YAAY,iDAAiD,EACpE;AAAA,EACC;AAAA,EACA;AACF,EACC,OAAO,yBAAyB,mCAAmC,EACnE,OAAO,OAAO,YAAY;AACzB,MAAI;AACF,UAAM,cAAc,QAAQ,IAAI;AAChC,UAAM,QAAQ,KAAK,aAAa,cAAc;AAE9C,QAAI,CAAC,WAAW,KAAK,GAAG;AACtB,gBAAU,OAAO,EAAE,WAAW,KAAK,CAAC;AAAA,IACtC;AAGA,QAAI,gBAAgB,kBAAkB;AACtC,UAAM,mBACJ,CAAC,cAAc,SAAS,WAAW,cAAc,SAAS;AAG5D,QAAI,QAAQ,UAAU,QAAQ,mBAAmB;AAE/C,cAAQ,IAAI,MAAM,KAAK,iCAAiC,CAAC;AAAA,IAC3D,WAAW,QAAQ,UAAU;AAE3B,YAAM,wBAAwB;AAAA,IAChC,WAAW,oBAAoB,QAAQ,MAAM,OAAO;AAElD,cAAQ,IAAI,MAAM,KAAK,yBAAyB,CAAC;AACjD,cAAQ,IAAI,MAAM,KAAK,2CAA2C,CAAC;AACnE,cAAQ,IAAI,MAAM,MAAM,qBAAqB,CAAC;AAC9C,cAAQ,IAAI,MAAM,KAAK,0BAA0B,CAAC;AAClD,cAAQ,IAAI,MAAM,KAAK,uBAAuB,CAAC;AAC/C,cAAQ,IAAI,MAAM,KAAK,kCAAkC,CAAC;AAC1D,cAAQ,IAAI,MAAM,MAAM,sBAAsB,CAAC;AAC/C,cAAQ,IAAI,MAAM,KAAK,2CAA2C,CAAC;AACnE,cAAQ,IAAI,MAAM,KAAK,+BAA+B,CAAC;AACvD,cAAQ,IAAI,MAAM,KAAK,mCAAmC,CAAC;AAE3D,YAAM,EAAE,aAAa,IAAI,MAAM,SAAS,OAAO;AAAA,QAC7C;AAAA,UACE,MAAM;AAAA,UACN,MAAM;AAAA,UACN,SAAS;AAAA,UACT,SAAS;AAAA,QACX;AAAA,MACF,CAAC;AAED,UAAI,cAAc;AAChB,cAAM,wBAAwB;AAAA,MAChC,OAAO;AACL,gBAAQ,IAAI,MAAM,KAAK,iCAAiC,CAAC;AAAA,MAC3D;AAAA,IACF;AAGA,UAAM,SAAS,KAAK,OAAO,YAAY;AACvC,UAAM,KAAK,IAAI,SAAS,MAAM;AAC9B,QAAI,aAAa,IAAI,aAAa;AAElC,WAAO,KAAK,wCAAwC,EAAE,YAAY,CAAC;AACnE,YAAQ;AAAA,MACN,MAAM,MAAM,mCAAmC;AAAA,MAC/C;AAAA,IACF;AAGA,oBAAgB,kBAAkB;AAClC,YAAQ,IAAI,MAAM,KAAK,iBAAiB,0BAA0B,CAAC,EAAE,CAAC;AAEtE,OAAG,MAAM;AAAA,EACX,SAAS,OAAgB;AACvB,WAAO,MAAM,oCAAoC,KAAc;AAC/D,YAAQ;AAAA,MACN,MAAM,IAAI,gCAAgC;AAAA,MACzC,MAAgB;AAAA,IACnB;AACA,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF,CAAC;AAKH,eAAe,0BAAyC;AACtD,QAAM,UAAU,MAAM,SAAS,OAAO;AAAA,IACpC;AAAA,MACE,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,UAAU,CAAC,UAAkB;AAC3B,YAAI,CAAC,SAAS,MAAM,KAAK,EAAE,WAAW,GAAG;AACvC,iBAAO;AAAA,QACT;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,IACA;AAAA,MACE,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,SAAS;AAAA,IACX;AAAA,EACF,CAAC;AAED,iBAAe;AAAA,IACb,QAAQ,QAAQ;AAAA,IAChB,QAAQ,QAAQ;AAAA,EAClB,CAAC;AAED,UAAQ,IAAI,MAAM,MAAM,4CAA4C,CAAC;AACrE,UAAQ;AAAA,IACN,MAAM,KAAK,qDAAqD;AAAA,EAClE;AACF;AAEA,QACG,QAAQ,QAAQ,EAChB,YAAY,iCAAiC,EAC7C,OAAO,SAAS,wCAAwC,EACxD,OAAO,aAAa,2CAA2C,EAC/D,OAAO,kBAAkB,kCAAkC,EAC3D,OAAO,OAAO,YAAY;AACzB,SAAO,MAAM,QAAQ,sBAAsB,SAAS,YAAY;AAC9D,QAAI;AACF,YAAM,cAAc,QAAQ,IAAI;AAChC,YAAM,SAAS,KAAK,aAAa,gBAAgB,YAAY;AAE7D,UAAI,CAAC,WAAW,MAAM,GAAG;AACvB,gBAAQ;AAAA,UACN;AAAA,QACF;AACA;AAAA,MACF;AAGA,YAAM,cAAc,gBAAgB,OAAO;AAG3C,YAAM,eAAe,WAAW;AAChC,YAAM,mBAAmB,WAAW;AAEpC,YAAM,UAAU,MAAM,eAAe,mBAAmB;AAAA,QACtD,aAAa;AAAA,QACb,WAAW,QAAQ;AAAA,MACrB,CAAC;AAGD,YAAM,mBAAmB,MAAM,mBAAmB,oBAAoB;AAGtE,UACE,iBAAiB,oBACjB,iBAAiB,eAAe,GAChC;AACA,gBAAQ,IAAI;AAAA,oCAAgC;AAC5C,gBAAQ;AAAA,UACN,MAAM,iBAAiB,YAAY;AAAA,QACrC;AAEA,YAAI,iBAAiB,eAAe,SAAS,GAAG;AAC9C,kBAAQ,IAAI,qBAAqB;AACjC,2BAAiB,eAAe,MAAM,GAAG,CAAC,EAAE,QAAQ,CAAC,MAAM;AACzD,oBAAQ;AAAA,cACN,eAAU,EAAE,IAAI,KAAK,EAAE,QAAQ,MAAM,GAAG,EAAE,CAAC,KAAK,EAAE,SAAS;AAAA,YAC7D;AAAA,UACF,CAAC;AAAA,QACH;AAEA,YAAI,iBAAiB,cAAc,SAAS,GAAG;AAC7C,kBAAQ;AAAA,YACN,qBAAqB,iBAAiB,cAAc,CAAC,EAAE,SAAS,MAAM,GAAG,EAAE,CAAC;AAAA,UAC9E;AAAA,QACF;AAAA,MACF;AAEA,YAAM,KAAK,IAAI,SAAS,MAAM;AAC9B,YAAM,eAAe,IAAI,aAAa,IAAI,QAAQ,SAAS;AAG3D,UAAI,QAAQ,KAAK;AACf,qBAAa,aAAa,eAAe,UAAU;AAAA,MACrD,WAAW,QAAQ,SAAS;AAC1B,qBAAa,aAAa,eAAe,cAAc;AAAA,MACzD;AAEA,YAAM,eAAe,aAAa,mBAAmB;AACrD,YAAM,aAAa,aAAa,cAAc;AAG9C,YAAM,aAAa,GAChB;AAAA,QACC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MASF,EACC,IAAI,QAAQ,SAAS;AAOxB,YAAM,eAAe,GAClB;AAAA,QACC;AAAA;AAAA;AAAA,MAGF,EACC,IAAI;AAEP,YAAM,aAAa,GAChB;AAAA,QACC;AAAA;AAAA;AAAA;AAAA;AAAA,MAKF,EACC,IAAI,QAAQ,SAAS;AAExB,cAAQ,IAAI,+BAAwB;AACpC,cAAQ;AAAA,QACN,eAAe,QAAQ,UAAU,MAAM,GAAG,CAAC,CAAC,KAAK,QAAQ,KAAK,KAAK,KAAK,OAAO,KAAK,IAAI,IAAI,QAAQ,aAAa,MAAO,EAAE,CAAC;AAAA,MAC7H;AACA,cAAQ,IAAI,eAAe,QAAQ,SAAS,EAAE;AAC9C,UAAI,QAAQ,QAAQ;AAClB,gBAAQ,IAAI,cAAc,QAAQ,MAAM,EAAE;AAAA,MAC5C;AAGA,cAAQ,IAAI;AAAA,uCAA0C;AACtD,cAAQ;AAAA,QACN,gBAAgB,WAAW,gBAAgB,CAAC,KAAK,WAAW,iBAAiB,CAAC,YAAY,WAAW,iBAAiB,CAAC;AAAA,MACzH;AACA,cAAQ,IAAI,gBAAgB,WAAW,SAAS,CAAC,EAAE;AACnD,cAAQ,IAAI,kBAAkB,WAAW,kBAAkB,CAAC,EAAE;AAC9D,cAAQ;AAAA,QACN,yBAAyB,aAAa,SAAS,CAAC;AAAA,MAClD;AAGA,YAAM,eAAe,GAClB;AAAA,QACC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAOF,EACC,IAAI,QAAQ,SAAS;AAOxB,UAAI,aAAa,SAAS,GAAG;AAC3B,gBAAQ,IAAI;AAAA,oBAAuB;AACnC,qBAAa,QAAQ,CAAC,MAAM;AAC1B,gBAAM,YAAY,EAAE,UAAU,WAAW,cAAO;AAChD,kBAAQ;AAAA,YACN,QAAQ,SAAS,IAAI,EAAE,IAAI,KAAK,EAAE,IAAI,OAAO,EAAE,OAAO;AAAA,UACxD;AAAA,QACF,CAAC;AAAA,MACH;AAEA,cAAQ,IAAI;AAAA,oBAAuB;AACnC,cAAQ,IAAI,qBAAqB,UAAU,EAAE;AAC7C,cAAQ,IAAI,uBAAuB,aAAa,MAAM,EAAE;AAExD,UAAI,aAAa,SAAS,GAAG;AAC3B,qBAAa,QAAQ,CAAC,OAAO,MAAM;AACjC,gBAAM,SAAS,UAAU,KAAK,OAAO,MAAM,SAAS,CAAC;AACrD,gBAAM,SAAS,MAAM,IAAI,iBAAO;AAChC,kBAAQ,IAAI,GAAG,MAAM,GAAG,MAAM,IAAI,MAAM,IAAI,KAAK,MAAM,IAAI,GAAG;AAAA,QAChE,CAAC;AAAA,MACH;AAGA,UAAI,CAAC,QAAQ,OAAO,CAAC,QAAQ,SAAS;AACpC,cAAM,gBAAgB,MAAM,eAAe,aAAa;AAAA,UACtD,WAAW,QAAQ;AAAA,UACnB,OAAO;AAAA,QACT,CAAC;AAED,cAAM,cAAc,cAAc;AAAA,UAChC,CAAC,MAAM,EAAE,cAAc,QAAQ;AAAA,QACjC;AACA,YAAI,YAAY,SAAS,GAAG;AAC1B,kBAAQ,IAAI;AAAA,yCAA4C;AACxD,sBAAY,QAAQ,CAAC,MAAM;AACzB,kBAAM,MAAM,KAAK;AAAA,eACd,KAAK,IAAI,IAAI,EAAE,gBAAgB,MAAO,KAAK;AAAA,YAC9C;AACA,oBAAQ;AAAA,cACN,UAAU,EAAE,UAAU,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,UAAU,MAAM,KAAK,GAAG;AAAA,YAClE;AAAA,UACF,CAAC;AACD,kBAAQ,IAAI;AAAA,gDAAmD;AAAA,QACjE;AAAA,MACF;AAEA,SAAG,MAAM;AAAA,IACX,SAAS,OAAgB;AACvB,aAAO,MAAM,wBAAwB,KAAc;AACnD,cAAQ,MAAM,+BAA2B,MAAgB,OAAO;AAChE,cAAQ,KAAK,CAAC;AAAA,IAChB;AAAA,EACF,CAAC;AACH,CAAC;AAEH,QACG,QAAQ,cAAc,EACtB,YAAY,+BAA+B,EAC3C,OAAO,YAAY;AAClB,MAAI;AACF,YAAQ,IAAI,mCAA4B;AACxC,UAAM,cAAc,WAAW,OAAO;AAAA,EACxC,SAAS,OAAgB;AACvB,WAAO,MAAM,uBAAuB,KAAc;AAClD,YAAQ,MAAM,+BAA2B,MAAgB,OAAO;AAChE,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF,CAAC;AAEH,QACG,QAAQ,UAAU,EAClB,YAAY,0CAA0C,EACtD,OAAO,YAAY;AAClB,MAAI;AACF,UAAM,cAAc,QAAQ,IAAI;AAChC,UAAM,SAAS,KAAK,aAAa,gBAAgB,YAAY;AAE7D,QAAI,CAAC,WAAW,MAAM,GAAG;AACvB,cAAQ;AAAA,QACN;AAAA,MACF;AACA;AAAA,IACF;AAEA,UAAM,WAAW,IAAI,gBAAgB,WAAW;AAChD,YAAQ,IAAI,SAAS,WAAW,CAAC;AAAA,EACnC,SAAS,OAAgB;AACvB,WAAO,MAAM,2BAA2B,KAAc;AACtD,YAAQ,MAAM,mCAA+B,MAAgB,OAAO;AACpE,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF,CAAC;AAEH,QACG,QAAQ,YAAY,EACpB,YAAY,iDAAiD,EAC7D,OAAO,wBAAwB,0BAA0B,QAAQ,IAAI,CAAC,EACtE,OAAO,OAAO,YAAY;AACzB,MAAI;AACF,UAAM,EAAE,aAAa,IAAI,MAAM,OAAO,+BAA+B;AAGrE,YAAQ,IAAI,cAAc,IAAI,QAAQ;AAEtC,YAAQ,IAAI,8CAAuC;AACnD,YAAQ,IAAI,eAAe,QAAQ,OAAO,EAAE;AAC5C,YAAQ,IAAI,eAAe,OAAO,EAAE;AAGpC,kBAAc,gBAAgB,SAAS,IAAI,EAAE,MAAM,MAAM;AAAA,IAAC,CAAC;AAG3D,UAAM,aAAa;AAAA,EACrB,SAAS,OAAgB;AACvB,WAAO,MAAM,8BAA8B,KAAc;AACzD,YAAQ,MAAM,6BAAyB,MAAgB,OAAO;AAC9D,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF,CAAC;AAGH,QACG,QAAQ,cAAc,EACtB,YAAY,oDAAoD,EAChE,OAAO,YAAY;AAClB,MAAI;AACF,UAAM,cAAc,QAAQ,IAAI;AAChC,UAAM,SAAS,KAAK,aAAa,gBAAgB,YAAY;AAE7D,QAAI,CAAC,WAAW,MAAM,GAAG;AACvB,cAAQ;AAAA,QACN;AAAA,MACF;AACA;AAAA,IACF;AAEA,UAAM,KAAK,IAAI,SAAS,MAAM;AAC9B,UAAM,eAAe,IAAI,aAAa,IAAI,aAAa;AAGvD,YAAQ,IAAI,2CAAoC;AAEhD,UAAM,YAAY,aAAa,YAAY;AAAA,MACzC,MAAM;AAAA,MACN,MAAM;AAAA,MACN,QAAQ,EAAE,MAAM,MAAM,YAAW,oBAAI,KAAK,GAAE,YAAY,EAAE;AAAA,IAC5D,CAAC;AAED,UAAM,YAAY,aAAa,YAAY;AAAA,MACzC,MAAM;AAAA,MACN,MAAM;AAAA,MACN,QAAQ,EAAE,aAAa,8BAA8B;AAAA,MACrD,eAAe;AAAA,IACjB,CAAC;AAED,UAAM,eAAe,aAAa,YAAY;AAAA,MAC5C,MAAM;AAAA,MACN,MAAM;AAAA,MACN,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE;AAAA,MAC3B,eAAe;AAAA,IACjB,CAAC;AAGD,iBAAa;AAAA,MACX;AAAA,MACA;AAAA,QACE,SAAS;AAAA,MACX;AAAA,MACA;AAAA,IACF;AAEA,YAAQ,IAAI,6BAAwB;AACpC,YAAQ,IAAI,0BAAmB,aAAa,cAAc,CAAC,EAAE;AAC7D,YAAQ;AAAA,MACN,4BAAqB,aAAa,mBAAmB,EAAE,MAAM;AAAA,IAC/D;AAGA,iBAAa,WAAW,YAAY;AACpC,YAAQ;AAAA,MACN,kDAA2C,aAAa,cAAc,CAAC;AAAA,IACzE;AAEA,OAAG,MAAM;AAAA,EACX,SAAS,OAAgB;AACvB,WAAO,MAAM,uBAAuB,KAAc;AAClD,YAAQ,MAAM,uBAAmB,MAAgB,OAAO;AACxD,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF,CAAC;AAIH,0BAA0B,OAAO;AACjC,sBAAsB,OAAO;AAC7B,qBAAqB,OAAO;AAC5B,sBAAsB,OAAO;AAC7B,mBAAmB,OAAO;AAC1B,wBAAwB,OAAO;AAC/B,yBAAyB,OAAO;AAGhC,uBAAuB,OAAO;AAG9B,QAAQ,WAAW,sBAAsB,CAAC;AAG1C,QAAQ,WAAW,mBAAmB,CAAC;AACvC,QAAQ,WAAW,oBAAoB,CAAC;AACxC,QAAQ,WAAW,iBAAiB,CAAC;AACrC,QAAQ,WAAW,sBAAsB,CAAC;AAC1C,QAAQ,WAAW,oBAAoB,CAAC;AACxC,QAAQ,WAAW,qBAAqB,CAAC;AACzC,QAAQ,WAAW,sBAAsB,CAAC;AAC1C,QAAQ,WAAW,oBAAoB,CAAC;AACxC,QAAQ,WAAW,qBAAqB,CAAC;AACzC,QAAQ,WAAW,oBAAoB,CAAC;AACxC,QAAQ,WAAW,kBAAkB,CAAC;AACtC,QAAQ,WAAW,YAAY;AAC/B,QAAQ,WAAW,sBAAsB,CAAC;AAC1C,QAAQ,WAAW,cAAc;AACjC,QAAQ,WAAW,cAAc;AACjC,QAAQ,WAAW,mBAAmB,CAAC;AACvC,QAAQ,WAAW,cAAc;AACjC,QAAQ,WAAW,mBAAmB,CAAC;AAGvC,QACG,QAAQ,WAAW,EACnB,YAAY,0CAA0C,EACtD,OAAO,eAAe,wBAAwB,EAC9C,OAAO,4BAA4B,+BAA+B,GAAG,EACrE,OAAO,OAAO,YAAY;AACzB,QAAM,EAAE,iBAAiB,IAAI,MAAM,OAAO,yBAAyB;AACnE,QAAM,iBAAiB,QAAQ,OAAO;AACxC,CAAC;AAGH,IAAI,QAAQ,KAAK,SAAS,GAAG;AAC3B,QAAM,UAAU,eAAe,YAAY;AAC3C,UAAQ,cAAc,EAAE,MAAM,MAAM;AAAA,EAEpC,CAAC;AACH;AAGA,MAAM,eACJ,YAAY,QAAQ,UAAU,QAAQ,KAAK,CAAC,CAAC,MAC7C,QAAQ,KAAK,CAAC,GAAG,SAAS,cAAc,KACxC,QAAQ,KAAK,CAAC,GAAG,SAAS,UAAU,KACpC,QAAQ,KAAK,CAAC,GAAG,SAAS,KAAK;AAEjC,IAAI,cAAc;AAChB,UAAQ,MAAM;AAChB;",
   "names": []
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stackmemoryai/stackmemory",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "description": "Lossless memory runtime for AI coding tools - organizes context as a call stack instead of linear chat logs, with team collaboration and infinite retention",
   "engines": {
     "node": ">=20.0.0",

package/scripts/install-sweep-hook.sh

@@ -0,0 +1,89 @@
+#!/bin/bash
+# Install Sweep prediction hook for Claude Code
+
+set -e
+
+HOOK_DIR="$HOME/.claude/hooks"
+SWEEP_DIR="$HOME/.stackmemory/sweep"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+REPO_DIR="$(dirname "$SCRIPT_DIR")"
+
+echo "Installing Sweep prediction hook for Claude Code..."
+
+# Create directories
+mkdir -p "$HOOK_DIR"
+mkdir -p "$SWEEP_DIR"
+
+# Copy hook script
+cp "$REPO_DIR/templates/claude-hooks/post-edit-sweep.js" "$HOOK_DIR/"
+chmod +x "$HOOK_DIR/post-edit-sweep.js"
+
+# Copy Python prediction script
+cp "$REPO_DIR/packages/sweep-addon/python/sweep_predict.py" "$SWEEP_DIR/"
+
+# Update hooks.json if it exists, otherwise create it
+HOOKS_JSON="$HOME/.claude/hooks.json"
+if [ -f "$HOOKS_JSON" ]; then
+    # Check if post-tool-use already configured
+    if grep -q "post-tool-use" "$HOOKS_JSON"; then
+        echo "Note: post-tool-use hook already configured in $HOOKS_JSON"
+        echo "You may need to manually add the sweep hook."
+    else
+        echo "Adding sweep hook to $HOOKS_JSON..."
+        # Use node to safely update JSON
+        node -e "
+const fs = require('fs');
+const hooks = JSON.parse(fs.readFileSync('$HOOKS_JSON', 'utf-8'));
+hooks['post-tool-use'] = '$HOOK_DIR/post-edit-sweep.js';
+fs.writeFileSync('$HOOKS_JSON', JSON.stringify(hooks, null, 2));
+console.log('Updated hooks.json');
+"
+    fi
+else
+    echo "Creating $HOOKS_JSON..."
+    cat > "$HOOKS_JSON" << 'EOF'
+{
+  "post-tool-use": "~/.claude/hooks/post-edit-sweep.js"
+}
+EOF
+fi
+
+# Check Python dependencies
+echo ""
+echo "Checking Python dependencies..."
+if python3 -c "import llama_cpp" 2>/dev/null; then
+    echo "  llama-cpp-python: installed"
+else
+    echo "  llama-cpp-python: NOT INSTALLED"
+    echo "  Run: pip install llama-cpp-python"
+fi
+
+if python3 -c "import huggingface_hub" 2>/dev/null; then
+    echo "  huggingface_hub: installed"
+else
+    echo "  huggingface_hub: NOT INSTALLED"
+    echo "  Run: pip install huggingface_hub"
+fi
+
+# Check model
+MODEL_PATH="$HOME/.stackmemory/models/sweep/sweep-next-edit-1.5b.q8_0.v2.gguf"
+if [ -f "$MODEL_PATH" ]; then
+    echo "  Model: downloaded"
+else
+    echo "  Model: NOT DOWNLOADED"
+    echo "  Run: stackmemory sweep setup --download"
+fi
+
+echo ""
+echo "Installation complete!"
+echo ""
+echo "Hook installed at: $HOOK_DIR/post-edit-sweep.js"
+echo "Python script at: $SWEEP_DIR/sweep_predict.py"
+echo ""
+echo "Usage:"
+echo "  - Hook runs automatically after Edit/Write operations"
+echo "  - Predictions appear after 2+ edits in session"
+echo "  - Check status: node $HOOK_DIR/post-edit-sweep.js --status"
+echo "  - Clear state: node $HOOK_DIR/post-edit-sweep.js --clear"
+echo "  - Disable: export SWEEP_ENABLED=false"
+echo ""

package/templates/claude-hooks/post-edit-sweep.js

@@ -0,0 +1,437 @@
+#!/usr/bin/env node
+
+/**
+ * Post-Edit Sweep Hook for Claude Code
+ *
+ * Runs Sweep 1.5B predictions after file edits to suggest next changes.
+ * Tracks recent diffs and provides context-aware predictions.
+ */
+
+import fs from 'fs';
+import path from 'path';
+import { spawn } from 'child_process';
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const CONFIG = {
+  enabled: process.env.SWEEP_ENABLED !== 'false',
+  maxRecentDiffs: 5,
+  predictionTimeout: 30000,
+  minEditSize: 10,
+  debounceMs: 2000,
+  minDiffsForPrediction: 2,
+  cooldownMs: 10000,
+  codeExtensions: [
+    '.ts',
+    '.tsx',
+    '.js',
+    '.jsx',
+    '.py',
+    '.go',
+    '.rs',
+    '.java',
+    '.c',
+    '.cpp',
+    '.h',
+    '.hpp',
+    '.cs',
+    '.rb',
+    '.php',
+    '.swift',
+    '.kt',
+    '.scala',
+    '.vue',
+    '.svelte',
+    '.astro',
+  ],
+  stateFile: path.join(
+    process.env.HOME || '/tmp',
+    '.stackmemory',
+    'sweep-state.json'
+  ),
+  logFile: path.join(
+    process.env.HOME || '/tmp',
+    '.stackmemory',
+    'sweep-predictions.log'
+  ),
+  pythonScript: path.join(
+    process.env.HOME || '/tmp',
+    '.stackmemory',
+    'sweep',
+    'sweep_predict.py'
+  ),
+};
+
+// Fallback locations for sweep_predict.py
+const SCRIPT_LOCATIONS = [
+  CONFIG.pythonScript,
+  path.join(
+    process.cwd(),
+    'packages',
+    'sweep-addon',
+    'python',
+    'sweep_predict.py'
+  ),
+  path.join(
+    process.cwd(),
+    'node_modules',
+    '@stackmemoryai',
+    'sweep-addon',
+    'python',
+    'sweep_predict.py'
+  ),
+];
+
+function findPythonScript() {
+  for (const loc of SCRIPT_LOCATIONS) {
+    if (fs.existsSync(loc)) {
+      return loc;
+    }
+  }
+  return null;
+}
+
+function loadState() {
+  try {
+    if (fs.existsSync(CONFIG.stateFile)) {
+      return JSON.parse(fs.readFileSync(CONFIG.stateFile, 'utf-8'));
+    }
+  } catch {
+    // Ignore errors
+  }
+  return {
+    recentDiffs: [],
+    lastPrediction: null,
+    pendingPrediction: null,
+    fileContents: {},
+  };
+}
+
+function saveState(state) {
+  try {
+    const dir = path.dirname(CONFIG.stateFile);
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+    fs.writeFileSync(CONFIG.stateFile, JSON.stringify(state, null, 2));
+  } catch {
+    // Ignore errors
+  }
+}
+
+function log(message, data = {}) {
+  try {
+    const dir = path.dirname(CONFIG.logFile);
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+    const entry = {
+      timestamp: new Date().toISOString(),
+      message,
+      ...data,
+    };
+    fs.appendFileSync(CONFIG.logFile, JSON.stringify(entry) + '\n');
+  } catch {
+    // Ignore
+  }
+}
+
+async function runPrediction(filePath, currentContent, recentDiffs) {
+  const scriptPath = findPythonScript();
+  if (!scriptPath) {
+    log('Sweep script not found');
+    return null;
+  }
+
+  const input = {
+    file_path: filePath,
+    current_content: currentContent,
+    recent_diffs: recentDiffs,
+  };
+
+  return new Promise((resolve) => {
+    const proc = spawn('python3', [scriptPath], {
+      stdio: ['pipe', 'pipe', 'pipe'],
+      timeout: CONFIG.predictionTimeout,
+    });
+
+    let stdout = '';
+    let stderr = '';
+
+    proc.stdout.on('data', (data) => (stdout += data));
+    proc.stderr.on('data', (data) => (stderr += data));
+
+    const timeout = setTimeout(() => {
+      proc.kill();
+      resolve(null);
+    }, CONFIG.predictionTimeout);
+
+    proc.on('close', (code) => {
+      clearTimeout(timeout);
+      try {
+        if (stdout.trim()) {
+          const result = JSON.parse(stdout.trim());
+          resolve(result);
+        } else {
+          resolve(null);
+        }
+      } catch {
+        resolve(null);
+      }
+    });
+
+    proc.on('error', () => {
+      clearTimeout(timeout);
+      resolve(null);
+    });
+
+    proc.stdin.write(JSON.stringify(input));
+    proc.stdin.end();
+  });
+}
+
+async function readInput() {
+  let input = '';
+  for await (const chunk of process.stdin) {
+    input += chunk;
+  }
+  return JSON.parse(input);
+}
+
+function isCodeFile(filePath) {
+  const ext = path.extname(filePath).toLowerCase();
+  return CONFIG.codeExtensions.includes(ext);
+}
+
+function shouldRunPrediction(state, filePath) {
+  if (state.recentDiffs.length < CONFIG.minDiffsForPrediction) {
+    return false;
+  }
+
+  if (state.lastPrediction) {
+    const timeSince = Date.now() - state.lastPrediction.timestamp;
+    if (timeSince < CONFIG.cooldownMs) {
+      return false;
+    }
+  }
+
+  if (state.pendingPrediction) {
+    const timeSince = Date.now() - state.pendingPrediction;
+    if (timeSince < CONFIG.debounceMs) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
+async function handleEdit(toolInput, toolResult) {
+  if (!CONFIG.enabled) return;
+
+  const { file_path, old_string, new_string } = toolInput;
+  if (!file_path || !old_string || !new_string) return;
+
+  if (!isCodeFile(file_path)) {
+    log('Skipping non-code file', { file_path });
+    return;
+  }
+
+  if (
+    new_string.length < CONFIG.minEditSize &&
+    old_string.length < CONFIG.minEditSize
+  ) {
+    return;
+  }
+
+  const state = loadState();
+
+  const diff = {
+    file_path,
+    original: old_string,
+    updated: new_string,
+    timestamp: Date.now(),
+  };
+
+  state.recentDiffs.unshift(diff);
+  state.recentDiffs = state.recentDiffs.slice(0, CONFIG.maxRecentDiffs);
+
+  try {
+    if (fs.existsSync(file_path)) {
+      state.fileContents[file_path] = fs.readFileSync(file_path, 'utf-8');
+    }
+  } catch {
+    // Ignore
+  }
+
+  saveState(state);
+  log('Edit recorded', { file_path, diffSize: new_string.length });
+
+  if (shouldRunPrediction(state, file_path)) {
+    state.pendingPrediction = Date.now();
+    saveState(state);
+
+    setTimeout(() => {
+      runPredictionAsync(file_path, loadState());
+    }, CONFIG.debounceMs);
+  }
+}
+
+async function runPredictionAsync(filePath, state) {
+  try {
+    const currentContent = state.fileContents[filePath] || '';
+    if (!currentContent) {
+      state.pendingPrediction = null;
+      saveState(state);
+      return;
+    }
+
+    const result = await runPrediction(
+      filePath,
+      currentContent,
+      state.recentDiffs
+    );
+
+    state.pendingPrediction = null;
+
+    if (result && result.success && result.predicted_content) {
+      state.lastPrediction = {
+        file_path: filePath,
+        prediction: result.predicted_content,
+        latency_ms: result.latency_ms,
+        timestamp: Date.now(),
+      };
+      saveState(state);
+
+      log('Prediction complete', {
+        file_path: filePath,
+        latency_ms: result.latency_ms,
+        tokens: result.tokens_generated,
+      });
+
+      const hint = formatPredictionHint(result);
+      if (hint) {
+        console.error(hint);
+      }
+    } else {
+      saveState(state);
+    }
+  } catch (error) {
+    state.pendingPrediction = null;
+    saveState(state);
+    log('Prediction error', { error: error.message });
+  }
+}
+
+function formatPredictionHint(result) {
+  if (!result.predicted_content || result.predicted_content.trim().length < 5) {
+    return null;
+  }
+
+  const preview = result.predicted_content
+    .trim()
+    .split('\n')
+    .slice(0, 3)
+    .join('\n');
+  const truncated = result.predicted_content.length > 200;
+
+  return `
+[Sweep Prediction] Next edit suggestion (${result.latency_ms}ms):
+${preview}${truncated ? '\n...' : ''}
+`;
+}
+
+async function handleWrite(toolInput, toolResult) {
+  if (!CONFIG.enabled) return;
+
+  const { file_path, content } = toolInput;
+  if (!file_path || !content) return;
+
+  if (!isCodeFile(file_path)) {
+    return;
+  }
+
+  const state = loadState();
+  state.fileContents[file_path] = content;
+  saveState(state);
+
+  log('Write recorded', { file_path, size: content.length });
+}
+
+async function main() {
+  try {
+    const input = await readInput();
+    const { tool_name, tool_input, tool_result, event_type } = input;
+
+    // Only handle post-tool-use events
+    if (event_type !== 'post_tool_use') {
+      process.exit(0);
+    }
+
+    // Handle different tools
+    switch (tool_name) {
+      case 'Edit':
+        await handleEdit(tool_input, tool_result);
+        break;
+      case 'Write':
+        await handleWrite(tool_input, tool_result);
+        break;
+    }
+
+    // Success
+    console.log(JSON.stringify({ status: 'ok' }));
+  } catch (error) {
+    log('Hook error', { error: error.message });
+    console.log(JSON.stringify({ status: 'error', message: error.message }));
+  }
+}
+
+// Handle info request
+if (process.argv.includes('--info')) {
+  console.log(
+    JSON.stringify({
+      hook: 'post-edit-sweep',
+      version: '1.0.0',
+      description: 'Runs Sweep 1.5B predictions after file edits',
+      config: {
+        enabled: CONFIG.enabled,
+        maxRecentDiffs: CONFIG.maxRecentDiffs,
+        predictionTimeout: CONFIG.predictionTimeout,
+      },
+    })
+  );
+  process.exit(0);
+}
+
+// Handle status request
+if (process.argv.includes('--status')) {
+  const state = loadState();
+  const scriptPath = findPythonScript();
+  console.log(
+    JSON.stringify(
+      {
+        enabled: CONFIG.enabled,
+        scriptFound: !!scriptPath,
+        scriptPath,
+        recentDiffs: state.recentDiffs.length,
+        lastPrediction: state.lastPrediction,
+      },
+      null,
+      2
+    )
+  );
+  process.exit(0);
+}
+
+// Handle clear request
+if (process.argv.includes('--clear')) {
+  saveState({ recentDiffs: [], lastPrediction: null, fileContents: {} });
+  console.log('Sweep state cleared');
+  process.exit(0);
+}
+
+main().catch((error) => {
+  console.error(JSON.stringify({ status: 'error', message: error.message }));
+  process.exit(1);
+});