@stackmemoryai/stackmemory 0.5.3 → 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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stackmemoryai/stackmemory",
-  "version": "0.5.3",
+  "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);
+});