@langchain/core 0.1.4 → 0.1.6

package/dist/runnables/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RunnableWithMessageHistory = exports.RunnableBranch = exports.RouterRunnable = exports.RunnableAssign = exports.RunnablePassthrough = exports._coerceToRunnable = exports.RunnableWithFallbacks = exports.RunnableLambda = exports.RunnableParallel = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = void 0;
+exports.RunnableWithMessageHistory = exports.RunnableBranch = exports.RouterRunnable = exports.RunnablePassthrough = exports.getCallbackManagerForConfig = exports._coerceToRunnable = exports.RunnablePick = exports.RunnableAssign = exports.RunnableWithFallbacks = exports.RunnableLambda = exports.RunnableParallel = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = void 0;
 var base_js_1 = require("./base.cjs");
 Object.defineProperty(exports, "Runnable", { enumerable: true, get: function () { return base_js_1.Runnable; } });
 Object.defineProperty(exports, "RunnableBinding", { enumerable: true, get: function () { return base_js_1.RunnableBinding; } });
@@ -11,10 +11,13 @@ Object.defineProperty(exports, "RunnableMap", { enumerable: true, get: function
 Object.defineProperty(exports, "RunnableParallel", { enumerable: true, get: function () { return base_js_1.RunnableParallel; } });
 Object.defineProperty(exports, "RunnableLambda", { enumerable: true, get: function () { return base_js_1.RunnableLambda; } });
 Object.defineProperty(exports, "RunnableWithFallbacks", { enumerable: true, get: function () { return base_js_1.RunnableWithFallbacks; } });
+Object.defineProperty(exports, "RunnableAssign", { enumerable: true, get: function () { return base_js_1.RunnableAssign; } });
+Object.defineProperty(exports, "RunnablePick", { enumerable: true, get: function () { return base_js_1.RunnablePick; } });
 Object.defineProperty(exports, "_coerceToRunnable", { enumerable: true, get: function () { return base_js_1._coerceToRunnable; } });
+var config_js_1 = require("./config.cjs");
+Object.defineProperty(exports, "getCallbackManagerForConfig", { enumerable: true, get: function () { return config_js_1.getCallbackManagerForConfig; } });
 var passthrough_js_1 = require("./passthrough.cjs");
 Object.defineProperty(exports, "RunnablePassthrough", { enumerable: true, get: function () { return passthrough_js_1.RunnablePassthrough; } });
-Object.defineProperty(exports, "RunnableAssign", { enumerable: true, get: function () { return passthrough_js_1.RunnableAssign; } });
 var router_js_1 = require("./router.cjs");
 Object.defineProperty(exports, "RouterRunnable", { enumerable: true, get: function () { return router_js_1.RouterRunnable; } });
 var branch_js_1 = require("./branch.cjs");

package/dist/runnables/index.d.ts

@@ -1,6 +1,6 @@
-export { type RunnableFunc, type RunnableLike, type RunnableBatchOptions, type RunnableRetryFailedAttemptHandler, Runnable, type RunnableInterface, type RunnableBindingArgs, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableParallel, RunnableLambda, RunnableWithFallbacks, _coerceToRunnable, } from "./base.js";
-export type { RunnableConfig, getCallbackMangerForConfig } from "./config.js";
-export { RunnablePassthrough, RunnableAssign } from "./passthrough.js";
+export { type RunnableFunc, type RunnableLike, type RunnableBatchOptions, type RunnableRetryFailedAttemptHandler, Runnable, type RunnableInterface, type RunnableBindingArgs, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableParallel, RunnableLambda, RunnableWithFallbacks, RunnableAssign, RunnablePick, _coerceToRunnable, } from "./base.js";
+export { type RunnableConfig, getCallbackManagerForConfig } from "./config.js";
+export { RunnablePassthrough } from "./passthrough.js";
 export { type RouterInput, RouterRunnable } from "./router.js";
 export { RunnableBranch, type Branch, type BranchLike } from "./branch.js";
 export { type RunnableWithMessageHistoryInputs, RunnableWithMessageHistory, } from "./history.js";

package/dist/runnables/index.js

@@ -1,5 +1,6 @@
-export { Runnable, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableParallel, RunnableLambda, RunnableWithFallbacks, _coerceToRunnable, } from "./base.js";
-export { RunnablePassthrough, RunnableAssign } from "./passthrough.js";
+export { Runnable, RunnableBinding, RunnableEach, RunnableRetry, RunnableSequence, RunnableMap, RunnableParallel, RunnableLambda, RunnableWithFallbacks, RunnableAssign, RunnablePick, _coerceToRunnable, } from "./base.js";
+export { getCallbackManagerForConfig } from "./config.js";
+export { RunnablePassthrough } from "./passthrough.js";
 export { RouterRunnable } from "./router.js";
 export { RunnableBranch } from "./branch.js";
 export { RunnableWithMessageHistory, } from "./history.js";

package/dist/runnables/passthrough.cjs

@@ -1,36 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RunnablePassthrough = exports.RunnableAssign = void 0;
+exports.RunnablePassthrough = void 0;
 const base_js_1 = require("./base.cjs");
-/**
- * A runnable that assigns key-value pairs to inputs of type `Record<string, unknown>`.
- */
-class RunnableAssign extends base_js_1.Runnable {
-    constructor(mapper) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain_core", "runnables"]
-        });
-        Object.defineProperty(this, "mapper", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.mapper = mapper;
-    }
-    async invoke(input, options) {
-        const mapperResult = await this.mapper.invoke(input, options);
-        return {
-            ...input,
-            ...mapperResult,
-        };
-    }
-}
-exports.RunnableAssign = RunnableAssign;
 /**
  * A runnable to passthrough inputs unchanged or with additional keys.
  *
@@ -79,6 +50,9 @@ class RunnablePassthrough extends base_js_1.Runnable {
     async invoke(input, options) {
         return this._callWithConfig((input) => Promise.resolve(input), input, options);
     }
+    transform(generator, options) {
+        return this._transformStreamWithConfig(generator, (input) => input, options);
+    }
     /**
      * A runnable that assigns key-value pairs to the input.
      *
@@ -110,7 +84,7 @@ class RunnablePassthrough extends base_js_1.Runnable {
     static assign(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     mapping) {
-        return new RunnableAssign(new base_js_1.RunnableMap({ steps: mapping }));
+        return new base_js_1.RunnableAssign(new base_js_1.RunnableMap({ steps: mapping }));
     }
 }
 exports.RunnablePassthrough = RunnablePassthrough;

package/dist/runnables/passthrough.d.ts

@@ -1,14 +1,5 @@
-import { Runnable, RunnableLike, RunnableMap } from "./base.js";
+import { Runnable, RunnableAssign, RunnableMapLike } from "./base.js";
 import type { RunnableConfig } from "./config.js";
-/**
- * A runnable that assigns key-value pairs to inputs of type `Record<string, unknown>`.
- */
-export declare class RunnableAssign<RunInput extends Record<string, any> = any, RunOutput extends Record<string, any> = any, CallOptions extends RunnableConfig = RunnableConfig> extends Runnable<RunInput, RunOutput> {
-    lc_namespace: string[];
-    mapper: RunnableMap<RunInput>;
-    constructor(mapper: RunnableMap<RunInput>);
-    invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
-}
 /**
  * A runnable to passthrough inputs unchanged or with additional keys.
  *
@@ -40,6 +31,7 @@ export declare class RunnablePassthrough<RunInput> extends Runnable<RunInput, Ru
     lc_namespace: string[];
     lc_serializable: boolean;
     invoke(input: RunInput, options?: Partial<RunnableConfig>): Promise<RunInput>;
+    transform(generator: AsyncGenerator<RunInput>, options: Partial<RunnableConfig>): AsyncGenerator<RunInput>;
     /**
      * A runnable that assigns key-value pairs to the input.
      *
@@ -68,5 +60,5 @@ export declare class RunnablePassthrough<RunInput> extends Runnable<RunInput, Ru
      * });
      * ```
      */
-    static assign(mapping: Record<string, RunnableLike<Record<string, unknown>, any>>): RunnableAssign<Record<string, unknown>, Record<string, unknown>>;
+    static assign(mapping: RunnableMapLike<Record<string, unknown>, Record<string, unknown>>): RunnableAssign<Record<string, unknown>, Record<string, unknown>>;
 }

package/dist/runnables/passthrough.js

@@ -1,32 +1,4 @@
-import { Runnable, RunnableMap } from "./base.js";
-/**
- * A runnable that assigns key-value pairs to inputs of type `Record<string, unknown>`.
- */
-export class RunnableAssign extends Runnable {
-    constructor(mapper) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain_core", "runnables"]
-        });
-        Object.defineProperty(this, "mapper", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.mapper = mapper;
-    }
-    async invoke(input, options) {
-        const mapperResult = await this.mapper.invoke(input, options);
-        return {
-            ...input,
-            ...mapperResult,
-        };
-    }
-}
+import { Runnable, RunnableAssign, RunnableMap, } from "./base.js";
 /**
  * A runnable to passthrough inputs unchanged or with additional keys.
  *
@@ -75,6 +47,9 @@ export class RunnablePassthrough extends Runnable {
     async invoke(input, options) {
         return this._callWithConfig((input) => Promise.resolve(input), input, options);
     }
+    transform(generator, options) {
+        return this._transformStreamWithConfig(generator, (input) => input, options);
+    }
     /**
      * A runnable that assigns key-value pairs to the input.
      *

package/dist/utils/stream.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.IterableReadableStream = void 0;
+exports.AsyncGeneratorWithSetup = exports.concat = exports.atee = exports.IterableReadableStream = void 0;
 /*
  * Support async iterator syntax for ReadableStreams in all environments.
  * Source: https://github.com/MattiasBuelens/web-streams-polyfill/pull/122#issuecomment-1627354490
@@ -98,3 +98,115 @@ class IterableReadableStream extends ReadableStream {
     }
 }
 exports.IterableReadableStream = IterableReadableStream;
+function atee(iter, length = 2) {
+    const buffers = Array.from({ length }, () => []);
+    return buffers.map(async function* makeIter(buffer) {
+        while (true) {
+            if (buffer.length === 0) {
+                const result = await iter.next();
+                for (const buffer of buffers) {
+                    buffer.push(result);
+                }
+            }
+            else if (buffer[0].done) {
+                return;
+            }
+            else {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                yield buffer.shift().value;
+            }
+        }
+    });
+}
+exports.atee = atee;
+function concat(first, second) {
+    if (Array.isArray(first) && Array.isArray(second)) {
+        return first.concat(second);
+    }
+    else if (typeof first === "string" && typeof second === "string") {
+        return (first + second);
+    }
+    else if (typeof first === "number" && typeof second === "number") {
+        return (first + second);
+    }
+    else if (
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    "concat" in first &&
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        typeof first.concat === "function") {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return first.concat(second);
+    }
+    else if (typeof first === "object" && typeof second === "object") {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const chunk = { ...first };
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        for (const [key, value] of Object.entries(second)) {
+            if (key in chunk) {
+                chunk[key] = concat(chunk[key], value);
+            }
+            else {
+                chunk[key] = value;
+            }
+        }
+        return chunk;
+    }
+    else {
+        throw new Error(`Cannot concat ${typeof first} and ${typeof second}`);
+    }
+}
+exports.concat = concat;
+class AsyncGeneratorWithSetup {
+    constructor(generator, startSetup) {
+        Object.defineProperty(this, "generator", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "setup", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "firstResult", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "firstResultUsed", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.generator = generator;
+        // setup is a promise that resolves only after the first iterator value
+        // is available. this is useful when setup of several piped generators
+        // needs to happen in logical order, ie. in the order in which input to
+        // to each generator is available.
+        this.setup = new Promise((resolve, reject) => {
+            this.firstResult = generator.next();
+            this.firstResult.then(startSetup).then(resolve, reject);
+        });
+    }
+    async next(...args) {
+        if (!this.firstResultUsed) {
+            this.firstResultUsed = true;
+            return this.firstResult;
+        }
+        return this.generator.next(...args);
+    }
+    async return(value) {
+        return this.generator.return(value);
+    }
+    async throw(e) {
+        return this.generator.throw(e);
+    }
+    [Symbol.asyncIterator]() {
+        return this;
+    }
+}
+exports.AsyncGeneratorWithSetup = AsyncGeneratorWithSetup;

package/dist/utils/stream.d.ts

@@ -16,3 +16,16 @@ export declare class IterableReadableStream<T> extends ReadableStream<T> impleme
     static fromReadableStream<T>(stream: ReadableStream<T>): IterableReadableStream<T>;
     static fromAsyncGenerator<T>(generator: AsyncGenerator<T>): IterableReadableStream<T>;
 }
+export declare function atee<T>(iter: AsyncGenerator<T>, length?: number): AsyncGenerator<T>[];
+export declare function concat<T extends Array<any> | string | number | Record<string, any> | any>(first: T, second: T): T;
+export declare class AsyncGeneratorWithSetup<S = unknown, T = unknown, TReturn = unknown, TNext = unknown> implements AsyncGenerator<T, TReturn, TNext> {
+    private generator;
+    setup: Promise<S>;
+    private firstResult;
+    private firstResultUsed;
+    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>>;
+    [Symbol.asyncIterator](): this;
+}

package/dist/utils/stream.js

@@ -94,3 +94,112 @@ export class IterableReadableStream extends ReadableStream {
         });
     }
 }
+export function atee(iter, length = 2) {
+    const buffers = Array.from({ length }, () => []);
+    return buffers.map(async function* makeIter(buffer) {
+        while (true) {
+            if (buffer.length === 0) {
+                const result = await iter.next();
+                for (const buffer of buffers) {
+                    buffer.push(result);
+                }
+            }
+            else if (buffer[0].done) {
+                return;
+            }
+            else {
+                // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                yield buffer.shift().value;
+            }
+        }
+    });
+}
+export function concat(first, second) {
+    if (Array.isArray(first) && Array.isArray(second)) {
+        return first.concat(second);
+    }
+    else if (typeof first === "string" && typeof second === "string") {
+        return (first + second);
+    }
+    else if (typeof first === "number" && typeof second === "number") {
+        return (first + second);
+    }
+    else if (
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    "concat" in first &&
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        typeof first.concat === "function") {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return first.concat(second);
+    }
+    else if (typeof first === "object" && typeof second === "object") {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const chunk = { ...first };
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        for (const [key, value] of Object.entries(second)) {
+            if (key in chunk) {
+                chunk[key] = concat(chunk[key], value);
+            }
+            else {
+                chunk[key] = value;
+            }
+        }
+        return chunk;
+    }
+    else {
+        throw new Error(`Cannot concat ${typeof first} and ${typeof second}`);
+    }
+}
+export class AsyncGeneratorWithSetup {
+    constructor(generator, startSetup) {
+        Object.defineProperty(this, "generator", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "setup", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "firstResult", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "firstResultUsed", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.generator = generator;
+        // setup is a promise that resolves only after the first iterator value
+        // is available. this is useful when setup of several piped generators
+        // needs to happen in logical order, ie. in the order in which input to
+        // to each generator is available.
+        this.setup = new Promise((resolve, reject) => {
+            this.firstResult = generator.next();
+            this.firstResult.then(startSetup).then(resolve, reject);
+        });
+    }
+    async next(...args) {
+        if (!this.firstResultUsed) {
+            this.firstResultUsed = true;
+            return this.firstResult;
+        }
+        return this.generator.next(...args);
+    }
+    async return(value) {
+        return this.generator.return(value);
+    }
+    async throw(e) {
+        return this.generator.throw(e);
+    }
+    [Symbol.asyncIterator]() {
+        return this;
+    }
+}

package/dist/utils/testing/index.cjs

@@ -91,6 +91,16 @@ class FakeLLM extends llms_js_1.LLM {
 }
 exports.FakeLLM = FakeLLM;
 class FakeStreamingLLM extends llms_js_1.LLM {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "sleep", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 50
+        });
+        this.sleep = fields.sleep ?? this.sleep;
+    }
     _llmType() {
         return "fake";
     }
@@ -99,7 +109,7 @@ class FakeStreamingLLM extends llms_js_1.LLM {
     }
     async *_streamResponseChunks(input) {
         for (const c of input) {
-            await new Promise((resolve) => setTimeout(resolve, 50));
+            await new Promise((resolve) => setTimeout(resolve, this.sleep));
             yield { text: c, generationInfo: {} };
         }
     }
@@ -328,5 +338,8 @@ class FakeListChatMessageHistory extends chat_history_js_1.BaseListChatMessageHi
     async addMessage(message) {
         this.messages.push(message);
     }
+    async getMessages() {
+        return this.messages;
+    }
 }
 exports.FakeListChatMessageHistory = FakeListChatMessageHistory;

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

@@ -36,6 +36,10 @@ export declare class FakeLLM extends LLM {
     _call(prompt: string, _options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
 }
 export declare class FakeStreamingLLM extends LLM {
+    sleep?: number;
+    constructor(fields: {
+        sleep?: number;
+    } & BaseLLMParams);
     _llmType(): string;
     _call(prompt: string): Promise<string>;
     _streamResponseChunks(input: string): AsyncGenerator<GenerationChunk, void, unknown>;
@@ -117,4 +121,5 @@ export declare class FakeListChatMessageHistory extends BaseListChatMessageHisto
     messages: Array<BaseMessage>;
     constructor();
     addMessage(message: BaseMessage): Promise<void>;
+    getMessages(): Promise<BaseMessage[]>;
 }

package/dist/utils/testing/index.js

@@ -85,6 +85,16 @@ export class FakeLLM extends LLM {
     }
 }
 export class FakeStreamingLLM extends LLM {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "sleep", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 50
+        });
+        this.sleep = fields.sleep ?? this.sleep;
+    }
     _llmType() {
         return "fake";
     }
@@ -93,7 +103,7 @@ export class FakeStreamingLLM extends LLM {
     }
     async *_streamResponseChunks(input) {
         for (const c of input) {
-            await new Promise((resolve) => setTimeout(resolve, 50));
+            await new Promise((resolve) => setTimeout(resolve, this.sleep));
             yield { text: c, generationInfo: {} };
         }
     }
@@ -317,4 +327,7 @@ export class FakeListChatMessageHistory extends BaseListChatMessageHistory {
     async addMessage(message) {
         this.messages.push(message);
     }
+    async getMessages() {
+        return this.messages;
+    }
 }

package/package.json

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