@langchain/core 0.2.1 → 0.2.2

package/dist/runnables/base.cjs

@@ -482,7 +482,7 @@ class Runnable extends serializable_js_1.Serializable {
     async *_streamEventsV2(input, options, streamOptions) {
         const eventStreamer = new event_stream_js_1.EventStreamCallbackHandler({
             ...streamOptions,
-            autoClose: false,
+            autoClose: true,
         });
         const config = (0, config_js_1.ensureConfig)(options);
         const runId = config.runId ?? (0, uuid_1.v4)();
@@ -504,16 +504,11 @@ class Runnable extends serializable_js_1.Serializable {
         // add each chunk to the output stream
         const outerThis = this;
         async function consumeRunnableStream() {
-            try {
-                const runnableStream = await outerThis.stream(input, config);
-                const tappedStream = eventStreamer.tapOutputIterable(runId, runnableStream);
-                // eslint-disable-next-line @typescript-eslint/no-unused-vars
-                for await (const _ of tappedStream) {
-                    // Just iterate so that the callback handler picks up events
-                }
-            }
-            finally {
-                await eventStreamer.writer.close();
+            const runnableStream = await outerThis.stream(input, config);
+            const tappedStream = eventStreamer.tapOutputIterable(runId, runnableStream);
+            // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            for await (const _ of tappedStream) {
+                // Just iterate so that the callback handler picks up events
             }
         }
         const runnableStreamConsumePromise = consumeRunnableStream();

package/dist/runnables/base.js

@@ -475,7 +475,7 @@ export class Runnable extends Serializable {
     async *_streamEventsV2(input, options, streamOptions) {
         const eventStreamer = new EventStreamCallbackHandler({
             ...streamOptions,
-            autoClose: false,
+            autoClose: true,
         });
         const config = ensureConfig(options);
         const runId = config.runId ?? uuidv4();
@@ -497,16 +497,11 @@ export class Runnable extends Serializable {
         // add each chunk to the output stream
         const outerThis = this;
         async function consumeRunnableStream() {
-            try {
-                const runnableStream = await outerThis.stream(input, config);
-                const tappedStream = eventStreamer.tapOutputIterable(runId, runnableStream);
-                // eslint-disable-next-line @typescript-eslint/no-unused-vars
-                for await (const _ of tappedStream) {
-                    // Just iterate so that the callback handler picks up events
-                }
-            }
-            finally {
-                await eventStreamer.writer.close();
+            const runnableStream = await outerThis.stream(input, config);
+            const tappedStream = eventStreamer.tapOutputIterable(runId, runnableStream);
+            // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            for await (const _ of tappedStream) {
+                // Just iterate so that the callback handler picks up events
             }
         }
         const runnableStreamConsumePromise = consumeRunnableStream();

package/dist/tracers/event_stream.cjs

@@ -205,6 +205,7 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
             finally {
                 // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
                 tappedPromiseResolver();
+                // Don't delete from the map to keep track of which runs have been tapped.
             }
         }
         else {
@@ -489,5 +490,18 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
             metadata: runInfo.metadata,
         }, runInfo);
     }
+    async onRunCreate(run) {
+        if (this.rootId === undefined) {
+            this.rootId = run.id;
+        }
+    }
+    async onRunUpdate(run) {
+        if (run.id === this.rootId && this.autoClose) {
+            const pendingPromises = [...this.tappedPromises.values()];
+            void Promise.all(pendingPromises).finally(() => {
+                void this.writer.close();
+            });
+        }
+    }
 }
 exports.EventStreamCallbackHandler = EventStreamCallbackHandler;

package/dist/tracers/event_stream.d.ts

@@ -133,5 +133,7 @@ export declare class EventStreamCallbackHandler extends BaseTracer {
     onToolEnd(run: Run): Promise<void>;
     onRetrieverStart(run: Run): Promise<void>;
     onRetrieverEnd(run: Run): Promise<void>;
+    onRunCreate(run: Run): Promise<void>;
+    onRunUpdate(run: Run): Promise<void>;
 }
 export {};

package/dist/tracers/event_stream.js

@@ -202,6 +202,7 @@ export class EventStreamCallbackHandler extends BaseTracer {
             finally {
                 // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
                 tappedPromiseResolver();
+                // Don't delete from the map to keep track of which runs have been tapped.
             }
         }
         else {
@@ -486,4 +487,17 @@ export class EventStreamCallbackHandler extends BaseTracer {
             metadata: runInfo.metadata,
         }, runInfo);
     }
+    async onRunCreate(run) {
+        if (this.rootId === undefined) {
+            this.rootId = run.id;
+        }
+    }
+    async onRunUpdate(run) {
+        if (run.id === this.rootId && this.autoClose) {
+            const pendingPromises = [...this.tappedPromises.values()];
+            void Promise.all(pendingPromises).finally(() => {
+                void this.writer.close();
+            });
+        }
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {