convex-durable-agents 0.2.5 → 0.2.6

package/src/component/messages.test.ts

@@ -0,0 +1,40 @@
+import { describe, expect, it } from "vitest";
+import { isToolInputAvailablePart } from "./messages";
+
+describe("isToolInputAvailablePart", () => {
+  it("returns true for tool input available objects", () => {
+    expect(
+      isToolInputAvailablePart({
+        toolCallId: "tool-call-1",
+        state: "input-available",
+      }),
+    ).toBe(true);
+  });
+
+  it("returns false for non-object values", () => {
+    expect(isToolInputAvailablePart(null)).toBe(false);
+    expect(isToolInputAvailablePart("tool-call-1")).toBe(false);
+    expect(isToolInputAvailablePart(123)).toBe(false);
+    expect(isToolInputAvailablePart(true)).toBe(false);
+  });
+
+  it("returns false when required fields are missing or invalid", () => {
+    expect(
+      isToolInputAvailablePart({
+        state: "input-available",
+      }),
+    ).toBe(false);
+    expect(
+      isToolInputAvailablePart({
+        toolCallId: 123,
+        state: "input-available",
+      }),
+    ).toBe(false);
+    expect(
+      isToolInputAvailablePart({
+        toolCallId: "tool-call-1",
+        state: "output-available",
+      }),
+    ).toBe(false);
+  });
+});

package/src/component/streams.test.ts

@@ -0,0 +1,118 @@
+import { describe, expect, it } from "vitest";
+import type { UIMessageChunk } from "ai";
+import { STREAM_HEARTBEAT_INTERVAL_MS, STREAM_LIVENESS_THRESHOLD_MS } from "../utils/streaming";
+import { isAlive, replacePartIds } from "./streams";
+
+describe("isAlive", () => {
+  it("uses a liveness threshold greater than the heartbeat interval", () => {
+    expect(STREAM_LIVENESS_THRESHOLD_MS).toBeGreaterThan(STREAM_HEARTBEAT_INTERVAL_MS);
+  });
+
+  it("returns true for a recent streaming heartbeat", () => {
+    const stream = {
+      state: {
+        kind: "streaming",
+        lockId: "lock-id",
+        lastHeartbeat: Date.now(),
+      },
+    } as Exclude<Parameters<typeof isAlive>[0], null>;
+    expect(isAlive(stream)).toBe(true);
+  });
+
+  it("returns false for an old streaming heartbeat", () => {
+    const stream = {
+      state: {
+        kind: "streaming",
+        lockId: "lock-id",
+        lastHeartbeat: Date.now() - STREAM_LIVENESS_THRESHOLD_MS - 1,
+      },
+    } as Exclude<Parameters<typeof isAlive>[0], null>;
+    expect(isAlive(stream)).toBe(false);
+  });
+});
+
+describe("replacePartIds", () => {
+  it("returns empty array for empty input", () => {
+    expect(replacePartIds([], new Map(), new Set())).toEqual([]);
+  });
+
+  it("assigns sequential ids starting from 0", () => {
+    const parts: UIMessageChunk[] = [
+      { type: "text-delta", id: "orig-a", delta: "Hello" },
+      { type: "text-delta", id: "orig-b", delta: " world" },
+    ];
+    const result = replacePartIds(parts, new Map(), new Set());
+    expect(result).toEqual([
+      { type: "text-delta", id: "0", delta: "Hello" },
+      { type: "text-delta", id: "1", delta: " world" },
+    ]);
+  });
+
+  it("reuses mapped id for parts with the same original id", () => {
+    const parts: UIMessageChunk[] = [
+      { type: "text-delta", id: "orig-a", delta: "Hello" },
+      { type: "text-delta", id: "orig-a", delta: " again" },
+    ];
+    const result = replacePartIds(parts, new Map(), new Set());
+    expect(result).toEqual([
+      { type: "text-delta", id: "0", delta: "Hello" },
+      { type: "text-delta", id: "0", delta: " again" },
+    ]);
+  });
+
+  it("uses pre-existing newIds mapping", () => {
+    const newIds = new Map([["orig-a", "42"]]);
+    const parts: UIMessageChunk[] = [{ type: "text-delta", id: "orig-a", delta: "Hello" }];
+    const result = replacePartIds(parts, newIds, new Set());
+    expect(result).toEqual([{ type: "text-delta", id: "42", delta: "Hello" }]);
+  });
+
+  it("skips ids that are already in prevIds", () => {
+    const prevIds = new Set(["0", "1"]);
+    const parts: UIMessageChunk[] = [{ type: "text-delta", id: "orig-a", delta: "Hello" }];
+    const result = replacePartIds(parts, new Map(), prevIds);
+    expect(result).toEqual([{ type: "text-delta", id: "2", delta: "Hello" }]);
+  });
+
+  it("adds generated ids to prevIds", () => {
+    const prevIds = new Set<string>();
+    const parts: UIMessageChunk[] = [{ type: "text-delta", id: "orig-a", delta: "Hello" }];
+    replacePartIds(parts, new Map(), prevIds);
+    expect(prevIds.has("0")).toBe(true);
+  });
+
+  it("records new mappings in newIds", () => {
+    const newIds = new Map<string, string>();
+    const parts: UIMessageChunk[] = [{ type: "text-delta", id: "orig-a", delta: "Hello" }];
+    replacePartIds(parts, newIds, new Set());
+    expect(newIds.get("orig-a")).toBe("0");
+  });
+
+  it("passes through parts without an id field unchanged", () => {
+    const parts: UIMessageChunk[] = [{ type: "tool-input-delta", toolCallId: "tc1", inputTextDelta: "some input" }];
+    const result = replacePartIds(parts, new Map(), new Set());
+    expect(result).toEqual([{ type: "tool-input-delta", toolCallId: "tc1", inputTextDelta: "some input" }]);
+  });
+
+  it("passes through parts with a falsy id unchanged", () => {
+    const parts: UIMessageChunk[] = [{ type: "text-delta", id: "", delta: "Hello" }];
+    const result = replacePartIds(parts, new Map(), new Set());
+    expect(result).toEqual([{ type: "text-delta", id: "", delta: "Hello" }]);
+  });
+
+  it("handles a mix of parts with and without ids", () => {
+    const parts: UIMessageChunk[] = [
+      { type: "text-delta", id: "a", delta: "Hello" },
+      { type: "tool-input-delta", toolCallId: "tc1", inputTextDelta: "input" },
+      { type: "reasoning-delta", id: "b", delta: "thinking" },
+      { type: "text-delta", id: "a", delta: " world" },
+    ];
+    const result = replacePartIds(parts, new Map(), new Set());
+    expect(result).toEqual([
+      { type: "text-delta", id: "0", delta: "Hello" },
+      { type: "tool-input-delta", toolCallId: "tc1", inputTextDelta: "input" },
+      { type: "reasoning-delta", id: "1", delta: "thinking" },
+      { type: "text-delta", id: "0", delta: " world" },
+    ]);
+  });
+});

package/src/component/threads.test.ts

@@ -0,0 +1,48 @@
+import { beforeEach, describe, expect, it } from "vitest";
+import { resetFinalizerMismatchAlertState, trackFinalizerMismatchRate } from "./threads";
+
+describe("trackFinalizerMismatchRate", () => {
+  beforeEach(() => {
+    resetFinalizerMismatchAlertState();
+  });
+
+  it("alerts when mismatch threshold is reached within the window", () => {
+    const now = 1_000_000;
+    expect(trackFinalizerMismatchRate("thread-a", now)).toEqual({
+      windowStartedAt: now,
+      count: 1,
+      shouldAlert: false,
+    });
+    expect(trackFinalizerMismatchRate("thread-a", now + 1_000)).toEqual({
+      windowStartedAt: now,
+      count: 2,
+      shouldAlert: false,
+    });
+    expect(trackFinalizerMismatchRate("thread-a", now + 2_000)).toEqual({
+      windowStartedAt: now,
+      count: 3,
+      shouldAlert: true,
+    });
+    expect(trackFinalizerMismatchRate("thread-a", now + 3_000)).toEqual({
+      windowStartedAt: now,
+      count: 4,
+      shouldAlert: false,
+    });
+  });
+
+  it("resets the counter outside the alert window and isolates by thread", () => {
+    const now = 2_000_000;
+    trackFinalizerMismatchRate("thread-a", now);
+    trackFinalizerMismatchRate("thread-a", now + 1_000);
+    expect(trackFinalizerMismatchRate("thread-b", now + 1_500)).toEqual({
+      windowStartedAt: now + 1_500,
+      count: 1,
+      shouldAlert: false,
+    });
+    expect(trackFinalizerMismatchRate("thread-a", now + 6 * 60 * 1_000)).toEqual({
+      windowStartedAt: now + 6 * 60 * 1_000,
+      count: 1,
+      shouldAlert: false,
+    });
+  });
+});

package/src/react/__fixtures__/01-early-streaming-start.json

@@ -0,0 +1,35 @@
+{
+  "description": "01-early-streaming-start",
+  "logLine": 31988,
+  "messages": [
+    {
+      "id": "LT8PGDxTlpf0lVZs",
+      "role": "user",
+      "parts": [
+        {
+          "text": "get me a weather forecast for this weekend",
+          "type": "text"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-LT8PGDxTlpf0lVZs",
+        "status": "success",
+        "_creationTime": 1770783020905.086
+      }
+    }
+  ],
+  "streamingUpdates": {
+    "messages": [
+      {
+        "msgId": "vsOhda95lumMg4QZ",
+        "parts": [
+          {
+            "messageId": "vsOhda95lumMg4QZ",
+            "seq": 1,
+            "type": "start"
+          }
+        ]
+      }
+    ]
+  }
+}

package/src/react/__fixtures__/02-reasoning-complete-tool-call.json

@@ -0,0 +1,85 @@
+{
+  "description": "02-reasoning-complete-tool-call",
+  "logLine": 31998,
+  "messages": [
+    {
+      "id": "LT8PGDxTlpf0lVZs",
+      "role": "user",
+      "parts": [
+        {
+          "text": "get me a weather forecast for this weekend",
+          "type": "text"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-LT8PGDxTlpf0lVZs",
+        "status": "success",
+        "_creationTime": 1770783020905.086
+      }
+    }
+  ],
+  "streamingUpdates": {
+    "messages": [
+      {
+        "msgId": "vsOhda95lumMg4QZ",
+        "parts": [
+          {
+            "messageId": "vsOhda95lumMg4QZ",
+            "seq": 1,
+            "type": "start"
+          },
+          {
+            "seq": 1,
+            "type": "start-step"
+          },
+          {
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-start"
+          },
+          {
+            "delta": "The user wants a weather forecast for this weekend. I should use the Weather capability which provides \"",
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-delta"
+          },
+          {
+            "delta": "current conditions, hourly and multi-day forecasts for an",
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-delta"
+          },
+          {
+            "delta": "y location worldwide\". That should have what I need",
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-delta"
+          },
+          {
+            "delta": " for a weekend forecast.\n\nLet me read the Weather CAPABILITY.md to see how to query it",
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-delta"
+          },
+          {
+            "delta": ".",
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-delta"
+          },
+          {
+            "id": "0",
+            "seq": 1,
+            "type": "reasoning-end"
+          },
+          {
+            "seq": 1,
+            "toolCallId": "tooluse_4X05H1krEnm7rFppLQw1pi",
+            "toolName": "readFile",
+            "type": "tool-input-start"
+          }
+        ]
+      }
+    ]
+  }
+}

package/src/react/__fixtures__/03-new-round-seq2.json

@@ -0,0 +1,89 @@
+{
+  "description": "03-new-round-seq2",
+  "logLine": 32006,
+  "messages": [
+    {
+      "id": "LT8PGDxTlpf0lVZs",
+      "role": "user",
+      "parts": [
+        {
+          "text": "get me a weather forecast for this weekend",
+          "type": "text"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-LT8PGDxTlpf0lVZs",
+        "status": "success",
+        "_creationTime": 1770783020905.086
+      }
+    },
+    {
+      "id": "vsOhda95lumMg4QZ",
+      "role": "assistant",
+      "parts": [
+        {
+          "type": "step-start"
+        },
+        {
+          "providerMetadata": {
+            "bedrock": {
+              "signature": "AxB2kT9pLmNcDfEgHiJkLmNoPqRsTuVwXyZ0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEF"
+            }
+          },
+          "state": "done",
+          "text": "The user wants a weather forecast for this weekend. I should use the Weather capability which provides \"current conditions, hourly and multi-day forecasts for any location worldwide\". That should have what I need for a weekend forecast.\n\nLet me read the Weather CAPABILITY.md to see how to query it.",
+          "type": "reasoning"
+        },
+        {
+          "input": {
+            "path": "capabilities/weather/CAPABILITY.md"
+          },
+          "output": "---\nname: Weather\ndescription: Weather provides current conditions, hourly and multi-day forecasts for any location worldwide\n---\n\n# Weather Service\n\nGlobal weather data provider with current observations and forecasts.\n\n## Data Overview\n\n- Current conditions: temperature, humidity, wind, visibility\n- Hourly forecasts: up to 48 hours ahead\n- Daily forecasts: up to 10 days ahead\n- Historical data: past 30 days\n\n## Examples\n\nGet a multi-day forecast:\n\n```typescript\nconst forecast = await getWeather(\"weatherprod\", {\n  location: \"San Francisco, CA\",\n  type: \"daily\",\n  days: 7,\n});\nconsole.log(forecast);\n```\n\nGet current conditions:\n\n```typescript\nconst current = await getWeather(\"weatherprod\", {\n  location: \"New York, NY\",\n  type: \"current\",\n});\nconsole.log(current);\n```\n\n## Guidelines\n\n- Always specify a location string with city and state/country.\n- Use type \"daily\" for multi-day forecasts and \"current\" for present conditions.\n",
+          "state": "output-available",
+          "toolCallId": "tooluse_4X05H1krEnm7rFppLQw1pi",
+          "type": "tool-readFile"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-vsOhda95lumMg4QZ",
+        "status": "success",
+        "_creationTime": 1770783024158.487,
+        "committedSeq": 1
+      }
+    }
+  ],
+  "streamingUpdates": {
+    "messages": [
+      {
+        "msgId": "vsOhda95lumMg4QZ",
+        "parts": [
+          {
+            "messageId": "vsOhda95lumMg4QZ",
+            "seq": 2,
+            "type": "start"
+          },
+          {
+            "seq": 2,
+            "type": "start-step"
+          },
+          {
+            "id": "0",
+            "seq": 2,
+            "type": "text-start"
+          },
+          {
+            "delta": "Now let me fetch the weekend forecast for your area:",
+            "id": "0",
+            "seq": 2,
+            "type": "text-delta"
+          },
+          {
+            "id": "0",
+            "seq": 2,
+            "type": "text-end"
+          }
+        ]
+      }
+    ]
+  }
+}

package/src/react/__fixtures__/04-tool-call-error-seq3.json

@@ -0,0 +1,145 @@
+{
+  "description": "04-tool-call-error-seq3",
+  "logLine": 32020,
+  "messages": [
+    {
+      "id": "LT8PGDxTlpf0lVZs",
+      "role": "user",
+      "parts": [
+        {
+          "text": "get me a weather forecast for this weekend",
+          "type": "text"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-LT8PGDxTlpf0lVZs",
+        "status": "success",
+        "_creationTime": 1770783020905.086
+      }
+    },
+    {
+      "id": "vsOhda95lumMg4QZ",
+      "role": "assistant",
+      "parts": [
+        {
+          "type": "step-start"
+        },
+        {
+          "providerMetadata": {
+            "bedrock": {
+              "signature": "AxB2kT9pLmNcDfEgHiJkLmNoPqRsTuVwXyZ0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEFghijklmnopqrstuvwxyz0123456789abcdefABCDEF"
+            }
+          },
+          "state": "done",
+          "text": "The user wants a weather forecast for this weekend. I should use the Weather capability which provides \"current conditions, hourly and multi-day forecasts for any location worldwide\". That should have what I need for a weekend forecast.\n\nLet me read the Weather CAPABILITY.md to see how to query it.",
+          "type": "reasoning"
+        },
+        {
+          "input": {
+            "path": "capabilities/weather/CAPABILITY.md"
+          },
+          "output": "---\nname: Weather\ndescription: Weather provides current conditions, hourly and multi-day forecasts for any location worldwide\n---\n\n# Weather Service\n\nGlobal weather data provider with current observations and forecasts.\n\n## Data Overview\n\n- Current conditions: temperature, humidity, wind, visibility\n- Hourly forecasts: up to 48 hours ahead\n- Daily forecasts: up to 10 days ahead\n- Historical data: past 30 days\n\n## Examples\n\nGet a multi-day forecast:\n\n```typescript\nconst forecast = await getWeather(\"weatherprod\", {\n  location: \"San Francisco, CA\",\n  type: \"daily\",\n  days: 7,\n});\nconsole.log(forecast);\n```\n\nGet current conditions:\n\n```typescript\nconst current = await getWeather(\"weatherprod\", {\n  location: \"New York, NY\",\n  type: \"current\",\n});\nconsole.log(current);\n```\n\n## Guidelines\n\n- Always specify a location string with city and state/country.\n- Use type \"daily\" for multi-day forecasts and \"current\" for present conditions.\n",
+          "state": "output-available",
+          "toolCallId": "tooluse_4X05H1krEnm7rFppLQw1pi",
+          "type": "tool-readFile"
+        },
+        {
+          "type": "step-start"
+        },
+        {
+          "state": "done",
+          "text": "Now let me fetch the weekend forecast for your area:",
+          "type": "text"
+        },
+        {
+          "errorText": "Code execution failed:\nWeather API request failed: Bad Request\nWeatherApiError: Weather API request failed: Bad Request\n    at getWeather (/workspace/services/executor/.cache/revision-envs/k970rr9s6zxgkh8rg4fdbbhw3d80yzhs/bundle.mjs:2141:15)\n    at async <anonymous> (script.ts:2:35)\n    at async executeTypeScript (/workspace/services/executor/src/exec.ts:507:14)\n    at processTicksAndRejections (native:7:39)\n{\"messages\":[{\"type\":\"FATAL\",\"text\":\"Invalid date range specified.\"}]}\n{\"headers\":{\"cache-control\":\"no-store, no-cache, must-revalidate, max-age=0\",\"content-length\":\"52\",\"content-type\":\"application/json; charset=UTF-8\",\"date\":\"Wed, 11 Feb 2026 04:10:27 GMT\"},\"status\":400}",
+          "input": {
+            "code": "\nconst forecast = await getWeather(\"weatherprod\", {\n  location: \"San Francisco, CA\",\n  type: \"daily\",\n  startDate: \"2026-02-14\",\n  endDate: \"2026-02-15\",\n});\nconsole.log(\"Weekend Forecast\");\nconsole.log(\"================\");\nconsole.log(JSON.stringify(forecast, null, 2));\n",
+            "description": "Get weekend weather forecast"
+          },
+          "state": "output-error",
+          "toolCallId": "tooluse_g3zTfupLSe8DI2Yla3gzG1",
+          "type": "tool-runCode"
+        },
+        {
+          "type": "step-start"
+        },
+        {
+          "type": "text",
+          "text": "Let me fix the query parameters:",
+          "state": "done"
+        },
+        {
+          "type": "tool-runCode",
+          "toolCallId": "tooluse_vGo3riHmTITFJfPvsE2y4k",
+          "state": "input-streaming"
+        }
+      ],
+      "metadata": {
+        "key": "jn7a2st357bqyzrcetv78j0fmx80yd8n-vsOhda95lumMg4QZ",
+        "status": "success",
+        "_creationTime": 1770783024158.487,
+        "committedSeq": 2
+      }
+    }
+  ],
+  "streamingUpdates": {
+    "messages": [
+      {
+        "msgId": "vsOhda95lumMg4QZ",
+        "parts": [
+          {
+            "messageId": "vsOhda95lumMg4QZ",
+            "seq": 3,
+            "type": "start"
+          },
+          {
+            "seq": 3,
+            "type": "start-step"
+          },
+          {
+            "id": "0",
+            "seq": 3,
+            "type": "text-start"
+          },
+          {
+            "delta": "Let me fix the query parameters:",
+            "id": "0",
+            "seq": 3,
+            "type": "text-delta"
+          },
+          {
+            "id": "0",
+            "seq": 3,
+            "type": "text-end"
+          },
+          {
+            "seq": 3,
+            "toolCallId": "tooluse_vGo3riHmTITFJfPvsE2y4k",
+            "toolName": "runCode",
+            "type": "tool-input-start"
+          },
+          {
+            "input": {
+              "code": "\nconst forecast = await getWeather(\"weatherprod\", {\n  location: \"San Francisco, CA\",\n  type: \"daily\",\n  days: 3,\n});\nconsole.log(\"Weekend Weather Forecast\");\nconsole.log(\"========================\");\nforecast.days.forEach((day: any) => {\n  console.log(`${day.date}: High ${day.high}°F / Low ${day.low}°F | ${day.condition} | Wind ${day.wind_mph} mph`);\n});\n",
+              "description": "Get weekend weather forecast"
+            },
+            "seq": 3,
+            "toolCallId": "tooluse_vGo3riHmTITFJfPvsE2y4k",
+            "toolName": "runCode",
+            "type": "tool-input-available"
+          },
+          {
+            "seq": 3,
+            "type": "finish-step"
+          },
+          {
+            "finishReason": "tool-calls",
+            "seq": 3,
+            "type": "finish"
+          }
+        ]
+      }
+    ]
+  }
+}