assistant-stream 0.0.24 → 0.0.26

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);
@@ -1066,7 +1099,7 @@ var ToolExecutionStream = class extends PipeableTransformStream {
           }
         },
         async flush() {
-          await Promise.all(toolCallPromises);
+          await Promise.all(toolCallPromises.values());
         }
       });
       return readable.pipeThrough(new AssistantMetaTransformStream()).pipeThrough(transform);
@@ -1508,54 +1541,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 +1715,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;