@langchain/core 0.1.62 → 0.1.63

package/dist/language_models/base.cjs

@@ -21,6 +21,9 @@ const getModelNameForTiktoken = (modelName) => {
     if (modelName.startsWith("gpt-4-")) {
         return "gpt-4";
     }
+    if (modelName.startsWith("gpt-4o")) {
+        return "gpt-4o";
+    }
     return modelName;
 };
 exports.getModelNameForTiktoken = getModelNameForTiktoken;

package/dist/language_models/base.js

@@ -18,6 +18,9 @@ export const getModelNameForTiktoken = (modelName) => {
     if (modelName.startsWith("gpt-4-")) {
         return "gpt-4";
     }
+    if (modelName.startsWith("gpt-4o")) {
+        return "gpt-4o";
+    }
     return modelName;
 };
 export const getEmbeddingContextSize = (modelName) => {

package/dist/runnables/base.cjs

@@ -17,6 +17,7 @@ const utils_js_1 = require("./utils.cjs");
 const index_js_1 = require("../singletons/index.cjs");
 const graph_js_1 = require("./graph.cjs");
 const wrappers_js_1 = require("./wrappers.cjs");
+const iter_js_1 = require("./iter.cjs");
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function _coerceToDict(value, defaultKey) {
     return value &&
@@ -1346,6 +1347,44 @@ class RunnableLambda extends Runnable {
                             recursionLimit: (childConfig.recursionLimit ?? config_js_1.DEFAULT_RECURSION_LIMIT) - 1,
                         });
                     }
+                    else if ((0, iter_js_1.isAsyncIterable)(output)) {
+                        let finalOutput;
+                        for await (const chunk of (0, iter_js_1.consumeAsyncIterableInContext)(childConfig, output)) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                // Make a best effort to gather, for any type that supports concat.
+                                try {
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                    finalOutput = (0, stream_js_1.concat)(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = chunk;
+                                }
+                            }
+                        }
+                        output = finalOutput;
+                    }
+                    else if ((0, iter_js_1.isIterator)(output)) {
+                        let finalOutput;
+                        for (const chunk of (0, iter_js_1.consumeIteratorInContext)(childConfig, output)) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                // Make a best effort to gather, for any type that supports concat.
+                                try {
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                    finalOutput = (0, stream_js_1.concat)(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = chunk;
+                                }
+                            }
+                        }
+                        output = finalOutput;
+                    }
                     resolve(output);
                 }
                 catch (e) {
@@ -1400,6 +1439,16 @@ class RunnableLambda extends Runnable {
                 yield chunk;
             }
         }
+        else if ((0, iter_js_1.isAsyncIterable)(output)) {
+            for await (const chunk of (0, iter_js_1.consumeAsyncIterableInContext)(config, output)) {
+                yield chunk;
+            }
+        }
+        else if ((0, iter_js_1.isIterator)(output)) {
+            for (const chunk of (0, iter_js_1.consumeIteratorInContext)(config, output)) {
+                yield chunk;
+            }
+        }
         else {
             yield output;
         }

package/dist/runnables/base.js

@@ -11,6 +11,7 @@ import { _RootEventFilter, isRunnableInterface } from "./utils.js";
 import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
 import { Graph } from "./graph.js";
 import { convertToHttpEventStream } from "./wrappers.js";
+import { consumeAsyncIterableInContext, consumeIteratorInContext, isAsyncIterable, isIterator, } from "./iter.js";
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export function _coerceToDict(value, defaultKey) {
     return value &&
@@ -1333,6 +1334,44 @@ export class RunnableLambda extends Runnable {
                             recursionLimit: (childConfig.recursionLimit ?? DEFAULT_RECURSION_LIMIT) - 1,
                         });
                     }
+                    else if (isAsyncIterable(output)) {
+                        let finalOutput;
+                        for await (const chunk of consumeAsyncIterableInContext(childConfig, output)) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                // Make a best effort to gather, for any type that supports concat.
+                                try {
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                    finalOutput = concat(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = chunk;
+                                }
+                            }
+                        }
+                        output = finalOutput;
+                    }
+                    else if (isIterator(output)) {
+                        let finalOutput;
+                        for (const chunk of consumeIteratorInContext(childConfig, output)) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                // Make a best effort to gather, for any type that supports concat.
+                                try {
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                    finalOutput = concat(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = chunk;
+                                }
+                            }
+                        }
+                        output = finalOutput;
+                    }
                     resolve(output);
                 }
                 catch (e) {
@@ -1387,6 +1426,16 @@ export class RunnableLambda extends Runnable {
                 yield chunk;
             }
         }
+        else if (isAsyncIterable(output)) {
+            for await (const chunk of consumeAsyncIterableInContext(config, output)) {
+                yield chunk;
+            }
+        }
+        else if (isIterator(output)) {
+            for (const chunk of consumeIteratorInContext(config, output)) {
+                yield chunk;
+            }
+        }
         else {
             yield output;
         }

package/dist/runnables/iter.cjs

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.consumeAsyncIterableInContext = exports.consumeIteratorInContext = exports.isAsyncIterable = exports.isIterator = void 0;
+const index_js_1 = require("../singletons/index.cjs");
+function isIterator(thing) {
+    return (typeof thing === "object" &&
+        thing !== null &&
+        typeof thing[Symbol.iterator] === "function" &&
+        // avoid detecting array/set as iterator
+        typeof thing.next === "function");
+}
+exports.isIterator = isIterator;
+function isAsyncIterable(thing) {
+    return (typeof thing === "object" &&
+        thing !== null &&
+        typeof thing[Symbol.asyncIterator] ===
+            "function");
+}
+exports.isAsyncIterable = isAsyncIterable;
+function* consumeIteratorInContext(context, iter) {
+    const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
+    while (true) {
+        const { value, done } = storage.run(context, iter.next.bind(iter));
+        if (done) {
+            break;
+        }
+        else {
+            yield value;
+        }
+    }
+}
+exports.consumeIteratorInContext = consumeIteratorInContext;
+async function* consumeAsyncIterableInContext(context, iter) {
+    const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
+    const iterator = iter[Symbol.asyncIterator]();
+    while (true) {
+        const { value, done } = await storage.run(context, iterator.next.bind(iter));
+        if (done) {
+            break;
+        }
+        else {
+            yield value;
+        }
+    }
+}
+exports.consumeAsyncIterableInContext = consumeAsyncIterableInContext;

package/dist/runnables/iter.d.ts

@@ -0,0 +1,5 @@
+import { RunnableConfig } from "./config.js";
+export declare function isIterator(thing: unknown): thing is IterableIterator<unknown>;
+export declare function isAsyncIterable(thing: unknown): thing is AsyncIterable<unknown>;
+export declare function consumeIteratorInContext<T>(context: Partial<RunnableConfig> | undefined, iter: IterableIterator<T>): IterableIterator<T>;
+export declare function consumeAsyncIterableInContext<T>(context: Partial<RunnableConfig> | undefined, iter: AsyncIterable<T>): AsyncIterableIterator<T>;

package/dist/runnables/iter.js

@@ -0,0 +1,39 @@
+import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
+export function isIterator(thing) {
+    return (typeof thing === "object" &&
+        thing !== null &&
+        typeof thing[Symbol.iterator] === "function" &&
+        // avoid detecting array/set as iterator
+        typeof thing.next === "function");
+}
+export function isAsyncIterable(thing) {
+    return (typeof thing === "object" &&
+        thing !== null &&
+        typeof thing[Symbol.asyncIterator] ===
+            "function");
+}
+export function* consumeIteratorInContext(context, iter) {
+    const storage = AsyncLocalStorageProviderSingleton.getInstance();
+    while (true) {
+        const { value, done } = storage.run(context, iter.next.bind(iter));
+        if (done) {
+            break;
+        }
+        else {
+            yield value;
+        }
+    }
+}
+export async function* consumeAsyncIterableInContext(context, iter) {
+    const storage = AsyncLocalStorageProviderSingleton.getInstance();
+    const iterator = iter[Symbol.asyncIterator]();
+    while (true) {
+        const { value, done } = await storage.run(context, iterator.next.bind(iter));
+        if (done) {
+            break;
+        }
+        else {
+            yield value;
+        }
+    }
+}

package/dist/singletons/index.cjs

@@ -7,7 +7,7 @@ class MockAsyncLocalStorage {
         return undefined;
     }
     run(_store, callback) {
-        callback();
+        return callback();
     }
 }
 exports.MockAsyncLocalStorage = MockAsyncLocalStorage;

package/dist/singletons/index.d.ts

@@ -1,10 +1,10 @@
 export interface AsyncLocalStorageInterface {
     getStore: () => any | undefined;
-    run: (store: any, callback: () => any) => any;
+    run: <T>(store: any, callback: () => T) => T;
 }
 export declare class MockAsyncLocalStorage implements AsyncLocalStorageInterface {
     getStore(): any;
-    run(_store: any, callback: () => any): any;
+    run<T>(_store: any, callback: () => T): T;
 }
 declare class AsyncLocalStorageProvider {
     private asyncLocalStorage;

package/dist/singletons/index.js

@@ -4,7 +4,7 @@ export class MockAsyncLocalStorage {
         return undefined;
     }
     run(_store, callback) {
-        callback();
+        return callback();
     }
 }
 class AsyncLocalStorageProvider {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.1.62",
+  "version": "0.1.63",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {
@@ -43,7 +43,7 @@
     "ansi-styles": "^5.0.0",
     "camelcase": "6",
     "decamelize": "1.2.0",
-    "js-tiktoken": "^1.0.8",
+    "js-tiktoken": "^1.0.12",
     "langsmith": "~0.1.7",
     "ml-distance": "^4.0.0",
     "mustache": "^4.2.0",