@vellumai/assistant 0.5.16 → 0.6.1

package/src/__tests__/conversation-runtime-workspace.test.ts

@@ -3,7 +3,6 @@ import { describe, expect, test } from "bun:test";
 import {
   applyRuntimeInjections,
   injectWorkspaceTopLevelContext,
-  stripWorkspaceTopLevelContext,
 } from "../daemon/conversation-runtime-assembly.js";
 import type { Message } from "../providers/types.js";
 
@@ -15,16 +14,12 @@ function userMsg(text: string): Message {
   return { role: "user", content: [{ type: "text", text }] };
 }
 
-function assistantMsg(text: string): Message {
-  return { role: "assistant", content: [{ type: "text", text }] };
-}
-
 // ---------------------------------------------------------------------------
 // Tests
 // ---------------------------------------------------------------------------
 
 const sampleContext =
-  "<workspace_top_level>\nRoot: /sandbox\nDirectories: src, lib, tests\n</workspace_top_level>";
+  "<workspace>\nRoot: /sandbox\nDirectories: src, lib, tests\n</workspace>";
 
 describe("Workspace top-level context — injection", () => {
   test("prepends workspace block to user message content", () => {
@@ -62,93 +57,6 @@ describe("Workspace top-level context — injection", () => {
   });
 });
 
-describe("Workspace top-level context — stripping", () => {
-  test("strips injected workspace block from user messages", () => {
-    const messages: Message[] = [
-      {
-        role: "user",
-        content: [
-          { type: "text", text: sampleContext },
-          { type: "text", text: "Hello" },
-        ],
-      },
-      assistantMsg("Hi there"),
-    ];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-
-    expect(stripped).toHaveLength(2);
-    expect(stripped[0].content).toHaveLength(1);
-    expect((stripped[0].content[0] as { text: string }).text).toBe("Hello");
-  });
-
-  test("does not strip non-workspace text blocks", () => {
-    const messages: Message[] = [
-      userMsg("Regular message"),
-      assistantMsg("Response"),
-    ];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-    expect(stripped).toEqual(messages);
-  });
-
-  test("removes user message entirely when only workspace block remains", () => {
-    const messages: Message[] = [
-      { role: "user", content: [{ type: "text", text: sampleContext }] },
-      assistantMsg("Response"),
-    ];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-    expect(stripped).toHaveLength(1);
-    expect(stripped[0].role).toBe("assistant");
-  });
-
-  test("does not strip from assistant messages", () => {
-    const messages: Message[] = [assistantMsg(sampleContext)];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-    expect(stripped).toHaveLength(1);
-    expect((stripped[0].content[0] as { text: string }).text).toBe(
-      sampleContext,
-    );
-  });
-
-  test("preserves tool_result blocks during stripping", () => {
-    const messages: Message[] = [
-      {
-        role: "user",
-        content: [
-          { type: "text", text: sampleContext },
-          {
-            type: "tool_result",
-            tool_use_id: "tu_1",
-            content: "result",
-            is_error: false,
-          },
-        ],
-      },
-    ];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-    expect(stripped).toHaveLength(1);
-    expect(stripped[0].content).toHaveLength(1);
-    expect(stripped[0].content[0].type).toBe("tool_result");
-  });
-
-  test("no empty-message artifacts after stripping", () => {
-    const messages: Message[] = [
-      userMsg("Before"),
-      { role: "user", content: [{ type: "text", text: sampleContext }] },
-      assistantMsg("After"),
-    ];
-
-    const stripped = stripWorkspaceTopLevelContext(messages);
-    expect(stripped).toHaveLength(2);
-    expect(stripped[0].role).toBe("user");
-    expect(stripped[1].role).toBe("assistant");
-  });
-});
-
 describe("applyRuntimeInjections — workspace top-level context", () => {
   test("injects workspace context when provided", () => {
     const messages: Message[] = [userMsg("Hello")];

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-tool-setup-app-refresh.test.ts

@@ -2,10 +2,12 @@
  * Regression tests for app surface refresh and eventing side effects in
  * createToolExecutor (conversation-tool-setup.ts).
  *
- * The app_refresh hook is the sole hook for app change refresh. These tests
- * verify that app_refresh, app_create, and app_delete hooks fire correctly,
- * and that removed hooks (app_update, app_file_edit, app_file_write) no
- * longer trigger side effects.
+ * Tests verify that app_refresh, app_create, and app_delete hooks fire
+ * correctly, and that removed hooks (app_update, app_file_edit,
+ * app_file_write) no longer trigger side effects.
+ *
+ * File-change detection for file_write/file_edit is handled by
+ * AppSourceWatcher (see app-source-watcher.test.ts).
  */
 
 import { beforeEach, describe, expect, mock, test } from "bun:test";
@@ -42,6 +44,16 @@ mock.module("../tools/browser/browser-screencast.js", () => ({
   registerConversationSender: mock(() => {}),
 }));
 
+// Mock app-store functions used by handleAppChange in tool-side-effects
+mock.module("../memory/app-store.js", () => ({
+  getApp: mock(() => null),
+  getAppDirPath: mock(() => "/tmp/test-apps/dummy"),
+  isMultifileApp: mock(() => false),
+  getAppsDir: mock(() => "/tmp/test-apps"),
+  resolveAppIdByDirName: mock(() => null),
+  resolveAppIdFromPath: mock(() => null),
+}));
+
 // ---------------------------------------------------------------------------
 // Import createToolExecutor after mocks are in place
 // ---------------------------------------------------------------------------
@@ -423,4 +435,5 @@ describe("session-tool-setup app refresh side effects", () => {
       expect(updatePublishedSpy).toHaveBeenCalledTimes(1);
     });
   });
+
 });

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();