@langchain/core 0.1.17 → 0.1.18

package/dist/runnables/base.cjs

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RunnablePick = exports.RunnableAssign = exports._coerceToRunnable = exports.RunnableWithFallbacks = exports.RunnableParallel = exports.RunnableLambda = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = void 0;
+exports.RunnablePick = exports.RunnableAssign = exports._coerceToRunnable = exports.RunnableWithFallbacks = exports.RunnableParallel = exports.RunnableLambda = exports.RunnableMap = exports.RunnableSequence = exports.RunnableRetry = exports.RunnableEach = exports.RunnableBinding = exports.Runnable = exports._coerceToDict = void 0;
 const p_retry_1 = __importDefault(require("p-retry"));
 const manager_js_1 = require("../callbacks/manager.cjs");
 const log_stream_js_1 = require("../tracers/log_stream.cjs");
@@ -22,6 +22,7 @@ function _coerceToDict(value, defaultKey) {
         ? value
         : { [defaultKey]: value };
 }
+exports._coerceToDict = _coerceToDict;
 /**
  * A Runnable is a generic unit of work that can be invoked, batched, streamed, and/or
  * transformed.

package/dist/runnables/base.d.ts

@@ -38,6 +38,7 @@ export type RunnableBatchOptions = {
     returnExceptions?: boolean;
 };
 export type RunnableRetryFailedAttemptHandler = (error: any) => any;
+export declare function _coerceToDict(value: any, defaultKey: string): any;
 /**
  * A Runnable is a generic unit of work that can be invoked, batched, streamed, and/or
  * transformed.

package/dist/runnables/base.js

@@ -7,7 +7,7 @@ import { DEFAULT_RECURSION_LIMIT, ensureConfig, getCallbackManagerForConfig, mer
 import { AsyncCaller } from "../utils/async_caller.js";
 import { RootListenersTracer } from "../tracers/root_listener.js";
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-function _coerceToDict(value, defaultKey) {
+export function _coerceToDict(value, defaultKey) {
     return value &&
         !Array.isArray(value) &&
         // eslint-disable-next-line no-instanceof/no-instanceof

package/dist/runnables/branch.cjs

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.RunnableBranch = void 0;
 const base_js_1 = require("./base.cjs");
 const config_js_1 = require("./config.cjs");
+const stream_js_1 = require("../utils/stream.cjs");
 /**
  * Class that represents a runnable branch. The RunnableBranch is
  * initialized with an array of branches and a default branch. When invoked,
@@ -129,7 +130,7 @@ class RunnableBranch extends base_js_1.Runnable {
         }
         if (!result) {
             result = await this.default.invoke(input, (0, config_js_1.patchConfig)(config, {
-                callbacks: runManager?.getChild("default"),
+                callbacks: runManager?.getChild("branch:default"),
             }));
         }
         return result;
@@ -137,5 +138,70 @@ class RunnableBranch extends base_js_1.Runnable {
     async invoke(input, config = {}) {
         return this._callWithConfig(this._invoke, input, config);
     }
+    async *_streamIterator(input, config) {
+        const callbackManager_ = await (0, config_js_1.getCallbackManagerForConfig)(config);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), (0, base_js_1._coerceToDict)(input, "input"), undefined, undefined, undefined, undefined, config?.runName);
+        let finalOutput;
+        let finalOutputSupported = true;
+        let stream;
+        try {
+            for (let i = 0; i < this.branches.length; i += 1) {
+                const [condition, branchRunnable] = this.branches[i];
+                const conditionValue = await condition.invoke(input, (0, config_js_1.patchConfig)(config, {
+                    callbacks: runManager?.getChild(`condition:${i + 1}`),
+                }));
+                if (conditionValue) {
+                    stream = await branchRunnable.stream(input, (0, config_js_1.patchConfig)(config, {
+                        callbacks: runManager?.getChild(`branch:${i + 1}`),
+                    }));
+                    for await (const chunk of stream) {
+                        yield chunk;
+                        if (finalOutputSupported) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                try {
+                                    finalOutput = (0, stream_js_1.concat)(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = undefined;
+                                    finalOutputSupported = false;
+                                }
+                            }
+                        }
+                    }
+                    break;
+                }
+            }
+            if (stream === undefined) {
+                stream = await this.default.stream(input, (0, config_js_1.patchConfig)(config, {
+                    callbacks: runManager?.getChild("branch:default"),
+                }));
+                for await (const chunk of stream) {
+                    yield chunk;
+                    if (finalOutputSupported) {
+                        if (finalOutput === undefined) {
+                            finalOutput = chunk;
+                        }
+                        else {
+                            try {
+                                finalOutput = (0, stream_js_1.concat)(finalOutput, chunk);
+                            }
+                            catch (e) {
+                                finalOutput = undefined;
+                                finalOutputSupported = false;
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        catch (e) {
+            await runManager?.handleChainError(e);
+            throw e;
+        }
+        await runManager?.handleChainEnd(finalOutput ?? {});
+    }
 }
 exports.RunnableBranch = RunnableBranch;

package/dist/runnables/branch.d.ts

@@ -91,4 +91,5 @@ export declare class RunnableBranch<RunInput = any, RunOutput = any> extends Run
     ]): RunnableBranch<RunInput, RunOutput>;
     _invoke(input: RunInput, config?: Partial<RunnableConfig>, runManager?: CallbackManagerForChainRun): Promise<RunOutput>;
     invoke(input: RunInput, config?: RunnableConfig): Promise<RunOutput>;
+    _streamIterator(input: RunInput, config?: Partial<RunnableConfig>): AsyncGenerator<Awaited<RunOutput>, void, unknown>;
 }

package/dist/runnables/branch.js

@@ -1,5 +1,6 @@
-import { Runnable, _coerceToRunnable } from "./base.js";
-import { patchConfig } from "./config.js";
+import { Runnable, _coerceToDict, _coerceToRunnable, } from "./base.js";
+import { getCallbackManagerForConfig, patchConfig, } from "./config.js";
+import { concat } from "../utils/stream.js";
 /**
  * Class that represents a runnable branch. The RunnableBranch is
  * initialized with an array of branches and a default branch. When invoked,
@@ -126,7 +127,7 @@ export class RunnableBranch extends Runnable {
         }
         if (!result) {
             result = await this.default.invoke(input, patchConfig(config, {
-                callbacks: runManager?.getChild("default"),
+                callbacks: runManager?.getChild("branch:default"),
             }));
         }
         return result;
@@ -134,4 +135,69 @@ export class RunnableBranch extends Runnable {
     async invoke(input, config = {}) {
         return this._callWithConfig(this._invoke, input, config);
     }
+    async *_streamIterator(input, config) {
+        const callbackManager_ = await getCallbackManagerForConfig(config);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, config?.runName);
+        let finalOutput;
+        let finalOutputSupported = true;
+        let stream;
+        try {
+            for (let i = 0; i < this.branches.length; i += 1) {
+                const [condition, branchRunnable] = this.branches[i];
+                const conditionValue = await condition.invoke(input, patchConfig(config, {
+                    callbacks: runManager?.getChild(`condition:${i + 1}`),
+                }));
+                if (conditionValue) {
+                    stream = await branchRunnable.stream(input, patchConfig(config, {
+                        callbacks: runManager?.getChild(`branch:${i + 1}`),
+                    }));
+                    for await (const chunk of stream) {
+                        yield chunk;
+                        if (finalOutputSupported) {
+                            if (finalOutput === undefined) {
+                                finalOutput = chunk;
+                            }
+                            else {
+                                try {
+                                    finalOutput = concat(finalOutput, chunk);
+                                }
+                                catch (e) {
+                                    finalOutput = undefined;
+                                    finalOutputSupported = false;
+                                }
+                            }
+                        }
+                    }
+                    break;
+                }
+            }
+            if (stream === undefined) {
+                stream = await this.default.stream(input, patchConfig(config, {
+                    callbacks: runManager?.getChild("branch:default"),
+                }));
+                for await (const chunk of stream) {
+                    yield chunk;
+                    if (finalOutputSupported) {
+                        if (finalOutput === undefined) {
+                            finalOutput = chunk;
+                        }
+                        else {
+                            try {
+                                finalOutput = concat(finalOutput, chunk);
+                            }
+                            catch (e) {
+                                finalOutput = undefined;
+                                finalOutputSupported = false;
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        catch (e) {
+            await runManager?.handleChainError(e);
+            throw e;
+        }
+        await runManager?.handleChainEnd(finalOutput ?? {});
+    }
 }

package/dist/tracers/base.cjs

@@ -225,7 +225,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
             throw new Error("No chain run to end.");
         }
         run.end_time = Date.now();
-        run.error = error.message + (error.stack ? `\n\n${error.stack}` : "");
+        run.error = error.message + (error?.stack ? `\n\n${error.stack}` : "");
         run.events.push({
             name: "error",
             time: new Date(run.end_time).toISOString(),

package/dist/tracers/base.js

@@ -222,7 +222,7 @@ export class BaseTracer extends BaseCallbackHandler {
             throw new Error("No chain run to end.");
         }
         run.end_time = Date.now();
-        run.error = error.message + (error.stack ? `\n\n${error.stack}` : "");
+        run.error = error.message + (error?.stack ? `\n\n${error.stack}` : "");
         run.events.push({
             name: "error",
             time: new Date(run.end_time).toISOString(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "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\" \"scripts\"",
-    "format:check": "prettier --check \"src\" \"scripts\""
+    "format": "prettier --config .prettierrc --write \"src\" \"scripts\"",
+    "format:check": "prettier --config .prettierrc --check \"src\" \"scripts\""
   },
   "author": "LangChain",
   "license": "MIT",