fix(langchain): support for BaseCallbackConfig in agent config (#9291)

Author: christian-bromann

libs/langchain/src/agents/runtime.ts

@@ -6,6 +6,7 @@ import type {
   PregelOptions,
 } from "@langchain/langgraph";
 import type { BaseMessage } from "@langchain/core/messages";
+import type { BaseCallbackConfig } from "@langchain/core/callbacks/manager";
 
 import type { ResponseFormatUndefined } from "./responses.js";
 
@@ -152,17 +153,24 @@ export type InvokeConfiguration<ContextSchema extends Record<string, any>> =
    * If the context schema is a default object, `context` can be optional
    */
   ContextSchema extends InteropZodDefault<any>
-    ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & {
-        context?: Partial<ContextSchema>;
-      }
+    ? BaseCallbackConfig &
+        Partial<
+          Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>
+        > & {
+          context?: Partial<ContextSchema>;
+        }
     : /**
      * If the context schema is all optional, `context` can be optional
      */
     IsAllOptional<ContextSchema> extends true
-    ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & {
-        context?: Partial<ContextSchema>;
-      }
-    : Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> &
+    ? BaseCallbackConfig &
+        Partial<
+          Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>
+        > & {
+          context?: Partial<ContextSchema>;
+        }
+    : BaseCallbackConfig &
+        Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> &
         WithMaybeContext<ContextSchema>;
 
 export type StreamConfiguration<ContextSchema extends Record<string, any>> =

libs/langchain/src/agents/tests/reactAgent.test-d.ts

@@ -189,4 +189,30 @@ describe("reactAgent", () => {
       });
     });
   });
+
+  it("supports base callback config", async () => {
+    const agent = createAgent({
+      model: "openai:gpt-4",
+    });
+    await agent.invoke(
+      {
+        messages: [new HumanMessage("Hello, world!")],
+      },
+      {
+        runName: "test",
+        metadata: {
+          test: "test",
+        },
+        callbacks: [
+          {
+            handleLLMStart: (input) => {
+              expectTypeOf({ id: input.id }).toMatchObjectType<{
+                id: string[];
+              }>();
+            },
+          },
+        ],
+      }
+    );
+  });
 });