@vellumai/assistant 0.6.0 → 0.6.2

package/src/skills/skill-memory.ts

@@ -1,25 +1,6 @@
-import { and, eq, sql } from "drizzle-orm";
-import { v4 as uuid } from "uuid";
-
-import { isAssistantFeatureFlagEnabled } from "../config/assistant-feature-flags.js";
-import { getConfig } from "../config/loader.js";
-import { resolveSkillStates } from "../config/skill-state.js";
-import { loadSkillCatalog, type SkillSummary } from "../config/skills.js";
-import { getDb } from "../memory/db.js";
-import { enqueueMemoryJob } from "../memory/jobs-store.js";
-import { memoryGraphNodes } from "../memory/schema.js";
-import { getLogger } from "../util/logger.js";
-import { getCachedCatalogSync } from "./catalog-cache.js";
+import type { SkillSummary } from "../config/skills.js";
 import type { CatalogSkill } from "./catalog-install.js";
 
-const log = getLogger("skill-memory");
-
-/** Escape SQL LIKE wildcards so they match literally.
- * Uses backslash as the escape character — callers must pair with ESCAPE '\\'. */
-function escapeLike(s: string): string {
-  return s.replace(/\\/g, "\\\\").replace(/%/g, "\\%").replace(/_/g, "\\_");
-}
-
 /**
  * Generic input for building capability statements.
  * Decoupled from CatalogSkill so other skill sources (e.g. bundled skills) can
@@ -61,276 +42,3 @@ export function fromCatalogSkill(entry: CatalogSkill): SkillCapabilityInput {
   };
 }
 
-/**
- * Build a semantically rich capability statement from a skill capability input.
- * Truncated to 500 chars max (matching the limit used by memory item extraction).
- */
-export function buildCapabilityStatement(input: SkillCapabilityInput): string {
-  const { displayName, activationHints, avoidWhen } = input;
-
-  let statement = `The "${displayName}" skill (${input.id}) is available. ${input.description}.`;
-  if (activationHints && activationHints.length > 0) {
-    statement += ` Use when: ${activationHints.join("; ")}.`;
-  }
-  if (avoidWhen && avoidWhen.length > 0) {
-    statement += ` Avoid when: ${avoidWhen.join("; ")}.`;
-  }
-
-  // Truncate to 500 chars max
-  if (statement.length > 500) {
-    statement = statement.slice(0, 500);
-  }
-
-  return statement;
-}
-
-/** Default emotional charge for capability graph nodes. */
-const DEFAULT_EMOTIONAL_CHARGE = JSON.stringify({
-  valence: 0,
-  intensity: 0.1,
-  decayCurve: "linear",
-  decayRate: 0.05,
-  originalIntensity: 0.1,
-});
-
-/**
- * Upsert a capability memory graph node for a skill.
- * Best-effort: errors are logged but never thrown.
- */
-export function upsertSkillCapabilityMemory(
-  skillId: string,
-  input: SkillCapabilityInput,
-): void {
-  try {
-    const db = getDb();
-    const statement = buildCapabilityStatement(input);
-    const content = `skill:${skillId}\n${statement}`;
-    const scopeId = "default";
-    const now = Date.now();
-
-    const existing = db
-      .select()
-      .from(memoryGraphNodes)
-      .where(
-        and(
-          eq(memoryGraphNodes.type, "procedural"),
-          sql`${memoryGraphNodes.content} LIKE ${'skill:' + escapeLike(skillId) + '\n%'} ESCAPE '\\'`,
-          eq(memoryGraphNodes.scopeId, scopeId),
-        ),
-      )
-      .get();
-
-    if (existing) {
-      if (
-        existing.content === content &&
-        existing.fidelity !== "gone"
-      ) {
-        // Same content — just touch lastAccessed
-        db.update(memoryGraphNodes)
-          .set({ lastAccessed: now })
-          .where(eq(memoryGraphNodes.id, existing.id))
-          .run();
-        return;
-      }
-
-      if (existing.fidelity !== "gone") {
-        // Content changed — update content
-        db.update(memoryGraphNodes)
-          .set({
-            content,
-            lastAccessed: now,
-          })
-          .where(eq(memoryGraphNodes.id, existing.id))
-          .run();
-        enqueueMemoryJob("embed_graph_node", { nodeId: existing.id });
-        return;
-      }
-
-      // fidelity === "gone" — reactivate
-      db.update(memoryGraphNodes)
-        .set({
-          fidelity: "vivid",
-          content,
-          created: now,
-          lastAccessed: now,
-        })
-        .where(eq(memoryGraphNodes.id, existing.id))
-        .run();
-      enqueueMemoryJob("embed_graph_node", { nodeId: existing.id });
-      log.info({ skillId, nodeId: existing.id }, "Reactivated skill capability memory");
-      return;
-    }
-
-    // No existing — insert new graph node
-    const id = uuid();
-    db.insert(memoryGraphNodes)
-      .values({
-        id,
-        content,
-        type: "procedural",
-        created: now,
-        lastAccessed: now,
-        lastConsolidated: now,
-        emotionalCharge: DEFAULT_EMOTIONAL_CHARGE,
-        fidelity: "vivid",
-        confidence: 1.0,
-        significance: 0.7,
-        stability: 14,
-        reinforcementCount: 0,
-        lastReinforced: now,
-        sourceConversations: JSON.stringify([]),
-        sourceType: "inferred",
-        narrativeRole: null,
-        partOfStory: null,
-        scopeId,
-      })
-      .run();
-    enqueueMemoryJob("embed_graph_node", { nodeId: id });
-    log.info({ skillId, nodeId: id }, "Created skill capability memory");
-  } catch (err) {
-    log.warn({ err, skillId }, "Failed to upsert skill capability memory");
-  }
-}
-
-/**
- * Soft-delete the capability memory graph node for a skill.
- * Best-effort: errors are logged but never thrown.
- */
-export function deleteSkillCapabilityMemory(skillId: string): void {
-  try {
-    const db = getDb();
-    const now = Date.now();
-
-    const existing = db
-      .select()
-      .from(memoryGraphNodes)
-      .where(
-        and(
-          eq(memoryGraphNodes.type, "procedural"),
-          sql`${memoryGraphNodes.content} LIKE ${'skill:' + escapeLike(skillId) + '\n%'} ESCAPE '\\'`,
-          eq(memoryGraphNodes.scopeId, "default"),
-          sql`${memoryGraphNodes.fidelity} != 'gone'`,
-        ),
-      )
-      .get();
-
-    if (existing) {
-      db.update(memoryGraphNodes)
-        .set({ fidelity: "gone", lastAccessed: now })
-        .where(eq(memoryGraphNodes.id, existing.id))
-        .run();
-    }
-  } catch (err) {
-    log.warn({ err, skillId }, "Failed to delete skill capability memory");
-  }
-}
-
-/**
- * Seed capability memory graph nodes for all enabled skills (bundled, managed, workspace, extra).
- * Prunes stale entries whose skills are no longer in the enabled set.
- * Best-effort: errors are logged but never thrown.
- */
-export function seedCatalogSkillMemories(): void {
-  try {
-    const catalog = loadSkillCatalog();
-    const config = getConfig();
-    const resolved = resolveSkillStates(catalog, config);
-    const enabled = resolved.filter((r) => r.state === "enabled");
-
-    const catalogIds = new Set<string>();
-    for (const { summary } of enabled) {
-      catalogIds.add(summary.id);
-      const input = fromSkillSummary(summary);
-
-      // Enrich mcp-setup description with configured server names
-      if (summary.id === "mcp-setup") {
-        const servers = config.mcp?.servers;
-        if (servers) {
-          const names = Object.keys(servers).filter(
-            (name) => servers[name]?.enabled !== false,
-          );
-          if (names.length > 0) {
-            input.description += ` Configured: ${names.join(", ")}`;
-          }
-        }
-      }
-
-      upsertSkillCapabilityMemory(summary.id, input);
-    }
-
-    // Prune stale capability memories for skills no longer in the enabled set
-    // and not available in the remote/local catalog.
-    const db = getDb();
-    const allCapabilities = db
-      .select()
-      .from(memoryGraphNodes)
-      .where(
-        and(
-          eq(memoryGraphNodes.type, "procedural"),
-          eq(memoryGraphNodes.scopeId, "default"),
-          sql`${memoryGraphNodes.fidelity} != 'gone'`,
-        ),
-      )
-      .all();
-
-    const allLocalSkillIds = new Set(catalog.map((s) => s.id));
-    const cachedCatalog = getCachedCatalogSync();
-    const cachedCatalogIds = new Set(cachedCatalog.map((s) => s.id));
-
-    const now = Date.now();
-    for (const item of allCapabilities) {
-      if (!item.content.startsWith("skill:")) continue;
-      const itemSkillId = item.content.split("\n")[0].replace("skill:", "");
-
-      // Keep enabled skills
-      if (catalogIds.has(itemSkillId)) continue;
-
-      // Keep uninstalled catalog skills that are still in the remote catalog
-      if (cachedCatalogIds.has(itemSkillId)) continue;
-
-      // If the catalog cache is empty (cold start, before async fetch),
-      // we can't tell whether an unknown skill is a stale entry or
-      // a valid uninstalled catalog skill. Only prune skills we can
-      // positively identify as local-but-disabled.
-      if (cachedCatalogIds.size === 0 && !allLocalSkillIds.has(itemSkillId)) continue;
-
-      log.info({ skillId: itemSkillId, nodeId: item.id, catalogSize: catalogIds.size, cacheSize: cachedCatalogIds.size }, "Pruning stale skill capability memory");
-      db.update(memoryGraphNodes)
-        .set({ fidelity: "gone", lastAccessed: now })
-        .where(eq(memoryGraphNodes.id, item.id))
-        .run();
-    }
-  } catch (err) {
-    log.warn({ err }, "Failed to seed catalog skill memories");
-  }
-}
-
-/**
- * Seed capability memories for catalog skills that are not yet installed.
- * This makes uninstalled skills discoverable via memory injection so the LLM
- * can auto-install them via skill_load when relevant.
- * Best-effort: errors are logged but never thrown.
- */
-export async function seedUninstalledCatalogSkillMemories(): Promise<void> {
-  try {
-    const { getCatalog } = await import("./catalog-cache.js");
-    const fullCatalog = await getCatalog();
-    if (fullCatalog.length === 0) return;
-
-    const installedCatalog = loadSkillCatalog();
-    const installedIds = new Set(installedCatalog.map((s) => s.id));
-
-    const config = getConfig();
-    for (const entry of fullCatalog) {
-      if (installedIds.has(entry.id)) continue;
-
-      const flagKey = entry.metadata?.vellum?.["feature-flag"];
-      if (flagKey && !isAssistantFeatureFlagEnabled(flagKey, config)) continue;
-
-      const input = fromCatalogSkill(entry);
-      upsertSkillCapabilityMemory(entry.id, input);
-    }
-  } catch (err) {
-    log.warn({ err }, "Failed to seed uninstalled catalog skill memories");
-  }
-}

package/src/subagent/index.ts

@@ -1,6 +1,16 @@
-export { SubagentManager } from "./manager.js";
-export type { SubagentConfig, SubagentState, SubagentStatus } from "./types.js";
-export { SUBAGENT_LIMITS, TERMINAL_STATUSES } from "./types.js";
+export { mergeSkillIds, SubagentManager } from "./manager.js";
+export type {
+  SubagentConfig,
+  SubagentRole,
+  SubagentRoleConfig,
+  SubagentState,
+  SubagentStatus,
+} from "./types.js";
+export {
+  SUBAGENT_LIMITS,
+  SUBAGENT_ROLE_REGISTRY,
+  TERMINAL_STATUSES,
+} from "./types.js";
 
 import { SubagentManager } from "./manager.js";