feat: add toolName to addToolResult callback (#735)

Yonom · Sep 6, 2024 · 31616d3 · 31616d3
1 parent 456d8f1
commit 31616d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/packages/react/src/context/stores/ThreadActions.ts b/packages/react/src/context/stores/ThreadActions.ts
@@ -6,6 +6,7 @@ import { SpeechSynthesisAdapter } from "../../runtimes/speech/SpeechAdapterTypes
 
 export type AddToolResultOptions = {
   messageId: string;
+  toolName: string;
   toolCallId: string;
   result: any;
 };

diff --git a/packages/react/src/primitives/message/MessageContent.tsx b/packages/react/src/primitives/message/MessageContent.tsx
@@ -106,6 +106,7 @@ const MessageContentPartComponent: FC<MessageContentPartComponentProps> = ({
       const addResult = (result: any) =>
         addToolResult({
           messageId: useMessage.getState().message.id,
+          toolName: part.toolName,
           toolCallId: part.toolCallId,
           result,
         });