From de6362602c9ff9fd7710976ef65e372843d455b9 Mon Sep 17 00:00:00 2001 From: Jacob Lee Date: Mon, 13 Jan 2025 14:14:07 -0800 Subject: [PATCH] fix(core): Add run name for trimMessages (#7517) --- langchain-core/src/messages/tests/message_utils.test.ts | 6 ++++-- langchain-core/src/messages/transformers.ts | 4 +++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/langchain-core/src/messages/tests/message_utils.test.ts b/langchain-core/src/messages/tests/message_utils.test.ts index 222d9b42372a..9788cbafc4b9 100644 --- a/langchain-core/src/messages/tests/message_utils.test.ts +++ b/langchain-core/src/messages/tests/message_utils.test.ts @@ -431,10 +431,12 @@ describe("trimMessages can trim", () => { "langchain_core", "runnables", ]); - expect("func" in trimmedMessages).toBeTruthy(); + expect("bound" in trimmedMessages).toBeTruthy(); + // eslint-disable-next-line @typescript-eslint/no-explicit-any + expect("func" in (trimmedMessages as any).bound).toBeTruthy(); // `func` is protected, so we need to cast it to any to access it // eslint-disable-next-line @typescript-eslint/no-explicit-any - expect(typeof (trimmedMessages as any).func).toBe("function"); + expect(typeof (trimmedMessages as any).bound.func).toBe("function"); }); }); diff --git a/langchain-core/src/messages/transformers.ts b/langchain-core/src/messages/transformers.ts index 81d0d7df6738..15389655d59c 100644 --- a/langchain-core/src/messages/transformers.ts +++ b/langchain-core/src/messages/transformers.ts @@ -643,7 +643,9 @@ export function trimMessages( const trimmerOptions = messagesOrOptions; return RunnableLambda.from((input: BaseMessage[]) => _trimMessagesHelper(input, trimmerOptions) - ); + ).withConfig({ + runName: "trim_messages", + }); } }