assistant-stream 0.3.20 → 0.3.22

package/dist/utils/json/is-json.js.map

@@ -1 +1 @@
-{"version":3,"file":"is-json.js","names":[],"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import type {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(value: unknown): value is ReadonlyJSONValue {\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"number\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  if (Array.isArray(value)) {\n    return value.every(isJSONValue);\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    !Array.isArray(value) &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";AAMA,SAAgB,YAAY,OAA4C;CACtE,IACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,WAEjB,OAAO;CAGT,IAAI,MAAM,QAAQ,KAAK,GACrB,OAAO,MAAM,MAAM,WAAW;CAGhC,IAAI,OAAO,UAAU,UACnB,OAAO,OAAO,QAAQ,KAAK,EAAE,OAC1B,CAAC,KAAK,SAAS,OAAO,QAAQ,YAAY,YAAY,GAAG,CAC5D;CAGF,OAAO;AACT;AAEA,SAAgB,YAAY,OAA4C;CACtE,OAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEA,SAAgB,aAAa,OAA6C;CACxE,OACE,SAAS,QACT,OAAO,UAAU,YACjB,CAAC,MAAM,QAAQ,KAAK,KACpB,OAAO,QAAQ,KAAK,EAAE,OACnB,CAAC,KAAK,SAAS,OAAO,QAAQ,YAAY,YAAY,GAAG,CAC5D;AAEJ"}
+{"version":3,"file":"is-json.js","names":[],"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import type {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(value: unknown): value is ReadonlyJSONValue {\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"number\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  if (Array.isArray(value)) {\n    return value.every(isJSONValue);\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    !Array.isArray(value) &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";AAMA,SAAgB,YAAY,OAA4C;CACtE,IACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,WAEjB,OAAO;CAGT,IAAI,MAAM,QAAQ,KAAK,GACrB,OAAO,MAAM,MAAM,WAAW;CAGhC,IAAI,OAAO,UAAU,UACnB,OAAO,OAAO,QAAQ,KAAK,CAAC,CAAC,OAC1B,CAAC,KAAK,SAAS,OAAO,QAAQ,YAAY,YAAY,GAAG,CAC5D;CAGF,OAAO;AACT;AAEA,SAAgB,YAAY,OAA4C;CACtE,OAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEA,SAAgB,aAAa,OAA6C;CACxE,OACE,SAAS,QACT,OAAO,UAAU,YACjB,CAAC,MAAM,QAAQ,KAAK,KACpB,OAAO,QAAQ,KAAK,CAAC,CAAC,OACnB,CAAC,KAAK,SAAS,OAAO,QAAQ,YAAY,YAAY,GAAG,CAC5D;AAEJ"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-stream",
-  "version": "0.3.20",
+  "version": "0.3.22",
   "description": "Streaming utilities for AI assistants",
   "keywords": [
     "ai",
@@ -64,10 +64,10 @@
   },
   "devDependencies": {
     "@types/json-schema": "^7.0.15",
-    "ioredis": "^5.11.0",
+    "ioredis": "^5.11.1",
     "redis": "^6.0.0",
-    "vitest": "^4.1.7",
-    "@assistant-ui/x-buildutils": "0.0.11"
+    "vitest": "^4.1.8",
+    "@assistant-ui/x-buildutils": "0.0.13"
   },
   "publishConfig": {
     "access": "public",

package/src/core/accumulators/assistant-message-accumulator.test.ts

@@ -101,6 +101,57 @@ describe("AssistantMessageAccumulator timing", () => {
     expect(last.metadata.timing!.toolCallCount).toBe(1);
   });
 
+  it("should record per-tool-call timing on the part", async () => {
+    const before = Date.now();
+    const chunks: AssistantStreamChunk[] = [
+      {
+        type: "part-start",
+        path: [0],
+        part: {
+          type: "tool-call",
+          toolCallId: "tc-1",
+          toolName: "search",
+        },
+      },
+      { type: "text-delta", path: [0], textDelta: '{"q":"test"}' },
+      { type: "tool-call-args-text-finish", path: [0] },
+      {
+        type: "result",
+        path: [0],
+        result: "found",
+        isError: false,
+      },
+      { type: "part-finish", path: [0] },
+      {
+        type: "message-finish",
+        path: [],
+        finishReason: "stop",
+        usage: { inputTokens: 0, outputTokens: 0 },
+      },
+    ];
+
+    const messages = await collectStream(chunks);
+    const after = Date.now();
+
+    const runningPart = messages
+      .find((m) =>
+        m.parts.some((p) => p.type === "tool-call" && p.state !== "result"),
+      )!
+      .parts.find((p) => p.type === "tool-call")!;
+    expect(runningPart.timing).toBeDefined();
+    expect(runningPart.timing!.startedAt).toBeGreaterThanOrEqual(before);
+    expect(runningPart.timing!.completedAt).toBeUndefined();
+
+    const settledPart = messages
+      .at(-1)!
+      .parts.find((p) => p.type === "tool-call")!;
+    expect(settledPart.timing).toBeDefined();
+    expect(settledPart.timing!.completedAt).toBeGreaterThanOrEqual(
+      settledPart.timing!.startedAt,
+    );
+    expect(settledPart.timing!.completedAt).toBeLessThanOrEqual(after);
+  });
+
   it("should include timing on flush when stream closes without message-finish", async () => {
     const chunks: AssistantStreamChunk[] = [
       { type: "part-start", path: [0], part: { type: "text" } },

package/src/core/accumulators/assistant-message-accumulator.ts

@@ -92,6 +92,7 @@ const handlePartStart = (
       toolName: partInit.toolName,
       argsText: "",
       args: {},
+      timing: { startedAt: Date.now() },
       ...(partInit.parentId && { parentId: partInit.parentId }),
     };
     return {
@@ -213,6 +214,14 @@ const handleResult = (
       return {
         ...part,
         state: "result",
+        ...(part.timing !== undefined
+          ? {
+              timing: {
+                ...part.timing,
+                completedAt: part.timing.completedAt ?? Date.now(),
+              },
+            }
+          : {}),
         ...(chunk.artifact !== undefined ? { artifact: chunk.artifact } : {}),
         result: chunk.result,
         isError: chunk.isError ?? false,

package/src/core/utils/types.ts

@@ -53,6 +53,19 @@ type ToolCallStatus =
       reason: "cancelled" | "length" | "content-filter" | "other";
     };
 
+/**
+ * Wall-clock timing of a tool call. Accumulator-populated timings are
+ * measured by the consuming accumulator, so resumed or replayed streams
+ * re-measure them; hosts that need authoritative timings supply the field
+ * themselves.
+ */
+export type ToolCallTiming = {
+  /** Epoch milliseconds when the tool call started streaming or executing. */
+  readonly startedAt: number;
+  /** Epoch milliseconds when the result landed. Absent while the call runs. */
+  readonly completedAt?: number;
+};
+
 type ToolCallPartBase = {
   type: "tool-call";
   status: ToolCallStatus;
@@ -60,6 +73,7 @@ type ToolCallPartBase = {
   toolName: string;
   argsText: string;
   args: ReadonlyJSONObject;
+  timing?: ToolCallTiming;
   artifact?: ReadonlyJSONValue;
   result?: ReadonlyJSONValue;
   modelContent?: readonly ToolModelContentPart[];

package/src/index.ts

@@ -33,6 +33,7 @@ export type {
   AssistantMessage,
   AssistantMessageTiming,
   DataPart,
+  ToolCallTiming,
 } from "./core/utils/types";
 
 export type {

package/src/resumable/__tests__/integration.test.ts

@@ -145,7 +145,17 @@ describe("resumable integration", () => {
       new AssistantTransportDecoder(),
     );
 
-    expect(replayMessage.parts).toEqual(producerMessage.parts);
+    // Tool-call timing is measured by the consuming accumulator, so a replay
+    // re-measures it; compare parts modulo timing.
+    const withoutTiming = (parts: typeof replayMessage.parts) =>
+      parts.map((part) =>
+        part.type === "tool-call"
+          ? (({ timing: _timing, ...rest }) => rest)(part)
+          : part,
+      );
+    expect(withoutTiming(replayMessage.parts)).toEqual(
+      withoutTiming(producerMessage.parts),
+    );
     expect(replayMessage.status).toEqual(producerMessage.status);
     const toolPart = replayMessage.parts.find((p) => p.type === "tool-call");
     expect(toolPart).toBeDefined();
@@ -154,6 +164,7 @@ describe("resumable integration", () => {
       toolCallId: "tool-1",
       args: { query: "weather" },
       result: { temperature: 72 },
+      timing: { startedAt: expect.any(Number) },
     });
   });
 });