@vellumai/assistant 0.5.16 → 0.6.0

package/src/__tests__/conversation-agent-loop.test.ts

@@ -445,6 +445,15 @@ function makeCtx(
       assistantMessageChannel: "vellum" as const,
     }),
 
+    graphMemory: {
+      onCompacted: () => {},
+      prepareMemory: async () => ({ runMessages: [], injectedTokens: 0, latencyMs: 0, mode: "none" as const }),
+      reinjectCachedMemory: (messages: Message[]) => ({
+        runMessages: messages,
+        injectedTokens: 0,
+      }),
+    } as unknown as AgentLoopConversationContext["graphMemory"],
+
     ...overrides,
   } as AgentLoopConversationContext;
 }

package/src/__tests__/conversation-attachments.test.ts

@@ -1,5 +1,5 @@
 import { existsSync } from "node:fs";
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 mock.module("../util/logger.js", () => ({
   getLogger: () =>
@@ -48,14 +48,10 @@ mock.module("../permissions/types.js", () => ({
 import type { AssistantAttachmentDraft } from "../daemon/assistant-attachments.js";
 import { getFilePathForAttachment } from "../memory/attachments-store.js";
 import { addMessage, createConversation } from "../memory/conversation-crud.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 function resetTables() {
   const db = getDb();
   db.run("DELETE FROM message_attachments");

package/src/__tests__/conversation-attention-store.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 mock.module("../util/logger.js", () => ({
   getLogger: () =>
@@ -17,7 +17,7 @@ import {
   projectAssistantMessage,
   recordConversationSeenSignal,
 } from "../memory/conversation-attention-store.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 import {
   conversationAssistantAttentionState,
   conversationAttentionEvents,
@@ -66,10 +66,6 @@ function insertAssistantMessage(
     .run();
 }
 
-afterAll(() => {
-  resetDb();
-});
-
 describe("conversation-attention-store", () => {
   beforeEach(() => {
     clearTables();

package/src/__tests__/conversation-clear-safety.test.ts

@@ -9,7 +9,7 @@
  * - lifecycle_events contains `conversations_clear_all` audit entry after clear
  */
 
-import { afterAll, describe, expect, mock, test } from "bun:test";
+import { describe, expect, mock, test } from "bun:test";
 
 mock.module("../util/logger.js", () => ({
   getLogger: () =>
@@ -31,17 +31,13 @@ import {
   createConversation,
   getConversation,
 } from "../memory/conversation-crud.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 import { enforcePolicy, getPolicy } from "../runtime/auth/route-policy.js";
 import type { AuthContext, Scope } from "../runtime/auth/types.js";
 import { conversationManagementRouteDefinitions } from "../runtime/routes/conversation-management-routes.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 /** Build a synthetic AuthContext for testing. */
 function buildAuthContext(overrides?: {
   principalType?: AuthContext["principalType"];

package/src/__tests__/conversation-delete-schedule-cleanup.test.ts

@@ -3,7 +3,7 @@
  * job also deletes the schedule, preventing orphaned scheduled automations.
  */
 
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 mock.module("../util/logger.js", () => ({
   getLogger: () =>
@@ -23,16 +23,12 @@ import {
   createConversation,
   getConversation,
 } from "../memory/conversation-crud.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 import { conversationManagementRouteDefinitions } from "../runtime/routes/conversation-management-routes.js";
 import { createSchedule, getSchedule } from "../schedule/schedule-store.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 function getRawDb(): Database {
   return (getDb() as unknown as { $client: Database }).$client;
 }
@@ -72,9 +68,8 @@ describe("DELETE /conversations/:id — schedule cleanup", () => {
   beforeEach(() => {
     getRawDb().run("DELETE FROM cron_runs");
     getRawDb().run("DELETE FROM cron_jobs");
-    getRawDb().run("DELETE FROM memory_item_sources");
+    getRawDb().run("DELETE FROM memory_graph_nodes");
     getRawDb().run("DELETE FROM memory_segments");
-    getRawDb().run("DELETE FROM memory_items");
     getRawDb().run("DELETE FROM memory_summaries");
     getRawDb().run("DELETE FROM memory_embeddings");
     getRawDb().run("DELETE FROM memory_jobs");
@@ -297,9 +292,8 @@ describe("POST /conversations/:id/wipe — schedule cleanup", () => {
   beforeEach(() => {
     getRawDb().run("DELETE FROM cron_runs");
     getRawDb().run("DELETE FROM cron_jobs");
-    getRawDb().run("DELETE FROM memory_item_sources");
+    getRawDb().run("DELETE FROM memory_graph_nodes");
     getRawDb().run("DELETE FROM memory_segments");
-    getRawDb().run("DELETE FROM memory_items");
     getRawDb().run("DELETE FROM memory_summaries");
     getRawDb().run("DELETE FROM memory_embeddings");
     getRawDb().run("DELETE FROM memory_jobs");

package/src/__tests__/conversation-disk-view-integration.test.ts

@@ -13,7 +13,7 @@ import {
   rmSync,
 } from "node:fs";
 import { join } from "node:path";
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 // ---------------------------------------------------------------------------
 // Mocks — must come before any imports that depend on them
@@ -59,14 +59,10 @@ import {
   getConversationDirPath,
   syncMessageToDisk,
 } from "../memory/conversation-disk-view.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 function resetTables() {
   const db = getDb();
   db.run("DELETE FROM message_attachments");

package/src/__tests__/conversation-disk-view.test.ts

@@ -8,7 +8,7 @@ import {
 } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 // ---------------------------------------------------------------------------
 // Mocks — must come before any imports that depend on them
@@ -61,14 +61,10 @@ import {
   syncMessageToDisk,
   updateMetaFile,
 } from "../memory/conversation-disk-view.js";
-import { getDb, initializeDb, rawRun, resetDb } from "../memory/db.js";
+import { getDb, initializeDb, rawRun } from "../memory/db.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 function resetTables() {
   const db = getDb();
   db.run("DELETE FROM message_attachments");

package/src/__tests__/conversation-error.test.ts

@@ -90,7 +90,6 @@ describe("classifyConversationError", () => {
       "rate limit exceeded",
       "Rate-limit hit",
       "too many requests",
-      "overloaded",
     ];
 
     for (const msg of cases) {
@@ -98,12 +97,44 @@ describe("classifyConversationError", () => {
         const result = classifyConversationError(new Error(msg), baseCtx);
         expect(result.code).toBe("PROVIDER_RATE_LIMIT");
         expect(result.retryable).toBe(true);
-        expect(result.userMessage).toContain("busy");
+        expect(result.userMessage).toContain("rate limited");
         expect(result.errorCategory).toBe("rate_limit");
       });
     }
   });
 
+  describe("provider overloaded errors", () => {
+    it('classifies "overloaded" as PROVIDER_OVERLOADED', () => {
+      const result = classifyConversationError(
+        new Error("overloaded"),
+        baseCtx,
+      );
+      expect(result.code).toBe("PROVIDER_OVERLOADED");
+      expect(result.retryable).toBe(true);
+      expect(result.userMessage).toContain("overloaded");
+      expect(result.errorCategory).toBe("provider_overloaded");
+    });
+
+    it("classifies Anthropic overloaded_error (no statusCode) as PROVIDER_OVERLOADED", () => {
+      const err = new ProviderError(
+        'Anthropic API error (undefined): {"type":"error","error":{"type":"overloaded_error","message":"Overloaded"}}',
+        "anthropic",
+      );
+      const result = classifyConversationError(err, baseCtx);
+      expect(result.code).toBe("PROVIDER_OVERLOADED");
+      expect(result.retryable).toBe(true);
+      expect(result.errorCategory).toBe("provider_overloaded");
+    });
+
+    it("classifies ProviderError with 529 as PROVIDER_OVERLOADED", () => {
+      const err = new ProviderError("Overloaded", "anthropic", 529);
+      const result = classifyConversationError(err, baseCtx);
+      expect(result.code).toBe("PROVIDER_OVERLOADED");
+      expect(result.retryable).toBe(true);
+      expect(result.errorCategory).toBe("provider_overloaded");
+    });
+  });
+
   describe("provider API errors", () => {
     const cases = [
       "HTTP 500 Internal Server Error",

package/src/__tests__/conversation-fork-crud.test.ts

@@ -1,6 +1,6 @@
 import { existsSync, readFileSync } from "node:fs";
 import { join } from "node:path";
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 import { eq, like } from "drizzle-orm";
 
@@ -39,7 +39,7 @@ import {
   PRIVATE_CONVERSATION_FORK_ERROR,
 } from "../memory/conversation-crud.js";
 import { getConversationDirPath } from "../memory/conversation-disk-view.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 import { getRequestLogsByMessageId } from "../memory/llm-request-log-store.js";
 import {
   channelInboundEvents,
@@ -71,10 +71,6 @@ function parseMetadata(metadata: string | null): unknown {
   return metadata == null ? null : JSON.parse(metadata);
 }
 
-afterAll(() => {
-  resetDb();
-});
-
 describe("forkConversation", () => {
   beforeEach(() => {
     resetTables();

package/src/__tests__/conversation-history-web-search.test.ts

@@ -782,6 +782,11 @@ describe("web_search_tool_result structural guard", () => {
     // Extracts tool results from persisted message content for work-item
     // display. web_search_tool_result blocks are not relevant here.
     "runtime/routes/work-items-routes.ts",
+
+    // Media token counting iterates tool_result.contentBlocks for nested
+    // image/file blocks. web_search_tool_result has opaque content with no
+    // contentBlocks property, so it cannot contain nested media.
+    "daemon/context-overflow-reducer.ts",
   ]);
 
   /**

package/src/__tests__/conversation-load-history-repair.test.ts

@@ -106,7 +106,7 @@ function makeConversation(): Conversation {
       stopReason: "end_turn",
     }),
   };
-  return new Conversation(
+  const conv = new Conversation(
     "conv-1",
     provider,
     "system prompt",
@@ -114,6 +114,10 @@ function makeConversation(): Conversation {
     () => {},
     "/tmp",
   );
+  // Default to guardian trust so history repair tests load all messages.
+  // Tests that exercise untrusted-actor filtering override this explicitly.
+  conv.setTrustContext({ trustClass: "guardian", sourceChannel: "vellum" });
+  return conv;
 }
 
 describe("loadFromDb history repair", () => {

package/src/__tests__/conversation-media-retry.test.ts

@@ -18,6 +18,15 @@ function makeImageBlock(
   };
 }
 
+function makeImageBlockWithSize(
+  dataLength: number,
+): Extract<ContentBlock, { type: "image" }> {
+  return {
+    type: "image",
+    source: { type: "base64", media_type: "image/png", data: "A".repeat(dataLength) },
+  };
+}
+
 function makeUserMessage(...blocks: ContentBlock[]): Message {
   return { role: "user", content: blocks };
 }
@@ -88,4 +97,86 @@ describe("stripMediaPayloadsForRetry", () => {
     const latestBlocks = result.messages[3].content;
     expect(latestBlocks.filter((b) => b.type === "image").length).toBe(1);
   });
+
+  // ---------------------------------------------------------------------------
+  // Budget-aware media retention
+  // ---------------------------------------------------------------------------
+
+  test("budget-aware: keeps images that fit within token budget", () => {
+    // Non-Anthropic estimation: estimateTextTokens(base64Data) + overhead (~19 tokens).
+    // Data length 4000 → 1000 data tokens + 19 overhead ≈ 1019 tokens/image.
+    // Budget of 3500 allows 3 images (3 * 1019 = 3057 <= 3500) but not 4.
+    const images = Array.from({ length: 5 }, () => makeImageBlockWithSize(4000));
+    const messages: Message[] = [
+      makeUserMessage({ type: "text", text: "describe these" }, ...images),
+    ];
+
+    const result = stripMediaPayloadsForRetry(messages, {
+      mediaTokenBudget: 3500,
+      providerName: "mock",
+    });
+    expect(result.modified).toBe(true);
+
+    const content = result.messages[0].content;
+    const keptImages = content.filter((b) => b.type === "image");
+    const stubs = content.filter(
+      (b) => b.type === "text" && (b as { text: string }).text.includes("Image omitted"),
+    );
+    expect(keptImages.length).toBe(3);
+    expect(stubs.length).toBe(2);
+  });
+
+  test("budget-aware: keeps all images when budget is generous", () => {
+    const images = Array.from({ length: 3 }, () => makeImageBlockWithSize(100));
+    const messages: Message[] = [
+      makeUserMessage({ type: "text", text: "describe these" }, ...images),
+    ];
+
+    const result = stripMediaPayloadsForRetry(messages, {
+      mediaTokenBudget: 100_000,
+      providerName: "mock",
+    });
+    expect(result.modified).toBe(false);
+    expect(result.replacedBlocks).toBe(0);
+
+    const content = result.messages[0].content;
+    const keptImages = content.filter((b) => b.type === "image");
+    expect(keptImages.length).toBe(3);
+  });
+
+  test("budget-aware: stubs all when budget is zero", () => {
+    const images = Array.from({ length: 3 }, () => makeImageBlockWithSize(100));
+    const messages: Message[] = [
+      makeUserMessage({ type: "text", text: "describe these" }, ...images),
+    ];
+
+    const result = stripMediaPayloadsForRetry(messages, {
+      mediaTokenBudget: 0,
+      providerName: "mock",
+    });
+    expect(result.modified).toBe(true);
+    expect(result.replacedBlocks).toBe(3);
+
+    const content = result.messages[0].content;
+    const keptImages = content.filter((b) => b.type === "image");
+    expect(keptImages.length).toBe(0);
+  });
+
+  test("no options falls back to hardcoded limit", () => {
+    const images = Array.from({ length: 5 }, () => makeImageBlockWithSize(100));
+    const messages: Message[] = [
+      makeUserMessage({ type: "text", text: "describe these" }, ...images),
+    ];
+
+    const result = stripMediaPayloadsForRetry(messages);
+    expect(result.modified).toBe(true);
+
+    const content = result.messages[0].content;
+    const keptImages = content.filter((b) => b.type === "image");
+    const stubs = content.filter(
+      (b) => b.type === "text" && (b as { text: string }).text.includes("Image omitted"),
+    );
+    expect(keptImages.length).toBe(3);
+    expect(stubs.length).toBe(2);
+  });
 });

package/src/__tests__/conversation-starter-routes.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 import { v4 as uuid } from "uuid";
 
@@ -9,7 +9,7 @@ mock.module("../util/logger.js", () => ({
     }),
 }));
 
-import { getSqlite, initializeDb, resetDb } from "../memory/db.js";
+import { getSqlite, initializeDb } from "../memory/db.js";
 import {
   conversationStarterRouteDefinitions,
   orderStrongestFirst,
@@ -17,10 +17,6 @@ import {
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 const routes = conversationStarterRouteDefinitions();
 
 function dispatch(path: string): Response | Promise<Response> {
@@ -41,7 +37,7 @@ function dispatch(path: string): Response | Promise<Response> {
 
 function clearTables() {
   getSqlite().run("DELETE FROM conversation_starters");
-  getSqlite().run("DELETE FROM memory_items");
+  getSqlite().run("DELETE FROM memory_graph_nodes");
   getSqlite().run("DELETE FROM memory_jobs");
   getSqlite().run("DELETE FROM memory_checkpoints");
 }
@@ -73,10 +69,23 @@ function insertStarter(overrides: {
 function insertMemoryItem(scopeId = "default") {
   const now = Date.now();
   getSqlite().run(
-    `INSERT INTO memory_items (
-      id, kind, subject, statement, status, confidence, fingerprint, scope_id, first_seen_at, last_seen_at
-    ) VALUES (?, 'fact', 'test', 'test statement', 'active', 0.9, ?, ?, ?, ?)`,
-    [uuid(), `fingerprint-${uuid()}`, scopeId, now, now],
+    `INSERT INTO memory_graph_nodes (
+      id, content, type, created, last_accessed, last_consolidated,
+      emotional_charge, fidelity, confidence, significance,
+      stability, reinforcement_count, last_reinforced,
+      source_conversations, source_type, scope_id
+    ) VALUES (?, ?, ?, ?, ?, ?, ?, 'vivid', 0.8, 0.5, 14, 0, ?, '[]', 'inferred', ?)`,
+    [
+      uuid(),
+      "test\ntest statement",
+      "semantic",
+      now,
+      now,
+      now,
+      '{"valence":0,"intensity":0.1,"decayCurve":"linear","decayRate":0.05,"originalIntensity":0.1}',
+      now,
+      scopeId,
+    ],
   );
 }
 

package/src/__tests__/conversation-store.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 mock.module("../util/logger.js", () => ({
   getLogger: () =>
@@ -24,15 +24,11 @@ import {
   getMessages,
 } from "../memory/conversation-crud.js";
 import { isLastUserMessageToolResult } from "../memory/conversation-queries.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 
 // Initialize db once before all tests
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 describe("deleteLastExchange", () => {
   beforeEach(() => {
     // Reset database between tests by dropping and recreating tables

package/src/__tests__/conversation-usage.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
+import { beforeEach, describe, expect, mock, test } from "bun:test";
 
 const updateConversationUsageCalls: Array<{
   conversationId: string;
@@ -37,17 +37,13 @@ mock.module("../memory/conversation-crud.js", () => ({
 }));
 
 import { recordUsage } from "../daemon/conversation-usage.js";
-import { getDb, initializeDb, resetDb } from "../memory/db.js";
+import { getDb, initializeDb } from "../memory/db.js";
 import { listUsageEvents } from "../memory/llm-usage-store.js";
 import type { PricingUsage } from "../usage/types.js";
 import { resolvePricingForUsageWithOverrides } from "../util/pricing.js";
 
 initializeDb();
 
-afterAll(() => {
-  resetDb();
-});
-
 describe("recordUsage", () => {
   beforeEach(() => {
     const db = getDb();