assistant-stream 0.0.25 → 0.0.27

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { a as AssistantStreamChunk, R as ReadonlyJSONValue, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-CEVTPU3I.js';
-export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-CEVTPU3I.js';
+import { a as AssistantStreamChunk, R as ReadonlyJSONValue, c as AssistantMessage, d as AssistantStreamEncoder, A as AssistantStream } from './assistant-stream-DVLo0jNu.js';
+export { e as createAssistantStream, f as createAssistantStreamResponse } from './assistant-stream-DVLo0jNu.js';
 
 declare class PipeableTransformStream<I, O> extends TransformStream<I, O> {
     constructor(transform: (readable: ReadableStream<I>) => ReadableStream<O>);

package/dist/index.js

@@ -234,6 +234,13 @@ var ToolCallStreamControllerImpl = class {
     return this._argsTextController;
   }
   _argsTextController;
+  unstable_setArtifact(artifact) {
+    this._controller.enqueue({
+      type: "artifact",
+      path: [],
+      artifact
+    });
+  }
   setResult(result, isError) {
     this._controller.enqueue({
       type: "result",
@@ -455,7 +462,7 @@ var AssistantMetaTransformStream = class extends TransformStream {
           controller.enqueue(chunk);
           return;
         }
-        if (chunk.type === "text-delta" || chunk.type === "result" || chunk.type === "part-finish" || chunk.type === "tool-call-args-text-finish") {
+        if (chunk.type === "text-delta" || chunk.type === "result" || chunk.type === "artifact" || chunk.type === "part-finish" || chunk.type === "tool-call-args-text-finish") {
           if (chunk.path.length !== 1) {
             controller.error(
               new Error(`${chunk.type} chunks must have a path of length 1`)
@@ -530,7 +537,7 @@ var DataStreamEncoder = class extends PipeableTransformStream {
                 }
                 case "tool-call": {
                   controller.enqueue({
-                    type: "c" /* ToolCallDelta */,
+                    type: "c" /* ToolCallArgsTextDelta */,
                     value: {
                       toolCallId: part.toolCallId,
                       argsTextDelta: chunk.textDelta
@@ -561,6 +568,22 @@ var DataStreamEncoder = class extends PipeableTransformStream {
               });
               break;
             }
+            case "artifact": {
+              const part = chunk.meta;
+              if (part.type !== "tool-call") {
+                throw new Error(
+                  `Artifact chunk on non-tool-call part not supported: ${part.type}`
+                );
+              }
+              controller.enqueue({
+                type: "aui-tool-artifact" /* AuiToolCallArtifact */,
+                value: {
+                  toolCallId: part.toolCallId,
+                  artifact: chunk.artifact
+                }
+              });
+              break;
+            }
             case "step-start": {
               const { type: type2, ...value } = chunk;
               controller.enqueue({
@@ -661,7 +684,7 @@ var DataStreamDecoder = class extends PipeableTransformStream {
               activeToolCallArgsText = toolCallController.argsText;
               break;
             }
-            case "c" /* ToolCallDelta */: {
+            case "c" /* ToolCallArgsTextDelta */: {
               const { toolCallId, argsTextDelta } = value;
               const toolCallController = toolCallControllers.get(toolCallId);
               if (!toolCallController)
@@ -671,6 +694,16 @@ var DataStreamDecoder = class extends PipeableTransformStream {
               toolCallController.argsText.append(argsTextDelta);
               break;
             }
+            case "aui-tool-artifact" /* AuiToolCallArtifact */: {
+              const { toolCallId, artifact } = value;
+              const toolCallController = toolCallControllers.get(toolCallId);
+              if (!toolCallController)
+                throw new Error(
+                  "Encountered tool call with unknown id: " + toolCallId
+                );
+              toolCallController.unstable_setArtifact(artifact);
+              break;
+            }
             case "a" /* ToolCallResult */: {
               const { toolCallId, result } = value;
               const toolCallController = toolCallControllers.get(toolCallId);
@@ -790,6 +823,10 @@ var AssistantStreamControllerImpl = class {
     this._closeSubscriber = callback;
   }
   _addPart(part, stream) {
+    if (this._append) {
+      this._append.controller.close();
+      this._append = void 0;
+    }
     this.enqueue({
       type: "part-start",
       part,
@@ -806,9 +843,6 @@ var AssistantStreamControllerImpl = class {
   }
   appendText(textDelta) {
     if (this._append?.kind !== "text") {
-      if (this._append) {
-        this._append.controller.close();
-      }
       this._append = {
         kind: "text",
         controller: this.addTextPart()
@@ -818,9 +852,6 @@ var AssistantStreamControllerImpl = class {
   }
   appendReasoning(textDelta) {
     if (this._append?.kind !== "reasoning") {
-      if (this._append) {
-        this._append.controller.close();
-      }
       this._append = {
         kind: "reasoning",
         controller: this.addReasoningPart()
@@ -1508,54 +1539,66 @@ var handlePartStart = (message, chunk) => {
   }
 };
 var handleToolCallArgsTextFinish = (message, chunk) => {
-  return updatePartForPath(message, chunk, (lastPart) => {
-    if (lastPart.type !== "tool-call") {
-      throw new Error("Last part is not a tool call");
+  return updatePartForPath(message, chunk, (part) => {
+    if (part.type !== "tool-call") {
+      throw new Error("Last is not a tool call");
     }
     return {
-      ...lastPart,
+      ...part,
       state: "call"
     };
   });
 };
 var handlePartFinish = (message, chunk) => {
-  return updatePartForPath(message, chunk, (lastPart) => ({
-    ...lastPart,
+  return updatePartForPath(message, chunk, (part) => ({
+    ...part,
     status: { type: "complete", reason: "unknown" }
   }));
 };
 var handleTextDelta = (message, chunk) => {
-  return updatePartForPath(message, chunk, (lastPart) => {
-    if (lastPart.type === "text") {
-      return { ...lastPart, text: lastPart.text + chunk.textDelta };
-    } else if (lastPart.type === "tool-call") {
-      const newArgsText = lastPart.argsText + chunk.textDelta;
+  return updatePartForPath(message, chunk, (part) => {
+    if (part.type === "text") {
+      return { ...part, text: part.text + chunk.textDelta };
+    } else if (part.type === "tool-call") {
+      const newArgsText = part.argsText + chunk.textDelta;
       let newArgs;
       try {
         newArgs = parsePartialJson(newArgsText);
       } catch (err) {
-        newArgs = lastPart.args;
+        newArgs = part.args;
       }
-      return { ...lastPart, argsText: newArgsText, args: newArgs };
+      return { ...part, argsText: newArgsText, args: newArgs };
     } else {
       throw new Error(
-        "text-delta received but last part is neither text nor tool-call"
+        "text-delta received but part is neither text nor tool-call"
       );
     }
   });
 };
 var handleResult = (message, chunk) => {
-  return updatePartForPath(message, chunk, (lastPart) => {
-    if (lastPart.type === "tool-call") {
+  return updatePartForPath(message, chunk, (part) => {
+    if (part.type === "tool-call") {
       return {
-        ...lastPart,
+        ...part,
         state: "result",
         result: chunk.result,
         isError: chunk.isError ?? false,
         status: { type: "complete", reason: "stop" }
       };
     } else {
-      throw new Error("Result chunk received but last part is not a tool-call");
+      throw new Error("Result chunk received but part is not a tool-call");
+    }
+  });
+};
+var handleArtifact = (message, chunk) => {
+  return updatePartForPath(message, chunk, (part) => {
+    if (part.type === "tool-call") {
+      return {
+        ...part,
+        artifact: chunk.artifact
+      };
+    } else {
+      throw new Error("Artifact chunk received but part is not a tool-call");
     }
   });
 };
@@ -1670,6 +1713,9 @@ var AssistantMessageAccumulator = class extends TransformStream {
           case "result":
             message = handleResult(message, chunk);
             break;
+          case "artifact":
+            message = handleArtifact(message, chunk);
+            break;
           case "message-finish":
             message = handleMessageFinish(message, chunk);
             break;