fix(core): Loosen return type of withConfig, fix tracing for nested t…

…ool calls (langchain-ai#6847)
FilipZmijewski · Sep 27, 2024 · 10b85b2 · 10b85b2
1 parent 012a6c7
commit 10b85b2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/langchain-core/src/runnables/base.ts b/langchain-core/src/runnables/base.ts
@@ -174,7 +174,7 @@ export abstract class Runnable<
    */
   withConfig(
     config: RunnableConfig
-  ): RunnableBinding<RunInput, RunOutput, CallOptions> {
+  ): Runnable<RunInput, RunOutput, CallOptions> {
     // eslint-disable-next-line @typescript-eslint/no-use-before-define
     return new RunnableBinding({
       bound: this,
@@ -1236,7 +1236,7 @@ export class RunnableBinding<
 
   withConfig(
     config: RunnableConfig
-  ): RunnableBinding<RunInput, RunOutput, CallOptions> {
+  ): Runnable<RunInput, RunOutput, CallOptions> {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     return new (this.constructor as any)({
       bound: this.bound,
@@ -2328,7 +2328,6 @@ export class RunnableLambda<RunInput, RunOutput> extends Runnable<
           try {
             let output = await this.func(input, {
               ...childConfig,
-              config: childConfig,
             });
             if (output && Runnable.isRunnable(output)) {
               if (config?.recursionLimit === 0) {

diff --git a/langchain-core/src/tools/index.ts b/langchain-core/src/tools/index.ts
@@ -606,7 +606,7 @@ export function tool<
         const childConfig = patchConfig(config, {
           callbacks: runManager?.getChild(),
         });
-        void AsyncLocalStorageProviderSingleton.getInstance().run(
+        void AsyncLocalStorageProviderSingleton.runWithConfig(
           childConfig,
           async () => {
             try {

diff --git a/langchain-core/src/tools/tests/tools.test.ts b/langchain-core/src/tools/tests/tools.test.ts
@@ -1,5 +1,6 @@
 import { test, expect } from "@jest/globals";
 import { z } from "zod";
+
 import { DynamicStructuredTool, tool } from "../index.js";
 import { ToolMessage } from "../../messages/tool.js";