class MockAsyncLocalStorage { getStore() { return undefined; } run(_, callback) { return callback(); } } const TRACING_ALS_KEY = Symbol.for("ls:tracing_async_local_storage"); const mockAsyncLocalStorage = new MockAsyncLocalStorage(); class AsyncLocalStorageProvider { getInstance() { // eslint-disable-next-line @typescript-eslint/no-explicit-any return globalThis[TRACING_ALS_KEY] ?? mockAsyncLocalStorage; } initializeGlobalInstance(instance) { // eslint-disable-next-line @typescript-eslint/no-explicit-any if (globalThis[TRACING_ALS_KEY] === undefined) { // eslint-disable-next-line @typescript-eslint/no-explicit-any globalThis[TRACING_ALS_KEY] = instance; } } } export const AsyncLocalStorageProviderSingleton = new AsyncLocalStorageProvider(); /** * Return the current run tree from within a traceable-wrapped function. * Will throw an error if called outside of a traceable function. * * @returns The run tree for the given context. */ export const getCurrentRunTree = () => { const runTree = AsyncLocalStorageProviderSingleton.getInstance().getStore(); if (runTree === undefined) { throw new Error([ "Could not get the current run tree.", "", "Please make sure you are calling this method within a traceable function or the tracing is enabled.", ].join("\n")); } return runTree; }; // eslint-disable-next-line @typescript-eslint/no-explicit-any export function withRunTree(runTree, fn) { const storage = AsyncLocalStorageProviderSingleton.getInstance(); return new Promise((resolve, reject) => { storage.run(runTree, () => void Promise.resolve(fn()).then(resolve).catch(reject)); }); } export const ROOT = Symbol.for("langsmith:traceable:root"); export function isTraceableFunction(x // eslint-disable-next-line @typescript-eslint/no-explicit-any ) { return typeof x === "function" && "langsmith:traceable" in x; }