fix(core): Add run name for trimMessages (#7517)

langchain-ai · Jan 13, 2025 · de63626 · de63626
1 parent dbda6d0
commit de63626
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git 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
@@ -643,7 +643,9 @@ export function trimMessages(
     const trimmerOptions = messagesOrOptions;
     return RunnableLambda.from((input: BaseMessage[]) =>
       _trimMessagesHelper(input, trimmerOptions)
-    );
+    ).withConfig({
+      runName: "trim_messages",
+    });
   }
 }