@vellumai/assistant 0.10.12 → 0.11.0-staging.1

package/src/plugins/defaults/memory/{graph → v1/graph}/injection.ts

@@ -1,12 +1,15 @@
 // ---------------------------------------------------------------------------
-// Memory Graph — Context assembly and injection tracking
+// Memory Graph — v1 context-block assembly and injection-image resolution
+//
+// Injection tracking (`InContextTracker`) lives in
+// `../../graph/in-context-tracker.ts` — it is kept on every tier.
 // ---------------------------------------------------------------------------
 
-import { optimizeImageForTransport } from "../../../../agent/image-optimize.js";
-import { getLogger } from "../logging.js";
-import { loadImageRefData } from "./image-ref-utils.js";
-import type { MemoryNode, ScoredNode } from "./types.js";
-import { isCapabilityNode } from "./types.js";
+import { optimizeImageForTransport } from "../../../../../agent/image-optimize.js";
+import { loadImageRefData } from "../../graph/image-ref-utils.js";
+import type { MemoryNode, ScoredNode } from "../../graph/types.js";
+import { isCapabilityNode } from "../../graph/types.js";
+import { getLogger } from "../../logging.js";
 
 // ---------------------------------------------------------------------------
 // Image injection budgets
@@ -25,111 +28,6 @@ export interface ResolvedImage {
   description: string;
 }
 
-// ---------------------------------------------------------------------------
-// InContextTracker — tracks which node IDs are visible to the LLM
-// ---------------------------------------------------------------------------
-
-export interface InjectionLogEntry {
-  nodeId: string;
-  turn: number;
-}
-
-export interface InContextTrackerSnapshot {
-  inContext: string[];
-  log: InjectionLogEntry[];
-  currentTurn: number;
-}
-
-/**
- * Tracks which memory graph nodes are currently in the LLM's context.
- * Handles:
- * - Deduplication: never re-inject a node already visible
- * - Compaction eviction: when context compaction removes turns,
- *   evict those nodes so they can be re-injected if relevant later
- */
-export class InContextTracker {
-  private inContext = new Set<string>();
-  private log: InjectionLogEntry[] = [];
-  private currentTurn = 0;
-
-  /** Mark nodes as loaded into context. */
-  add(nodeIds: string[]): void {
-    for (const id of nodeIds) {
-      this.inContext.add(id);
-      this.log.push({ nodeId: id, turn: this.currentTurn });
-    }
-  }
-
-  /** Check if a node is already in context. */
-  isInContext(nodeId: string): boolean {
-    return this.inContext.has(nodeId);
-  }
-
-  /** Filter candidates to only those not already in context. */
-  filterNew(candidates: ScoredNode[]): ScoredNode[] {
-    return candidates.filter((c) => !this.inContext.has(c.node.id));
-  }
-
-  /** Advance the turn counter. Called before each retrieval step. */
-  advanceTurn(): void {
-    this.currentTurn++;
-  }
-
-  /**
-   * Evict nodes that were injected in compacted turns.
-   * Called when context compaction removes message history.
-   */
-  evictCompactedTurns(upToTurn: number): void {
-    const evicted: string[] = [];
-    this.log = this.log.filter((entry) => {
-      if (entry.turn <= upToTurn) {
-        evicted.push(entry.nodeId);
-        return false;
-      }
-      return true;
-    });
-
-    // Only evict if the node isn't also loaded in a later turn
-    const stillPresent = new Set(this.log.map((e) => e.nodeId));
-    for (const id of evicted) {
-      if (!stillPresent.has(id)) {
-        this.inContext.delete(id);
-      }
-    }
-  }
-
-  /** Get all node IDs currently in context. Useful for extraction. */
-  getActiveNodeIds(): string[] {
-    return [...this.inContext];
-  }
-
-  /** Get the injection log. Useful for debugging. */
-  getLog(): InjectionLogEntry[] {
-    return [...this.log];
-  }
-
-  /** Current turn number. */
-  getTurn(): number {
-    return this.currentTurn;
-  }
-
-  /** Serialize tracker state for persistence across eviction. */
-  toJSON(): InContextTrackerSnapshot {
-    return {
-      inContext: [...this.inContext],
-      log: [...this.log],
-      currentTurn: this.currentTurn,
-    };
-  }
-
-  /** Restore tracker state from a persisted snapshot. Replaces current state. */
-  restoreFrom(snapshot: InContextTrackerSnapshot): void {
-    this.inContext = new Set(snapshot.inContext);
-    this.log = [...snapshot.log];
-    this.currentTurn = snapshot.currentTurn;
-  }
-}
-
 // ---------------------------------------------------------------------------
 // Context assembly — programmatic, not LLM
 //

package/src/plugins/defaults/memory/{graph → v1/graph}/narrative.ts

@@ -11,12 +11,12 @@
 
 import { getConfiguredProvider } from "@vellumai/plugin-api";
 
-import type { AssistantConfig } from "../../../../config/types.js";
-import { BackendUnavailableError } from "../host-utils.js";
-import { extractToolUse, userMessage } from "../llm-helpers.js";
-import { getLogger } from "../logging.js";
-import { queryNodes, updateNode } from "./store.js";
-import type { MemoryNode } from "./types.js";
+import type { AssistantConfig } from "../../../../../config/types.js";
+import { queryNodes, updateNode } from "../../graph/store.js";
+import type { MemoryNode } from "../../graph/types.js";
+import { BackendUnavailableError } from "../../host-utils.js";
+import { extractToolUse, userMessage } from "../../llm-helpers.js";
+import { getLogger } from "../../logging.js";
 
 const log = getLogger("graph-narrative");
 

package/src/plugins/defaults/memory/{graph → v1/graph}/pattern-scan.ts

@@ -10,11 +10,11 @@
 
 import { getConfiguredProvider } from "@vellumai/plugin-api";
 
-import type { AssistantConfig } from "../../../../config/types.js";
-import { BackendUnavailableError } from "../host-utils.js";
-import { extractToolUse, userMessage } from "../llm-helpers.js";
-import { getLogger } from "../logging.js";
-import { createEdge, createNode, queryNodes } from "./store.js";
+import type { AssistantConfig } from "../../../../../config/types.js";
+import { createEdge, createNode, queryNodes } from "../../graph/store.js";
+import { BackendUnavailableError } from "../../host-utils.js";
+import { extractToolUse, userMessage } from "../../llm-helpers.js";
+import { getLogger } from "../../logging.js";
 
 const log = getLogger("graph-pattern-scan");
 

package/src/plugins/defaults/memory/{graph → v1/graph}/retriever.test.ts

@@ -23,7 +23,7 @@ let embedCallCount = 0;
 // mock behavior: takes the first text input and returns a matching vector.
 let embedRouter: ((text: string) => number[]) | null = null;
 
-mock.module("../../../../persistence/embeddings/embed.js", () => ({
+mock.module("../../../../../persistence/embeddings/embed.js", () => ({
   embedWithRetry: async (
     _config: unknown,
     texts: unknown[],
@@ -45,9 +45,12 @@ mock.module("../../../../persistence/embeddings/embed.js", () => ({
   },
 }));
 
-mock.module("../../../../persistence/embeddings/embedding-backend.js", () => ({
-  selectedBackendSupportsMultimodal: async () => false,
-}));
+mock.module(
+  "../../../../../persistence/embeddings/embedding-backend.js",
+  () => ({
+    selectedBackendSupportsMultimodal: async () => false,
+  }),
+);
 
 // Optional input-aware search router. When set, chooses a candidate list
 // based on the query vector's identity (vector equality on the first 3 dims).
@@ -72,15 +75,15 @@ mock.module("@vellumai/plugin-api", () => ({
   getConfiguredProvider: async () => null,
 }));
 
-import { resetDbForTesting } from "../../../../__tests__/db-test-helpers.js";
-import { DEFAULT_CONFIG } from "../../../../config/defaults.js";
-import type { AssistantConfig } from "../../../../config/types.js";
-import { getMemorySqlite } from "../../../../persistence/db-connection.js";
-import { initializeDb } from "../../../../persistence/db-init.js";
-import { InContextTracker } from "./injection.js";
+import { resetDbForTesting } from "../../../../../__tests__/db-test-helpers.js";
+import { DEFAULT_CONFIG } from "../../../../../config/defaults.js";
+import type { AssistantConfig } from "../../../../../config/types.js";
+import { getMemorySqlite } from "../../../../../persistence/db-connection.js";
+import { initializeDb } from "../../../../../persistence/db-init.js";
+import { InContextTracker } from "../../graph/in-context-tracker.js";
+import { createNode } from "../../graph/store.js";
+import type { NewNode } from "../../graph/types.js";
 import { loadContextMemory, retrieveForTurn } from "./retriever.js";
-import { createNode } from "./store.js";
-import type { NewNode } from "./types.js";
 
 // These tests exercise v1 retrieval. `memory.v2.enabled` (default `true`)
 // makes `loadContextMemory` short-circuit, so disable it here to keep the
@@ -93,6 +96,16 @@ const TEST_CONFIG: AssistantConfig = {
   },
 };
 
+// Concept-page substrate active through the v3 arm: memory on, the v2
+// injection engine off, memory-v3 live.
+const V3_LIVE_CONFIG: AssistantConfig = {
+  ...TEST_CONFIG,
+  memory: {
+    ...TEST_CONFIG.memory,
+    v3: { ...TEST_CONFIG.memory.v3, live: true },
+  },
+};
+
 function makeCapabilityNode(content: string, capId: string): NewNode {
   const now = Date.now();
   return {
@@ -123,6 +136,36 @@ function makeCapabilityNode(content: string, capId: string): NewNode {
   };
 }
 
+function makeEpisodicNode(content: string): NewNode {
+  const now = Date.now();
+  return {
+    content,
+    type: "episodic",
+    created: now,
+    lastAccessed: now,
+    lastConsolidated: now,
+    eventDate: null,
+    emotionalCharge: {
+      valence: 0,
+      intensity: 0,
+      decayCurve: "linear",
+      decayRate: 0,
+      originalIntensity: 0,
+    },
+    fidelity: "vivid",
+    confidence: 1,
+    significance: 0.5,
+    stability: 14,
+    reinforcementCount: 0,
+    lastReinforced: now,
+    sourceConversations: [],
+    sourceType: "direct",
+    narrativeRole: null,
+    partOfStory: null,
+    imageRefs: null,
+  };
+}
+
 describe("loadContextMemory — query/sparse vector surfacing", () => {
   beforeAll(async () => {
     await initializeDb();
@@ -245,6 +288,86 @@ describe("retrieveForTurn — query/sparse vector surfacing", () => {
   });
 });
 
+describe("retrieveForTurn — concept-page substrate short-circuit", () => {
+  let seededNodeId = "";
+
+  beforeAll(async () => {
+    await initializeDb();
+  }, 30_000);
+
+  beforeEach(async () => {
+    embedShouldThrow = false;
+    embedVector = [0.5, 0.5, 0.5];
+    embedCallCount = 0;
+    embedRouter = null;
+    // Any embedded query resolves to the seeded node, so a retrieval that
+    // actually runs is unmistakable in the result.
+    searchRouter = () => [{ nodeId: seededNodeId, score: 0.95 }];
+    resetDbForTesting();
+    await initializeDb();
+    seededNodeId = createNode(
+      makeEpisodicNode("Notes on the quarterly planning meeting."),
+    ).id;
+  }, 30_000);
+
+  test("returns the empty result and never embeds under memory-v3", async () => {
+    const tracker = new InContextTracker();
+    const result = await retrieveForTurn({
+      assistantLastMessage: "What did we decide yesterday?",
+      userLastMessage: "We decided to ship on Friday.",
+      config: V3_LIVE_CONFIG,
+      tracker,
+    });
+
+    // No embedding round-trip, so no per-turn spend on a v3 assistant.
+    expect(embedCallCount).toBe(0);
+    expect(result.nodes).toEqual([]);
+    expect(result.serendipityNodes).toEqual([]);
+    expect(result.triggeredNodes).toEqual([]);
+    expect(result.queryVector).toBeUndefined();
+    expect(result.sparseVector).toBeUndefined();
+    expect(result.metrics.semanticHits).toBe(0);
+    expect(result.metrics.selectedCount).toBe(0);
+    expect(result.metrics.topCandidates).toEqual([]);
+    expect(result.metrics.embeddingProvider).toBeNull();
+  });
+
+  test("returns the empty result and never embeds when the v2 engine is enabled", async () => {
+    const tracker = new InContextTracker();
+    const result = await retrieveForTurn({
+      assistantLastMessage: "What did we decide yesterday?",
+      userLastMessage: "We decided to ship on Friday.",
+      // Stock config: `memory.v2.enabled` defaults true, which is the other
+      // arm of the concept-page substrate.
+      config: DEFAULT_CONFIG,
+      tracker,
+    });
+
+    expect(embedCallCount).toBe(0);
+    expect(result.nodes).toEqual([]);
+    expect(result.metrics.topCandidates).toEqual([]);
+  });
+
+  test("still retrieves on a v1 config (memory on, v2 disabled, v3 not live)", async () => {
+    const tracker = new InContextTracker();
+    const result = await retrieveForTurn({
+      assistantLastMessage: "What did we decide yesterday?",
+      userLastMessage: "We decided to ship on Friday.",
+      config: TEST_CONFIG,
+      tracker,
+    });
+
+    // The guard is a substrate gate, not a kill switch: v1 assistants still
+    // embed the last exchange and score the candidates it surfaces.
+    expect(embedCallCount).toBe(1);
+    expect(result.queryVector).toEqual([0.5, 0.5, 0.5]);
+    const candidateIds = new Set(
+      result.metrics.topCandidates.map((c) => c.nodeId),
+    );
+    expect(candidateIds.has(seededNodeId)).toBe(true);
+  });
+});
+
 describe("retrieveForTurn — topic-pivot recovery", () => {
   // Build a one-hot keyword router so the combined assistant+user query and
   // the user-only query produce distinct vectors that can be routed to
@@ -280,36 +403,6 @@ describe("retrieveForTurn — topic-pivot recovery", () => {
     return [];
   }
 
-  function makeEpisodicNode(content: string): NewNode {
-    const now = Date.now();
-    return {
-      content,
-      type: "episodic",
-      created: now,
-      lastAccessed: now,
-      lastConsolidated: now,
-      eventDate: null,
-      emotionalCharge: {
-        valence: 0,
-        intensity: 0,
-        decayCurve: "linear",
-        decayRate: 0,
-        originalIntensity: 0,
-      },
-      fidelity: "vivid",
-      confidence: 1,
-      significance: 0.5,
-      stability: 14,
-      reinforcementCount: 0,
-      lastReinforced: now,
-      sourceConversations: [],
-      sourceType: "direct",
-      narrativeRole: null,
-      partOfStory: null,
-      imageRefs: null,
-    };
-  }
-
   beforeAll(async () => {
     await initializeDb();
   }, 30_000);

package/src/plugins/defaults/memory/{graph → v1/graph}/retriever.ts

@@ -13,15 +13,29 @@ import {
 } from "@vellumai/plugin-api";
 import { selectedBackendSupportsMultimodal } from "@vellumai/plugin-api";
 
-import { usesConceptPageMemory } from "../../../../config/memory-v3-gate.js";
-import type { AssistantConfig } from "../../../../config/types.js";
-import { embedWithRetry } from "../../../../persistence/embeddings/embed.js";
-import { generateSparseEmbedding } from "../../../../persistence/embeddings/embedding-backend.js";
-import type { QdrantSparseVector } from "../../../../persistence/embeddings/qdrant-client.js";
-import { extractToolUse, userMessage } from "../llm-helpers.js";
-import { getLogger } from "../logging.js";
+import { usesConceptPageMemory } from "../../../../../config/memory-v3-gate.js";
+import type { AssistantConfig } from "../../../../../config/types.js";
+import { embedWithRetry } from "../../../../../persistence/embeddings/embed.js";
+import { generateSparseEmbedding } from "../../../../../persistence/embeddings/embedding-backend.js";
+import type { QdrantSparseVector } from "../../../../../persistence/embeddings/qdrant-client.js";
+import type { InContextTracker } from "../../graph/in-context-tracker.js";
+import {
+  getEdgesForNode,
+  getNodesByIds,
+  queryCapabilityNodes,
+  queryNodes,
+} from "../../graph/store.js";
+import { getActiveTriggersByType } from "../../graph/store.js";
+import type {
+  MemoryEdge,
+  MemoryNode,
+  RetrievalMetrics,
+  ScoredNode,
+} from "../../graph/types.js";
+import { isCapabilityNode } from "../../graph/types.js";
+import { extractToolUse, userMessage } from "../../llm-helpers.js";
+import { getLogger } from "../../logging.js";
 import { searchGraphNodes } from "./graph-search.js";
-import type { InContextTracker } from "./injection.js";
 import {
   computeActivationSpread,
   computeEffectiveSignificance,
@@ -32,26 +46,12 @@ import {
   weightsForContextLoad,
 } from "./scoring.js";
 import { sampleSerendipity } from "./serendipity.js";
-import {
-  getEdgesForNode,
-  getNodesByIds,
-  queryCapabilityNodes,
-  queryNodes,
-} from "./store.js";
-import { getActiveTriggersByType } from "./store.js";
 import {
   evaluateEventTriggers,
   evaluateSemanticTriggers,
   evaluateTemporalTriggers,
   type TriggeredResult,
 } from "./triggers.js";
-import type {
-  MemoryEdge,
-  MemoryNode,
-  RetrievalMetrics,
-  ScoredNode,
-} from "./types.js";
-import { isCapabilityNode } from "./types.js";
 
 const log = getLogger("graph-retriever");
 
@@ -991,6 +991,24 @@ export async function retrieveForTurn(
     topCandidates: [],
   };
 
+  // Concept-page memory owns the read path when active, so the v1 per-turn
+  // retriever contributes nothing under the substrate: a v2/v3 assistant never
+  // pays for the per-turn embedding round-trip, the Qdrant search, or trigger
+  // evaluation, and no v1 graph node is ever injected. The same guard fronts
+  // `loadContextMemory`, so the invariant holds for every caller of either
+  // entry point rather than depending on an upstream gate.
+  if (usesConceptPageMemory(opts.config.memory)) {
+    return {
+      nodes: [],
+      serendipityNodes: [],
+      triggeredNodes: [],
+      latencyMs: 0,
+      metrics: ZERO_METRICS,
+      queryVector: undefined,
+      sparseVector: undefined,
+    };
+  }
+
   // 1. Build query from last exchange
   const queryText = [opts.assistantLastMessage, opts.userLastMessage]
     .filter((m) => m.length > 0)

package/src/plugins/defaults/memory/{graph → v1/graph}/scoring.test.ts

@@ -1,5 +1,6 @@
 import { describe, expect, test } from "bun:test";
 
+import type { MemoryNode } from "../../graph/types.js";
 import {
   computeActivationSpread,
   computeEffectiveSignificance,
@@ -12,7 +13,6 @@ import {
   type ScoringWeights,
   weightsForContextLoad,
 } from "./scoring.js";
-import type { MemoryNode } from "./types.js";
 
 // ---------------------------------------------------------------------------
 // Test helpers

package/src/plugins/defaults/memory/{graph → v1/graph}/scoring.ts

@@ -2,7 +2,7 @@
 // Memory Graph — Scoring functions for retrieval
 // ---------------------------------------------------------------------------
 
-import type { MemoryEdge, MemoryNode, ScoredNode } from "./types.js";
+import type { MemoryEdge, MemoryNode, ScoredNode } from "../../graph/types.js";
 
 // ---------------------------------------------------------------------------
 // Temporal boost — cyclical similarity on hour / dayOfWeek / month

package/src/plugins/defaults/memory/{graph → v1/graph}/serendipity.ts

@@ -6,7 +6,7 @@
 // creates unexpected associations that can spark new connections.
 // ---------------------------------------------------------------------------
 
-import type { ScoredNode } from "./types.js";
+import type { ScoredNode } from "../../graph/types.js";
 
 /**
  * Sample `reserveSlots` candidates from the 30th–70th percentile band

package/src/plugins/defaults/memory/{graph → v1/graph}/triggers.test.ts

@@ -1,11 +1,11 @@
 import { describe, expect, test } from "bun:test";
 
+import type { MemoryTrigger } from "../../graph/types.js";
 import {
   evaluateEventTriggers,
   evaluateSemanticTriggers,
   evaluateTemporalTriggers,
 } from "./triggers.js";
-import type { MemoryTrigger } from "./types.js";
 
 // ---------------------------------------------------------------------------
 // Helpers

package/src/plugins/defaults/memory/{graph → v1/graph}/triggers.ts

@@ -2,7 +2,7 @@
 // Memory Graph — Trigger evaluation
 // ---------------------------------------------------------------------------
 
-import type { MemoryTrigger } from "./types.js";
+import type { MemoryTrigger } from "../../graph/types.js";
 
 export interface TriggeredResult {
   trigger: MemoryTrigger;

package/src/plugins/defaults/memory/{identity-context.ts → v1/identity-context.ts}

@@ -1,7 +1,7 @@
-import { resolveGuardianPersona } from "../../../prompts/persona-resolver.js";
-import { readPromptFile } from "../../../prompts/system-prompt.js";
-import { isTemplateContent } from "../../../prompts/template-detection.js";
-import { getWorkspacePromptPath } from "./paths.js";
+import { resolveGuardianPersona } from "../../../../prompts/persona-resolver.js";
+import { readPromptFile } from "../../../../prompts/system-prompt.js";
+import { isTemplateContent } from "../../../../prompts/template-detection.js";
+import { getWorkspacePromptPath } from "../paths.js";
 
 const IDENTITY_PROMPT_FILES = ["IDENTITY.md", "SOUL.md"] as const;
 

package/src/plugins/defaults/memory/{job-handlers → v1/job-handlers}/backfill.ts

@@ -1,19 +1,19 @@
 import { parseMessageMetadata } from "@vellumai/plugin-api";
 import { and, asc, eq, gt, or } from "drizzle-orm";
 
-import type { AssistantConfig } from "../../../../config/types.js";
+import type { AssistantConfig } from "../../../../../config/types.js";
 import {
   readMessageCursorCheckpoint,
   resetMessageCursorCheckpoint,
   writeMessageCursorCheckpoint,
-} from "../../../../persistence/checkpoints.js";
-import { getDb } from "../../../../persistence/db-connection.js";
+} from "../../../../../persistence/checkpoints.js";
+import { getDb } from "../../../../../persistence/db-connection.js";
 import {
   enqueueMemoryJob,
   type MemoryJob,
-} from "../../../../persistence/jobs-store.js";
-import { messages } from "../../../../persistence/schema/index.js";
-import { indexMessageNow } from "../indexer.js";
+} from "../../../../../persistence/jobs-store.js";
+import { messages } from "../../../../../persistence/schema/index.js";
+import { indexMessageNow } from "../../indexer.js";
 
 const BACKFILL_CHECKPOINT_KEY = "memory:backfill:last_created_at";
 const BACKFILL_CHECKPOINT_ID_KEY = "memory:backfill:last_message_id";

package/src/plugins/defaults/memory/{job-handlers → v1/job-handlers}/embedding.test.ts

@@ -13,7 +13,7 @@ const embedAndUpsertCalls: Array<{
   extraPayload: unknown;
 }> = [];
 
-mock.module("../../../../persistence/job-utils.js", () => ({
+mock.module("../../../../../persistence/job-utils.js", () => ({
   asString: (value: unknown) =>
     typeof value === "string" && value.length > 0 ? value : null,
   embedAndUpsert: async (
@@ -33,11 +33,11 @@ mock.module("../../../../persistence/job-utils.js", () => ({
   },
 }));
 
-import { resetDbForTesting } from "../../../../__tests__/db-test-helpers.js";
-import { getDb } from "../../../../persistence/db-connection.js";
-import { initializeDb } from "../../../../persistence/db-init.js";
-import type { MemoryJob } from "../../../../persistence/jobs-store.js";
-import { mediaAssets } from "../../../../persistence/schema/index.js";
+import { resetDbForTesting } from "../../../../../__tests__/db-test-helpers.js";
+import { getDb } from "../../../../../persistence/db-connection.js";
+import { initializeDb } from "../../../../../persistence/db-init.js";
+import type { MemoryJob } from "../../../../../persistence/jobs-store.js";
+import { mediaAssets } from "../../../../../persistence/schema/index.js";
 import { embedMediaJob } from "./embedding.js";
 
 describe("embedMediaJob", () => {

package/src/plugins/defaults/memory/{job-handlers → v1/job-handlers}/embedding.ts

@@ -3,17 +3,17 @@ import { readFile } from "node:fs/promises";
 import { embedAndUpsert } from "@vellumai/plugin-api";
 import { and, eq } from "drizzle-orm";
 
-import { getDb } from "../../../../persistence/db-connection.js";
-import type { EmbeddingInput } from "../../../../persistence/embeddings/embedding-types.js";
-import { asString } from "../../../../persistence/job-utils.js";
-import type { MemoryJob } from "../../../../persistence/jobs-store.js";
+import { getDb } from "../../../../../persistence/db-connection.js";
+import type { EmbeddingInput } from "../../../../../persistence/embeddings/embedding-types.js";
+import { asString } from "../../../../../persistence/job-utils.js";
+import type { MemoryJob } from "../../../../../persistence/jobs-store.js";
 import {
   mediaAssets,
   memorySegments,
   memorySummaries,
   messages,
-} from "../../../../persistence/schema/index.js";
-import { extractMediaBlocks } from "../message-media.js";
+} from "../../../../../persistence/schema/index.js";
+import { extractMediaBlocks } from "../../message-media.js";
 
 export async function embedSegmentJob(job: MemoryJob): Promise<void> {
   const segmentId = asString(job.payload.segmentId);