@stackmemoryai/stackmemory 0.5.58 → 0.5.59

package/dist/src/features/sweep/status-bar.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../../src/features/sweep/status-bar.ts"],
+  "sourcesContent": ["/**\n * Sweep Status Bar\n *\n * Renders a 1-row prediction status bar at the bottom of the terminal\n * using ANSI escape sequences. Preserves cursor position.\n */\n\nimport { basename } from 'path';\n\n// ANSI escape sequences\nconst ESC = '\\x1b';\nconst SAVE_CURSOR = `${ESC}7`;\nconst RESTORE_CURSOR = `${ESC}8`;\nconst CLEAR_LINE = `${ESC}[2K`;\nconst RESET = `${ESC}[0m`;\nconst BG_DARK = `${ESC}[48;5;236m`; // dark gray bg\nconst FG_GRAY = `${ESC}[38;5;244m`; // gray text\nconst FG_CYAN = `${ESC}[38;5;37m`; // cyan for label\nconst FG_DIM = `${ESC}[2m`; // dim\nconst BOLD = `${ESC}[1m`;\n\nfunction moveTo(row: number, col: number): string {\n  return `${ESC}[${row};${col}H`;\n}\n\nexport class StatusBar {\n  private rows: number;\n  private cols: number;\n  private visible = false;\n  private currentPrediction = '';\n  private currentFile = '';\n\n  constructor() {\n    this.rows = process.stdout.rows || 24;\n    this.cols = process.stdout.columns || 80;\n  }\n\n  show(prediction: string, filePath: string, latencyMs: number): void {\n    this.currentPrediction = prediction;\n    this.currentFile = filePath;\n    this.visible = true;\n\n    const file = basename(filePath);\n    const preview = this.truncatePreview(prediction);\n    const latency = `${latencyMs}ms`;\n\n    // Build status bar content\n    const label = `${FG_CYAN}${BOLD}[Sweep]${RESET}${BG_DARK}`;\n    const fileInfo = `${FG_GRAY} ${file}${RESET}${BG_DARK}`;\n    const content = `${FG_GRAY} ${preview}${RESET}${BG_DARK}`;\n    const time = `${FG_DIM}${BG_DARK} ${latency}${RESET}${BG_DARK}`;\n    const keys = `${BOLD}${BG_DARK}  [Tab]${RESET}${BG_DARK}${FG_GRAY} Accept  ${BOLD}${BG_DARK}[Esc]${RESET}${BG_DARK}${FG_GRAY} Dismiss${RESET}`;\n\n    const bar = `${BG_DARK}${label}${fileInfo}${content}${time}${keys}${RESET}`;\n\n    this.render(bar);\n  }\n\n  showLoading(): void {\n    this.visible = true;\n    const bar = `${BG_DARK}${FG_CYAN}${BOLD}[Sweep]${RESET}${BG_DARK}${FG_GRAY} Predicting next edit...${RESET}`;\n    this.render(bar);\n  }\n\n  hide(): void {\n    if (!this.visible) return;\n    this.visible = false;\n    this.currentPrediction = '';\n    this.currentFile = '';\n\n    const output =\n      SAVE_CURSOR + moveTo(this.rows, 1) + CLEAR_LINE + RESTORE_CURSOR;\n\n    process.stdout.write(output);\n  }\n\n  resize(rows: number, cols: number): void {\n    this.rows = rows;\n    this.cols = cols;\n    if (this.visible && this.currentPrediction) {\n      this.show(this.currentPrediction, this.currentFile, 0);\n    }\n  }\n\n  isVisible(): boolean {\n    return this.visible;\n  }\n\n  private render(content: string): void {\n    if (!process.stdout.isTTY) return;\n\n    const output =\n      SAVE_CURSOR +\n      moveTo(this.rows, 1) +\n      CLEAR_LINE +\n      content +\n      RESTORE_CURSOR;\n\n    process.stdout.write(output);\n  }\n\n  private truncatePreview(prediction: string): string {\n    // Take first non-empty line\n    const lines = prediction.trim().split('\\n');\n    let preview = lines[0] || '';\n\n    // Max preview length: cols minus label/keys overhead (~50 chars)\n    const maxLen = Math.max(10, this.cols - 55);\n    if (preview.length > maxLen) {\n      preview = preview.slice(0, maxLen - 3) + '...';\n    }\n\n    // Indicate more lines\n    if (lines.length > 1) {\n      preview += ` (+${lines.length - 1} lines)`;\n    }\n\n    return preview;\n  }\n}\n"],
+  "mappings": ";;;;AAOA,SAAS,gBAAgB;AAGzB,MAAM,MAAM;AACZ,MAAM,cAAc,GAAG,GAAG;AAC1B,MAAM,iBAAiB,GAAG,GAAG;AAC7B,MAAM,aAAa,GAAG,GAAG;AACzB,MAAM,QAAQ,GAAG,GAAG;AACpB,MAAM,UAAU,GAAG,GAAG;AACtB,MAAM,UAAU,GAAG,GAAG;AACtB,MAAM,UAAU,GAAG,GAAG;AACtB,MAAM,SAAS,GAAG,GAAG;AACrB,MAAM,OAAO,GAAG,GAAG;AAEnB,SAAS,OAAO,KAAa,KAAqB;AAChD,SAAO,GAAG,GAAG,IAAI,GAAG,IAAI,GAAG;AAC7B;AAEO,MAAM,UAAU;AAAA,EACb;AAAA,EACA;AAAA,EACA,UAAU;AAAA,EACV,oBAAoB;AAAA,EACpB,cAAc;AAAA,EAEtB,cAAc;AACZ,SAAK,OAAO,QAAQ,OAAO,QAAQ;AACnC,SAAK,OAAO,QAAQ,OAAO,WAAW;AAAA,EACxC;AAAA,EAEA,KAAK,YAAoB,UAAkB,WAAyB;AAClE,SAAK,oBAAoB;AACzB,SAAK,cAAc;AACnB,SAAK,UAAU;AAEf,UAAM,OAAO,SAAS,QAAQ;AAC9B,UAAM,UAAU,KAAK,gBAAgB,UAAU;AAC/C,UAAM,UAAU,GAAG,SAAS;AAG5B,UAAM,QAAQ,GAAG,OAAO,GAAG,IAAI,UAAU,KAAK,GAAG,OAAO;AACxD,UAAM,WAAW,GAAG,OAAO,IAAI,IAAI,GAAG,KAAK,GAAG,OAAO;AACrD,UAAM,UAAU,GAAG,OAAO,IAAI,OAAO,GAAG,KAAK,GAAG,OAAO;AACvD,UAAM,OAAO,GAAG,MAAM,GAAG,OAAO,IAAI,OAAO,GAAG,KAAK,GAAG,OAAO;AAC7D,UAAM,OAAO,GAAG,IAAI,GAAG,OAAO,UAAU,KAAK,GAAG,OAAO,GAAG,OAAO,YAAY,IAAI,GAAG,OAAO,QAAQ,KAAK,GAAG,OAAO,GAAG,OAAO,WAAW,KAAK;AAE5I,UAAM,MAAM,GAAG,OAAO,GAAG,KAAK,GAAG,QAAQ,GAAG,OAAO,GAAG,IAAI,GAAG,IAAI,GAAG,KAAK;AAEzE,SAAK,OAAO,GAAG;AAAA,EACjB;AAAA,EAEA,cAAoB;AAClB,SAAK,UAAU;AACf,UAAM,MAAM,GAAG,OAAO,GAAG,OAAO,GAAG,IAAI,UAAU,KAAK,GAAG,OAAO,GAAG,OAAO,2BAA2B,KAAK;AAC1G,SAAK,OAAO,GAAG;AAAA,EACjB;AAAA,EAEA,OAAa;AACX,QAAI,CAAC,KAAK,QAAS;AACnB,SAAK,UAAU;AACf,SAAK,oBAAoB;AACzB,SAAK,cAAc;AAEnB,UAAM,SACJ,cAAc,OAAO,KAAK,MAAM,CAAC,IAAI,aAAa;AAEpD,YAAQ,OAAO,MAAM,MAAM;AAAA,EAC7B;AAAA,EAEA,OAAO,MAAc,MAAoB;AACvC,SAAK,OAAO;AACZ,SAAK,OAAO;AACZ,QAAI,KAAK,WAAW,KAAK,mBAAmB;AAC1C,WAAK,KAAK,KAAK,mBAAmB,KAAK,aAAa,CAAC;AAAA,IACvD;AAAA,EACF;AAAA,EAEA,YAAqB;AACnB,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ,OAAO,SAAuB;AACpC,QAAI,CAAC,QAAQ,OAAO,MAAO;AAE3B,UAAM,SACJ,cACA,OAAO,KAAK,MAAM,CAAC,IACnB,aACA,UACA;AAEF,YAAQ,OAAO,MAAM,MAAM;AAAA,EAC7B;AAAA,EAEQ,gBAAgB,YAA4B;AAElD,UAAM,QAAQ,WAAW,KAAK,EAAE,MAAM,IAAI;AAC1C,QAAI,UAAU,MAAM,CAAC,KAAK;AAG1B,UAAM,SAAS,KAAK,IAAI,IAAI,KAAK,OAAO,EAAE;AAC1C,QAAI,QAAQ,SAAS,QAAQ;AAC3B,gBAAU,QAAQ,MAAM,GAAG,SAAS,CAAC,IAAI;AAAA,IAC3C;AAGA,QAAI,MAAM,SAAS,GAAG;AACpB,iBAAW,MAAM,MAAM,SAAS,CAAC;AAAA,IACnC;AAEA,WAAO;AAAA,EACT;AACF;",
+  "names": []
+}

package/dist/src/features/sweep/sweep-server-manager.js

@@ -0,0 +1,226 @@
+import { fileURLToPath as __fileURLToPath } from 'url';
+import { dirname as __pathDirname } from 'path';
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+import { spawn, execSync } from "child_process";
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from "fs";
+import { join, dirname } from "path";
+import {
+  DEFAULT_SERVER_CONFIG
+} from "./types.js";
+import { createPredictionClient } from "./prediction-client.js";
+import { logger } from "../../core/monitoring/logger.js";
+const HOME = process.env["HOME"] || "/tmp";
+const PID_FILE = join(HOME, ".stackmemory", "sweep", "server.pid");
+const LOG_FILE = join(HOME, ".stackmemory", "sweep", "server.log");
+class SweepServerManager {
+  config;
+  process = null;
+  constructor(config = {}) {
+    this.config = { ...DEFAULT_SERVER_CONFIG, ...config };
+    if (!this.config.modelPath) {
+      this.config.modelPath = join(
+        HOME,
+        ".stackmemory",
+        "models",
+        "sweep",
+        "sweep-next-edit-1.5b.q8_0.v2.gguf"
+      );
+    }
+  }
+  /**
+   * Find llama-server executable
+   */
+  findLlamaServer() {
+    const candidates = [
+      "llama-server",
+      "llama.cpp/llama-server",
+      "/usr/local/bin/llama-server",
+      "/opt/homebrew/bin/llama-server",
+      join(HOME, ".local", "bin", "llama-server")
+    ];
+    for (const cmd of candidates) {
+      try {
+        execSync(`which ${cmd}`, { stdio: "ignore" });
+        return cmd;
+      } catch {
+        if (existsSync(cmd)) {
+          return cmd;
+        }
+      }
+    }
+    return null;
+  }
+  /**
+   * Start the llama-server
+   */
+  async startServer() {
+    const status = await this.getStatus();
+    if (status.running) {
+      return status;
+    }
+    if (!existsSync(this.config.modelPath)) {
+      throw new Error(
+        `Model not found: ${this.config.modelPath}
+Download with: huggingface-cli download sweepai/sweep-next-edit-1.5B sweep-next-edit-1.5b.q8_0.v2.gguf --local-dir ~/.stackmemory/models/sweep`
+      );
+    }
+    const llamaServer = this.findLlamaServer();
+    if (!llamaServer) {
+      throw new Error(
+        "llama-server not found. Install with:\n  brew install llama.cpp\nor build from source: https://github.com/ggerganov/llama.cpp"
+      );
+    }
+    const logDir = dirname(LOG_FILE);
+    if (!existsSync(logDir)) {
+      mkdirSync(logDir, { recursive: true });
+    }
+    const args = [
+      "-m",
+      this.config.modelPath,
+      "--port",
+      String(this.config.port),
+      "--host",
+      this.config.host,
+      "-c",
+      String(this.config.contextSize)
+    ];
+    if (this.config.threads) {
+      args.push("-t", String(this.config.threads));
+    }
+    if (this.config.gpuLayers && this.config.gpuLayers > 0) {
+      args.push("-ngl", String(this.config.gpuLayers));
+    }
+    logger.info("Starting Sweep server", { llamaServer, args });
+    this.process = spawn(llamaServer, args, {
+      detached: true,
+      stdio: ["ignore", "pipe", "pipe"]
+    });
+    if (this.process.pid) {
+      const pidDir = dirname(PID_FILE);
+      if (!existsSync(pidDir)) {
+        mkdirSync(pidDir, { recursive: true });
+      }
+      writeFileSync(
+        PID_FILE,
+        JSON.stringify({
+          pid: this.process.pid,
+          port: this.config.port,
+          host: this.config.host,
+          modelPath: this.config.modelPath,
+          startedAt: Date.now()
+        })
+      );
+    }
+    this.process.unref();
+    const ready = await this.waitForReady(1e4);
+    if (!ready) {
+      await this.stopServer();
+      throw new Error("Server failed to start within timeout");
+    }
+    return this.getStatus();
+  }
+  /**
+   * Wait for server to be ready
+   */
+  async waitForReady(timeoutMs) {
+    const client = createPredictionClient({
+      port: this.config.port,
+      host: this.config.host
+    });
+    const startTime = Date.now();
+    while (Date.now() - startTime < timeoutMs) {
+      if (await client.checkHealth()) {
+        return true;
+      }
+      await new Promise((resolve) => setTimeout(resolve, 500));
+    }
+    return false;
+  }
+  /**
+   * Stop the server
+   */
+  async stopServer() {
+    const status = await this.getStatus();
+    if (!status.running || !status.pid) {
+      return;
+    }
+    try {
+      process.kill(status.pid, "SIGTERM");
+      await new Promise((resolve) => {
+        const checkInterval = setInterval(() => {
+          try {
+            process.kill(status.pid, 0);
+          } catch {
+            clearInterval(checkInterval);
+            resolve();
+          }
+        }, 100);
+        setTimeout(() => {
+          clearInterval(checkInterval);
+          try {
+            process.kill(status.pid, "SIGKILL");
+          } catch {
+          }
+          resolve();
+        }, 5e3);
+      });
+    } catch (error) {
+      logger.warn("Error stopping server", { error });
+    }
+    try {
+      if (existsSync(PID_FILE)) {
+        const { unlinkSync } = await import("fs");
+        unlinkSync(PID_FILE);
+      }
+    } catch {
+    }
+  }
+  /**
+   * Get server status
+   */
+  async getStatus() {
+    if (!existsSync(PID_FILE)) {
+      return { running: false };
+    }
+    try {
+      const data = JSON.parse(readFileSync(PID_FILE, "utf-8"));
+      const { pid, port, host, modelPath, startedAt } = data;
+      try {
+        process.kill(pid, 0);
+      } catch {
+        return { running: false };
+      }
+      const client = createPredictionClient({ port, host });
+      const healthy = await client.checkHealth();
+      return {
+        running: healthy,
+        pid,
+        port,
+        host,
+        modelPath,
+        startedAt
+      };
+    } catch {
+      return { running: false };
+    }
+  }
+  /**
+   * Check server health
+   */
+  async checkHealth() {
+    const client = createPredictionClient({
+      port: this.config.port,
+      host: this.config.host
+    });
+    return client.checkHealth();
+  }
+}
+function createServerManager(config) {
+  return new SweepServerManager(config);
+}
+export {
+  SweepServerManager,
+  createServerManager
+};
+//# sourceMappingURL=sweep-server-manager.js.map

package/dist/src/features/sweep/sweep-server-manager.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../../src/features/sweep/sweep-server-manager.ts"],
+  "sourcesContent": ["/**\n * Sweep Server Manager\n *\n * Manages the llama-server process for Sweep predictions.\n */\n\nimport { spawn, ChildProcess, execSync } from 'child_process';\nimport { existsSync, readFileSync, writeFileSync, mkdirSync } from 'fs';\nimport { join, dirname } from 'path';\nimport {\n  SweepServerConfig,\n  SweepServerStatus,\n  DEFAULT_SERVER_CONFIG,\n} from './types.js';\nimport { createPredictionClient } from './prediction-client.js';\nimport { logger } from '../../core/monitoring/logger.js';\n\nconst HOME = process.env['HOME'] || '/tmp';\nconst PID_FILE = join(HOME, '.stackmemory', 'sweep', 'server.pid');\nconst LOG_FILE = join(HOME, '.stackmemory', 'sweep', 'server.log');\n\nexport class SweepServerManager {\n  private config: SweepServerConfig;\n  private process: ChildProcess | null = null;\n\n  constructor(config: Partial<SweepServerConfig> = {}) {\n    this.config = { ...DEFAULT_SERVER_CONFIG, ...config };\n\n    // Set default model path if not provided\n    if (!this.config.modelPath) {\n      this.config.modelPath = join(\n        HOME,\n        '.stackmemory',\n        'models',\n        'sweep',\n        'sweep-next-edit-1.5b.q8_0.v2.gguf'\n      );\n    }\n  }\n\n  /**\n   * Find llama-server executable\n   */\n  private findLlamaServer(): string | null {\n    const candidates = [\n      'llama-server',\n      'llama.cpp/llama-server',\n      '/usr/local/bin/llama-server',\n      '/opt/homebrew/bin/llama-server',\n      join(HOME, '.local', 'bin', 'llama-server'),\n    ];\n\n    for (const cmd of candidates) {\n      try {\n        execSync(`which ${cmd}`, { stdio: 'ignore' });\n        return cmd;\n      } catch {\n        if (existsSync(cmd)) {\n          return cmd;\n        }\n      }\n    }\n\n    return null;\n  }\n\n  /**\n   * Start the llama-server\n   */\n  async startServer(): Promise<SweepServerStatus> {\n    // Check if already running\n    const status = await this.getStatus();\n    if (status.running) {\n      return status;\n    }\n\n    // Check model exists\n    if (!existsSync(this.config.modelPath)) {\n      throw new Error(\n        `Model not found: ${this.config.modelPath}\\n` +\n          'Download with: huggingface-cli download sweepai/sweep-next-edit-1.5B sweep-next-edit-1.5b.q8_0.v2.gguf --local-dir ~/.stackmemory/models/sweep'\n      );\n    }\n\n    // Find llama-server\n    const llamaServer = this.findLlamaServer();\n    if (!llamaServer) {\n      throw new Error(\n        'llama-server not found. Install with:\\n' +\n          '  brew install llama.cpp\\n' +\n          'or build from source: https://github.com/ggerganov/llama.cpp'\n      );\n    }\n\n    // Ensure log directory exists\n    const logDir = dirname(LOG_FILE);\n    if (!existsSync(logDir)) {\n      mkdirSync(logDir, { recursive: true });\n    }\n\n    // Build command args\n    const args = [\n      '-m',\n      this.config.modelPath,\n      '--port',\n      String(this.config.port),\n      '--host',\n      this.config.host,\n      '-c',\n      String(this.config.contextSize),\n    ];\n\n    if (this.config.threads) {\n      args.push('-t', String(this.config.threads));\n    }\n\n    if (this.config.gpuLayers && this.config.gpuLayers > 0) {\n      args.push('-ngl', String(this.config.gpuLayers));\n    }\n\n    logger.info('Starting Sweep server', { llamaServer, args });\n\n    // Start the process\n    this.process = spawn(llamaServer, args, {\n      detached: true,\n      stdio: ['ignore', 'pipe', 'pipe'],\n    });\n\n    // Write PID file\n    if (this.process.pid) {\n      const pidDir = dirname(PID_FILE);\n      if (!existsSync(pidDir)) {\n        mkdirSync(pidDir, { recursive: true });\n      }\n      writeFileSync(\n        PID_FILE,\n        JSON.stringify({\n          pid: this.process.pid,\n          port: this.config.port,\n          host: this.config.host,\n          modelPath: this.config.modelPath,\n          startedAt: Date.now(),\n        })\n      );\n    }\n\n    // Unref to allow parent to exit\n    this.process.unref();\n\n    // Wait for server to be ready\n    const ready = await this.waitForReady(10000);\n    if (!ready) {\n      await this.stopServer();\n      throw new Error('Server failed to start within timeout');\n    }\n\n    return this.getStatus();\n  }\n\n  /**\n   * Wait for server to be ready\n   */\n  private async waitForReady(timeoutMs: number): Promise<boolean> {\n    const client = createPredictionClient({\n      port: this.config.port,\n      host: this.config.host,\n    });\n\n    const startTime = Date.now();\n    while (Date.now() - startTime < timeoutMs) {\n      if (await client.checkHealth()) {\n        return true;\n      }\n      await new Promise((resolve) => setTimeout(resolve, 500));\n    }\n\n    return false;\n  }\n\n  /**\n   * Stop the server\n   */\n  async stopServer(): Promise<void> {\n    const status = await this.getStatus();\n\n    if (!status.running || !status.pid) {\n      return;\n    }\n\n    try {\n      process.kill(status.pid, 'SIGTERM');\n\n      // Wait for process to exit\n      await new Promise<void>((resolve) => {\n        const checkInterval = setInterval(() => {\n          try {\n            process.kill(status.pid!, 0); // Check if still running\n          } catch {\n            clearInterval(checkInterval);\n            resolve();\n          }\n        }, 100);\n\n        // Force kill after 5 seconds\n        setTimeout(() => {\n          clearInterval(checkInterval);\n          try {\n            process.kill(status.pid!, 'SIGKILL');\n          } catch {\n            // Already dead\n          }\n          resolve();\n        }, 5000);\n      });\n    } catch (error) {\n      // Process may already be dead\n      logger.warn('Error stopping server', { error });\n    }\n\n    // Clean up PID file\n    try {\n      if (existsSync(PID_FILE)) {\n        const { unlinkSync } = await import('fs');\n        unlinkSync(PID_FILE);\n      }\n    } catch {\n      // Ignore\n    }\n  }\n\n  /**\n   * Get server status\n   */\n  async getStatus(): Promise<SweepServerStatus> {\n    // Check PID file\n    if (!existsSync(PID_FILE)) {\n      return { running: false };\n    }\n\n    try {\n      const data = JSON.parse(readFileSync(PID_FILE, 'utf-8'));\n      const { pid, port, host, modelPath, startedAt } = data;\n\n      // Check if process is still running\n      try {\n        process.kill(pid, 0);\n      } catch {\n        // Process not running\n        return { running: false };\n      }\n\n      // Verify server is responsive\n      const client = createPredictionClient({ port, host });\n      const healthy = await client.checkHealth();\n\n      return {\n        running: healthy,\n        pid,\n        port,\n        host,\n        modelPath,\n        startedAt,\n      };\n    } catch {\n      return { running: false };\n    }\n  }\n\n  /**\n   * Check server health\n   */\n  async checkHealth(): Promise<boolean> {\n    const client = createPredictionClient({\n      port: this.config.port,\n      host: this.config.host,\n    });\n    return client.checkHealth();\n  }\n}\n\n/**\n * Create a server manager with default config\n */\nexport function createServerManager(\n  config?: Partial<SweepServerConfig>\n): SweepServerManager {\n  return new SweepServerManager(config);\n}\n"],
+  "mappings": ";;;;AAMA,SAAS,OAAqB,gBAAgB;AAC9C,SAAS,YAAY,cAAc,eAAe,iBAAiB;AACnE,SAAS,MAAM,eAAe;AAC9B;AAAA,EAGE;AAAA,OACK;AACP,SAAS,8BAA8B;AACvC,SAAS,cAAc;AAEvB,MAAM,OAAO,QAAQ,IAAI,MAAM,KAAK;AACpC,MAAM,WAAW,KAAK,MAAM,gBAAgB,SAAS,YAAY;AACjE,MAAM,WAAW,KAAK,MAAM,gBAAgB,SAAS,YAAY;AAE1D,MAAM,mBAAmB;AAAA,EACtB;AAAA,EACA,UAA+B;AAAA,EAEvC,YAAY,SAAqC,CAAC,GAAG;AACnD,SAAK,SAAS,EAAE,GAAG,uBAAuB,GAAG,OAAO;AAGpD,QAAI,CAAC,KAAK,OAAO,WAAW;AAC1B,WAAK,OAAO,YAAY;AAAA,QACtB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,kBAAiC;AACvC,UAAM,aAAa;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,KAAK,MAAM,UAAU,OAAO,cAAc;AAAA,IAC5C;AAEA,eAAW,OAAO,YAAY;AAC5B,UAAI;AACF,iBAAS,SAAS,GAAG,IAAI,EAAE,OAAO,SAAS,CAAC;AAC5C,eAAO;AAAA,MACT,QAAQ;AACN,YAAI,WAAW,GAAG,GAAG;AACnB,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,cAA0C;AAE9C,UAAM,SAAS,MAAM,KAAK,UAAU;AACpC,QAAI,OAAO,SAAS;AAClB,aAAO;AAAA,IACT;AAGA,QAAI,CAAC,WAAW,KAAK,OAAO,SAAS,GAAG;AACtC,YAAM,IAAI;AAAA,QACR,oBAAoB,KAAK,OAAO,SAAS;AAAA;AAAA,MAE3C;AAAA,IACF;AAGA,UAAM,cAAc,KAAK,gBAAgB;AACzC,QAAI,CAAC,aAAa;AAChB,YAAM,IAAI;AAAA,QACR;AAAA,MAGF;AAAA,IACF;AAGA,UAAM,SAAS,QAAQ,QAAQ;AAC/B,QAAI,CAAC,WAAW,MAAM,GAAG;AACvB,gBAAU,QAAQ,EAAE,WAAW,KAAK,CAAC;AAAA,IACvC;AAGA,UAAM,OAAO;AAAA,MACX;AAAA,MACA,KAAK,OAAO;AAAA,MACZ;AAAA,MACA,OAAO,KAAK,OAAO,IAAI;AAAA,MACvB;AAAA,MACA,KAAK,OAAO;AAAA,MACZ;AAAA,MACA,OAAO,KAAK,OAAO,WAAW;AAAA,IAChC;AAEA,QAAI,KAAK,OAAO,SAAS;AACvB,WAAK,KAAK,MAAM,OAAO,KAAK,OAAO,OAAO,CAAC;AAAA,IAC7C;AAEA,QAAI,KAAK,OAAO,aAAa,KAAK,OAAO,YAAY,GAAG;AACtD,WAAK,KAAK,QAAQ,OAAO,KAAK,OAAO,SAAS,CAAC;AAAA,IACjD;AAEA,WAAO,KAAK,yBAAyB,EAAE,aAAa,KAAK,CAAC;AAG1D,SAAK,UAAU,MAAM,aAAa,MAAM;AAAA,MACtC,UAAU;AAAA,MACV,OAAO,CAAC,UAAU,QAAQ,MAAM;AAAA,IAClC,CAAC;AAGD,QAAI,KAAK,QAAQ,KAAK;AACpB,YAAM,SAAS,QAAQ,QAAQ;AAC/B,UAAI,CAAC,WAAW,MAAM,GAAG;AACvB,kBAAU,QAAQ,EAAE,WAAW,KAAK,CAAC;AAAA,MACvC;AACA;AAAA,QACE;AAAA,QACA,KAAK,UAAU;AAAA,UACb,KAAK,KAAK,QAAQ;AAAA,UAClB,MAAM,KAAK,OAAO;AAAA,UAClB,MAAM,KAAK,OAAO;AAAA,UAClB,WAAW,KAAK,OAAO;AAAA,UACvB,WAAW,KAAK,IAAI;AAAA,QACtB,CAAC;AAAA,MACH;AAAA,IACF;AAGA,SAAK,QAAQ,MAAM;AAGnB,UAAM,QAAQ,MAAM,KAAK,aAAa,GAAK;AAC3C,QAAI,CAAC,OAAO;AACV,YAAM,KAAK,WAAW;AACtB,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,WAAO,KAAK,UAAU;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,aAAa,WAAqC;AAC9D,UAAM,SAAS,uBAAuB;AAAA,MACpC,MAAM,KAAK,OAAO;AAAA,MAClB,MAAM,KAAK,OAAO;AAAA,IACpB,CAAC;AAED,UAAM,YAAY,KAAK,IAAI;AAC3B,WAAO,KAAK,IAAI,IAAI,YAAY,WAAW;AACzC,UAAI,MAAM,OAAO,YAAY,GAAG;AAC9B,eAAO;AAAA,MACT;AACA,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,IACzD;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,aAA4B;AAChC,UAAM,SAAS,MAAM,KAAK,UAAU;AAEpC,QAAI,CAAC,OAAO,WAAW,CAAC,OAAO,KAAK;AAClC;AAAA,IACF;AAEA,QAAI;AACF,cAAQ,KAAK,OAAO,KAAK,SAAS;AAGlC,YAAM,IAAI,QAAc,CAAC,YAAY;AACnC,cAAM,gBAAgB,YAAY,MAAM;AACtC,cAAI;AACF,oBAAQ,KAAK,OAAO,KAAM,CAAC;AAAA,UAC7B,QAAQ;AACN,0BAAc,aAAa;AAC3B,oBAAQ;AAAA,UACV;AAAA,QACF,GAAG,GAAG;AAGN,mBAAW,MAAM;AACf,wBAAc,aAAa;AAC3B,cAAI;AACF,oBAAQ,KAAK,OAAO,KAAM,SAAS;AAAA,UACrC,QAAQ;AAAA,UAER;AACA,kBAAQ;AAAA,QACV,GAAG,GAAI;AAAA,MACT,CAAC;AAAA,IACH,SAAS,OAAO;AAEd,aAAO,KAAK,yBAAyB,EAAE,MAAM,CAAC;AAAA,IAChD;AAGA,QAAI;AACF,UAAI,WAAW,QAAQ,GAAG;AACxB,cAAM,EAAE,WAAW,IAAI,MAAM,OAAO,IAAI;AACxC,mBAAW,QAAQ;AAAA,MACrB;AAAA,IACF,QAAQ;AAAA,IAER;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAwC;AAE5C,QAAI,CAAC,WAAW,QAAQ,GAAG;AACzB,aAAO,EAAE,SAAS,MAAM;AAAA,IAC1B;AAEA,QAAI;AACF,YAAM,OAAO,KAAK,MAAM,aAAa,UAAU,OAAO,CAAC;AACvD,YAAM,EAAE,KAAK,MAAM,MAAM,WAAW,UAAU,IAAI;AAGlD,UAAI;AACF,gBAAQ,KAAK,KAAK,CAAC;AAAA,MACrB,QAAQ;AAEN,eAAO,EAAE,SAAS,MAAM;AAAA,MAC1B;AAGA,YAAM,SAAS,uBAAuB,EAAE,MAAM,KAAK,CAAC;AACpD,YAAM,UAAU,MAAM,OAAO,YAAY;AAEzC,aAAO;AAAA,QACL,SAAS;AAAA,QACT;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF,QAAQ;AACN,aAAO,EAAE,SAAS,MAAM;AAAA,IAC1B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,cAAgC;AACpC,UAAM,SAAS,uBAAuB;AAAA,MACpC,MAAM,KAAK,OAAO;AAAA,MAClB,MAAM,KAAK,OAAO;AAAA,IACpB,CAAC;AACD,WAAO,OAAO,YAAY;AAAA,EAC5B;AACF;AAKO,SAAS,oBACd,QACoB;AACpB,SAAO,IAAI,mBAAmB,MAAM;AACtC;",
+  "names": []
+}

package/dist/src/features/sweep/tab-interceptor.js

@@ -0,0 +1,38 @@
+import { fileURLToPath as __fileURLToPath } from 'url';
+import { dirname as __pathDirname } from 'path';
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+const TAB = 9;
+const ESC = 27;
+class TabInterceptor {
+  predictionActive = false;
+  callbacks;
+  constructor(callbacks) {
+    this.callbacks = callbacks;
+  }
+  setPredictionActive(active) {
+    this.predictionActive = active;
+  }
+  isPredictionActive() {
+    return this.predictionActive;
+  }
+  process(data) {
+    if (!this.predictionActive) {
+      this.callbacks.onPassthrough(data);
+      return;
+    }
+    if (data.length === 1 && data[0] === TAB) {
+      this.callbacks.onAccept();
+      return;
+    }
+    if (data.length === 1 && data[0] === ESC) {
+      this.callbacks.onDismiss();
+      return;
+    }
+    this.callbacks.onPassthrough(data);
+  }
+}
+export {
+  TabInterceptor
+};
+//# sourceMappingURL=tab-interceptor.js.map

package/dist/src/features/sweep/tab-interceptor.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../../src/features/sweep/tab-interceptor.ts"],
+  "sourcesContent": ["/**\n * Tab Interceptor\n *\n * Intercepts Tab and Esc keystrokes when a Sweep prediction is active.\n * All other input passes through to the PTY child unchanged.\n */\n\nconst TAB = 0x09;\nconst ESC = 0x1b;\n\nexport interface TabInterceptorCallbacks {\n  onAccept: () => void;\n  onDismiss: () => void;\n  onPassthrough: (data: Buffer) => void;\n}\n\nexport class TabInterceptor {\n  private predictionActive = false;\n  private callbacks: TabInterceptorCallbacks;\n\n  constructor(callbacks: TabInterceptorCallbacks) {\n    this.callbacks = callbacks;\n  }\n\n  setPredictionActive(active: boolean): void {\n    this.predictionActive = active;\n  }\n\n  isPredictionActive(): boolean {\n    return this.predictionActive;\n  }\n\n  process(data: Buffer): void {\n    if (!this.predictionActive) {\n      this.callbacks.onPassthrough(data);\n      return;\n    }\n\n    // Tab key: accept prediction\n    if (data.length === 1 && data[0] === TAB) {\n      this.callbacks.onAccept();\n      return;\n    }\n\n    // Bare Escape: dismiss prediction\n    // Distinguish from escape sequences (arrow keys, etc.) by length.\n    // Bare Esc is a single byte; escape sequences are multi-byte.\n    if (data.length === 1 && data[0] === ESC) {\n      this.callbacks.onDismiss();\n      return;\n    }\n\n    // Everything else passes through (including escape sequences)\n    this.callbacks.onPassthrough(data);\n  }\n}\n"],
+  "mappings": ";;;;AAOA,MAAM,MAAM;AACZ,MAAM,MAAM;AAQL,MAAM,eAAe;AAAA,EAClB,mBAAmB;AAAA,EACnB;AAAA,EAER,YAAY,WAAoC;AAC9C,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,oBAAoB,QAAuB;AACzC,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,qBAA8B;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,QAAQ,MAAoB;AAC1B,QAAI,CAAC,KAAK,kBAAkB;AAC1B,WAAK,UAAU,cAAc,IAAI;AACjC;AAAA,IACF;AAGA,QAAI,KAAK,WAAW,KAAK,KAAK,CAAC,MAAM,KAAK;AACxC,WAAK,UAAU,SAAS;AACxB;AAAA,IACF;AAKA,QAAI,KAAK,WAAW,KAAK,KAAK,CAAC,MAAM,KAAK;AACxC,WAAK,UAAU,UAAU;AACzB;AAAA,IACF;AAGA,SAAK,UAAU,cAAc,IAAI;AAAA,EACnC;AACF;",
+  "names": []
+}

package/dist/src/features/sweep/types.js

@@ -0,0 +1,18 @@
+import { fileURLToPath as __fileURLToPath } from 'url';
+import { dirname as __pathDirname } from 'path';
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+const DEFAULT_SERVER_CONFIG = {
+  port: 8766,
+  host: "127.0.0.1",
+  modelPath: "",
+  contextSize: 8192,
+  threads: void 0,
+  gpuLayers: 0
+};
+const SWEEP_STOP_TOKENS = ["<|file_sep|>", "</s>"];
+export {
+  DEFAULT_SERVER_CONFIG,
+  SWEEP_STOP_TOKENS
+};
+//# sourceMappingURL=types.js.map

package/dist/src/features/sweep/types.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../../src/features/sweep/types.ts"],
+  "sourcesContent": ["/**\n * Sweep Next-Edit Types\n *\n * Types for the Sweep 1.5B model server integration.\n */\n\nexport interface SweepServerConfig {\n  port: number;\n  host: string;\n  modelPath: string;\n  contextSize: number;\n  threads?: number;\n  gpuLayers?: number;\n}\n\nexport interface SweepServerStatus {\n  running: boolean;\n  pid?: number;\n  port?: number;\n  host?: string;\n  startedAt?: number;\n  modelPath?: string;\n}\n\nexport interface SweepPredictInput {\n  file_path: string;\n  current_content: string;\n  original_content?: string;\n  context_files?: Record<string, string>;\n  recent_diffs?: DiffEntry[];\n  max_tokens?: number;\n  temperature?: number;\n  top_k?: number;\n}\n\nexport interface DiffEntry {\n  file_path: string;\n  original: string;\n  updated: string;\n  timestamp?: number;\n}\n\nexport interface 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\nexport interface SweepPromptInput {\n  filePath: string;\n  originalContent: string;\n  currentContent: string;\n  recentDiffs: DiffEntry[];\n  contextFiles?: Record<string, string>;\n}\n\nexport interface CompletionRequest {\n  model: string;\n  prompt: string;\n  max_tokens: number;\n  temperature: number;\n  top_k?: number;\n  stop?: string[];\n  stream?: boolean;\n}\n\nexport interface CompletionResponse {\n  id: string;\n  object: string;\n  created: number;\n  model: string;\n  choices: Array<{\n    text: string;\n    index: number;\n    logprobs: null;\n    finish_reason: string;\n  }>;\n  usage: {\n    prompt_tokens: number;\n    completion_tokens: number;\n    total_tokens: number;\n  };\n}\n\nexport const DEFAULT_SERVER_CONFIG: SweepServerConfig = {\n  port: 8766,\n  host: '127.0.0.1',\n  modelPath: '',\n  contextSize: 8192,\n  threads: undefined,\n  gpuLayers: 0,\n};\n\nexport const SWEEP_STOP_TOKENS = ['<|file_sep|>', '</s>'];\n"],
+  "mappings": ";;;;AAwFO,MAAM,wBAA2C;AAAA,EACtD,MAAM;AAAA,EACN,MAAM;AAAA,EACN,WAAW;AAAA,EACX,aAAa;AAAA,EACb,SAAS;AAAA,EACT,WAAW;AACb;AAEO,MAAM,oBAAoB,CAAC,gBAAgB,MAAM;",
+  "names": []
+}