fix(ai-sdk): DataStream interop without tool call streaming (#1039)

Co-authored-by: Abdulaziz Alaboudi <[email protected]>
Yonom · Oct 21, 2024 · a1bfd26 · a1bfd26
1 parent b38165d
commit a1bfd26
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/.changeset/honest-monkeys-doubt.md b/.changeset/honest-monkeys-doubt.md
@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+fix(ai-sdk): DataStream interop without tool call streaming
diff --git a/packages/react/src/runtimes/edge/streams/assistantDecoderStream.ts b/packages/react/src/runtimes/edge/streams/assistantDecoderStream.ts
@@ -42,6 +42,7 @@ export function assistantDecoderStream() {
         case AssistantStreamChunkType.ToolCallBegin: {
           const { toolCallId: id, toolName: name } = value;
           toolCallNames.set(id, name);
+
           currentToolCall = { id, name, argsText: "" };
 
           controller.enqueue({
@@ -96,6 +97,8 @@ export function assistantDecoderStream() {
 
         case AssistantStreamChunkType.ToolCall: {
           const { toolCallId, toolName, args } = value;
+          toolCallNames.set(toolCallId, toolName);
+
           const argsText = JSON.stringify(args);
           controller.enqueue({
             type: "tool-call-delta",

diff --git a/packages/react/src/runtimes/edge/streams/runResultStream.ts b/packages/react/src/runtimes/edge/streams/runResultStream.ts
@@ -100,7 +100,7 @@ const appendOrUpdateToolCall = (
   argsTextDelta: string,
 ): ChatModelRunResult => {
   let contentParts = message.content ?? [];
-  let contentPartIdx = contentParts.findIndex(
+  const contentPartIdx = contentParts.findIndex(
     (c) => c.type === "tool-call" && c.toolCallId === toolCallId,
   );
   let contentPart =