Merge branch 'main' into brace/mistral-llm-completions

langchain-ai · May 29, 2024 · 6a17555 · 6a17555
2 parents 6d5a259 + 9f0b1c3
commit 6a17555
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/langchain-core/src/singletons/index.ts b/langchain-core/src/singletons/index.ts
@@ -16,20 +16,19 @@ export class MockAsyncLocalStorage implements AsyncLocalStorageInterface {
   }
 }
 
-class AsyncLocalStorageProvider {
-  private asyncLocalStorage: AsyncLocalStorageInterface =
-    new MockAsyncLocalStorage();
-
-  private hasBeenInitialized = false;
+const mockAsyncLocalStorage = new MockAsyncLocalStorage();
 
+class AsyncLocalStorageProvider {
   getInstance(): AsyncLocalStorageInterface {
-    return this.asyncLocalStorage;
+    return (
+      (globalThis as any).__lc_tracing_async_local_storage ??
+      mockAsyncLocalStorage
+    );
   }
 
   initializeGlobalInstance(instance: AsyncLocalStorageInterface) {
-    if (!this.hasBeenInitialized) {
-      this.hasBeenInitialized = true;
-      this.asyncLocalStorage = instance;
+    if ((globalThis as any).__lc_tracing_async_local_storage === undefined) {
+      (globalThis as any).__lc_tracing_async_local_storage = instance;
     }
   }
 }