@emblemvault/hustle-react 1.4.1 → 1.4.3

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 export { HustleProvider, useHustle } from './providers/index.cjs';
 export { HustleChat, HustleChatProps, HustleChatWidget, HustleChatWidgetConfig, HustleChatWidgetProps, MarkdownContent, WidgetPosition, WidgetSize } from './components/index.cjs';
-export { A as Attachment, C as ChatMessage, b as ChatOptions, d as ChatResponse, H as HustleConfig, h as HustleContextValue, i as HustleProviderProps, M as Model, c as StreamChunk, g as StreamEndEvent, S as StreamOptions, T as ToolCall, f as ToolEndEvent, a as ToolResult, e as ToolStartEvent } from './hustle-S48t4lTZ.cjs';
+export { A as Attachment, C as ChatMessage, b as ChatOptions, d as ChatResponse, H as HustleConfig, h as HustleContextValue, i as HustleProviderProps, M as Model, c as StreamChunk, g as StreamEndEvent, S as StreamOptions, T as ToolCall, f as ToolEndEvent, a as ToolResult, e as ToolStartEvent } from './hustle-C0Ltl5k4.cjs';
 import { S as StoredPlugin, H as HustlePlugin, a as HydratedPlugin } from './plugin-COr42J6-.cjs';
 export { C as ClientToolDefinition, E as ErrorContext, d as HustleRequest, J as JSONSchema, b as JSONSchemaProperty, e as PluginHooks, P as ProcessedResponse, f as SerializedHooks, c as SerializedToolDefinition, T as ToolExecutor } from './plugin-COr42J6-.cjs';
 export { UsePluginsReturn, usePlugins } from './hooks/index.cjs';

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 export { HustleProvider, useHustle } from './providers/index.js';
 export { HustleChat, HustleChatProps, HustleChatWidget, HustleChatWidgetConfig, HustleChatWidgetProps, MarkdownContent, WidgetPosition, WidgetSize } from './components/index.js';
-export { A as Attachment, C as ChatMessage, b as ChatOptions, d as ChatResponse, H as HustleConfig, h as HustleContextValue, i as HustleProviderProps, M as Model, c as StreamChunk, g as StreamEndEvent, S as StreamOptions, T as ToolCall, f as ToolEndEvent, a as ToolResult, e as ToolStartEvent } from './hustle-S48t4lTZ.js';
+export { A as Attachment, C as ChatMessage, b as ChatOptions, d as ChatResponse, H as HustleConfig, h as HustleContextValue, i as HustleProviderProps, M as Model, c as StreamChunk, g as StreamEndEvent, S as StreamOptions, T as ToolCall, f as ToolEndEvent, a as ToolResult, e as ToolStartEvent } from './hustle-C0Ltl5k4.js';
 import { S as StoredPlugin, H as HustlePlugin, a as HydratedPlugin } from './plugin-COr42J6-.js';
 export { C as ClientToolDefinition, E as ErrorContext, d as HustleRequest, J as JSONSchema, b as JSONSchemaProperty, e as PluginHooks, P as ProcessedResponse, f as SerializedHooks, c as SerializedToolDefinition, T as ToolExecutor } from './plugin-COr42J6-.js';
 export { UsePluginsReturn, usePlugins } from './hooks/index.js';

package/dist/index.js

@@ -4023,7 +4023,14 @@ function HustleChat({
     setIsStreaming(true);
     setCurrentToolCalls([]);
     try {
-      const chatMessages = messagesRef.current.filter((m) => !m.isStreaming).map((m) => ({ role: m.role, content: m.content }));
+      const chatMessages = messagesRef.current.filter((m) => !m.isStreaming).map((m) => {
+        const msg = { role: m.role, content: m.content };
+        if (m.role === "assistant" && m.toolInvocations && m.toolInvocations.length > 0) {
+          msg.toolInvocations = m.toolInvocations;
+          msg.parts = m.parts;
+        }
+        return msg;
+      });
       chatMessages.push({ role: "user", content: "continue" });
       const stream = chatStream({
         messages: chatMessages,
@@ -4031,6 +4038,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;
@@ -4042,22 +4052,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(
-              (m) => m.id === assistantMessage.id ? { ...m, toolCalls: [...toolCallsAccumulated] } : m
+              (m) => m.id === assistantMessage.id ? {
+                ...m,
+                toolCalls: [...toolCallsAccumulated],
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m
             )
           );
           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(
+              (m) => m.id === assistantMessage.id ? {
+                ...m,
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m
+            )
+          );
         } 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(
-          (m) => m.id === assistantMessage.id ? { ...m, isStreaming: false, content: finalContent } : m
+          (m) => m.id === assistantMessage.id ? {
+            ...m,
+            isStreaming: false,
+            content: finalContent,
+            toolInvocations: toolInvocationsAccumulated.length > 0 ? toolInvocationsAccumulated : void 0,
+            parts: toolInvocationsAccumulated.length > 0 ? finalParts : void 0
+          } : m
         )
       );
       onResponse?.(finalContent);
@@ -4119,7 +4186,14 @@ function HustleChat({
     setIsStreaming(true);
     setCurrentToolCalls([]);
     try {
-      const chatMessages = messages.filter((m) => !m.isStreaming).map((m) => ({ role: m.role, content: m.content }));
+      const chatMessages = messages.filter((m) => !m.isStreaming).map((m) => {
+        const msg = { role: m.role, content: m.content };
+        if (m.role === "assistant" && m.toolInvocations && m.toolInvocations.length > 0) {
+          msg.toolInvocations = m.toolInvocations;
+          msg.parts = m.parts;
+        }
+        return msg;
+      });
       chatMessages.push({ role: "user", content });
       const stream = chatStream({
         messages: chatMessages,
@@ -4129,6 +4203,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;
@@ -4140,22 +4217,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(
-              (m) => m.id === assistantMessage.id ? { ...m, toolCalls: [...toolCallsAccumulated] } : m
+              (m) => m.id === assistantMessage.id ? {
+                ...m,
+                toolCalls: [...toolCallsAccumulated],
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m
             )
           );
           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(
+              (m) => m.id === assistantMessage.id ? {
+                ...m,
+                toolInvocations: [...toolInvocationsAccumulated],
+                parts: [...partsAccumulated]
+              } : m
+            )
+          );
         } 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(
-          (m) => m.id === assistantMessage.id ? { ...m, isStreaming: false, content: finalContent } : m
+          (m) => m.id === assistantMessage.id ? {
+            ...m,
+            isStreaming: false,
+            content: finalContent,
+            toolInvocations: toolInvocationsAccumulated.length > 0 ? toolInvocationsAccumulated : void 0,
+            parts: toolInvocationsAccumulated.length > 0 ? finalParts : void 0
+          } : m
         )
       );
       onResponse?.(finalContent);