langchain 0.0.151 → 0.0.153

package/dist/schema/output_parser.d.ts

@@ -118,11 +118,26 @@ export declare class BytesOutputParser extends BaseTransformOutputParser<Uint8Ar
     getFormatInstructions(): string;
 }
 /**
- * Custom error class used to handle exceptions related to output parsing.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
+ * Exception that output parsers should raise to signify a parsing error.
+ *
+ * This exists to differentiate parsing errors from other code or execution errors
+ * that also may arise inside the output parser. OutputParserExceptions will be
+ * available to catch and handle in ways to fix the parsing error, while other
+ * errors will be raised.
+ *
+ * @param message - The error that's being re-raised or an error message.
+ * @param llmOutput - String model output which is error-ing.
+ * @param observation - String explanation of error which can be passed to a
+ *     model to try and remediate the issue.
+ * @param sendToLLM - Whether to send the observation and llm_output back to an Agent
+ *     after an OutputParserException has been raised. This gives the underlying
+ *     model driving the agent the context that the previous output was improperly
+ *     structured, in the hopes that it will update the output to the correct
+ *     format.
  */
 export declare class OutputParserException extends Error {
-    output?: string;
-    constructor(message: string, output?: string);
+    llmOutput?: string;
+    observation?: string;
+    sendToLLM: boolean;
+    constructor(message: string, llmOutput?: string, observation?: string, sendToLLM?: boolean);
 }

package/dist/schema/output_parser.js

@@ -154,19 +154,51 @@ export class BytesOutputParser extends BaseTransformOutputParser {
     }
 }
 /**
- * Custom error class used to handle exceptions related to output parsing.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
+ * Exception that output parsers should raise to signify a parsing error.
+ *
+ * This exists to differentiate parsing errors from other code or execution errors
+ * that also may arise inside the output parser. OutputParserExceptions will be
+ * available to catch and handle in ways to fix the parsing error, while other
+ * errors will be raised.
+ *
+ * @param message - The error that's being re-raised or an error message.
+ * @param llmOutput - String model output which is error-ing.
+ * @param observation - String explanation of error which can be passed to a
+ *     model to try and remediate the issue.
+ * @param sendToLLM - Whether to send the observation and llm_output back to an Agent
+ *     after an OutputParserException has been raised. This gives the underlying
+ *     model driving the agent the context that the previous output was improperly
+ *     structured, in the hopes that it will update the output to the correct
+ *     format.
  */
 export class OutputParserException extends Error {
-    constructor(message, output) {
+    constructor(message, llmOutput, observation, sendToLLM = false) {
         super(message);
-        Object.defineProperty(this, "output", {
+        Object.defineProperty(this, "llmOutput", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        this.output = output;
+        Object.defineProperty(this, "observation", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sendToLLM", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llmOutput = llmOutput;
+        this.observation = observation;
+        this.sendToLLM = sendToLLM;
+        if (sendToLLM) {
+            if (observation === undefined || llmOutput === undefined) {
+                throw new Error("Arguments 'observation' & 'llmOutput' are required if 'sendToLlm' is true");
+            }
+        }
     }
 }

package/dist/schema/runnable/base.cjs

@@ -37,7 +37,7 @@ class Runnable extends serializable_js_1.Serializable {
      */
     bind(kwargs) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
-        return new RunnableBinding({ bound: this, kwargs });
+        return new RunnableBinding({ bound: this, kwargs, config: {} });
     }
     /**
      * Return a new Runnable that maps a list of inputs to a list of outputs,
@@ -48,19 +48,33 @@ class Runnable extends serializable_js_1.Serializable {
         return new RunnableEach({ bound: this });
     }
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
         return new RunnableRetry({
             bound: this,
             kwargs: {},
+            config: {},
             maxAttemptNumber: fields?.stopAfterAttempt,
             ...fields,
         });
     }
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config) {
+        // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        return new RunnableBinding({
+            bound: this,
+            config,
+            kwargs: {},
+        });
+    }
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -312,6 +326,12 @@ class RunnableBinding extends Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "config", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "kwargs", {
             enumerable: true,
             configurable: true,
@@ -320,35 +340,70 @@ class RunnableBinding extends Runnable {
         });
         this.bound = fields.bound;
         this.kwargs = fields.kwargs;
+        this.config = fields.config;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _mergeConfig(options) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const copy = { ...this.config };
+        if (options) {
+            for (const key of Object.keys(options)) {
+                if (key === "metadata") {
+                    copy[key] = { ...copy[key], ...options[key] };
+                }
+                else if (key === "tags") {
+                    copy[key] = (copy[key] ?? []).concat(options[key] ?? []);
+                }
+                else {
+                    copy[key] = options[key] ?? copy[key];
+                }
+            }
+        }
+        return copy;
     }
     bind(kwargs) {
-        return new RunnableBinding({
+        return this.constructor({
             bound: this.bound,
             kwargs: { ...this.kwargs, ...kwargs },
+            config: this.config,
+        });
+    }
+    withConfig(config) {
+        return this.constructor({
+            bound: this.bound,
+            kwargs: this.kwargs,
+            config: { ...this.config, ...config },
+        });
+    }
+    withRetry(fields) {
+        return this.constructor({
+            bound: this.bound.withRetry(fields),
+            kwargs: this.kwargs,
+            config: this.config,
         });
     }
     async invoke(input, options) {
-        return this.bound.invoke(input, { ...options, ...this.kwargs });
+        return this.bound.invoke(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async batch(inputs, options, batchOptions) {
         const mergedOptions = Array.isArray(options)
-            ? options.map((individualOption) => ({
+            ? options.map((individualOption) => this._mergeConfig({
                 ...individualOption,
                 ...this.kwargs,
             }))
-            : { ...options, ...this.kwargs };
+            : this._mergeConfig({ ...options, ...this.kwargs });
         return this.bound.batch(inputs, mergedOptions, batchOptions);
     }
     async *_streamIterator(input, options) {
-        yield* this.bound._streamIterator(input, { ...options, ...this.kwargs });
+        yield* this.bound._streamIterator(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async stream(input, options) {
-        return this.bound.stream(input, { ...options, ...this.kwargs });
+        return this.bound.stream(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async *transform(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     generator, options) {
-        yield* this.bound.transform(generator, options);
+        yield* this.bound.transform(generator, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     static isRunnableBinding(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/schema/runnable/base.d.ts

@@ -33,14 +33,20 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      */
     map(): Runnable<RunInput[], RunOutput[], CallOptions>;
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields?: {
         stopAfterAttempt?: number;
         onFailedAttempt?: RunnableRetryFailedAttemptHandler;
     }): RunnableRetry<RunInput, RunOutput, CallOptions>;
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config: RunnableConfig): RunnableBinding<RunInput, RunOutput, CallOptions>;
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -122,18 +128,26 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
 export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends RunnableConfig> = {
     bound: Runnable<RunInput, RunOutput, CallOptions>;
     kwargs: Partial<CallOptions>;
+    config: RunnableConfig;
 };
 /**
  * A runnable that delegates calls to another runnable with a set of kwargs.
  */
-export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends BaseCallbackConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
+export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends RunnableConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
     bound: Runnable<RunInput, RunOutput, CallOptions>;
+    config: RunnableConfig;
     protected kwargs: Partial<CallOptions>;
     constructor(fields: RunnableBindingArgs<RunInput, RunOutput, CallOptions>);
+    _mergeConfig(options?: Record<string, any>): Partial<CallOptions>;
     bind(kwargs: Partial<CallOptions>): RunnableBinding<RunInput, RunOutput, CallOptions>;
+    withConfig(config: RunnableConfig): RunnableBinding<RunInput, RunOutput, CallOptions>;
+    withRetry(fields?: {
+        stopAfterAttempt?: number;
+        onFailedAttempt?: RunnableRetryFailedAttemptHandler;
+    }): RunnableRetry<RunInput, RunOutput, CallOptions>;
     invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
     batch(inputs: RunInput[], options?: Partial<CallOptions> | Partial<CallOptions>[], batchOptions?: RunnableBatchOptions & {
         returnExceptions?: false;

package/dist/schema/runnable/base.js

@@ -31,7 +31,7 @@ export class Runnable extends Serializable {
      */
     bind(kwargs) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
-        return new RunnableBinding({ bound: this, kwargs });
+        return new RunnableBinding({ bound: this, kwargs, config: {} });
     }
     /**
      * Return a new Runnable that maps a list of inputs to a list of outputs,
@@ -42,19 +42,33 @@ export class Runnable extends Serializable {
         return new RunnableEach({ bound: this });
     }
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
         return new RunnableRetry({
             bound: this,
             kwargs: {},
+            config: {},
             maxAttemptNumber: fields?.stopAfterAttempt,
             ...fields,
         });
     }
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config) {
+        // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        return new RunnableBinding({
+            bound: this,
+            config,
+            kwargs: {},
+        });
+    }
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -305,6 +319,12 @@ export class RunnableBinding extends Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "config", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "kwargs", {
             enumerable: true,
             configurable: true,
@@ -313,35 +333,70 @@ export class RunnableBinding extends Runnable {
         });
         this.bound = fields.bound;
         this.kwargs = fields.kwargs;
+        this.config = fields.config;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _mergeConfig(options) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const copy = { ...this.config };
+        if (options) {
+            for (const key of Object.keys(options)) {
+                if (key === "metadata") {
+                    copy[key] = { ...copy[key], ...options[key] };
+                }
+                else if (key === "tags") {
+                    copy[key] = (copy[key] ?? []).concat(options[key] ?? []);
+                }
+                else {
+                    copy[key] = options[key] ?? copy[key];
+                }
+            }
+        }
+        return copy;
     }
     bind(kwargs) {
-        return new RunnableBinding({
+        return this.constructor({
             bound: this.bound,
             kwargs: { ...this.kwargs, ...kwargs },
+            config: this.config,
+        });
+    }
+    withConfig(config) {
+        return this.constructor({
+            bound: this.bound,
+            kwargs: this.kwargs,
+            config: { ...this.config, ...config },
+        });
+    }
+    withRetry(fields) {
+        return this.constructor({
+            bound: this.bound.withRetry(fields),
+            kwargs: this.kwargs,
+            config: this.config,
         });
     }
     async invoke(input, options) {
-        return this.bound.invoke(input, { ...options, ...this.kwargs });
+        return this.bound.invoke(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async batch(inputs, options, batchOptions) {
         const mergedOptions = Array.isArray(options)
-            ? options.map((individualOption) => ({
+            ? options.map((individualOption) => this._mergeConfig({
                 ...individualOption,
                 ...this.kwargs,
             }))
-            : { ...options, ...this.kwargs };
+            : this._mergeConfig({ ...options, ...this.kwargs });
         return this.bound.batch(inputs, mergedOptions, batchOptions);
     }
     async *_streamIterator(input, options) {
-        yield* this.bound._streamIterator(input, { ...options, ...this.kwargs });
+        yield* this.bound._streamIterator(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async stream(input, options) {
-        return this.bound.stream(input, { ...options, ...this.kwargs });
+        return this.bound.stream(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async *transform(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     generator, options) {
-        yield* this.bound.transform(generator, options);
+        yield* this.bound.transform(generator, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     static isRunnableBinding(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/stores/message/cloudflare_d1.cjs

@@ -0,0 +1,134 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CloudflareD1MessageHistory = void 0;
+const uuid_1 = require("uuid");
+const index_js_1 = require("../../schema/index.cjs");
+const utils_js_1 = require("./utils.cjs");
+/**
+ * Class for storing and retrieving chat message history from a
+ * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
+ */
+class CloudflareD1MessageHistory extends index_js_1.BaseListChatMessageHistory {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cloudflare_d1"]
+        });
+        Object.defineProperty(this, "database", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tableInitialized", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const { sessionId, database, tableName } = fields;
+        if (database) {
+            this.database = database;
+        }
+        else {
+            throw new Error("Either a client or config must be provided to CloudflareD1MessageHistory");
+        }
+        this.tableName = tableName || "langchain_chat_histories";
+        this.tableInitialized = false;
+        this.sessionId = sessionId;
+    }
+    /**
+     * Private method to ensure that the necessary table exists in the
+     * Cloudflare D1 database before performing any operations. If the table
+     * does not exist, it is created.
+     * @returns Promise that resolves to void.
+     */
+    async ensureTable() {
+        if (this.tableInitialized) {
+            return;
+        }
+        const query = `CREATE TABLE IF NOT EXISTS ${this.tableName} (id TEXT PRIMARY KEY, session_id TEXT, type TEXT, content TEXT, role TEXT, name TEXT, additional_kwargs TEXT);`;
+        await this.database.prepare(query).bind().all();
+        const idIndexQuery = `CREATE INDEX IF NOT EXISTS id_index ON ${this.tableName} (id);`;
+        await this.database.prepare(idIndexQuery).bind().all();
+        const sessionIdIndexQuery = `CREATE INDEX IF NOT EXISTS session_id_index ON ${this.tableName} (session_id);`;
+        await this.database.prepare(sessionIdIndexQuery).bind().all();
+        this.tableInitialized = true;
+    }
+    /**
+     * Method to retrieve all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to an array of BaseMessage objects.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const query = `SELECT * FROM ${this.tableName} WHERE session_id = ?`;
+        const rawStoredMessages = await this.database
+            .prepare(query)
+            .bind(this.sessionId)
+            .all();
+        const storedMessagesObject = rawStoredMessages.results;
+        const orderedMessages = storedMessagesObject.map((message) => {
+            const data = {
+                content: message.content,
+                additional_kwargs: JSON.parse(message.additional_kwargs),
+            };
+            if (message.role) {
+                data.role = message.role;
+            }
+            if (message.name) {
+                data.name = message.name;
+            }
+            return {
+                type: message.type,
+                data,
+            };
+        });
+        return (0, utils_js_1.mapStoredMessagesToChatMessages)(orderedMessages);
+    }
+    /**
+     * Method to add a new message to the Cloudflare D1 database for the current
+     * session.
+     * @param message The BaseMessage object to be added to the database.
+     * @returns Promise that resolves to void.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messageToAdd = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
+        const query = `INSERT INTO ${this.tableName} (id, session_id, type, content, role, name, additional_kwargs) VALUES(?, ?, ?, ?, ?, ?, ?)`;
+        const id = (0, uuid_1.v4)();
+        await this.database
+            .prepare(query)
+            .bind(id, this.sessionId, messageToAdd[0].type || null, messageToAdd[0].data.content || null, messageToAdd[0].data.role || null, messageToAdd[0].data.name || null, JSON.stringify(messageToAdd[0].data.additional_kwargs))
+            .all();
+    }
+    /**
+     * Method to delete all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to void.
+     */
+    async clear() {
+        await this.ensureTable();
+        const query = `DELETE FROM ? WHERE session_id = ? `;
+        await this.database
+            .prepare(query)
+            .bind(this.tableName, this.sessionId)
+            .all();
+    }
+}
+exports.CloudflareD1MessageHistory = CloudflareD1MessageHistory;

package/dist/stores/message/cloudflare_d1.d.ts

@@ -0,0 +1,49 @@
+import type { D1Database } from "@cloudflare/workers-types";
+import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+/**
+ * Type definition for the input parameters required when instantiating a
+ * CloudflareD1MessageHistory object.
+ */
+export type CloudflareD1MessageHistoryInput = {
+    tableName?: string;
+    sessionId: string;
+    database?: D1Database;
+};
+/**
+ * Class for storing and retrieving chat message history from a
+ * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
+ */
+export declare class CloudflareD1MessageHistory extends BaseListChatMessageHistory {
+    lc_namespace: string[];
+    database: D1Database;
+    private tableName;
+    private sessionId;
+    private tableInitialized;
+    constructor(fields: CloudflareD1MessageHistoryInput);
+    /**
+     * Private method to ensure that the necessary table exists in the
+     * Cloudflare D1 database before performing any operations. If the table
+     * does not exist, it is created.
+     * @returns Promise that resolves to void.
+     */
+    private ensureTable;
+    /**
+     * Method to retrieve all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to an array of BaseMessage objects.
+     */
+    getMessages(): Promise<BaseMessage[]>;
+    /**
+     * Method to add a new message to the Cloudflare D1 database for the current
+     * session.
+     * @param message The BaseMessage object to be added to the database.
+     * @returns Promise that resolves to void.
+     */
+    addMessage(message: BaseMessage): Promise<void>;
+    /**
+     * Method to delete all messages from the Cloudflare D1 database for the
+     * current session.
+     * @returns Promise that resolves to void.
+     */
+    clear(): Promise<void>;
+}