@vellumai/assistant 0.5.16 → 0.6.1

package/src/__tests__/memory-context-benchmark.benchmark.test.ts

@@ -1,297 +0,0 @@
-/**
- * Memory Context Benchmark Fixture
- *
- * Baseline (first green run target ranges):
- * - compaction.summaryCalls: 2-6
- * - compaction.estimatedInputTokens: < previousEstimatedInputTokens
- * - recall.injectedTokens: <= computed dynamic budget
- * - recall.enabled: true
- */
-import {
-  afterAll,
-  beforeAll,
-  beforeEach,
-  describe,
-  expect,
-  mock,
-  test,
-} from "bun:test";
-
-import { DEFAULT_CONFIG } from "../config/defaults.js";
-import { estimatePromptTokens } from "../context/token-estimator.js";
-import { ContextWindowManager } from "../context/window-manager.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
-import { computeRecallBudget } from "../memory/retrieval-budget.js";
-import { buildMemoryRecall } from "../memory/retriever.js";
-import { conversations, memorySegments, messages } from "../memory/schema.js";
-import type { Message, Provider } from "../providers/types.js";
-
-mock.module("../util/logger.js", () => ({
-  getLogger: () =>
-    new Proxy({} as Record<string, unknown>, {
-      get: () => () => {},
-    }),
-}));
-
-// Stub the local embedding backend so the real ONNX model never loads.
-mock.module("../memory/embedding-local.js", () => ({
-  LocalEmbeddingBackend: class {
-    readonly provider = "local" as const;
-    readonly model: string;
-    constructor(model: string) {
-      this.model = model;
-    }
-    async embed(texts: string[]): Promise<number[][]> {
-      return texts.map(() => new Array(384).fill(0));
-    }
-  },
-}));
-
-// Dynamic Qdrant mock so the benchmark can inject high-scoring results
-let mockQdrantResults: Array<{
-  id: string;
-  score: number;
-  payload: Record<string, unknown>;
-}> = [];
-
-mock.module("../memory/qdrant-client.js", () => ({
-  getQdrantClient: () => ({
-    searchWithFilter: async () => mockQdrantResults,
-    hybridSearch: async () => mockQdrantResults,
-    upsertPoints: async () => {},
-    deletePoints: async () => {},
-  }),
-  initQdrantClient: () => {},
-}));
-
-function makeLongMessages(turns: number): Message[] {
-  const rows: Message[] = [];
-  const userTail =
-    "Need fast tests, deterministic memory recall, and stable prompt budgeting in long sessions.";
-  const assistantTail =
-    "Tracked: compaction boundaries, recall sources, and quality constraints for incremental rollout.";
-  for (let i = 0; i < turns; i++) {
-    rows.push({
-      role: "user",
-      content: [
-        {
-          type: "text",
-          text: `[U${i}] ${userTail} Topic ${i % 9} branch codex/memory-${i}.`,
-        },
-      ],
-    });
-    rows.push({
-      role: "assistant",
-      content: [
-        {
-          type: "text",
-          text: `[A${i}] ${assistantTail} Result ${
-            i % 7
-          } with additional diagnostics.`,
-        },
-      ],
-    });
-  }
-  return rows;
-}
-
-function makeSummaryProvider(counter: { calls: number }): Provider {
-  return {
-    name: "mock",
-    async sendMessage() {
-      counter.calls += 1;
-      return {
-        content: [
-          {
-            type: "text",
-            text: `## Goals\n- Preserve rollout state\n## Constraints\n- Keep PRs small\n## Decisions\n- Call ${counter.calls}`,
-          },
-        ],
-        model: "mock-context-model",
-        usage: { inputTokens: 420, outputTokens: 85 },
-        stopReason: "end_turn",
-      };
-    },
-  };
-}
-
-function seedRecallConversation(conversationId: string, now: number): void {
-  const db = getDb();
-  db.insert(conversations)
-    .values({
-      id: conversationId,
-      title: null,
-      createdAt: now,
-      updatedAt: now,
-      totalInputTokens: 0,
-      totalOutputTokens: 0,
-      totalEstimatedCost: 0,
-      contextSummary: null,
-      contextCompactedMessageCount: 0,
-      contextCompactedAt: null,
-    })
-    .run();
-
-  for (let i = 0; i < 160; i++) {
-    const msgId = `msg-bench-${i}`;
-    const createdAt = now + i;
-    const text =
-      i % 5 === 0
-        ? `Decision ${i}: use Bun test fixtures for memory regressions and recall ranking checks.`
-        : `Progress ${i}: conversation indexing and retrieval diagnostics are stable.`;
-
-    db.insert(messages)
-      .values({
-        id: msgId,
-        conversationId,
-        role: i % 2 === 0 ? "user" : "assistant",
-        content: JSON.stringify([{ type: "text", text }]),
-        createdAt,
-      })
-      .run();
-
-    db.insert(memorySegments)
-      .values({
-        id: `seg-bench-${i}`,
-        messageId: msgId,
-        conversationId,
-        role: i % 2 === 0 ? "user" : "assistant",
-        segmentIndex: 0,
-        text,
-        tokenEstimate: 20,
-        scopeId: "default",
-        createdAt,
-        updatedAt: createdAt,
-      })
-      .run();
-  }
-}
-
-describe("Memory context benchmark", () => {
-  beforeAll(() => {
-    initializeDb();
-  });
-
-  beforeEach(() => {
-    const db = getDb();
-    db.run("DELETE FROM memory_item_sources");
-    db.run("DELETE FROM memory_embeddings");
-    db.run("DELETE FROM memory_items");
-
-    db.run("DELETE FROM memory_segments");
-    db.run("DELETE FROM messages");
-    db.run("DELETE FROM conversations");
-    db.run("DELETE FROM memory_jobs");
-    db.run("DELETE FROM memory_checkpoints");
-    mockQdrantResults = [];
-  });
-
-  afterAll(() => {
-    resetDb();
-  });
-
-  test("long-session compaction + recall benchmark remains within expected bounds", async () => {
-    const conversationId = "conv-memory-benchmark";
-    const now = 1_700_500_000_000;
-
-    seedRecallConversation(conversationId, now);
-
-    const longMessages = makeLongMessages(90); // 180 messages
-    const summaryCounter = { calls: 0 };
-    const manager = new ContextWindowManager({
-      provider: makeSummaryProvider(summaryCounter),
-      systemPrompt: "system prompt for compaction benchmark",
-      config: {
-        ...DEFAULT_CONFIG.contextWindow,
-        maxInputTokens: 6000,
-        targetBudgetRatio: 0.58,
-        compactThreshold: 0.6,
-        summaryBudgetRatio: 0.05,
-      },
-    });
-
-    const compacted = await manager.maybeCompact(longMessages);
-    expect(compacted.compacted).toBe(true);
-    expect(compacted.summaryCalls).toBeGreaterThan(0);
-    expect(compacted.summaryCalls).toBe(summaryCounter.calls);
-    expect(compacted.estimatedInputTokens).toBeLessThan(
-      compacted.previousEstimatedInputTokens,
-    );
-    expect(compacted.compactedMessages).toBeGreaterThan(0);
-
-    const recallConfig = {
-      ...DEFAULT_CONFIG,
-      contextWindow: { ...DEFAULT_CONFIG.contextWindow, maxInputTokens: 6000 },
-      memory: {
-        ...DEFAULT_CONFIG.memory,
-        embeddings: {
-          ...DEFAULT_CONFIG.memory.embeddings,
-          provider: "local" as const,
-          required: false,
-        },
-        retrieval: {
-          ...DEFAULT_CONFIG.memory.retrieval,
-          maxInjectTokens: 750,
-          dynamicBudget: {
-            enabled: true,
-            minInjectTokens: 160,
-            maxInjectTokens: 750,
-            targetHeadroomTokens: 900,
-          },
-        },
-      },
-    };
-
-    const recallBudget = computeRecallBudget({
-      estimatedPromptTokens: estimatePromptTokens(
-        compacted.messages,
-        "system prompt for compaction benchmark",
-        { providerName: "mock" },
-      ),
-      maxInputTokens: recallConfig.contextWindow.maxInputTokens,
-      targetHeadroomTokens:
-        recallConfig.memory.retrieval.dynamicBudget.targetHeadroomTokens,
-      minInjectTokens:
-        recallConfig.memory.retrieval.dynamicBudget.minInjectTokens,
-      maxInjectTokens:
-        recallConfig.memory.retrieval.dynamicBudget.maxInjectTokens,
-    });
-
-    // Seed Qdrant mock with a representative decision segment so
-    // the benchmark validates content quality, not just pipeline completion.
-    mockQdrantResults = [
-      {
-        id: "emb-bench-decision",
-        score: 0.9,
-        payload: {
-          target_type: "segment",
-          target_id: "seg-bench-0",
-          text: "Decision 0: use Bun test fixtures for memory regressions and recall ranking checks.",
-          kind: "segment",
-          created_at: now,
-          last_seen_at: now,
-        },
-      },
-    ];
-
-    const recall = await buildMemoryRecall(
-      "What decisions did we make about Bun tests and retrieval diagnostics?",
-      conversationId,
-      recallConfig,
-      { maxInjectTokensOverride: recallBudget },
-    );
-
-    expect(recall.enabled).toBe(true);
-    // With Qdrant mock returning a high-scoring result, content should be injected.
-    expect(recall.selectedCount).toBeGreaterThan(0);
-    expect(recall.injectedText).toContain("Bun test fixtures");
-    expect(recall.injectedTokens).toBeGreaterThan(0);
-    expect(recall.injectedTokens).toBeLessThanOrEqual(recallBudget);
-    expect(recallBudget).toBeGreaterThanOrEqual(
-      recallConfig.memory.retrieval.dynamicBudget.minInjectTokens,
-    );
-    expect(recallBudget).toBeLessThanOrEqual(
-      recallConfig.memory.retrieval.dynamicBudget.maxInjectTokens,
-    );
-  });
-});