@langchain/core 0.2.19 → 0.2.20

package/dist/runnables/config.cjs

@@ -24,6 +24,31 @@ function mergeConfigs(...configs) {
             else if (key === "configurable") {
                 copy[key] = { ...copy[key], ...options[key] };
             }
+            else if (key === "timeout") {
+                if (copy.timeout === undefined) {
+                    copy.timeout = options.timeout;
+                }
+                else if (options.timeout !== undefined) {
+                    copy.timeout = Math.min(copy.timeout, options.timeout);
+                }
+            }
+            else if (key === "signal") {
+                if (copy.signal === undefined) {
+                    copy.signal = options.signal;
+                }
+                else if (options.signal !== undefined) {
+                    if ("any" in AbortSignal) {
+                        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                        copy.signal = AbortSignal.any([
+                            copy.signal,
+                            options.signal,
+                        ]);
+                    }
+                    else {
+                        copy.signal = options.signal;
+                    }
+                }
+            }
             else if (key === "callbacks") {
                 const baseCallbacks = copy.callbacks;
                 const providedCallbacks = options.callbacks;
@@ -130,6 +155,22 @@ function ensureConfig(config) {
             }
         }
     }
+    if (empty.timeout !== undefined) {
+        if (empty.timeout <= 0) {
+            throw new Error("Timeout must be a positive number");
+        }
+        const timeoutSignal = AbortSignal.timeout(empty.timeout);
+        if (empty.signal !== undefined) {
+            if ("any" in AbortSignal) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                empty.signal = AbortSignal.any([empty.signal, timeoutSignal]);
+            }
+        }
+        else {
+            empty.signal = timeoutSignal;
+        }
+        delete empty.timeout;
+    }
     return empty;
 }
 exports.ensureConfig = ensureConfig;

package/dist/runnables/config.js

@@ -20,6 +20,31 @@ export function mergeConfigs(...configs) {
             else if (key === "configurable") {
                 copy[key] = { ...copy[key], ...options[key] };
             }
+            else if (key === "timeout") {
+                if (copy.timeout === undefined) {
+                    copy.timeout = options.timeout;
+                }
+                else if (options.timeout !== undefined) {
+                    copy.timeout = Math.min(copy.timeout, options.timeout);
+                }
+            }
+            else if (key === "signal") {
+                if (copy.signal === undefined) {
+                    copy.signal = options.signal;
+                }
+                else if (options.signal !== undefined) {
+                    if ("any" in AbortSignal) {
+                        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                        copy.signal = AbortSignal.any([
+                            copy.signal,
+                            options.signal,
+                        ]);
+                    }
+                    else {
+                        copy.signal = options.signal;
+                    }
+                }
+            }
             else if (key === "callbacks") {
                 const baseCallbacks = copy.callbacks;
                 const providedCallbacks = options.callbacks;
@@ -125,6 +150,22 @@ export function ensureConfig(config) {
             }
         }
     }
+    if (empty.timeout !== undefined) {
+        if (empty.timeout <= 0) {
+            throw new Error("Timeout must be a positive number");
+        }
+        const timeoutSignal = AbortSignal.timeout(empty.timeout);
+        if (empty.signal !== undefined) {
+            if ("any" in AbortSignal) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                empty.signal = AbortSignal.any([empty.signal, timeoutSignal]);
+            }
+        }
+        else {
+            empty.signal = timeoutSignal;
+        }
+        delete empty.timeout;
+    }
     return empty;
 }
 /**

package/dist/runnables/remote.cjs

@@ -186,9 +186,10 @@ function deserialize(str) {
     const obj = JSON.parse(str);
     return revive(obj);
 }
-function removeCallbacks(options) {
+function removeCallbacksAndSignal(options) {
     const rest = { ...options };
     delete rest.callbacks;
+    delete rest.signal;
     return rest;
 }
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -253,7 +254,7 @@ class RemoteRunnable extends base_js_1.Runnable {
         this.url = url.replace(/\/$/, ""); // remove trailing slash
         this.options = options;
     }
-    async post(path, body) {
+    async post(path, body, signal) {
         return fetch(`${this.url}${path}`, {
             method: "POST",
             body: JSON.stringify(serialize(body)),
@@ -261,16 +262,16 @@ class RemoteRunnable extends base_js_1.Runnable {
                 "Content-Type": "application/json",
                 ...this.options?.headers,
             },
-            signal: AbortSignal.timeout(this.options?.timeout ?? 60000),
+            signal: signal ?? AbortSignal.timeout(this.options?.timeout ?? 60000),
         });
     }
     async _invoke(input, options, _) {
         const [config, kwargs] = this._separateRunnableConfigFromCallOptions(options);
         const response = await this.post("/invoke", {
             input,
-            config: removeCallbacks(config),
+            config: removeCallbacksAndSignal(config),
             kwargs: kwargs ?? {},
-        });
+        }, config.signal);
         if (!response.ok) {
             throw new Error(`${response.status} Error: ${await response.text()}`);
         }
@@ -291,10 +292,10 @@ class RemoteRunnable extends base_js_1.Runnable {
         const response = await this.post("/batch", {
             inputs,
             config: (configs ?? [])
-                .map(removeCallbacks)
+                .map(removeCallbacksAndSignal)
                 .map((config) => ({ ...config, ...batchOptions })),
             kwargs,
-        });
+        }, options?.[0]?.signal);
         if (!response.ok) {
             throw new Error(`${response.status} Error: ${await response.text()}`);
         }
@@ -319,9 +320,9 @@ class RemoteRunnable extends base_js_1.Runnable {
         try {
             const response = await this.post("/stream", {
                 input,
-                config: removeCallbacks(config),
+                config: removeCallbacksAndSignal(config),
                 kwargs,
-            });
+            }, config.signal);
             if (!response.ok) {
                 const json = await response.json();
                 const error = new Error(`RemoteRunnable call failed with status code ${response.status}: ${json.message}`);
@@ -378,11 +379,11 @@ class RemoteRunnable extends base_js_1.Runnable {
         try {
             const response = await this.post("/stream_log", {
                 input,
-                config: removeCallbacks(config),
+                config: removeCallbacksAndSignal(config),
                 kwargs,
                 ...camelCaseStreamOptions,
                 diff: false,
-            });
+            }, config.signal);
             const { body, ok } = response;
             if (!ok) {
                 throw new Error(`${response.status} Error: ${await response.text()}`);
@@ -430,11 +431,11 @@ class RemoteRunnable extends base_js_1.Runnable {
             try {
                 const response = await outerThis.post("/stream_events", {
                     input,
-                    config: removeCallbacks(config),
+                    config: removeCallbacksAndSignal(config),
                     kwargs,
                     ...camelCaseStreamOptions,
                     diff: false,
-                });
+                }, config.signal);
                 const { body, ok } = response;
                 if (!ok) {
                     throw new Error(`${response.status} Error: ${await response.text()}`);

package/dist/runnables/remote.js

@@ -183,9 +183,10 @@ function deserialize(str) {
     const obj = JSON.parse(str);
     return revive(obj);
 }
-function removeCallbacks(options) {
+function removeCallbacksAndSignal(options) {
     const rest = { ...options };
     delete rest.callbacks;
+    delete rest.signal;
     return rest;
 }
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -250,7 +251,7 @@ export class RemoteRunnable extends Runnable {
         this.url = url.replace(/\/$/, ""); // remove trailing slash
         this.options = options;
     }
-    async post(path, body) {
+    async post(path, body, signal) {
         return fetch(`${this.url}${path}`, {
             method: "POST",
             body: JSON.stringify(serialize(body)),
@@ -258,16 +259,16 @@ export class RemoteRunnable extends Runnable {
                 "Content-Type": "application/json",
                 ...this.options?.headers,
             },
-            signal: AbortSignal.timeout(this.options?.timeout ?? 60000),
+            signal: signal ?? AbortSignal.timeout(this.options?.timeout ?? 60000),
         });
     }
     async _invoke(input, options, _) {
         const [config, kwargs] = this._separateRunnableConfigFromCallOptions(options);
         const response = await this.post("/invoke", {
             input,
-            config: removeCallbacks(config),
+            config: removeCallbacksAndSignal(config),
             kwargs: kwargs ?? {},
-        });
+        }, config.signal);
         if (!response.ok) {
             throw new Error(`${response.status} Error: ${await response.text()}`);
         }
@@ -288,10 +289,10 @@ export class RemoteRunnable extends Runnable {
         const response = await this.post("/batch", {
             inputs,
             config: (configs ?? [])
-                .map(removeCallbacks)
+                .map(removeCallbacksAndSignal)
                 .map((config) => ({ ...config, ...batchOptions })),
             kwargs,
-        });
+        }, options?.[0]?.signal);
         if (!response.ok) {
             throw new Error(`${response.status} Error: ${await response.text()}`);
         }
@@ -316,9 +317,9 @@ export class RemoteRunnable extends Runnable {
         try {
             const response = await this.post("/stream", {
                 input,
-                config: removeCallbacks(config),
+                config: removeCallbacksAndSignal(config),
                 kwargs,
-            });
+            }, config.signal);
             if (!response.ok) {
                 const json = await response.json();
                 const error = new Error(`RemoteRunnable call failed with status code ${response.status}: ${json.message}`);
@@ -375,11 +376,11 @@ export class RemoteRunnable extends Runnable {
         try {
             const response = await this.post("/stream_log", {
                 input,
-                config: removeCallbacks(config),
+                config: removeCallbacksAndSignal(config),
                 kwargs,
                 ...camelCaseStreamOptions,
                 diff: false,
-            });
+            }, config.signal);
             const { body, ok } = response;
             if (!ok) {
                 throw new Error(`${response.status} Error: ${await response.text()}`);
@@ -427,11 +428,11 @@ export class RemoteRunnable extends Runnable {
             try {
                 const response = await outerThis.post("/stream_events", {
                     input,
-                    config: removeCallbacks(config),
+                    config: removeCallbacksAndSignal(config),
                     kwargs,
                     ...camelCaseStreamOptions,
                     diff: false,
-                });
+                }, config.signal);
                 const { body, ok } = response;
                 if (!ok) {
                     throw new Error(`${response.status} Error: ${await response.text()}`);

package/dist/runnables/types.d.ts

@@ -53,4 +53,14 @@ export interface RunnableConfig extends BaseCallbackConfig {
     recursionLimit?: number;
     /** Maximum number of parallel calls to make. */
     maxConcurrency?: number;
+    /**
+     * Timeout for this call in milliseconds.
+     */
+    timeout?: number;
+    /**
+     * Abort signal for this call.
+     * If provided, the call will be aborted when the signal is aborted.
+     * @see https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal
+     */
+    signal?: AbortSignal;
 }

package/dist/utils/math.cjs

@@ -1,7 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.maximalMarginalRelevance = exports.euclideanDistance = exports.innerProduct = exports.cosineSimilarity = exports.normalize = exports.matrixFunc = void 0;
-const ml_distance_1 = require("ml-distance");
+const similarities_js_1 = require("./ml-distance/similarities.cjs");
+const distances_js_1 = require("./ml-distance/distances.cjs");
+const euclidean_js_1 = require("./ml-distance-euclidean/euclidean.cjs");
 /**
  * Apply a row-wise function between two matrices with the same number of columns.
  *
@@ -45,15 +47,15 @@ exports.normalize = normalize;
  * @returns {number[][] | [[]]} A matrix where each row represents the cosine similarity values between the corresponding rows of X and Y.
  */
 function cosineSimilarity(X, Y) {
-    return matrixFunc(X, Y, ml_distance_1.similarity.cosine);
+    return matrixFunc(X, Y, similarities_js_1.cosine);
 }
 exports.cosineSimilarity = cosineSimilarity;
 function innerProduct(X, Y) {
-    return matrixFunc(X, Y, ml_distance_1.distance.innerProduct);
+    return matrixFunc(X, Y, distances_js_1.innerProduct);
 }
 exports.innerProduct = innerProduct;
 function euclideanDistance(X, Y) {
-    return matrixFunc(X, Y, ml_distance_1.distance.euclidean);
+    return matrixFunc(X, Y, euclidean_js_1.euclidean);
 }
 exports.euclideanDistance = euclideanDistance;
 /**

package/dist/utils/math.js

@@ -1,4 +1,6 @@
-import { similarity as ml_distance_similarity, distance as ml_distance, } from "ml-distance";
+import { cosine } from "./ml-distance/similarities.js";
+import { innerProduct as innerProductDistance } from "./ml-distance/distances.js";
+import { euclidean } from "./ml-distance-euclidean/euclidean.js";
 /**
  * Apply a row-wise function between two matrices with the same number of columns.
  *
@@ -40,13 +42,13 @@ export function normalize(M, similarity = false) {
  * @returns {number[][] | [[]]} A matrix where each row represents the cosine similarity values between the corresponding rows of X and Y.
  */
 export function cosineSimilarity(X, Y) {
-    return matrixFunc(X, Y, ml_distance_similarity.cosine);
+    return matrixFunc(X, Y, cosine);
 }
 export function innerProduct(X, Y) {
-    return matrixFunc(X, Y, ml_distance.innerProduct);
+    return matrixFunc(X, Y, innerProductDistance);
 }
 export function euclideanDistance(X, Y) {
-    return matrixFunc(X, Y, ml_distance.euclidean);
+    return matrixFunc(X, Y, euclidean);
 }
 /**
  * This function implements the Maximal Marginal Relevance algorithm

package/dist/utils/ml-distance/distances.cjs

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.innerProduct = void 0;
+/**
+ *Returns the Inner Product similarity between vectors a and b
+ * @link [Inner Product Similarity algorithm](https://www.naun.org/main/NAUN/ijmmas/mmmas-49.pdf)
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+function innerProduct(a, b) {
+    let ans = 0;
+    for (let i = 0; i < a.length; i++) {
+        ans += a[i] * b[i];
+    }
+    return ans;
+}
+exports.innerProduct = innerProduct;

package/dist/utils/ml-distance/distances.d.ts

@@ -0,0 +1,8 @@
+/**
+ *Returns the Inner Product similarity between vectors a and b
+ * @link [Inner Product Similarity algorithm](https://www.naun.org/main/NAUN/ijmmas/mmmas-49.pdf)
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+export declare function innerProduct(a: number[], b: number[]): number;

package/dist/utils/ml-distance/distances.js

@@ -0,0 +1,14 @@
+/**
+ *Returns the Inner Product similarity between vectors a and b
+ * @link [Inner Product Similarity algorithm](https://www.naun.org/main/NAUN/ijmmas/mmmas-49.pdf)
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+export function innerProduct(a, b) {
+    let ans = 0;
+    for (let i = 0; i < a.length; i++) {
+        ans += a[i] * b[i];
+    }
+    return ans;
+}

package/dist/utils/ml-distance/similarities.cjs

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cosine = void 0;
+/**
+ * Returns the average of cosine distances between vectors a and b
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+function cosine(a, b) {
+    let p = 0;
+    let p2 = 0;
+    let q2 = 0;
+    for (let i = 0; i < a.length; i++) {
+        p += a[i] * b[i];
+        p2 += a[i] * a[i];
+        q2 += b[i] * b[i];
+    }
+    return p / (Math.sqrt(p2) * Math.sqrt(q2));
+}
+exports.cosine = cosine;

package/dist/utils/ml-distance/similarities.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Returns the average of cosine distances between vectors a and b
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+export declare function cosine(a: number[], b: number[]): number;

package/dist/utils/ml-distance/similarities.js

@@ -0,0 +1,17 @@
+/**
+ * Returns the average of cosine distances between vectors a and b
+ * @param a - first vector
+ * @param b - second vector
+ *
+ */
+export function cosine(a, b) {
+    let p = 0;
+    let p2 = 0;
+    let q2 = 0;
+    for (let i = 0; i < a.length; i++) {
+        p += a[i] * b[i];
+        p2 += a[i] * a[i];
+        q2 += b[i] * b[i];
+    }
+    return p / (Math.sqrt(p2) * Math.sqrt(q2));
+}

package/dist/utils/ml-distance-euclidean/euclidean.cjs

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.euclidean = exports.squaredEuclidean = void 0;
+function squaredEuclidean(p, q) {
+    let d = 0;
+    for (let i = 0; i < p.length; i++) {
+        d += (p[i] - q[i]) * (p[i] - q[i]);
+    }
+    return d;
+}
+exports.squaredEuclidean = squaredEuclidean;
+function euclidean(p, q) {
+    return Math.sqrt(squaredEuclidean(p, q));
+}
+exports.euclidean = euclidean;

package/dist/utils/ml-distance-euclidean/euclidean.d.ts

@@ -0,0 +1,2 @@
+export declare function squaredEuclidean(p: number[], q: number[]): number;
+export declare function euclidean(p: number[], q: number[]): number;

package/dist/utils/ml-distance-euclidean/euclidean.js

@@ -0,0 +1,10 @@
+export function squaredEuclidean(p, q) {
+    let d = 0;
+    for (let i = 0; i < p.length; i++) {
+        d += (p[i] - q[i]) * (p[i] - q[i]);
+    }
+    return d;
+}
+export function euclidean(p, q) {
+    return Math.sqrt(squaredEuclidean(p, q));
+}

package/dist/utils/signal.cjs

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.raceWithSignal = void 0;
+async function raceWithSignal(promise, signal) {
+    if (signal === undefined) {
+        return promise;
+    }
+    return Promise.race([
+        promise.catch((err) => {
+            if (!signal?.aborted) {
+                throw err;
+            }
+            else {
+                return undefined;
+            }
+        }),
+        new Promise((_, reject) => {
+            signal.addEventListener("abort", () => {
+                reject(new Error("Aborted"));
+            });
+            // Must be here inside the promise to avoid a race condition
+            if (signal.aborted) {
+                reject(new Error("Aborted"));
+            }
+        }),
+    ]);
+}
+exports.raceWithSignal = raceWithSignal;

package/dist/utils/signal.d.ts

@@ -0,0 +1 @@
+export declare function raceWithSignal<T>(promise: Promise<T>, signal?: AbortSignal): Promise<T>;

package/dist/utils/signal.js

@@ -0,0 +1,24 @@
+export async function raceWithSignal(promise, signal) {
+    if (signal === undefined) {
+        return promise;
+    }
+    return Promise.race([
+        promise.catch((err) => {
+            if (!signal?.aborted) {
+                throw err;
+            }
+            else {
+                return undefined;
+            }
+        }),
+        new Promise((_, reject) => {
+            signal.addEventListener("abort", () => {
+                reject(new Error("Aborted"));
+            });
+            // Must be here inside the promise to avoid a race condition
+            if (signal.aborted) {
+                reject(new Error("Aborted"));
+            }
+        }),
+    ]);
+}

package/dist/utils/stream.cjs

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.pipeGeneratorWithSetup = exports.AsyncGeneratorWithSetup = exports.concat = exports.atee = exports.IterableReadableStream = void 0;
 const index_js_1 = require("../singletons/index.cjs");
+const signal_js_1 = require("./signal.cjs");
 /*
  * Support async iterator syntax for ReadableStreams in all environments.
  * Source: https://github.com/MattiasBuelens/web-streams-polyfill/pull/122#issuecomment-1627354490
@@ -187,6 +188,12 @@ class AsyncGeneratorWithSetup {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "signal", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "firstResult", {
             enumerable: true,
             configurable: true,
@@ -201,6 +208,8 @@ class AsyncGeneratorWithSetup {
         });
         this.generator = params.generator;
         this.config = params.config;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        this.signal = params.signal ?? this.config?.signal;
         // 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
@@ -218,13 +227,18 @@ class AsyncGeneratorWithSetup {
         });
     }
     async next(...args) {
+        this.signal?.throwIfAborted();
         if (!this.firstResultUsed) {
             this.firstResultUsed = true;
             return this.firstResult;
         }
-        return index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(this.config, async () => {
-            return this.generator.next(...args);
-        }, true);
+        return index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(this.config, this.signal
+            ? async () => {
+                return (0, signal_js_1.raceWithSignal)(this.generator.next(...args), this.signal);
+            }
+            : async () => {
+                return this.generator.next(...args);
+            }, true);
     }
     async return(value) {
         return this.generator.return(value);
@@ -237,10 +251,11 @@ class AsyncGeneratorWithSetup {
     }
 }
 exports.AsyncGeneratorWithSetup = AsyncGeneratorWithSetup;
-async function pipeGeneratorWithSetup(to, generator, startSetup, ...args) {
+async function pipeGeneratorWithSetup(to, generator, startSetup, signal, ...args) {
     const gen = new AsyncGeneratorWithSetup({
         generator,
         startSetup,
+        signal,
     });
     const setup = await gen.setup;
     return { output: to(gen, setup, ...args), setup };

package/dist/utils/stream.d.ts

@@ -15,19 +15,21 @@ export declare class AsyncGeneratorWithSetup<S = unknown, T = unknown, TReturn =
     private generator;
     setup: Promise<S>;
     config?: unknown;
+    signal?: AbortSignal;
     private firstResult;
     private firstResultUsed;
     constructor(params: {
         generator: AsyncGenerator<T>;
         startSetup?: () => Promise<S>;
         config?: unknown;
+        signal?: AbortSignal;
     });
     next(...args: [] | [TNext]): Promise<IteratorResult<T>>;
     return(value: TReturn | PromiseLike<TReturn>): Promise<IteratorResult<T>>;
     throw(e: Error): Promise<IteratorResult<T>>;
     [Symbol.asyncIterator](): this;
 }
-export declare function pipeGeneratorWithSetup<S, A extends unknown[], T, TReturn, TNext, U, UReturn, UNext>(to: (g: AsyncGenerator<T, TReturn, TNext>, s: S, ...args: A) => AsyncGenerator<U, UReturn, UNext>, generator: AsyncGenerator<T, TReturn, TNext>, startSetup: () => Promise<S>, ...args: A): Promise<{
+export declare function pipeGeneratorWithSetup<S, A extends unknown[], T, TReturn, TNext, U, UReturn, UNext>(to: (g: AsyncGenerator<T, TReturn, TNext>, s: S, ...args: A) => AsyncGenerator<U, UReturn, UNext>, generator: AsyncGenerator<T, TReturn, TNext>, startSetup: () => Promise<S>, signal: AbortSignal | undefined, ...args: A): Promise<{
     output: AsyncGenerator<U, UReturn, UNext>;
     setup: Awaited<S>;
 }>;