@langchain/core 0.1.16 → 0.1.17-rc.0

package/dist/runnables/base.cjs

@@ -154,7 +154,11 @@ class Runnable extends serializable_js_1.Serializable {
      * @returns A readable stream that is also an iterable.
      */
     async stream(input, options) {
-        return stream_js_1.IterableReadableStream.fromAsyncGenerator(this._streamIterator(input, options));
+        // Buffer the first streamed chunk to allow for initial errors
+        // to surface immediately.
+        const wrappedGenerator = new stream_js_1.AsyncGeneratorWithSetup(this._streamIterator(input, options));
+        await wrappedGenerator.setup;
+        return stream_js_1.IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
     _separateRunnableConfigFromCallOptions(options = {}) {
         const runnableConfig = (0, config_js_1.ensureConfig)({
@@ -1043,7 +1047,9 @@ class RunnableMap extends Runnable {
         async function* generator() {
             yield input;
         }
-        return stream_js_1.IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new stream_js_1.AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return stream_js_1.IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 exports.RunnableMap = RunnableMap;
@@ -1132,7 +1138,9 @@ class RunnableLambda extends Runnable {
         async function* generator() {
             yield input;
         }
-        return stream_js_1.IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new stream_js_1.AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return stream_js_1.IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 exports.RunnableLambda = RunnableLambda;
@@ -1329,7 +1337,9 @@ class RunnableAssign extends Runnable {
         async function* generator() {
             yield input;
         }
-        return stream_js_1.IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new stream_js_1.AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return stream_js_1.IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 exports.RunnableAssign = RunnableAssign;
@@ -1395,7 +1405,9 @@ class RunnablePick extends Runnable {
         async function* generator() {
             yield input;
         }
-        return stream_js_1.IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new stream_js_1.AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return stream_js_1.IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 exports.RunnablePick = RunnablePick;

package/dist/runnables/base.js

@@ -2,7 +2,7 @@ import pRetry from "p-retry";
 import { CallbackManager, } from "../callbacks/manager.js";
 import { LogStreamCallbackHandler, RunLogPatch, } from "../tracers/log_stream.js";
 import { Serializable } from "../load/serializable.js";
-import { IterableReadableStream, concat, atee, pipeGeneratorWithSetup, } from "../utils/stream.js";
+import { IterableReadableStream, concat, atee, pipeGeneratorWithSetup, AsyncGeneratorWithSetup, } from "../utils/stream.js";
 import { DEFAULT_RECURSION_LIMIT, ensureConfig, getCallbackManagerForConfig, mergeConfigs, patchConfig, } from "./config.js";
 import { AsyncCaller } from "../utils/async_caller.js";
 import { RootListenersTracer } from "../tracers/root_listener.js";
@@ -148,7 +148,11 @@ export class Runnable extends Serializable {
      * @returns A readable stream that is also an iterable.
      */
     async stream(input, options) {
-        return IterableReadableStream.fromAsyncGenerator(this._streamIterator(input, options));
+        // Buffer the first streamed chunk to allow for initial errors
+        // to surface immediately.
+        const wrappedGenerator = new AsyncGeneratorWithSetup(this._streamIterator(input, options));
+        await wrappedGenerator.setup;
+        return IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
     _separateRunnableConfigFromCallOptions(options = {}) {
         const runnableConfig = ensureConfig({
@@ -1032,7 +1036,9 @@ export class RunnableMap extends Runnable {
         async function* generator() {
             yield input;
         }
-        return IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 /**
@@ -1120,7 +1126,9 @@ export class RunnableLambda extends Runnable {
         async function* generator() {
             yield input;
         }
-        return IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 export class RunnableParallel extends RunnableMap {
@@ -1313,7 +1321,9 @@ export class RunnableAssign extends Runnable {
         async function* generator() {
             yield input;
         }
-        return IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }
 /**
@@ -1378,6 +1388,8 @@ export class RunnablePick extends Runnable {
         async function* generator() {
             yield input;
         }
-        return IterableReadableStream.fromAsyncGenerator(this.transform(generator(), options));
+        const wrappedGenerator = new AsyncGeneratorWithSetup(this.transform(generator(), options));
+        await wrappedGenerator.setup;
+        return IterableReadableStream.fromAsyncGenerator(wrappedGenerator);
     }
 }

package/dist/utils/stream.cjs

@@ -196,7 +196,12 @@ class AsyncGeneratorWithSetup {
         // to each generator is available.
         this.setup = new Promise((resolve, reject) => {
             this.firstResult = generator.next();
-            this.firstResult.then(startSetup).then(resolve, reject);
+            if (startSetup) {
+                this.firstResult.then(startSetup).then(resolve, reject);
+            }
+            else {
+                this.firstResult.then((_result) => resolve(undefined), reject);
+            }
         });
     }
     async next(...args) {

package/dist/utils/stream.d.ts

@@ -17,7 +17,7 @@ export declare class AsyncGeneratorWithSetup<S = unknown, T = unknown, TReturn =
     setup: Promise<S>;
     private firstResult;
     private firstResultUsed;
-    constructor(generator: AsyncGenerator<T>, startSetup: () => Promise<S>);
+    constructor(generator: AsyncGenerator<T>, startSetup?: () => Promise<S>);
     next(...args: [] | [TNext]): Promise<IteratorResult<T>>;
     return(value: TReturn | PromiseLike<TReturn>): Promise<IteratorResult<T>>;
     throw(e: Error): Promise<IteratorResult<T>>;

package/dist/utils/stream.js

@@ -190,7 +190,12 @@ export class AsyncGeneratorWithSetup {
         // to each generator is available.
         this.setup = new Promise((resolve, reject) => {
             this.firstResult = generator.next();
-            this.firstResult.then(startSetup).then(resolve, reject);
+            if (startSetup) {
+                this.firstResult.then(startSetup).then(resolve, reject);
+            }
+            else {
+                this.firstResult.then((_result) => resolve(undefined), reject);
+            }
         });
     }
     async next(...args) {

package/dist/utils/testing/index.cjs

@@ -106,18 +106,31 @@ class FakeStreamingLLM extends llms_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "thrownErrorString", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.sleep = fields.sleep ?? this.sleep;
         this.responses = fields.responses;
+        this.thrownErrorString = fields.thrownErrorString;
     }
     _llmType() {
         return "fake";
     }
     async _call(prompt) {
+        if (this.thrownErrorString) {
+            throw new Error(this.thrownErrorString);
+        }
         const response = this.responses?.[0];
         this.responses = this.responses?.slice(1);
         return response ?? prompt;
     }
     async *_streamResponseChunks(input) {
+        if (this.thrownErrorString) {
+            throw new Error(this.thrownErrorString);
+        }
         const response = this.responses?.[0];
         this.responses = this.responses?.slice(1);
         for (const c of response ?? input) {

package/dist/utils/testing/index.d.ts

@@ -42,9 +42,11 @@ export declare class FakeLLM extends LLM {
 export declare class FakeStreamingLLM extends LLM {
     sleep?: number;
     responses?: string[];
+    thrownErrorString?: string;
     constructor(fields: {
         sleep?: number;
         responses?: string[];
+        thrownErrorString?: string;
     } & BaseLLMParams);
     _llmType(): string;
     _call(prompt: string): Promise<string>;

package/dist/utils/testing/index.js

@@ -100,18 +100,31 @@ export class FakeStreamingLLM extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "thrownErrorString", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.sleep = fields.sleep ?? this.sleep;
         this.responses = fields.responses;
+        this.thrownErrorString = fields.thrownErrorString;
     }
     _llmType() {
         return "fake";
     }
     async _call(prompt) {
+        if (this.thrownErrorString) {
+            throw new Error(this.thrownErrorString);
+        }
         const response = this.responses?.[0];
         this.responses = this.responses?.slice(1);
         return response ?? prompt;
     }
     async *_streamResponseChunks(input) {
+        if (this.thrownErrorString) {
+            throw new Error(this.thrownErrorString);
+        }
         const response = this.responses?.[0];
         this.responses = this.responses?.slice(1);
         for (const c of response ?? input) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.1.16",
+  "version": "0.1.17-rc.0",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {
@@ -28,8 +28,8 @@
     "test": "NODE_OPTIONS=--experimental-vm-modules jest --testPathIgnorePatterns=\\.int\\.test.ts --testTimeout 30000 --maxWorkers=50%",
     "test:watch": "NODE_OPTIONS=--experimental-vm-modules jest --watch --testPathIgnorePatterns=\\.int\\.test.ts",
     "test:single": "NODE_OPTIONS=--experimental-vm-modules yarn run jest --config jest.config.cjs --testTimeout 100000",
-    "format": "prettier --write \"src\"",
-    "format:check": "prettier --check \"src\""
+    "format": "prettier --write \"src\" \"scripts\"",
+    "format:check": "prettier --check \"src\" \"scripts\""
   },
   "author": "LangChain",
   "license": "MIT",