@emblemvault/hustle-react 1.4.1 → 1.4.3

package/dist/browser/hustle-react.js

@@ -1,4 +1,3 @@
-'use client';
 import { createContext, useState, useEffect, useCallback, useRef, useMemo, useContext } from 'react';
 import { HustleIncognitoClient } from 'hustle-incognito';
 import { useEmblemAuthOptional } from '@emblemvault/emblem-auth-react';
@@ -15179,7 +15178,14 @@ function HustleChat({
     setIsStreaming(true);
     setCurrentToolCalls([]);
     try {
-      const chatMessages = messagesRef.current.filter((m2) => !m2.isStreaming).map((m2) => ({ role: m2.role, content: m2.content }));
+      const chatMessages = messagesRef.current.filter((m2) => !m2.isStreaming).map((m2) => {
+        const msg = { role: m2.role, content: m2.content };
+        if (m2.role === "assistant" && m2.toolInvocations && m2.toolInvocations.length > 0) {
+          msg.toolInvocations = m2.toolInvocations;
+          msg.parts = m2.parts;
+        }
+        return msg;
+      });
       chatMessages.push({ role: "user", content: "continue" });
       const stream = chatStream({
         messages: chatMessages,
@@ -15187,6 +15193,9 @@ function HustleChat({
       });
       let fullContent = "";
       const toolCallsAccumulated = [];
+      const toolInvocationsAccumulated = [];
+      const partsAccumulated = [{ type: "step-start" }];
+      let stepCounter = 0;
       for await (const chunk of stream) {
         if (chunk.type === "text") {
           fullContent += chunk.value;
@@ -15198,22 +15207,79 @@ function HustleChat({
         } else if (chunk.type === "tool_call") {
           const toolCall = chunk.value;
           toolCallsAccumulated.push(toolCall);
+          const invocation = {
+            state: "call",
+            step: stepCounter++,
+            toolCallId: toolCall.toolCallId,
+            toolName: toolCall.toolName,
+            args: toolCall.args
+          };
+          toolInvocationsAccumulated.push(invocation);
+          partsAccumulated.push({ type: "tool-invocation", toolInvocation: invocation });
           setCurrentToolCalls([...toolCallsAccumulated]);
           setMessages(
             (prev) => prev.map(
-              (m2) => m2.id === assistantMessage.id ? { ...m2, toolCalls: [...toolCallsAccumulated] } : m2
+              (m2) => m2.id === assistantMessage.id ? {
+                ...m2,
+                toolCalls: [...toolCallsAccumulated],
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m2
             )
           );
           onToolCall?.(toolCall);
+        } else if (chunk.type === "tool_result") {
+          const toolResult = chunk.value;
+          const invocationIndex = toolInvocationsAccumulated.findIndex(
+            (inv) => inv.toolCallId === toolResult.toolCallId
+          );
+          if (invocationIndex !== -1) {
+            toolInvocationsAccumulated[invocationIndex] = {
+              ...toolInvocationsAccumulated[invocationIndex],
+              state: "result",
+              result: toolResult.result
+            };
+            const partIndex = partsAccumulated.findIndex(
+              (p) => p.type === "tool-invocation" && p.toolInvocation.toolCallId === toolResult.toolCallId
+            );
+            if (partIndex !== -1) {
+              partsAccumulated[partIndex] = {
+                type: "tool-invocation",
+                toolInvocation: toolInvocationsAccumulated[invocationIndex]
+              };
+            }
+          }
+          setMessages(
+            (prev) => prev.map(
+              (m2) => m2.id === assistantMessage.id ? {
+                ...m2,
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m2
+            )
+          );
         } else if (chunk.type === "error") {
           console.error("Stream error:", chunk.value);
         }
       }
       const processedResponse = await stream.response;
       const finalContent = processedResponse?.content || fullContent || "(No response)";
+      const finalParts = [{ type: "step-start" }];
+      if (finalContent) {
+        finalParts.push({ type: "text", text: finalContent });
+      }
+      for (const inv of toolInvocationsAccumulated) {
+        finalParts.push({ type: "tool-invocation", toolInvocation: inv });
+      }
       setMessages(
         (prev) => prev.map(
-          (m2) => m2.id === assistantMessage.id ? { ...m2, isStreaming: false, content: finalContent } : m2
+          (m2) => m2.id === assistantMessage.id ? {
+            ...m2,
+            isStreaming: false,
+            content: finalContent,
+            toolInvocations: toolInvocationsAccumulated.length > 0 ? toolInvocationsAccumulated : void 0,
+            parts: toolInvocationsAccumulated.length > 0 ? finalParts : void 0
+          } : m2
         )
       );
       onResponse?.(finalContent);
@@ -15275,7 +15341,14 @@ function HustleChat({
     setIsStreaming(true);
     setCurrentToolCalls([]);
     try {
-      const chatMessages = messages.filter((m2) => !m2.isStreaming).map((m2) => ({ role: m2.role, content: m2.content }));
+      const chatMessages = messages.filter((m2) => !m2.isStreaming).map((m2) => {
+        const msg = { role: m2.role, content: m2.content };
+        if (m2.role === "assistant" && m2.toolInvocations && m2.toolInvocations.length > 0) {
+          msg.toolInvocations = m2.toolInvocations;
+          msg.parts = m2.parts;
+        }
+        return msg;
+      });
       chatMessages.push({ role: "user", content });
       const stream = chatStream({
         messages: chatMessages,
@@ -15285,6 +15358,9 @@ function HustleChat({
       setAttachments([]);
       let fullContent = "";
       const toolCallsAccumulated = [];
+      const toolInvocationsAccumulated = [];
+      const partsAccumulated = [{ type: "step-start" }];
+      let stepCounter = 0;
       for await (const chunk of stream) {
         if (chunk.type === "text") {
           fullContent += chunk.value;
@@ -15296,22 +15372,79 @@ function HustleChat({
         } else if (chunk.type === "tool_call") {
           const toolCall = chunk.value;
           toolCallsAccumulated.push(toolCall);
+          const invocation = {
+            state: "call",
+            step: stepCounter++,
+            toolCallId: toolCall.toolCallId,
+            toolName: toolCall.toolName,
+            args: toolCall.args
+          };
+          toolInvocationsAccumulated.push(invocation);
+          partsAccumulated.push({ type: "tool-invocation", toolInvocation: invocation });
           setCurrentToolCalls([...toolCallsAccumulated]);
           setMessages(
             (prev) => prev.map(
-              (m2) => m2.id === assistantMessage.id ? { ...m2, toolCalls: [...toolCallsAccumulated] } : m2
+              (m2) => m2.id === assistantMessage.id ? {
+                ...m2,
+                toolCalls: [...toolCallsAccumulated],
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m2
             )
           );
           onToolCall?.(toolCall);
+        } else if (chunk.type === "tool_result") {
+          const toolResult = chunk.value;
+          const invocationIndex = toolInvocationsAccumulated.findIndex(
+            (inv) => inv.toolCallId === toolResult.toolCallId
+          );
+          if (invocationIndex !== -1) {
+            toolInvocationsAccumulated[invocationIndex] = {
+              ...toolInvocationsAccumulated[invocationIndex],
+              state: "result",
+              result: toolResult.result
+            };
+            const partIndex = partsAccumulated.findIndex(
+              (p) => p.type === "tool-invocation" && p.toolInvocation.toolCallId === toolResult.toolCallId
+            );
+            if (partIndex !== -1) {
+              partsAccumulated[partIndex] = {
+                type: "tool-invocation",
+                toolInvocation: toolInvocationsAccumulated[invocationIndex]
+              };
+            }
+          }
+          setMessages(
+            (prev) => prev.map(
+              (m2) => m2.id === assistantMessage.id ? {
+                ...m2,
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m2
+            )
+          );
         } else if (chunk.type === "error") {
           console.error("Stream error:", chunk.value);
         }
       }
       const processedResponse = await stream.response;
       const finalContent = processedResponse?.content || fullContent || "(No response)";
+      const finalParts = [{ type: "step-start" }];
+      if (finalContent) {
+        finalParts.push({ type: "text", text: finalContent });
+      }
+      for (const inv of toolInvocationsAccumulated) {
+        finalParts.push({ type: "tool-invocation", toolInvocation: inv });
+      }
       setMessages(
         (prev) => prev.map(
-          (m2) => m2.id === assistantMessage.id ? { ...m2, isStreaming: false, content: finalContent } : m2
+          (m2) => m2.id === assistantMessage.id ? {
+            ...m2,
+            isStreaming: false,
+            content: finalContent,
+            toolInvocations: toolInvocationsAccumulated.length > 0 ? toolInvocationsAccumulated : void 0,
+            parts: toolInvocationsAccumulated.length > 0 ? finalParts : void 0
+          } : m2
         )
       );
       onResponse?.(finalContent);