@langchain/core 0.2.17 → 0.2.18

package/dist/tracers/tracer_langchain.js

@@ -1,4 +1,5 @@
 import { Client } from "langsmith";
+import { RunTree } from "langsmith/run_trees";
 import { getCurrentRunTree } from "langsmith/singletons/traceable";
 import { getEnvironmentVariable, getRuntimeEnvironment } from "../utils/env.js";
 import { BaseTracer } from "./base.js";
@@ -36,37 +37,9 @@ export class LangChainTracer extends BaseTracer {
                 getEnvironmentVariable("LANGCHAIN_SESSION");
         this.exampleId = exampleId;
         this.client = client ?? new Client({});
-        // if we're inside traceable, we can obtain the traceable tree
-        // and populate the run map, which is used to correctly
-        // infer dotted order and execution order
-        const traceableTree = this.getTraceableRunTree();
+        const traceableTree = LangChainTracer.getTraceableRunTree();
         if (traceableTree) {
-            let rootRun = traceableTree;
-            const visited = new Set();
-            while (rootRun.parent_run) {
-                if (visited.has(rootRun.id))
-                    break;
-                visited.add(rootRun.id);
-                if (!rootRun.parent_run)
-                    break;
-                rootRun = rootRun.parent_run;
-            }
-            visited.clear();
-            const queue = [rootRun];
-            while (queue.length > 0) {
-                const current = queue.shift();
-                if (!current || visited.has(current.id))
-                    continue;
-                visited.add(current.id);
-                // @ts-expect-error Types of property 'events' are incompatible.
-                this.runMap.set(current.id, current);
-                if (current.child_runs) {
-                    queue.push(...current.child_runs);
-                }
-            }
-            this.client = traceableTree.client ?? this.client;
-            this.projectName = traceableTree.project_name ?? this.projectName;
-            this.exampleId = traceableTree.reference_example_id ?? this.exampleId;
+            this.updateFromRunTree(traceableTree);
         }
     }
     async _convertToCreate(run, example_id = undefined) {
@@ -102,7 +75,76 @@ export class LangChainTracer extends BaseTracer {
     getRun(id) {
         return this.runMap.get(id);
     }
-    getTraceableRunTree() {
+    updateFromRunTree(runTree) {
+        let rootRun = runTree;
+        const visited = new Set();
+        while (rootRun.parent_run) {
+            if (visited.has(rootRun.id))
+                break;
+            visited.add(rootRun.id);
+            if (!rootRun.parent_run)
+                break;
+            rootRun = rootRun.parent_run;
+        }
+        visited.clear();
+        const queue = [rootRun];
+        while (queue.length > 0) {
+            const current = queue.shift();
+            if (!current || visited.has(current.id))
+                continue;
+            visited.add(current.id);
+            // @ts-expect-error Types of property 'events' are incompatible.
+            this.runMap.set(current.id, current);
+            if (current.child_runs) {
+                queue.push(...current.child_runs);
+            }
+        }
+        this.client = runTree.client ?? this.client;
+        this.projectName = runTree.project_name ?? this.projectName;
+        this.exampleId = runTree.reference_example_id ?? this.exampleId;
+    }
+    convertToRunTree(id) {
+        const runTreeMap = {};
+        const runTreeList = [];
+        for (const [id, run] of this.runMap) {
+            // by converting the run map to a run tree, we are doing a copy
+            // thus, any mutation performed on the run tree will not be reflected
+            // back in the run map
+            // TODO: Stop using `this.runMap` in favour of LangSmith's `RunTree`
+            const runTree = new RunTree({
+                ...run,
+                child_runs: [],
+                parent_run: undefined,
+                // inherited properties
+                client: this.client,
+                project_name: this.projectName,
+                reference_example_id: this.exampleId,
+                tracingEnabled: true,
+            });
+            runTreeMap[id] = runTree;
+            runTreeList.push([id, run.dotted_order]);
+        }
+        runTreeList.sort((a, b) => {
+            if (!a[1] || !b[1])
+                return 0;
+            return a[1].localeCompare(b[1]);
+        });
+        for (const [id] of runTreeList) {
+            const run = this.runMap.get(id);
+            const runTree = runTreeMap[id];
+            if (!run || !runTree)
+                continue;
+            if (run.parent_run_id) {
+                const parentRunTree = runTreeMap[run.parent_run_id];
+                if (parentRunTree) {
+                    parentRunTree.child_runs.push(runTree);
+                    runTree.parent_run = parentRunTree;
+                }
+            }
+        }
+        return runTreeMap[id];
+    }
+    static getTraceableRunTree() {
         try {
             return getCurrentRunTree();
         }

package/dist/utils/stream.cjs

@@ -1,8 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.pipeGeneratorWithSetup = exports.AsyncGeneratorWithSetup = exports.concat = exports.atee = exports.IterableReadableStream = void 0;
-// Make this a type to override ReadableStream's async iterator type in case
-// the popular web-streams-polyfill is imported - the supplied types
 const index_js_1 = require("../singletons/index.cjs");
 /*
  * Support async iterator syntax for ReadableStreams in all environments.
@@ -208,8 +206,7 @@ class AsyncGeneratorWithSetup {
         // needs to happen in logical order, ie. in the order in which input to
         // to each generator is available.
         this.setup = new Promise((resolve, reject) => {
-            const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
-            void storage.run(params.config, async () => {
+            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(params.config, async () => {
                 this.firstResult = params.generator.next();
                 if (params.startSetup) {
                     this.firstResult.then(params.startSetup).then(resolve, reject);
@@ -217,7 +214,7 @@ class AsyncGeneratorWithSetup {
                 else {
                     this.firstResult.then((_result) => resolve(undefined), reject);
                 }
-            });
+            }, true);
         });
     }
     async next(...args) {
@@ -225,10 +222,9 @@ class AsyncGeneratorWithSetup {
             this.firstResultUsed = true;
             return this.firstResult;
         }
-        const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
-        return storage.run(this.config, async () => {
+        return index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(this.config, async () => {
             return this.generator.next(...args);
-        });
+        }, true);
     }
     async return(value) {
         return this.generator.return(value);
@@ -242,7 +238,10 @@ class AsyncGeneratorWithSetup {
 }
 exports.AsyncGeneratorWithSetup = AsyncGeneratorWithSetup;
 async function pipeGeneratorWithSetup(to, generator, startSetup, ...args) {
-    const gen = new AsyncGeneratorWithSetup({ generator, startSetup });
+    const gen = new AsyncGeneratorWithSetup({
+        generator,
+        startSetup,
+    });
     const setup = await gen.setup;
     return { output: to(gen, setup, ...args), setup };
 }

package/dist/utils/stream.js

@@ -1,5 +1,3 @@
-// Make this a type to override ReadableStream's async iterator type in case
-// the popular web-streams-polyfill is imported - the supplied types
 import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
 /*
  * Support async iterator syntax for ReadableStreams in all environments.
@@ -202,8 +200,7 @@ export class AsyncGeneratorWithSetup {
         // needs to happen in logical order, ie. in the order in which input to
         // to each generator is available.
         this.setup = new Promise((resolve, reject) => {
-            const storage = AsyncLocalStorageProviderSingleton.getInstance();
-            void storage.run(params.config, async () => {
+            void AsyncLocalStorageProviderSingleton.runWithConfig(params.config, async () => {
                 this.firstResult = params.generator.next();
                 if (params.startSetup) {
                     this.firstResult.then(params.startSetup).then(resolve, reject);
@@ -211,7 +208,7 @@ export class AsyncGeneratorWithSetup {
                 else {
                     this.firstResult.then((_result) => resolve(undefined), reject);
                 }
-            });
+            }, true);
         });
     }
     async next(...args) {
@@ -219,10 +216,9 @@ export class AsyncGeneratorWithSetup {
             this.firstResultUsed = true;
             return this.firstResult;
         }
-        const storage = AsyncLocalStorageProviderSingleton.getInstance();
-        return storage.run(this.config, async () => {
+        return AsyncLocalStorageProviderSingleton.runWithConfig(this.config, async () => {
             return this.generator.next(...args);
-        });
+        }, true);
     }
     async return(value) {
         return this.generator.return(value);
@@ -235,7 +231,10 @@ export class AsyncGeneratorWithSetup {
     }
 }
 export async function pipeGeneratorWithSetup(to, generator, startSetup, ...args) {
-    const gen = new AsyncGeneratorWithSetup({ generator, startSetup });
+    const gen = new AsyncGeneratorWithSetup({
+        generator,
+        startSetup,
+    });
     const setup = await gen.setup;
     return { output: to(gen, setup, ...args), setup };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.2.17",
+  "version": "0.2.18",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {
@@ -45,7 +45,7 @@
     "camelcase": "6",
     "decamelize": "1.2.0",
     "js-tiktoken": "^1.0.12",
-    "langsmith": "~0.1.30",
+    "langsmith": "~0.1.39",
     "ml-distance": "^4.0.0",
     "mustache": "^4.2.0",
     "p-queue": "^6.6.2",