@agentmemory/agentmemory 0.7.2 → 0.7.3

package/src/auth.ts

@@ -1,12 +0,0 @@
-import { timingSafeEqual, createHmac, randomBytes } from "node:crypto";
-
-const hmacKey = randomBytes(32);
-
-export function timingSafeCompare(a: string, b: string): boolean {
-  const hmacA = createHmac("sha256", hmacKey).update(a).digest();
-  const hmacB = createHmac("sha256", hmacKey).update(b).digest();
-  return timingSafeEqual(hmacA, hmacB);
-}
-
-export const VIEWER_CSP =
-  "default-src 'none'; script-src 'unsafe-inline'; style-src 'unsafe-inline'; connect-src 'self' ws://localhost:* wss://localhost:*; img-src 'self'; font-src 'self'";

package/src/cli.ts

@@ -1,251 +0,0 @@
-#!/usr/bin/env node
-
-import { spawn, execFileSync, execSync } from "node:child_process";
-import { existsSync } from "node:fs";
-import { join, dirname } from "node:path";
-import { fileURLToPath } from "node:url";
-import * as p from "@clack/prompts";
-
-const __dirname = dirname(fileURLToPath(import.meta.url));
-const args = process.argv.slice(2);
-
-if (args.includes("--help") || args.includes("-h")) {
-  console.log(`
-agentmemory — persistent memory for AI coding agents
-
-Usage: agentmemory [options]
-
-Options:
-  --help, -h         Show this help
-  --tools all|core   Tool visibility (default: core = 7 tools)
-  --no-engine        Skip auto-starting iii-engine
-  --port <N>         Override REST port (default: 3111)
-
-Environment:
-  AGENTMEMORY_TOOLS=all        Expose all 41 MCP tools
-  AGENTMEMORY_SECRET=xxx       Auth secret for REST/MCP
-  CONSOLIDATION_ENABLED=true   Enable auto-consolidation (off by default)
-  OBSIDIAN_AUTO_EXPORT=true    Auto-export on consolidation
-
-Quick start:
-  npx @agentmemory/agentmemory    # installs iii if missing, starts everything
-  npx agentmemory-mcp             # standalone MCP server (no engine needed)
-`);
-  process.exit(0);
-}
-
-const toolsIdx = args.indexOf("--tools");
-if (toolsIdx !== -1 && args[toolsIdx + 1]) {
-  process.env["AGENTMEMORY_TOOLS"] = args[toolsIdx + 1];
-}
-
-const portIdx = args.indexOf("--port");
-if (portIdx !== -1 && args[portIdx + 1]) {
-  process.env["III_REST_PORT"] = args[portIdx + 1];
-}
-
-const skipEngine = args.includes("--no-engine");
-
-function getRestPort(): number {
-  return parseInt(process.env["III_REST_PORT"] || "3111", 10) || 3111;
-}
-
-async function isEngineRunning(): Promise<boolean> {
-  try {
-    const res = await fetch(`http://localhost:${getRestPort()}/agentmemory/livez`, {
-      signal: AbortSignal.timeout(2000),
-    });
-    return res.ok;
-  } catch {
-    return false;
-  }
-}
-
-function findIiiConfig(): string {
-  const candidates = [
-    join(__dirname, "iii-config.yaml"),
-    join(__dirname, "..", "iii-config.yaml"),
-    join(process.cwd(), "iii-config.yaml"),
-  ];
-  for (const c of candidates) {
-    if (existsSync(c)) return c;
-  }
-  return "";
-}
-
-function whichBinary(name: string): string | null {
-  const cmd = process.platform === "win32" ? "where" : "which";
-  try {
-    return execFileSync(cmd, [name], { encoding: "utf-8" }).trim().split("\n")[0];
-  } catch {
-    return null;
-  }
-}
-
-async function installIii(): Promise<boolean> {
-  if (process.platform === "win32") {
-    p.log.warn("Automatic iii-engine install is not supported on Windows.");
-    p.log.info("Install manually: https://iii.dev/docs");
-    return false;
-  }
-
-  const curlBin = whichBinary("curl");
-  if (!curlBin) {
-    p.log.warn("curl not found — cannot auto-install iii-engine.");
-    return false;
-  }
-
-  const shouldInstall = await p.confirm({
-    message: "iii-engine is not installed. Install it now?",
-    initialValue: true,
-  });
-
-  if (p.isCancel(shouldInstall) || !shouldInstall) {
-    return false;
-  }
-
-  const s = p.spinner();
-  s.start("Installing iii-engine...");
-
-  try {
-    execSync("curl -fsSL https://install.iii.dev/iii/main/install.sh | sh", {
-      stdio: ["pipe", "pipe", "pipe"],
-      timeout: 120000,
-    });
-
-    const installed = whichBinary("iii");
-    if (installed) {
-      s.stop("iii-engine installed successfully");
-      return true;
-    }
-
-    s.stop("Installation completed but iii not found in PATH");
-    p.log.warn("You may need to restart your shell or add iii to your PATH.");
-
-    const iiiPaths = [
-      join(process.env["HOME"] || "", ".local", "bin", "iii"),
-      "/usr/local/bin/iii",
-    ];
-    for (const iiiPath of iiiPaths) {
-      if (existsSync(iiiPath)) {
-        p.log.info(`Found iii at: ${iiiPath}`);
-        process.env["PATH"] = `${dirname(iiiPath)}:${process.env["PATH"]}`;
-        return true;
-      }
-    }
-
-    return false;
-  } catch (err) {
-    s.stop("Failed to install iii-engine");
-    p.log.error(err instanceof Error ? err.message : String(err));
-    return false;
-  }
-}
-
-async function startEngine(): Promise<boolean> {
-  const configPath = findIiiConfig();
-  let iiiBin = whichBinary("iii");
-
-  if (!iiiBin) {
-    const installed = await installIii();
-    if (installed) {
-      iiiBin = whichBinary("iii");
-    }
-  }
-
-  if (iiiBin && configPath) {
-    const s = p.spinner();
-    s.start(`Starting iii-engine: ${iiiBin}`);
-    const child = spawn(iiiBin, ["--config", configPath], {
-      detached: true,
-      stdio: "ignore",
-    });
-    child.unref();
-    s.stop("iii-engine process started");
-    return true;
-  }
-
-  const dockerBin = whichBinary("docker");
-  const dockerCompose = join(__dirname, "..", "docker-compose.yml");
-  const dcExists = existsSync(dockerCompose) || existsSync(join(process.cwd(), "docker-compose.yml"));
-
-  if (dockerBin && dcExists) {
-    const composeFile = existsSync(dockerCompose) ? dockerCompose : join(process.cwd(), "docker-compose.yml");
-    const s = p.spinner();
-    s.start("Starting iii-engine via Docker...");
-    const child = spawn(dockerBin, ["compose", "-f", composeFile, "up", "-d"], {
-      detached: true,
-      stdio: "ignore",
-    });
-    child.unref();
-    s.stop("Docker compose started");
-    return true;
-  }
-
-  return false;
-}
-
-async function waitForEngine(timeoutMs: number): Promise<boolean> {
-  const start = Date.now();
-  while (Date.now() - start < timeoutMs) {
-    if (await isEngineRunning()) return true;
-    await new Promise((r) => setTimeout(r, 500));
-  }
-  return false;
-}
-
-async function main() {
-  p.intro("agentmemory");
-
-  if (skipEngine) {
-    p.log.info("Skipping engine check (--no-engine)");
-    await import("./index.js");
-    return;
-  }
-
-  if (await isEngineRunning()) {
-    p.log.success("iii-engine is running");
-    await import("./index.js");
-    return;
-  }
-
-  const started = await startEngine();
-  if (!started) {
-    p.log.error("Could not start iii-engine.");
-    p.note(
-      [
-        "Install iii-engine (pick one):",
-        "  curl -fsSL https://install.iii.dev/iii/main/install.sh | sh",
-        "  cargo install iii-engine",
-        "",
-        "Or use Docker:",
-        "  docker pull iiidev/iii:latest",
-        "",
-        "Docs: https://iii.dev/docs",
-        "",
-        "Or skip with: agentmemory --no-engine",
-      ].join("\n"),
-      "Setup required",
-    );
-    process.exit(1);
-  }
-
-  const s = p.spinner();
-  s.start("Waiting for iii-engine to be ready...");
-
-  const ready = await waitForEngine(15000);
-  if (!ready) {
-    const port = getRestPort();
-    s.stop("iii-engine did not become ready within 15s");
-    p.log.error(`Check that ports ${port}, ${port + 1}, 49134 are available.`);
-    process.exit(1);
-  }
-
-  s.stop("iii-engine is ready");
-  await import("./index.js");
-}
-
-main().catch((err) => {
-  p.log.error(err instanceof Error ? err.message : String(err));
-  process.exit(1);
-});

package/src/config.ts

@@ -1,221 +0,0 @@
-import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
-import { homedir } from "node:os";
-import type {
-  AgentMemoryConfig,
-  ProviderConfig,
-  EmbeddingConfig,
-  FallbackConfig,
-  ClaudeBridgeConfig,
-  TeamConfig,
-} from "./types.js";
-
-function safeParseInt(value: string | undefined, fallback: number): number {
-  if (!value) return fallback;
-  const parsed = parseInt(value, 10);
-  return Number.isNaN(parsed) ? fallback : parsed;
-}
-
-const DATA_DIR = join(homedir(), ".agentmemory");
-const ENV_FILE = join(DATA_DIR, ".env");
-
-function loadEnvFile(): Record<string, string> {
-  if (!existsSync(ENV_FILE)) return {};
-  const content = readFileSync(ENV_FILE, "utf-8");
-  const vars: Record<string, string> = {};
-  for (const line of content.split("\n")) {
-    const trimmed = line.trim();
-    if (!trimmed || trimmed.startsWith("#")) continue;
-    const eqIdx = trimmed.indexOf("=");
-    if (eqIdx === -1) continue;
-    const key = trimmed.slice(0, eqIdx).trim();
-    let val = trimmed.slice(eqIdx + 1).trim();
-    if (
-      (val.startsWith('"') && val.endsWith('"')) ||
-      (val.startsWith("'") && val.endsWith("'"))
-    ) {
-      val = val.slice(1, -1);
-    }
-    vars[key] = val;
-  }
-  return vars;
-}
-
-function detectProvider(env: Record<string, string>): ProviderConfig {
-  const maxTokens = parseInt(env["MAX_TOKENS"] || "4096", 10);
-
-  if (env["ANTHROPIC_API_KEY"]) {
-    return {
-      provider: "anthropic",
-      model: env["ANTHROPIC_MODEL"] || "claude-sonnet-4-20250514",
-      maxTokens,
-    };
-  }
-  if (env["GEMINI_API_KEY"]) {
-    return {
-      provider: "gemini",
-      model: env["GEMINI_MODEL"] || "gemini-2.0-flash",
-      maxTokens,
-    };
-  }
-  if (env["OPENROUTER_API_KEY"]) {
-    return {
-      provider: "openrouter",
-      model: env["OPENROUTER_MODEL"] || "anthropic/claude-sonnet-4-20250514",
-      maxTokens,
-    };
-  }
-  return {
-    provider: "agent-sdk",
-    model: "claude-sonnet-4-20250514",
-    maxTokens: 4096,
-  };
-}
-
-export function loadConfig(): AgentMemoryConfig {
-  const env = getMergedEnv();
-
-  const provider = detectProvider(env);
-
-  return {
-    engineUrl: env["III_ENGINE_URL"] || "ws://localhost:49134",
-    restPort: parseInt(env["III_REST_PORT"] || "3111", 10) || 3111,
-    streamsPort: parseInt(env["III_STREAMS_PORT"] || "3112", 10) || 3112,
-    provider,
-    tokenBudget: safeParseInt(env["TOKEN_BUDGET"], 2000),
-    maxObservationsPerSession: safeParseInt(env["MAX_OBS_PER_SESSION"], 500),
-    compressionModel: provider.model,
-    dataDir: DATA_DIR,
-  };
-}
-
-function getMergedEnv(
-  overrides?: Record<string, string>,
-): Record<string, string> {
-  const fileEnv = loadEnvFile();
-  return { ...fileEnv, ...process.env, ...overrides } as Record<string, string>;
-}
-
-export function getEnvVar(key: string): string | undefined {
-  return getMergedEnv()[key];
-}
-
-export function loadEmbeddingConfig(): EmbeddingConfig {
-  const env = getMergedEnv();
-  let bm25Weight = parseFloat(env["BM25_WEIGHT"] || "0.4");
-  let vectorWeight = parseFloat(env["VECTOR_WEIGHT"] || "0.6");
-  bm25Weight =
-    isNaN(bm25Weight) || bm25Weight < 0 ? 0.4 : Math.min(bm25Weight, 1);
-  vectorWeight =
-    isNaN(vectorWeight) || vectorWeight < 0 ? 0.6 : Math.min(vectorWeight, 1);
-  return {
-    provider: env["EMBEDDING_PROVIDER"] || undefined,
-    bm25Weight,
-    vectorWeight,
-  };
-}
-
-export function detectEmbeddingProvider(
-  env?: Record<string, string>,
-): string | null {
-  const source = env ?? getMergedEnv();
-  const forced = source["EMBEDDING_PROVIDER"];
-  if (forced) return forced;
-
-  if (source["GEMINI_API_KEY"]) return "gemini";
-  if (source["OPENAI_API_KEY"]) return "openai";
-  if (source["VOYAGE_API_KEY"]) return "voyage";
-  if (source["COHERE_API_KEY"]) return "cohere";
-  if (source["OPENROUTER_API_KEY"]) return "openrouter";
-  return null;
-}
-
-export function loadClaudeBridgeConfig(): ClaudeBridgeConfig {
-  const env = getMergedEnv();
-  const enabled = env["CLAUDE_MEMORY_BRIDGE"] === "true";
-  const projectPath = env["CLAUDE_PROJECT_PATH"] || "";
-  const lineBudget = safeParseInt(env["CLAUDE_MEMORY_LINE_BUDGET"], 200);
-  let memoryFilePath = "";
-  if (enabled && projectPath) {
-    const safePath = projectPath.replace(/[/\\]/g, "-").replace(/^-/, "");
-    memoryFilePath = join(
-      homedir(),
-      ".claude",
-      "projects",
-      safePath,
-      "memory",
-      "MEMORY.md",
-    );
-  }
-  return { enabled, projectPath, memoryFilePath, lineBudget };
-}
-
-export function loadTeamConfig(): TeamConfig | null {
-  const env = getMergedEnv();
-  const teamId = env["TEAM_ID"];
-  const userId = env["USER_ID"];
-  if (!teamId || !userId) return null;
-  const mode = env["TEAM_MODE"] === "shared" ? "shared" : "private";
-  return { teamId, userId, mode };
-}
-
-export function loadSnapshotConfig(): {
-  enabled: boolean;
-  interval: number;
-  dir: string;
-} {
-  const env = getMergedEnv();
-  return {
-    enabled: env["SNAPSHOT_ENABLED"] === "true",
-    interval: safeParseInt(env["SNAPSHOT_INTERVAL"], 3600),
-    dir: env["SNAPSHOT_DIR"] || join(homedir(), ".agentmemory", "snapshots"),
-  };
-}
-
-export function isGraphExtractionEnabled(): boolean {
-  return getMergedEnv()["GRAPH_EXTRACTION_ENABLED"] === "true";
-}
-
-export function getGraphBatchSize(): number {
-  return safeParseInt(getMergedEnv()["GRAPH_EXTRACTION_BATCH_SIZE"], 10);
-}
-
-export function isConsolidationEnabled(): boolean {
-  return getMergedEnv()["CONSOLIDATION_ENABLED"] === "true";
-}
-
-export function getConsolidationDecayDays(): number {
-  return safeParseInt(getMergedEnv()["CONSOLIDATION_DECAY_DAYS"], 30);
-}
-
-export function isStandaloneMcp(): boolean {
-  return getMergedEnv()["STANDALONE_MCP"] === "true";
-}
-
-export function getStandalonePersistPath(): string {
-  const env = getMergedEnv();
-  return (
-    env["STANDALONE_PERSIST_PATH"] ||
-    join(homedir(), ".agentmemory", "standalone.json")
-  );
-}
-
-const VALID_PROVIDERS = new Set([
-  "anthropic",
-  "gemini",
-  "openrouter",
-  "agent-sdk",
-]);
-
-export function loadFallbackConfig(): FallbackConfig {
-  const env = getMergedEnv();
-  const raw = env["FALLBACK_PROVIDERS"] || "";
-  const providers = raw
-    .split(",")
-    .map((p) => p.trim())
-    .filter(
-      (p): p is FallbackConfig["providers"][number] =>
-        Boolean(p) && VALID_PROVIDERS.has(p),
-    );
-  return { providers };
-}

package/src/eval/metrics-store.ts

@@ -1,65 +0,0 @@
-import type { FunctionMetrics } from "../types.js";
-import type { StateKV } from "../state/kv.js";
-import { KV } from "../state/schema.js";
-
-export class MetricsStore {
-  private cache = new Map<string, FunctionMetrics>();
-  private qualityCallCounts = new Map<string, number>();
-
-  constructor(private kv: StateKV) {}
-
-  async record(
-    functionId: string,
-    latencyMs: number,
-    success: boolean,
-    qualityScore?: number,
-  ): Promise<void> {
-    let m = this.cache.get(functionId);
-    if (!m) {
-      m = (await this.kv.get<FunctionMetrics>(KV.metrics, functionId)) ?? {
-        functionId,
-        totalCalls: 0,
-        successCount: 0,
-        failureCount: 0,
-        avgLatencyMs: 0,
-        avgQualityScore: 0,
-      };
-    }
-
-    const prev = m.totalCalls;
-    m.totalCalls += 1;
-    m.avgLatencyMs = (m.avgLatencyMs * prev + latencyMs) / m.totalCalls;
-    if (success) {
-      m.successCount += 1;
-    } else {
-      m.failureCount += 1;
-    }
-    if (qualityScore !== undefined) {
-      const prevQualityCalls = this.qualityCallCounts.get(functionId) || 0;
-      m.avgQualityScore =
-        (m.avgQualityScore * prevQualityCalls + qualityScore) /
-        (prevQualityCalls + 1);
-      this.qualityCallCounts.set(functionId, prevQualityCalls + 1);
-    }
-
-    this.cache.set(functionId, m);
-    await this.kv.set(KV.metrics, functionId, m).catch(() => {});
-  }
-
-  async get(functionId: string): Promise<FunctionMetrics | null> {
-    return (
-      this.cache.get(functionId) ??
-      (await this.kv.get<FunctionMetrics>(KV.metrics, functionId))
-    );
-  }
-
-  async getAll(): Promise<FunctionMetrics[]> {
-    const kvMetrics = await this.kv
-      .list<FunctionMetrics>(KV.metrics)
-      .catch(() => []);
-    const merged = new Map<string, FunctionMetrics>();
-    for (const m of kvMetrics) merged.set(m.functionId, m);
-    for (const [id, m] of this.cache) merged.set(id, m);
-    return Array.from(merged.values());
-  }
-}

package/src/eval/quality.ts

@@ -1,51 +0,0 @@
-export function scoreCompression(obs: {
-  type?: string;
-  title?: string;
-  facts?: string[];
-  narrative?: string;
-  concepts?: string[];
-  importance?: number;
-}): number {
-  let score = 0;
-  if (obs.facts && obs.facts.length > 0) score += 25;
-  if (obs.facts && obs.facts.length >= 3) score += 10;
-  if (obs.narrative && obs.narrative.length >= 20) score += 20;
-  if (obs.narrative && obs.narrative.length >= 50) score += 5;
-  if (obs.title && obs.title.length >= 5 && obs.title.length <= 120) score += 15;
-  if (obs.concepts && obs.concepts.length > 0) score += 15;
-  if (obs.importance && obs.importance >= 1 && obs.importance <= 10) score += 10;
-  return Math.min(100, score);
-}
-
-export function scoreSummary(summary: {
-  title?: string;
-  narrative?: string;
-  keyDecisions?: string[];
-  filesModified?: string[];
-  concepts?: string[];
-}): number {
-  let score = 0;
-  if (summary.title && summary.title.length >= 5) score += 20;
-  if (summary.narrative && summary.narrative.length >= 20) score += 25;
-  if (summary.narrative && summary.narrative.length >= 100) score += 5;
-  if (summary.keyDecisions && summary.keyDecisions.length > 0) score += 20;
-  if (summary.filesModified && summary.filesModified.length > 0) score += 15;
-  if (summary.concepts && summary.concepts.length > 0) score += 15;
-  return Math.min(100, score);
-}
-
-export function scoreContextRelevance(
-  context: string,
-  project: string,
-): number {
-  let score = 0;
-  if (context.length > 0) score += 20;
-  if (project && context.toLowerCase().includes(project.toLowerCase())) score += 20;
-  if (context.includes("<")) score += 15;
-  const sectionCount = (context.match(/<\w+>/g) || []).length;
-  if (sectionCount >= 2) score += 15;
-  if (sectionCount >= 4) score += 10;
-  if (context.length >= 100) score += 10;
-  if (context.length >= 500) score += 10;
-  return Math.min(100, score);
-}