core[patch]: Populate tool name in ToolMessage return value (#6066)

* core[patch]: Populate tool name in ToolMessage return value * cr
langchain-ai · Jul 15, 2024 · 7e47915 · 7e47915
1 parent 4644dd3
commit 7e47915
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/langchain-core/src/tools/index.ts b/langchain-core/src/tools/index.ts
@@ -247,6 +247,7 @@ export abstract class StructuredTool<
       content,
       artifact,
       toolCallId,
+      name: this.name,
     });
     await runManager?.handleToolEnd(formattedOutput);
     return formattedOutput;
@@ -528,6 +529,7 @@ function _isToolCall(toolCall?: unknown): toolCall is ToolCall {
 
 function _formatToolOutput(params: {
   content: unknown;
+  name: string;
   artifact?: unknown;
   toolCallId?: string;
 }): ToolReturnType {
@@ -542,12 +544,14 @@ function _formatToolOutput(params: {
         content,
         artifact,
         tool_call_id: toolCallId,
+        name: params.name,
       });
     } else {
       return new ToolMessage({
         content: _stringify(content),
         artifact,
         tool_call_id: toolCallId,
+        name: params.name,
       });
     }
   } else {

diff --git a/langchain-core/src/tools/tests/tools.test.ts b/langchain-core/src/tools/tests/tools.test.ts
@@ -97,4 +97,5 @@ test("Returns tool message if responseFormat is content_and_artifact and returns
   expect(toolResult).toBeInstanceOf(ToolMessage);
   expect(toolResult.content).toBe("msg_content");
   expect(toolResult.artifact).toEqual({ location: "San Francisco" });
+  expect(toolResult.name).toBe("weather");
 });