@vellumai/assistant 0.6.2 → 0.6.4

package/src/workspace/migrations/017-seed-persona-dirs.ts

@@ -13,10 +13,72 @@ import { desc, eq } from "drizzle-orm";
 import { generateUserFileSlug } from "../../contacts/contact-store.js";
 import { getDb } from "../../memory/db.js";
 import { contacts } from "../../memory/schema/contacts.js";
-import { isTemplateContent } from "../../prompts/system-prompt.js";
-import { stripCommentLines } from "../../util/strip-comment-lines.js";
 import type { WorkspaceMigration } from "./types.js";
 
+// ── Inlined helpers ───────────────────────────────────────────────
+//
+// Per migrations/AGENTS.md, migrations must be self-contained. The
+// helpers below are duplicated inline (rather than imported from
+// `util/strip-comment-lines.js` or `prompts/system-prompt.js`) so this
+// migration does not regress if those modules change — or, in the case
+// of the legacy `templates/USER.md` template file, disappear entirely.
+// Migration 031-drop-user-md deletes that template file, and previously
+// this migration's unmodified-template check silently started copying
+// bare scaffolds once the template was gone.
+
+/**
+ * Strip lines starting with `_` (comment convention for prompt .md files)
+ * and collapse any resulting consecutive blank lines. Copied from
+ * `util/strip-comment-lines.ts` to keep this migration self-contained.
+ */
+function stripCommentLines(content: string): string {
+  const normalized = content.replace(/\r\n/g, "\n");
+  let openFenceChar: string | null = null;
+  const filtered = normalized.split("\n").filter((line) => {
+    const fenceMatch = line.match(/^ {0,3}(`{3,}|~{3,})/);
+    if (fenceMatch) {
+      const char = fenceMatch[1][0];
+      if (!openFenceChar) {
+        openFenceChar = char;
+      } else if (char === openFenceChar) {
+        openFenceChar = null;
+      }
+    }
+    if (openFenceChar) return true;
+    return !line.trimStart().startsWith("_");
+  });
+  return filtered
+    .join("\n")
+    .replace(/\n{3,}/g, "\n\n")
+    .trim();
+}
+
+/**
+ * Frozen snapshot of the legacy `templates/USER.md` contents that
+ * shipped at the time this migration was authored. Used to detect
+ * unmodified template installs so we don't copy a useless scaffold
+ * into `users/<slug>.md`. The template file itself is deleted by
+ * migration 031, so we cannot read it from disk here.
+ */
+const LEGACY_USER_MD_TEMPLATE = `_ Lines starting with _ are comments - they won't appear in the system prompt
+
+# USER.md
+
+Store details about your user here. Edit freely - build this over time as you learn about them. Don't be pushy about seeking details, but when you learn something, write it down. More context makes you more useful.
+
+- Preferred name/reference:
+- Pronouns:
+- Locale:
+- Work role:
+- Goals:
+- Hobbies/fun:
+- Daily tools:
+`;
+
+const LEGACY_USER_MD_TEMPLATE_STRIPPED = stripCommentLines(
+  LEGACY_USER_MD_TEMPLATE,
+);
+
 export const seedPersonaDirsMigration: WorkspaceMigration = {
   id: "017-seed-persona-dirs",
   description:
@@ -55,8 +117,10 @@ export const seedPersonaDirsMigration: WorkspaceMigration = {
     const content = stripCommentLines(rawContent);
     if (!content) return;
 
-    // Skip if the content is the unmodified template
-    if (isTemplateContent(content, "USER.md")) return;
+    // Skip if the content is the unmodified legacy template. We compare
+    // against an inlined snapshot rather than reading the bundled
+    // template from disk, since migration 031 deletes that template file.
+    if (content === LEGACY_USER_MD_TEMPLATE_STRIPPED) return;
 
     // Determine destination filename based on guardian contact
     let destFilename = "guardian.md";

package/src/workspace/migrations/029-seed-pkb.ts

@@ -7,7 +7,7 @@ const INDEX_TEMPLATE = `_ Lines starting with _ are comments - they won't appear
 
 # Knowledge Base
 
-**Remember aggressively.** When you learn ANY fact — a preference, a name, a date, a habit, a plan, an opinion — call \`remember\` immediately. Don't filter, don't judge importance. Remembering too much costs nothing. Forgetting something that mattered costs trust.
+**Remember aggressively.** Capture anything concrete about your user — preferences, names, dates, habits, plans, opinions, health details, commitments. Default to remembering; only skip obvious noise (small talk, hypotheticals). Don't judge importance — filing decides that later. Call \`remember\` immediately, multiple times per conversation. Remembering too much costs nothing. Forgetting something that mattered costs trust.
 
 ## Always Loaded
 - essentials.md — Core facts, patterns, and biographical info

package/src/workspace/migrations/031-drop-user-md.ts

@@ -0,0 +1,317 @@
+import {
+  copyFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  statSync,
+  unlinkSync,
+  writeFileSync,
+} from "node:fs";
+import { basename, join } from "node:path";
+
+import { eq } from "drizzle-orm";
+
+import {
+  findGuardianForChannel,
+  generateUserFileSlug,
+  listGuardianChannels,
+} from "../../contacts/contact-store.js";
+import { getDb } from "../../memory/db.js";
+import { contacts } from "../../memory/schema/contacts.js";
+import { getLogger } from "../../util/logger.js";
+import type { WorkspaceMigration } from "./types.js";
+
+const log = getLogger("workspace-migration-031-drop-user-md");
+
+// ── Inlined helpers ───────────────────────────────────────────────
+//
+// Per AGENTS.md, migrations should minimize cross-module imports so
+// they remain stable as code around them evolves. The helpers below
+// are duplicated inline (rather than imported from
+// `util/strip-comment-lines.js` and `prompts/system-prompt.js`) so
+// this migration does not regress if those modules change later.
+
+/**
+ * Strip lines starting with `_` (comment convention for prompt .md files)
+ * and collapse any resulting consecutive blank lines. Copied from
+ * `util/strip-comment-lines.ts` to keep this migration self-contained.
+ */
+function stripCommentLines(content: string): string {
+  const normalized = content.replace(/\r\n/g, "\n");
+  let openFenceChar: string | null = null;
+  const filtered = normalized.split("\n").filter((line) => {
+    const fenceMatch = line.match(/^ {0,3}(`{3,}|~{3,})/);
+    if (fenceMatch) {
+      const char = fenceMatch[1][0];
+      if (!openFenceChar) {
+        openFenceChar = char;
+      } else if (char === openFenceChar) {
+        openFenceChar = null;
+      }
+    }
+    if (openFenceChar) return true;
+    return !line.trimStart().startsWith("_");
+  });
+  return filtered
+    .join("\n")
+    .replace(/\n{3,}/g, "\n\n")
+    .trim();
+}
+
+/**
+ * Frozen snapshot of the legacy `templates/USER.md` contents shipped
+ * before this migration deletes it. Used to detect unmodified template
+ * installs so we don't copy a useless scaffold into `users/<slug>.md`.
+ */
+const LEGACY_USER_MD_TEMPLATE = `_ Lines starting with _ are comments - they won't appear in the system prompt
+
+# USER.md
+
+Store details about your user here. Edit freely - build this over time as you learn about them. Don't be pushy about seeking details, but when you learn something, write it down. More context makes you more useful.
+
+- Preferred name/reference:
+- Pronouns:
+- Locale:
+- Work role:
+- Goals:
+- Hobbies/fun:
+- Daily tools:
+`;
+
+const LEGACY_USER_MD_TEMPLATE_STRIPPED = stripCommentLines(
+  LEGACY_USER_MD_TEMPLATE,
+);
+
+/**
+ * Current `GUARDIAN_PERSONA_TEMPLATE` text from `prompts/persona-resolver.ts`,
+ * duplicated here for the same self-containment reason. Written when we
+ * need to seed an empty `users/<slug>.md` so new installs stay consistent
+ * with `ensureGuardianPersonaFile`.
+ */
+const GUARDIAN_PERSONA_TEMPLATE = `_ Lines starting with _ are comments - they won't appear in the system prompt
+
+# User Profile
+
+Store details about your user here. Edit freely - build this over time as you learn about them. Don't be pushy about seeking details, but when you learn something, write it down. More context makes you more useful.
+
+- Preferred name/reference:
+- Pronouns:
+- Locale:
+- Work role:
+- Goals:
+- Hobbies/fun:
+- Daily tools:
+`;
+
+function isLegacyTemplateContent(raw: string): boolean {
+  const stripped = stripCommentLines(raw);
+  if (stripped.length === 0) return true;
+  return stripped === LEGACY_USER_MD_TEMPLATE_STRIPPED;
+}
+
+function destFileIsMissingOrEmpty(destPath: string): boolean {
+  if (!existsSync(destPath)) return true;
+  try {
+    const raw = readFileSync(destPath, "utf-8");
+    return stripCommentLines(raw).length === 0;
+  } catch {
+    return true;
+  }
+}
+
+function isValidSlug(slug: string): boolean {
+  return basename(slug) === slug && slug !== "." && slug !== "..";
+}
+
+/**
+ * Delete the legacy `USER.md` at the workspace root after migrating
+ * any customized content into `users/<slug>.md`.
+ *
+ * Handles four relevant states:
+ *   1. Fresh install, no guardian → no-op (nothing to migrate yet;
+ *      `USER.md` is no longer seeded post PR 11).
+ *   2. Pre-017 customized `USER.md`, guardian has no `userFile` →
+ *      backfill the slug, copy `USER.md` → `users/<slug>.md`, delete `USER.md`.
+ *   3. Post-017 state where `users/<slug>.md` already has content →
+ *      do NOT overwrite; delete lingering `USER.md` regardless of content.
+ *   4. Missing `users/<slug>.md` after guardian is resolved → seed a bare
+ *      `GUARDIAN_PERSONA_TEMPLATE` scaffold so downstream readers have a file.
+ */
+export const dropUserMdMigration: WorkspaceMigration = {
+  id: "031-drop-user-md",
+  description:
+    "Delete legacy workspace-root USER.md after migrating content to users/<slug>.md",
+
+  run(workspaceDir: string): void {
+    const userMdPath = join(workspaceDir, "USER.md");
+
+    // Resolve the guardian contact. Prefer the vellum-channel binding
+    // (the canonical local/native guardian); fall back to whichever
+    // guardian has the most recently verified active channel.
+    let guardian: { id: string; displayName: string; userFile: string | null };
+    try {
+      const vellumGuardian = findGuardianForChannel("vellum");
+      if (vellumGuardian) {
+        guardian = {
+          id: vellumGuardian.contact.id,
+          displayName: vellumGuardian.contact.displayName,
+          userFile: vellumGuardian.contact.userFile ?? null,
+        };
+      } else {
+        const anyGuardian = listGuardianChannels();
+        if (!anyGuardian) {
+          // Fresh install or pre-onboarding. If a stale USER.md somehow
+          // remains on disk (e.g. leftover from an older build), best-
+          // effort remove it so future first runs are clean.
+          if (existsSync(userMdPath)) {
+            try {
+              unlinkSync(userMdPath);
+              log.info(
+                { path: userMdPath },
+                "Deleted stale pre-onboarding USER.md with no guardian",
+              );
+            } catch (err) {
+              log.warn(
+                { err, path: userMdPath },
+                "Failed to delete pre-onboarding USER.md; leaving in place",
+              );
+            }
+          }
+          return;
+        }
+        guardian = {
+          id: anyGuardian.contact.id,
+          displayName: anyGuardian.contact.displayName,
+          userFile: anyGuardian.contact.userFile ?? null,
+        };
+      }
+    } catch (err) {
+      // DB not ready or query failed — leave USER.md alone. The next
+      // startup after DB init will try again.
+      log.warn(
+        { err },
+        "Failed to resolve guardian contact; deferring USER.md cleanup",
+      );
+      return;
+    }
+
+    // Backfill a userFile slug on the guardian contact if one isn't set.
+    if (!guardian.userFile) {
+      try {
+        const slug = generateUserFileSlug(guardian.displayName);
+        const db = getDb();
+        db.update(contacts)
+          .set({ userFile: slug })
+          .where(eq(contacts.id, guardian.id))
+          .run();
+        guardian.userFile = slug;
+        log.info(
+          { contactId: guardian.id, slug },
+          "Backfilled missing guardian.userFile",
+        );
+      } catch (err) {
+        log.warn(
+          { err, contactId: guardian.id },
+          "Failed to backfill guardian.userFile; deferring USER.md cleanup",
+        );
+        return;
+      }
+    }
+
+    const userFile = guardian.userFile;
+    if (!userFile || !isValidSlug(userFile)) {
+      log.warn(
+        { userFile },
+        "Guardian userFile is missing or not a safe basename; deferring USER.md cleanup",
+      );
+      return;
+    }
+
+    const usersDir = join(workspaceDir, "users");
+    mkdirSync(usersDir, { recursive: true });
+
+    const destPath = join(usersDir, userFile);
+
+    // Read USER.md if it exists and classify its content.
+    let userMdRaw: string | null = null;
+    let userMdIsCustomized = false;
+    if (existsSync(userMdPath)) {
+      try {
+        // Guard against USER.md being a directory (hostile state).
+        if (statSync(userMdPath).isFile()) {
+          userMdRaw = readFileSync(userMdPath, "utf-8");
+          userMdIsCustomized = !isLegacyTemplateContent(userMdRaw);
+        }
+      } catch (err) {
+        log.warn(
+          { err, path: userMdPath },
+          "Failed to read USER.md; treating as unreadable",
+        );
+      }
+    }
+
+    // Copy customized USER.md content into users/<slug>.md when the
+    // destination is missing or effectively empty. Post-017 installs
+    // that already populated users/<slug>.md are left untouched.
+    if (
+      userMdIsCustomized &&
+      userMdRaw !== null &&
+      destFileIsMissingOrEmpty(destPath)
+    ) {
+      try {
+        copyFileSync(userMdPath, destPath);
+        log.info(
+          { src: userMdPath, dest: destPath },
+          "Copied customized USER.md content into users/<slug>.md",
+        );
+      } catch (err) {
+        log.warn(
+          { err, src: userMdPath, dest: destPath },
+          "Failed to copy USER.md; deferring USER.md cleanup",
+        );
+        return;
+      }
+    }
+
+    // Seed the guardian persona scaffold when the destination file
+    // still doesn't exist (e.g. no USER.md and no post-017 content).
+    // This keeps parity with `ensureGuardianPersonaFile` for new
+    // installs so downstream readers always find a file.
+    if (!existsSync(destPath)) {
+      try {
+        writeFileSync(destPath, GUARDIAN_PERSONA_TEMPLATE, "utf-8");
+        log.info(
+          { dest: destPath },
+          "Seeded guardian persona scaffold at users/<slug>.md",
+        );
+      } catch (err) {
+        log.warn(
+          { err, dest: destPath },
+          "Failed to seed guardian persona scaffold; continuing with USER.md deletion",
+        );
+      }
+    }
+
+    // Finally, delete the legacy USER.md if it still exists. Template
+    // or customized, it has no remaining consumer now that PR 11
+    // dropped the seed/fallback read path.
+    if (existsSync(userMdPath)) {
+      try {
+        unlinkSync(userMdPath);
+        log.info({ path: userMdPath }, "Deleted legacy workspace USER.md");
+      } catch (err) {
+        log.warn(
+          { err, path: userMdPath },
+          "Failed to delete legacy USER.md",
+        );
+      }
+    }
+  },
+
+  down(_workspaceDir: string): void {
+    // No-op: deletion is irreversible. Any content that was present in
+    // `USER.md` was copied into `users/<slug>.md` during `run()`, so
+    // rolling back this migration cannot restore the original file.
+    // Downstream migrations (if any) read from `users/<slug>.md`.
+  },
+};

package/src/workspace/migrations/031-llm-log-retention-zero-to-null.ts

@@ -0,0 +1,73 @@
+import { existsSync, readFileSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+
+import type { WorkspaceMigration } from "./types.js";
+
+/**
+ * Convert `memory.cleanup.llmRequestLogRetentionMs: 0` to `null`.
+ *
+ * Under the old semantics `0` meant "keep forever" (never prune). The new
+ * semantics use `null` for "keep forever" and `0` for "prune immediately".
+ * Without this migration, users who previously set the retention to "keep
+ * forever" would be silently switched to "prune everything on next cleanup
+ * run" after upgrading.
+ */
+export const llmLogRetentionZeroToNullMigration: WorkspaceMigration = {
+  id: "031-llm-log-retention-zero-to-null",
+  description:
+    "Convert llmRequestLogRetentionMs: 0 (old 'keep forever') to null (new 'keep forever')",
+  run(workspaceDir: string): void {
+    const configPath = join(workspaceDir, "config.json");
+    if (!existsSync(configPath)) return;
+
+    let config: Record<string, unknown>;
+    try {
+      const raw = JSON.parse(readFileSync(configPath, "utf-8"));
+      if (!raw || typeof raw !== "object" || Array.isArray(raw)) return;
+      config = raw as Record<string, unknown>;
+    } catch {
+      return; // Malformed config — skip
+    }
+
+    const memory = config.memory;
+    if (!memory || typeof memory !== "object" || Array.isArray(memory)) return;
+    const memoryObj = memory as Record<string, unknown>;
+
+    const cleanup = memoryObj.cleanup;
+    if (!cleanup || typeof cleanup !== "object" || Array.isArray(cleanup))
+      return;
+    const cleanupObj = cleanup as Record<string, unknown>;
+
+    if (cleanupObj.llmRequestLogRetentionMs !== 0) return;
+
+    cleanupObj.llmRequestLogRetentionMs = null;
+    writeFileSync(configPath, JSON.stringify(config, null, 2) + "\n");
+  },
+  down(workspaceDir: string): void {
+    const configPath = join(workspaceDir, "config.json");
+    if (!existsSync(configPath)) return;
+
+    let config: Record<string, unknown>;
+    try {
+      const raw = JSON.parse(readFileSync(configPath, "utf-8"));
+      if (!raw || typeof raw !== "object" || Array.isArray(raw)) return;
+      config = raw as Record<string, unknown>;
+    } catch {
+      return;
+    }
+
+    const memory = config.memory;
+    if (!memory || typeof memory !== "object" || Array.isArray(memory)) return;
+    const memoryObj = memory as Record<string, unknown>;
+
+    const cleanup = memoryObj.cleanup;
+    if (!cleanup || typeof cleanup !== "object" || Array.isArray(cleanup))
+      return;
+    const cleanupObj = cleanup as Record<string, unknown>;
+
+    if (cleanupObj.llmRequestLogRetentionMs !== null) return;
+
+    cleanupObj.llmRequestLogRetentionMs = 0;
+    writeFileSync(configPath, JSON.stringify(config, null, 2) + "\n");
+  },
+};