@vellumai/assistant 0.11.2 → 0.11.3-staging.2

package/src/__tests__/list-messages-attachments.test.ts

@@ -103,10 +103,14 @@ describe("handleListMessages attachments", () => {
       "user",
       JSON.stringify([{ type: "text", text: "check this image" }]),
     );
-    const stored = uploadAttachment("photo.png", "image/png", IMAGE_BASE64);
+    const stored = await uploadAttachment(
+      "photo.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     expect(body.messages).toHaveLength(1);
@@ -124,14 +128,14 @@ describe("handleListMessages attachments", () => {
       "user",
       JSON.stringify([{ type: "text", text: "check this doc" }]),
     );
-    const stored = uploadAttachment(
+    const stored = await uploadAttachment(
       "report.pdf",
       "application/pdf",
       DOC_BASE64,
     );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     expect(body.messages).toHaveLength(1);
@@ -150,10 +154,14 @@ describe("handleListMessages attachments", () => {
       "assistant",
       JSON.stringify([{ type: "text", text: "here is an image" }]),
     );
-    const stored = uploadAttachment("result.png", "image/png", IMAGE_BASE64);
+    const stored = await uploadAttachment(
+      "result.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     expect(body.messages).toHaveLength(1);
@@ -172,8 +180,12 @@ describe("handleListMessages attachments", () => {
       "user",
       JSON.stringify([{ type: "text", text: "here are files" }]),
     );
-    const imgStored = uploadAttachment("photo.png", "image/png", IMAGE_BASE64);
-    const docStored = uploadAttachment(
+    const imgStored = await uploadAttachment(
+      "photo.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
+    const docStored = await uploadAttachment(
       "doc.pdf",
       "application/pdf",
       DOC_BASE64,
@@ -181,7 +193,7 @@ describe("handleListMessages attachments", () => {
     linkAttachmentToMessage(msg.id, imgStored.id, 0);
     linkAttachmentToMessage(msg.id, docStored.id, 1);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     const attachments = body.messages[0].attachments!;
@@ -200,7 +212,11 @@ describe("handleListMessages attachments", () => {
     // duplicating it doubles every image in the response and in client
     // memory.
     const conv = createConversation();
-    const stored = uploadAttachment("photo.png", "image/png", IMAGE_BASE64);
+    const stored = await uploadAttachment(
+      "photo.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
     const msg = await addMessage(
       conv.id,
       "user",
@@ -216,7 +232,7 @@ describe("handleListMessages attachments", () => {
     );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: {
         attachments?: AttachmentPayload[];
@@ -257,10 +273,14 @@ describe("handleListMessages attachments", () => {
       "assistant",
       JSON.stringify([{ type: "text", text: "" }]),
     );
-    const stored = uploadAttachment("output.png", "image/png", IMAGE_BASE64);
+    const stored = await uploadAttachment(
+      "output.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: {
         attachments?: AttachmentPayload[];
@@ -294,10 +314,14 @@ describe("handleListMessages attachments", () => {
       "assistant",
       JSON.stringify([{ type: "text", text: "" }]),
     );
-    const stored = uploadAttachment("output.png", "image/png", IMAGE_BASE64);
+    const stored = await uploadAttachment(
+      "output.png",
+      "image/png",
+      IMAGE_BASE64,
+    );
     linkAttachmentToMessage(msg.id, stored.id, 0);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: {
         attachments?: AttachmentPayload[];
@@ -335,7 +359,7 @@ describe("handleListMessages HEIC display normalization", () => {
     const heicB64 = fakeHeifHeaderBytes().toString("base64");
     insertLegacyAttachmentRow(msg.id, "IMG_1.HEIC", "image/heic", heicB64);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     const attachments = body.messages[0].attachments!;
@@ -364,7 +388,7 @@ describe("handleListMessages HEIC display normalization", () => {
       "document",
     );
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as { messages: MessagePayload[] };
 
     const attachments = body.messages[0].attachments!;
@@ -394,7 +418,7 @@ describe("handleListMessages HEIC display normalization", () => {
           heic!.toString("base64"),
         );
 
-        const response = handleListMessages(createTestArgs(conv.id));
+        const response = await handleListMessages(createTestArgs(conv.id));
         const body = response as { messages: MessagePayload[] };
 
         const attachments = body.messages[0].attachments!;
@@ -428,7 +452,7 @@ describe("handleListMessages HEIC display normalization", () => {
           "document",
         );
 
-        const response = handleListMessages(createTestArgs(conv.id));
+        const response = await handleListMessages(createTestArgs(conv.id));
         const body = response as { messages: MessagePayload[] };
 
         const attachments = body.messages[0].attachments!;
@@ -455,7 +479,7 @@ describe("handleListMessages no_response filtering", () => {
       JSON.stringify([{ type: "text", text: "<no_response/>" }]),
     );
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: { textSegments?: string[] }[];
     };
@@ -476,7 +500,7 @@ describe("handleListMessages no_response filtering", () => {
       ]),
     );
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: { textSegments?: string[] }[];
     };
@@ -504,7 +528,7 @@ describe("handleListMessages no_response filtering", () => {
       ]),
     );
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: {
         textSegments: string[];
@@ -527,7 +551,7 @@ describe("handleListMessages no_response filtering", () => {
       JSON.stringify([{ type: "text", text: "What does <no_response/> do?" }]),
     );
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as {
       messages: { textSegments?: string[] }[];
     };
@@ -588,7 +612,7 @@ describe("handleListMessages pagination", () => {
     const conv = createConversation();
     await insertMessages(conv.id, 5);
 
-    const response = handleListMessages(createTestArgs(conv.id));
+    const response = await handleListMessages(createTestArgs(conv.id));
     const body = response as unknown as PaginatedResponse;
 
     expect(body.messages).toHaveLength(5);
@@ -602,7 +626,7 @@ describe("handleListMessages pagination", () => {
     await insertMessages(conv.id, 5);
 
     const args = createPaginatedArgs(conv.id, { limit: "3" });
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     // Option A: without beforeTimestamp, all messages are returned regardless of limit
@@ -619,7 +643,7 @@ describe("handleListMessages pagination", () => {
       beforeTimestamp: String(msgs[7].createdAt),
       limit: "3",
     });
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     expect(body.messages).toHaveLength(3);
@@ -640,7 +664,7 @@ describe("handleListMessages pagination", () => {
       beforeTimestamp: String(msgs[1].createdAt),
       limit: "10",
     });
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     const ids = body.messages.map((m) => m.id);
@@ -658,7 +682,7 @@ describe("handleListMessages pagination", () => {
       beforeTimestamp: String(msgs[4].createdAt + 1),
       limit: "10",
     });
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     expect(body.messages).toHaveLength(5);
@@ -674,7 +698,7 @@ describe("handleListMessages pagination", () => {
       beforeTimestamp: String(msgs[4].createdAt + 1),
       limit: "3",
     });
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     expect(body.messages).toHaveLength(3);
@@ -685,7 +709,7 @@ describe("handleListMessages pagination", () => {
 
   test("empty / nonexistent conversation → empty messages, no pagination metadata", async () => {
     const args = createPaginatedArgs("nonexistent-conv-id");
-    const response = handleListMessages(args);
+    const response = await handleListMessages(args);
     const body = response as unknown as PaginatedResponse;
 
     expect(body.messages).toEqual([]);
@@ -698,7 +722,7 @@ describe("handleListMessages pagination", () => {
     const conv = createConversation();
     const args = createPaginatedArgs(conv.id, { limit: "abc" });
 
-    expect(() => handleListMessages(args)).toThrow(
+    await expect(handleListMessages(args)).rejects.toThrow(
       "limit must be a valid number",
     );
   });
@@ -707,7 +731,7 @@ describe("handleListMessages pagination", () => {
     const conv = createConversation();
     const args = createPaginatedArgs(conv.id, { beforeTimestamp: "abc" });
 
-    expect(() => handleListMessages(args)).toThrow(
+    await expect(handleListMessages(args)).rejects.toThrow(
       "beforeTimestamp must be a valid number",
     );
   });

package/src/__tests__/list-messages-background-tool-completion.test.ts

@@ -85,9 +85,9 @@ describe("handleListMessages background-tool completion projection", () => {
       },
     );
 
-    const response = handleListMessages({
+    const response = (await handleListMessages({
       queryParams: { conversationId: conv.id },
-    }) as { messages: ProjectedMessage[] };
+    })) as { messages: ProjectedMessage[] };
 
     // Every projected message validates against the wire schema.
     for (const message of response.messages) {
@@ -115,9 +115,9 @@ describe("handleListMessages background-tool completion projection", () => {
       JSON.stringify([{ type: "text", text: "hi there" }]),
     );
 
-    const response = handleListMessages({
+    const response = (await handleListMessages({
       queryParams: { conversationId: conv.id },
-    }) as { messages: ProjectedMessage[] };
+    })) as { messages: ProjectedMessage[] };
 
     expect(response.messages).toHaveLength(2);
     for (const message of response.messages) {

package/src/__tests__/list-messages-client-message-id.test.ts

@@ -48,7 +48,7 @@ describe("handleListMessages clientMessageId", () => {
     );
 
     // WHEN the messages snapshot is built
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };
@@ -70,7 +70,7 @@ describe("handleListMessages clientMessageId", () => {
     );
 
     // WHEN the messages snapshot is built
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };

package/src/__tests__/list-messages-hidden-metadata.test.ts

@@ -70,7 +70,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
       JSON.stringify([{ type: "text", text: "second visible" }]),
     );
 
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };
@@ -109,7 +109,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
     );
 
     // WHEN the UI history list is serialized
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };
@@ -140,7 +140,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
       { metadata: { hidden: false } },
     );
 
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };
@@ -177,9 +177,9 @@ describe("handleListMessages metadata.hidden filtering", () => {
       );
     }
 
-    const latest = handleListMessages({
+    const latest = (await handleListMessages({
       queryParams: { conversationId: conv.id, page: "latest", limit: "2" },
-    }) as {
+    })) as {
       messages: MessagePayload[];
       hasMore: boolean;
       oldestTimestamp: number | null;
@@ -196,13 +196,13 @@ describe("handleListMessages metadata.hidden filtering", () => {
 
     // Older page request — anchored before the latest page's oldest row —
     // should skip the hidden block entirely and return the next 2 visible rows.
-    const older = handleListMessages({
+    const older = (await handleListMessages({
       queryParams: {
         conversationId: conv.id,
         beforeTimestamp: String(latest.oldestTimestamp),
         limit: "2",
       },
-    }) as {
+    })) as {
       messages: MessagePayload[];
       hasMore: boolean;
     };
@@ -214,7 +214,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
     expect(older.hasMore).toBe(true);
   });
 
-  test("pagination keeps hasMore + a cursor when a >cap hidden block truncates the scan", () => {
+  test("pagination keeps hasMore + a cursor when a >cap hidden block truncates the scan", async () => {
     // Exercise cap-truncation with a small cap + a few hundred rows rather than
     // >10k: the large seed made the post-test resetTables DELETE slow enough to
     // time out the next test's beforeEach under parallel CI load.
@@ -258,9 +258,9 @@ describe("handleListMessages metadata.hidden filtering", () => {
         }
       });
 
-      const latest = handleListMessages({
+      const latest = (await handleListMessages({
         queryParams: { conversationId: conv.id, page: "latest", limit: "2" },
-      }) as {
+      })) as {
         messages: MessagePayload[];
         hasMore: boolean;
         oldestTimestamp: number | null;
@@ -274,13 +274,13 @@ describe("handleListMessages metadata.hidden filtering", () => {
 
       // Resuming from the surfaced cursor drains the remaining hidden rows and
       // surfaces the visible ones below the cap.
-      const older = handleListMessages({
+      const older = (await handleListMessages({
         queryParams: {
           conversationId: conv.id,
           beforeTimestamp: String(latest.oldestTimestamp),
           limit: "2",
         },
-      }) as { messages: MessagePayload[]; hasMore: boolean };
+      })) as { messages: MessagePayload[]; hasMore: boolean };
 
       expect(older.messages.map(plainText)).toEqual(["visible 0", "visible 1"]);
       expect(older.hasMore).toBe(false);
@@ -311,9 +311,9 @@ describe("handleListMessages metadata.hidden filtering", () => {
       );
     }
 
-    const latest = handleListMessages({
+    const latest = (await handleListMessages({
       queryParams: { conversationId: conv.id, page: "latest", limit: "2" },
-    }) as {
+    })) as {
       messages: MessagePayload[];
       hasMore: boolean;
       oldestTimestamp: number | null;
@@ -357,7 +357,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
       JSON.stringify([{ type: "text", text: "retrospective review" }]),
     );
 
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };
@@ -405,7 +405,7 @@ describe("handleListMessages metadata.hidden filtering", () => {
       JSON.stringify([{ type: "text", text: "retrospective review" }]),
     );
 
-    const response = handleListMessages({
+    const response = await handleListMessages({
       queryParams: { conversationId: conv.id },
     });
     const body = response as { messages: MessagePayload[] };