@convex-dev/agent 0.2.6-alpha.0 → 0.2.6

package/src/client/messages.ts

@@ -1,6 +1,6 @@
 import type { ModelMessage } from "ai";
 import type { PaginationOptions, PaginationResult } from "convex/server";
-import type { MessageDoc } from "../component/schema.js";
+import type { MessageDoc } from "../validators.js";
 import { validateVectorDimension } from "../component/vector/tables.js";
 import {
   vMessageWithMetadata,

package/src/client/mock.json

@@ -0,0 +1,68 @@
+[
+  { "type": "start" },
+  { "type": "start-step" },
+  { "type": "reasoning-start", "id": "0-reasoning" },
+  {
+    "type": "reasoning-delta",
+    "id": "0-reasoning",
+    "delta": "I'm",
+    "providerMetadata": {
+      "mockProvider": { "mock": { "reasoningDetails": null } }
+    }
+  },
+  {
+    "type": "reasoning-delta",
+    "id": "0-reasoning",
+    "delta": " thinking",
+    "providerMetadata": {
+      "mockProvider": { "mock": { "reasoningDetails": null } }
+    }
+  },
+  { "type": "reasoning-end", "id": "0-reasoning" },
+  {
+    "type": "tool-input-available",
+    "toolCallId": "1",
+    "toolName": "say",
+    "input": { "question": "What is the best flavor of ice cream?" }
+  },
+  {
+    "type": "tool-output-available",
+    "toolCallId": "1",
+    "output": "I'm sorry I can't help you. Stop asking me questions."
+  },
+  {
+    "type": "tool-output-available",
+    "toolCallId": "1",
+    "output": "I'm sorry I can't help you. Stop asking me questions."
+  },
+  { "type": "reasoning-start", "id": "3-reasoning" },
+  {
+    "type": "reasoning-delta",
+    "id": "3-reasoning",
+    "delta": "I'm",
+    "providerMetadata": {
+      "mockProvider": { "mock": { "reasoningDetails": null } }
+    }
+  },
+  {
+    "type": "reasoning-delta",
+    "id": "3-reasoning",
+    "delta": " thinking",
+    "providerMetadata": {
+      "mockProvider": { "mock": { "reasoningDetails": null } }
+    }
+  },
+  { "type": "reasoning-end", "id": "3-reasoning" },
+  { "type": "text-start", "id": "4-text" },
+  { "type": "text-delta", "id": "4-text", "delta": "The" },
+  { "type": "text-delta", "id": "4-text", "delta": " best" },
+  { "type": "text-delta", "id": "4-text", "delta": " flavor" },
+  { "type": "text-delta", "id": "4-text", "delta": " of" },
+  { "type": "text-delta", "id": "4-text", "delta": " ice" },
+  { "type": "text-delta", "id": "4-text", "delta": " cream" },
+  { "type": "text-delta", "id": "4-text", "delta": " is" },
+  { "type": "text-delta", "id": "4-text", "delta": " vanilla." },
+  { "type": "text-end", "id": "4-text" },
+  { "type": "finish-step" },
+  { "type": "finish" }
+]

package/src/client/mockModel.ts

@@ -1,17 +1,17 @@
 import type {
   LanguageModelV2,
+  LanguageModelV2Content,
   LanguageModelV2StreamPart,
 } from "@ai-sdk/provider";
-import type { ReasoningPart, TextPart } from "@ai-sdk/provider-utils";
 import { simulateReadableStream } from "ai";
 
-const longDefaultText = `
+export const DEFAULT_TEXT = `
 A A A A A A A A A A A A A A A
 B B B B B B B B B B B B B B B
 C C C C C C C C C C C C C C C
 D D D D D D D D D D D D D D D
 `;
-const defaultUsage = { outputTokens: 10, inputTokens: 3, totalTokens: 13 };
+const DEFAULT_USAGE = { outputTokens: 10, inputTokens: 3, totalTokens: 13 };
 
 export type MockModelArgs = {
   provider?: LanguageModelV2["provider"];
@@ -22,7 +22,7 @@ export type MockModelArgs = {
   chunkDelayInMs?: number;
   initialDelayInMs?: number;
   // provide either content or doGenerate & doStream
-  content?: (TextPart | ReasoningPart)[];
+  content?: LanguageModelV2Content[];
   doGenerate?: LanguageModelV2["doGenerate"];
   doStream?: LanguageModelV2["doStream"];
   fail?:
@@ -55,9 +55,9 @@ export class MockLanguageModel implements LanguageModelV2 {
     this.provider = args.provider || "mock-provider";
     this.modelId = args.modelId || "mock-model-id";
     const {
-      content = [{ type: "text", text: longDefaultText }],
-      chunkDelayInMs = 200,
-      initialDelayInMs = 1000,
+      content = [{ type: "text", text: DEFAULT_TEXT }],
+      chunkDelayInMs = 0,
+      initialDelayInMs = 0,
       supportedUrls = {},
     } = args;
     const fail =
@@ -73,21 +73,24 @@ export class MockLanguageModel implements LanguageModelV2 {
       { type: "stream-start", warnings: [] },
     ];
     chunks.push(
-      ...content.flatMap((c, ci) => {
+      ...content.flatMap((c, ci): LanguageModelV2StreamPart[] => {
+        if (c.type !== "text" && c.type !== "reasoning") {
+          return [c];
+        }
         const deltas = c.text.split(" ");
-        let parts: LanguageModelV2StreamPart[] = [];
+        const parts: LanguageModelV2StreamPart[] = [];
         if (c.type === "reasoning") {
           parts.push({
             type: "reasoning-start",
-            id: `${ci}-reasoning-start`,
+            id: `reasoning-${ci}`,
           });
           parts.push(
             ...deltas.map(
               (delta, di) =>
                 ({
                   type: "reasoning-delta",
-                  delta,
-                  id: `${ci}-reasoning-${di}`,
+                  delta: (di ? " " : "") + delta,
+                  id: `reasoning-${ci}`,
                   providerMetadata: {
                     mockProvider: { mock: { reasoningDetails: null } },
                   },
@@ -96,21 +99,26 @@ export class MockLanguageModel implements LanguageModelV2 {
           );
           parts.push({
             type: "reasoning-end",
-            id: `${ci}-reasoning-end`,
+            id: `reasoning-${ci}`,
           });
         } else if (c.type === "text") {
           parts.push({
             type: "text-start",
-            id: `${ci}-text-start`,
+            id: `txt-${ci}`,
           });
-          parts = deltas.map((delta, di) => ({
-            type: "text-delta",
-            delta,
-            id: `${ci}-text-${di}`,
-          }));
+          parts.push(
+            ...deltas.map(
+              (delta, di) =>
+                ({
+                  type: "text-delta",
+                  delta: (di ? " " : "") + delta,
+                  id: `txt-${ci}`,
+                }) satisfies LanguageModelV2StreamPart,
+            ),
+          );
           parts.push({
             type: "text-end",
-            id: `${ci}-text-end`,
+            id: `txt-${ci}`,
           });
         }
         return parts;
@@ -125,7 +133,7 @@ export class MockLanguageModel implements LanguageModelV2 {
     chunks.push({
       type: "finish",
       finishReason: fail ? "error" : "stop",
-      usage: defaultUsage,
+      usage: DEFAULT_USAGE,
       providerMetadata: {
         mockProvider: { mock: "mock metadata" },
       },
@@ -144,7 +152,7 @@ export class MockLanguageModel implements LanguageModelV2 {
         return {
           content,
           finishReason: "stop",
-          usage: defaultUsage,
+          usage: DEFAULT_USAGE,
           providerMetadata: { mockProvider: { mock: "mock metadata" } },
           warnings: [],
         };
@@ -169,8 +177,11 @@ export class MockLanguageModel implements LanguageModelV2 {
           initialDelayInMs,
           chunkDelayInMs,
         });
+
         if (options.abortSignal) {
-          throw new Error("abortSignal in mock model");
+          options.abortSignal.addEventListener("abort", () => {
+            console.warn("abortSignal in mock model not supported");
+          });
         }
         return {
           stream,