cr

Author: christian-bromann

libs/langchain/src/agents/middleware/tests/modelCallLimit.test.ts

@@ -112,7 +112,7 @@ describe("ModelCallLimitMiddleware", () => {
     }
   );
 
-  describe.each(["throw" /*, "end"*/] as const)(
+  describe.each(["throw", "end"] as const)(
     "thread limit with exit behavior %s",
     (exitBehavior) => {
       const checkpointer = new MemorySaver();
@@ -126,7 +126,7 @@ describe("ModelCallLimitMiddleware", () => {
         exitBehavior,
       });
 
-      it.only("should not throw if the thread limit is not exceeded", async () => {
+      it("should not throw if the thread limit is not exceeded", async () => {
         const model = new FakeToolCallingChatModel({
           responses: [toolCallMessage1],
         });
@@ -151,7 +151,6 @@ describe("ModelCallLimitMiddleware", () => {
             { messages: ["Hello, world!"] },
             config
           );
-          console.log("result", result);
           await expect(result.runModelCallCount).toBe(3);
           await expect(result.threadModelCallCount).toBe(3);
         } else {

libs/langchain/src/agents/utils.ts

@@ -499,6 +499,7 @@ function chainToolCallHandlers(
  * the error message.
  *
  * @param middleware list of middleware passed to the agent
+ * @param state state of the agent
  * @returns single wrap function
  */
 export function wrapToolCall(