@emblemvault/hustle-react 1.4.1 → 1.4.3

package/dist/hooks/index.d.cts

@@ -1,5 +1,5 @@
 export { HustleProvider, useHustle } from '../providers/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, a as HydratedPlugin, H as HustlePlugin } from '../plugin-COr42J6-.cjs';
 import 'react/jsx-runtime';
 import 'hustle-incognito';

package/dist/hooks/index.d.ts

@@ -1,5 +1,5 @@
 export { HustleProvider, useHustle } from '../providers/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, a as HydratedPlugin, H as HustlePlugin } from '../plugin-COr42J6-.js';
 import 'react/jsx-runtime';
 import 'hustle-incognito';

package/dist/{hustle-S48t4lTZ.d.cts → hustle-C0Ltl5k4.d.cts}

@@ -5,12 +5,39 @@ import { HustleIncognitoClient } from 'hustle-incognito';
  * These types mirror the HustleIncognitoClient types for use in React components
  */
 
+/**
+ * Tool invocation with state tracking (AI SDK format)
+ */
+interface ToolInvocation {
+    state: 'call' | 'result';
+    step: number;
+    toolCallId: string;
+    toolName: string;
+    args?: Record<string, unknown>;
+    result?: unknown;
+}
+/**
+ * Message part for AI SDK compatibility
+ */
+type MessagePart = {
+    type: 'step-start';
+} | {
+    type: 'text';
+    text: string;
+} | {
+    type: 'tool-invocation';
+    toolInvocation: ToolInvocation;
+};
 /**
  * Chat message structure
  */
 interface ChatMessage {
     role: 'system' | 'user' | 'assistant';
     content: string;
+    /** Tool invocations for this message (AI SDK format) */
+    toolInvocations?: ToolInvocation[];
+    /** Message parts for AI SDK compatibility */
+    parts?: MessagePart[];
 }
 /**
  * Tool call information

package/dist/{hustle-S48t4lTZ.d.ts → hustle-C0Ltl5k4.d.ts}

@@ -5,12 +5,39 @@ import { HustleIncognitoClient } from 'hustle-incognito';
  * These types mirror the HustleIncognitoClient types for use in React components
  */
 
+/**
+ * Tool invocation with state tracking (AI SDK format)
+ */
+interface ToolInvocation {
+    state: 'call' | 'result';
+    step: number;
+    toolCallId: string;
+    toolName: string;
+    args?: Record<string, unknown>;
+    result?: unknown;
+}
+/**
+ * Message part for AI SDK compatibility
+ */
+type MessagePart = {
+    type: 'step-start';
+} | {
+    type: 'text';
+    text: string;
+} | {
+    type: 'tool-invocation';
+    toolInvocation: ToolInvocation;
+};
 /**
  * Chat message structure
  */
 interface ChatMessage {
     role: 'system' | 'user' | 'assistant';
     content: string;
+    /** Tool invocations for this message (AI SDK format) */
+    toolInvocations?: ToolInvocation[];
+    /** Message parts for AI SDK compatibility */
+    parts?: MessagePart[];
 }
 /**
  * Tool call information

package/dist/index.cjs

@@ -4049,7 +4049,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,
@@ -4057,6 +4064,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;
@@ -4068,22 +4078,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);
@@ -4145,7 +4212,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,
@@ -4155,6 +4229,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;
@@ -4166,22 +4243,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);