pi-automem-bridge 0.2.0 → 0.2.2

package/src/config.ts

@@ -1,251 +1,262 @@
-/**
- * config.ts - Config loading, defaults, validation, and env-var resolution.
- */
-
-import { readFileSync, existsSync } from "node:fs";
-import { homedir } from "node:os";
-import { resolve } from "node:path";
-
-// ---------------------------------------------------------------------------
-// Types
-// ---------------------------------------------------------------------------
-
-export type MemoryType =
-  | "Decision"
-  | "Pattern"
-  | "Preference"
-  | "Style"
-  | "Habit"
-  | "Insight"
-  | "Context";
-
-export type RelationshipType =
-  | "RELATES_TO"
-  | "LEADS_TO"
-  | "OCCURRED_BEFORE"
-  | "PREFERS_OVER"
-  | "EXEMPLIFIES"
-  | "CONTRADICTS"
-  | "REINFORCES"
-  | "INVALIDATED_BY"
-  | "EVOLVED_INTO"
-  | "DERIVED_FROM"
-  | "PART_OF";
-
-export interface ProjectRecallOverride {
-  limit?: number;
-  maxBytes?: number;
-  contextTypes?: MemoryType[];
-  expandRelations?: boolean;
-  expandEntities?: boolean;
-}
-
-export interface AutoMemConfig {
-  mcpServerName: string;
-  startupRecall: {
-    enabled: boolean;
-    queries: string[];
-    tags: string[];
-    tagMode: "any" | "all";
-    limit: number;
-    maxBytes: number;
-    showStatus: boolean;
-  };
-  turnRecall: {
-    enabled: boolean;
-    limit: number;
-    maxBytes: number;
-    contextTypes: MemoryType[];
-    expandRelations: boolean;
-    expandEntities: boolean;
-  };
-  projectDetection: {
-    enabled: boolean;
-    tagPrefix: string;
-    folderTags: Record<string, string[]>;
-    gitRepoToTag: Record<string, string>;
-  };
-  projectOverrides?: Record<string, ProjectRecallOverride>;
-  writePolicy: {
-    mode: "off" | "propose" | "safe-auto" | "confirm-all";
-    autoWriteCategories: string[];
-    confirmCategories: string[];
-    blockedCategories: string[];
-    defaultSource: string;
-    machineTag: boolean;
-    alwaysTag: string[];
-    minImportanceToWrite: number;
-    dedupeBeforeWrite: boolean;
-    dedupeLimit: number;
-  };
-  behavior: {
-    injectSystemPrompt: boolean;
-    displayRecall: "full" | "summary" | "hidden";
-    maxContentLength: number;
-    preferredContentLength: number;
-  };
-  viewer: {
-    enabled: boolean;
-    mode: "standalone" | "embedded";
-    port: number;
-  };
-}
-
-// ---------------------------------------------------------------------------
-// Defaults
-// ---------------------------------------------------------------------------
-
-export const DEFAULT_CONFIG: AutoMemConfig = {
-  mcpServerName: "automem",
-  startupRecall: {
-    enabled: true,
-    queries: ["user preferences working style environment"],
-    tags: [],
-    tagMode: "any",
-    limit: 8,
-    maxBytes: 6000,
-    showStatus: true,
-  },
-  turnRecall: {
-    enabled: true,
-    limit: 6,
-    maxBytes: 4000,
-    contextTypes: ["Preference", "Decision", "Pattern", "Insight", "Context"],
-    expandRelations: true,
-    expandEntities: true,
-  },
-  projectDetection: {
-    enabled: true,
-    tagPrefix: "project:",
-    folderTags: {},
-    gitRepoToTag: {},
-  },
-  writePolicy: {
-    mode: "safe-auto",
-    autoWriteCategories: ["technical-decision", "agent-pattern", "bug-fix", "tooling-lesson"],
-    confirmCategories: ["personal", "financial", "private", "identity"],
-    blockedCategories: ["secret", "credential", "api-key", "raw-transcript"],
-    defaultSource: "pi-session",
-    machineTag: true,
-    alwaysTag: ["source:pi"],
-    minImportanceToWrite: 0.7,
-    dedupeBeforeWrite: true,
-    dedupeLimit: 3,
-  },
-  behavior: {
-    injectSystemPrompt: true,
-    displayRecall: "summary",
-    maxContentLength: 2000,
-    preferredContentLength: 500,
-  },
-  viewer: {
-    enabled: false,
-    mode: "standalone",
-    port: 3000,
-  },
-};
-
-// ---------------------------------------------------------------------------
-// Config path resolution
-// ---------------------------------------------------------------------------
-
-export function resolveConfigPath(): string {
-  const envPath = process.env.AUTOMEM_CONFIG_PATH;
-  if (envPath) return resolve(envPath);
-  return resolve(homedir(), ".pi", "agent", "automem.json");
-}
-
-// ---------------------------------------------------------------------------
-// Env-var interpolation
-// ---------------------------------------------------------------------------
-
-export function resolveEnvVars(value: string): string {
-  return value.replace(/\$\{([^}]+)\}/g, function(_match: string, name: string) {
-    const v = process.env[name];
-    if (v === undefined) {
-      console.warn('[automem] env var "' + name + '" referenced in config but not set');
-      return "";
-    }
-    return v;
-  });
-}
-
-// ---------------------------------------------------------------------------
-// Deep merge
-// ---------------------------------------------------------------------------
-
-function deepMerge(base: any, override: any): any {
-  const result = Object.assign({}, base);
-  const keys = Object.keys(override);
-  for (let i = 0; i < keys.length; i++) {
-    const key = keys[i];
-    const bVal = (base as any)[key];
-    const oVal = override[key];
-    if (
-      oVal !== undefined &&
-      typeof oVal === "object" &&
-      oVal !== null &&
-      !Array.isArray(oVal) &&
-      typeof bVal === "object" &&
-      bVal !== null &&
-      !Array.isArray(bVal)
-    ) {
-      (result as any)[key] = deepMerge(bVal, oVal);
-    } else if (oVal !== undefined) {
-      (result as any)[key] = oVal;
-    }
-  }
-  return result;
-}
-
-// ---------------------------------------------------------------------------
-// Load + validate
-// ---------------------------------------------------------------------------
-
-export function loadConfig(): AutoMemConfig {
-  const configPath = resolveConfigPath();
-
-  if (!existsSync(configPath)) {
-    console.log("[automem] no config at " + configPath + ", using defaults");
-    return DEFAULT_CONFIG;
-  }
-
-  let raw: any;
-  try {
-    const text = readFileSync(configPath, "utf8");
-    raw = JSON.parse(text);
-  } catch (err) {
-    console.error("[automem] failed to read/parse config: " + err);
-    return DEFAULT_CONFIG;
-  }
-
-  if (typeof raw !== "object" || raw === null) {
-    console.error("[automem] config root must be an object, using defaults");
-    return DEFAULT_CONFIG;
-  }
-
-  const config = deepMerge(DEFAULT_CONFIG, raw) as AutoMemConfig;
-
-  if (config.startupRecall.limit < 1 || config.startupRecall.limit > 20) {
-    console.warn("[automem] startupRecall.limit out of range (1-20), clamping to 8");
-    config.startupRecall.limit = 8;
-  }
-  if (config.turnRecall.limit < 1 || config.turnRecall.limit > 20) {
-    console.warn("[automem] turnRecall.limit out of range (1-20), clamping to 6");
-    config.turnRecall.limit = 6;
-  }
-
-  const validDisplayModes = ["full", "summary", "hidden"];
-  if (!validDisplayModes.includes(config.behavior.displayRecall)) {
-    console.warn("[automem] unknown behavior.displayRecall \"" + config.behavior.displayRecall + "\", valid values: full, summary, hidden. Defaulting to \"summary\"");
-    config.behavior.displayRecall = "summary";
-  }
-
-  const validWriteModes = ["off", "propose", "safe-auto", "confirm-all"];
-  if (!validWriteModes.includes(config.writePolicy.mode)) {
-    console.warn("[automem] unknown writePolicy.mode \"" + config.writePolicy.mode + "\", valid values: off, propose, safe-auto, confirm-all. Defaulting to \"propose\"");
-    config.writePolicy.mode = "propose";
-  }
-
-  return config;
-}
+/**
+ * config.ts - Config loading, defaults, validation, and env-var resolution.
+ */
+
+import { readFileSync, existsSync } from "node:fs";
+import { homedir } from "node:os";
+import { resolve } from "node:path";
+
+// ---------------------------------------------------------------------------
+// Types
+// ---------------------------------------------------------------------------
+
+export type MemoryType =
+  | "Decision"
+  | "Pattern"
+  | "Preference"
+  | "Style"
+  | "Habit"
+  | "Insight"
+  | "Context";
+
+export type RelationshipType =
+  | "RELATES_TO"
+  | "LEADS_TO"
+  | "OCCURRED_BEFORE"
+  | "PREFERS_OVER"
+  | "EXEMPLIFIES"
+  | "CONTRADICTS"
+  | "REINFORCES"
+  | "INVALIDATED_BY"
+  | "EVOLVED_INTO"
+  | "DERIVED_FROM"
+  | "PART_OF";
+
+export interface ProjectRecallOverride {
+  limit?: number;
+  maxBytes?: number;
+  contextTypes?: MemoryType[];
+  expandRelations?: boolean;
+  expandEntities?: boolean;
+}
+
+export interface AutoMemConfig {
+  mcpServerName: string;
+  startupRecall: {
+    enabled: boolean;
+    queries: string[];
+    tags: string[];
+    tagMode: "any" | "all";
+    limit: number;
+    maxBytes: number;
+    showStatus: boolean;
+    timeoutMs: number;
+  };
+  turnRecall: {
+    enabled: boolean;
+    limit: number;
+    maxBytes: number;
+    contextTypes: MemoryType[];
+    expandRelations: boolean;
+    expandEntities: boolean;
+    timeoutMs: number;
+  };
+  projectDetection: {
+    enabled: boolean;
+    tagPrefix: string;
+    folderTags: Record<string, string[]>;
+    gitRepoToTag: Record<string, string>;
+  };
+  projectOverrides?: Record<string, ProjectRecallOverride>;
+  writePolicy: {
+    mode: "off" | "propose" | "safe-auto" | "confirm-all";
+    autoWriteCategories: string[];
+    confirmCategories: string[];
+    blockedCategories: string[];
+    defaultSource: string;
+    machineTag: boolean;
+    alwaysTag: string[];
+    minImportanceToWrite: number;
+    dedupeBeforeWrite: boolean;
+    dedupeLimit: number;
+  };
+  behavior: {
+    injectSystemPrompt: boolean;
+    displayRecall: "full" | "summary" | "hidden";
+    maxContentLength: number;
+    preferredContentLength: number;
+  };
+  viewer: {
+    enabled: boolean;
+    mode: "standalone" | "embedded";
+    port: number;
+  };
+}
+
+// ---------------------------------------------------------------------------
+// Defaults
+// ---------------------------------------------------------------------------
+
+export const DEFAULT_CONFIG: AutoMemConfig = {
+  mcpServerName: "automem",
+  startupRecall: {
+    enabled: true,
+    queries: ["user preferences working style environment"],
+    tags: [],
+    tagMode: "any",
+    limit: 8,
+    maxBytes: 6000,
+    showStatus: true,
+    // Bound startup recall so an unreachable sidecar can't stall session start
+    // for the full 30s MCP timeout per query.
+    timeoutMs: 15000,
+  },
+  turnRecall: {
+    enabled: true,
+    limit: 6,
+    maxBytes: 4000,
+    contextTypes: ["Preference", "Decision", "Pattern", "Insight", "Context"],
+    expandRelations: true,
+    expandEntities: true,
+    // Turn recall is best-effort enrichment on the prompt hot path; bound it
+    // tightly so a slow/stalled sidecar can't block every prompt for the full
+    // 30s MCP timeout. Recall failure degrades gracefully to no injection.
+    timeoutMs: 8000,
+  },
+  projectDetection: {
+    enabled: true,
+    tagPrefix: "project:",
+    folderTags: {},
+    gitRepoToTag: {},
+  },
+  writePolicy: {
+    mode: "safe-auto",
+    autoWriteCategories: ["technical-decision", "agent-pattern", "bug-fix", "tooling-lesson"],
+    confirmCategories: ["personal", "financial", "private", "identity"],
+    blockedCategories: ["secret", "credential", "api-key", "raw-transcript"],
+    defaultSource: "pi-session",
+    machineTag: true,
+    alwaysTag: ["source:pi"],
+    minImportanceToWrite: 0.7,
+    dedupeBeforeWrite: true,
+    dedupeLimit: 3,
+  },
+  behavior: {
+    injectSystemPrompt: true,
+    displayRecall: "summary",
+    maxContentLength: 2000,
+    preferredContentLength: 500,
+  },
+  viewer: {
+    enabled: false,
+    mode: "standalone",
+    port: 3000,
+  },
+};
+
+// ---------------------------------------------------------------------------
+// Config path resolution
+// ---------------------------------------------------------------------------
+
+export function resolveConfigPath(): string {
+  const envPath = process.env.AUTOMEM_CONFIG_PATH;
+  if (envPath) return resolve(envPath);
+  return resolve(homedir(), ".pi", "agent", "automem.json");
+}
+
+// ---------------------------------------------------------------------------
+// Env-var interpolation
+// ---------------------------------------------------------------------------
+
+export function resolveEnvVars(value: string): string {
+  return value.replace(/\$\{([^}]+)\}/g, function(_match: string, name: string) {
+    const v = process.env[name];
+    if (v === undefined) {
+      console.warn('[automem] env var "' + name + '" referenced in config but not set');
+      return "";
+    }
+    return v;
+  });
+}
+
+// ---------------------------------------------------------------------------
+// Deep merge
+// ---------------------------------------------------------------------------
+
+function deepMerge(base: any, override: any): any {
+  const result = Object.assign({}, base);
+  const keys = Object.keys(override);
+  for (let i = 0; i < keys.length; i++) {
+    const key = keys[i];
+    // Never merge prototype-mutating keys from an untrusted config file.
+    if (key === "__proto__" || key === "constructor" || key === "prototype") continue;
+    const bVal = (base as any)[key];
+    const oVal = override[key];
+    if (
+      oVal !== undefined &&
+      typeof oVal === "object" &&
+      oVal !== null &&
+      !Array.isArray(oVal) &&
+      typeof bVal === "object" &&
+      bVal !== null &&
+      !Array.isArray(bVal)
+    ) {
+      (result as any)[key] = deepMerge(bVal, oVal);
+    } else if (oVal !== undefined) {
+      (result as any)[key] = oVal;
+    }
+  }
+  return result;
+}
+
+// ---------------------------------------------------------------------------
+// Load + validate
+// ---------------------------------------------------------------------------
+
+export function loadConfig(): AutoMemConfig {
+  const configPath = resolveConfigPath();
+
+  if (!existsSync(configPath)) {
+    console.log("[automem] no config at " + configPath + ", using defaults");
+    return DEFAULT_CONFIG;
+  }
+
+  let raw: any;
+  try {
+    const text = readFileSync(configPath, "utf8");
+    raw = JSON.parse(text);
+  } catch (err) {
+    console.error("[automem] failed to read/parse config: " + err);
+    return DEFAULT_CONFIG;
+  }
+
+  if (typeof raw !== "object" || raw === null) {
+    console.error("[automem] config root must be an object, using defaults");
+    return DEFAULT_CONFIG;
+  }
+
+  const config = deepMerge(DEFAULT_CONFIG, raw) as AutoMemConfig;
+
+  if (config.startupRecall.limit < 1 || config.startupRecall.limit > 20) {
+    console.warn("[automem] startupRecall.limit out of range (1-20), clamping to 8");
+    config.startupRecall.limit = 8;
+  }
+  if (config.turnRecall.limit < 1 || config.turnRecall.limit > 20) {
+    console.warn("[automem] turnRecall.limit out of range (1-20), clamping to 6");
+    config.turnRecall.limit = 6;
+  }
+
+  const validDisplayModes = ["full", "summary", "hidden"];
+  if (!validDisplayModes.includes(config.behavior.displayRecall)) {
+    console.warn("[automem] unknown behavior.displayRecall \"" + config.behavior.displayRecall + "\", valid values: full, summary, hidden. Defaulting to \"summary\"");
+    config.behavior.displayRecall = "summary";
+  }
+
+  const validWriteModes = ["off", "propose", "safe-auto", "confirm-all"];
+  if (!validWriteModes.includes(config.writePolicy.mode)) {
+    console.warn("[automem] unknown writePolicy.mode \"" + config.writePolicy.mode + "\", valid values: off, propose, safe-auto, confirm-all. Defaulting to \"propose\"");
+    config.writePolicy.mode = "propose";
+  }
+
+  return config;
+}