feat(js): Validate that store values are RunTree instances (#1088)

langchain-ai · Oct 11, 2024 · 3c697ab · 3c697ab
1 parent 2bf5482
commit 3c697ab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/js/src/singletons/traceable.ts b/js/src/singletons/traceable.ts
@@ -1,4 +1,4 @@
-import { RunTree } from "../run_trees.js";
+import { isRunTree, RunTree } from "../run_trees.js";
 import { TraceableFunction } from "./types.js";
 
 interface AsyncLocalStorageInterface {
@@ -47,7 +47,7 @@ export const AsyncLocalStorageProviderSingleton =
  */
 export const getCurrentRunTree = () => {
   const runTree = AsyncLocalStorageProviderSingleton.getInstance().getStore();
-  if (runTree === undefined) {
+  if (!isRunTree(runTree)) {
     throw new Error(
       [
         "Could not get the current run tree.",

diff --git a/js/src/traceable.ts b/js/src/traceable.ts
@@ -406,7 +406,7 @@ export function traceable<Func extends (...args: any[]) => any>(
       // Node.JS uses AsyncLocalStorage (ALS) and AsyncResource
       // to allow storing context
       const prevRunFromStore = asyncLocalStorage.getStore();
-      if (prevRunFromStore) {
+      if (isRunTree(prevRunFromStore)) {
         return [
           getTracingRunTree(
             prevRunFromStore.createChild(ensuredConfig),