langchain 0.0.178 → 0.0.180

package/dist/schema/prompt_template.cjs

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatDocument = void 0;
+/**
+ * Formats a document using a given prompt template.
+ *
+ * @async
+ * @param {Document} document - The document to format.
+ * @param {BasePromptTemplate} prompt - The prompt template to use for formatting.
+ * @returns {Promise<string>} A Promise that resolves to the formatted document as a string.
+ * @throws {Error} If the document is missing required metadata variables specified in the prompt template.
+ */
+const formatDocument = async (document, prompt) => {
+    const baseInfo = {
+        pageContent: document.pageContent,
+        ...document.metadata,
+    };
+    const variables = new Set(prompt.inputVariables);
+    const requiredMetadata = new Set(prompt.inputVariables
+        .map((v) => (v !== "pageContent" ? v : null))
+        .filter((v) => v !== null));
+    const missingMetadata = [];
+    for (const variable of variables) {
+        if (!(variable in baseInfo) && variable !== "pageContent") {
+            missingMetadata.push(variable);
+        }
+    }
+    if (missingMetadata.length) {
+        throw new Error(`Document prompt requires documents to have metadata variables: ${JSON.stringify(requiredMetadata)}. Received document with missing metadata: ${JSON.stringify(missingMetadata)}`);
+    }
+    return prompt.format(baseInfo);
+};
+exports.formatDocument = formatDocument;

package/dist/schema/prompt_template.d.ts

@@ -0,0 +1,12 @@
+import { Document } from "../document.js";
+import { BasePromptTemplate } from "../prompts/base.js";
+/**
+ * Formats a document using a given prompt template.
+ *
+ * @async
+ * @param {Document} document - The document to format.
+ * @param {BasePromptTemplate} prompt - The prompt template to use for formatting.
+ * @returns {Promise<string>} A Promise that resolves to the formatted document as a string.
+ * @throws {Error} If the document is missing required metadata variables specified in the prompt template.
+ */
+export declare const formatDocument: (document: Document, prompt: BasePromptTemplate) => Promise<string>;

package/dist/schema/prompt_template.js

@@ -0,0 +1,29 @@
+/**
+ * Formats a document using a given prompt template.
+ *
+ * @async
+ * @param {Document} document - The document to format.
+ * @param {BasePromptTemplate} prompt - The prompt template to use for formatting.
+ * @returns {Promise<string>} A Promise that resolves to the formatted document as a string.
+ * @throws {Error} If the document is missing required metadata variables specified in the prompt template.
+ */
+export const formatDocument = async (document, prompt) => {
+    const baseInfo = {
+        pageContent: document.pageContent,
+        ...document.metadata,
+    };
+    const variables = new Set(prompt.inputVariables);
+    const requiredMetadata = new Set(prompt.inputVariables
+        .map((v) => (v !== "pageContent" ? v : null))
+        .filter((v) => v !== null));
+    const missingMetadata = [];
+    for (const variable of variables) {
+        if (!(variable in baseInfo) && variable !== "pageContent") {
+            missingMetadata.push(variable);
+        }
+    }
+    if (missingMetadata.length) {
+        throw new Error(`Document prompt requires documents to have metadata variables: ${JSON.stringify(requiredMetadata)}. Received document with missing metadata: ${JSON.stringify(missingMetadata)}`);
+    }
+    return prompt.format(baseInfo);
+};

package/dist/storage/convex.d.ts

@@ -21,12 +21,33 @@ export type ConvexKVStoreConfig<DataModel extends GenericDataModel, TableName ex
     key: string;
 }>> = {
     readonly ctx: GenericActionCtx<DataModel>;
+    /**
+     * Defaults to "cache"
+     */
     readonly table?: TableName;
+    /**
+     * Defaults to "byKey"
+     */
     readonly index?: IndexName;
+    /**
+     * Defaults to "key"
+     */
     readonly keyField?: KeyFieldName;
+    /**
+     * Defaults to "value"
+     */
     readonly valueField?: ValueFieldName;
+    /**
+     * Defaults to `internal.langchain.db.upsert`
+     */
     readonly upsert?: UpsertMutation;
+    /**
+     * Defaults to `internal.langchain.db.lookup`
+     */
     readonly lookup?: LookupQuery;
+    /**
+     * Defaults to `internal.langchain.db.deleteMany`
+     */
     readonly deleteMany?: DeleteManyMutation;
 };
 /**

package/dist/stores/message/cassandra.cjs

@@ -0,0 +1,135 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CassandraChatMessageHistory = void 0;
+const cassandra_driver_1 = require("cassandra-driver");
+const index_js_1 = require("../../schema/index.cjs");
+const utils_js_1 = require("./utils.cjs");
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+class CassandraChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
+    constructor(options) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cassandra"]
+        });
+        Object.defineProperty(this, "keyspace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "table", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            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, "tableExists", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queries", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = new cassandra_driver_1.Client(options);
+        this.keyspace = options.keyspace;
+        this.table = options.table;
+        this.sessionId = options.sessionId;
+        this.tableExists = false;
+        this.options = options;
+    }
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const resultSet = await this.client.execute(this.queries.select, [this.sessionId], { prepare: true });
+        const storedMessages = resultSet.rows.map((row) => ({
+            type: row.message_type,
+            data: JSON.parse(row.data),
+        }));
+        const baseMessages = (0, utils_js_1.mapStoredMessagesToChatMessages)(storedMessages);
+        return baseMessages;
+    }
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messages = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
+        const { type, data } = messages[0];
+        return this.client
+            .execute(this.queries.insert, [this.sessionId, type, JSON.stringify(data)], { prepare: true, ...this.options })
+            .then(() => { });
+    }
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    async clear() {
+        await this.ensureTable();
+        return this.client
+            .execute(this.queries.delete, [this.sessionId], {
+            prepare: true,
+            ...this.options,
+        })
+            .then(() => { });
+    }
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    async ensureTable() {
+        if (this.tableExists) {
+            return;
+        }
+        await this.client.execute(`
+    CREATE TABLE IF NOT EXISTS ${this.keyspace}.${this.table} (
+        session_id text,
+        message_ts timestamp,
+        message_type text,
+        data text,
+        PRIMARY KEY ((session_id), message_ts)
+      );
+    `);
+        this.queries = {
+            insert: `INSERT INTO ${this.keyspace}.${this.table} (session_id, message_ts, message_type, data) VALUES (?, toTimestamp(now()), ?, ?);`,
+            select: `SELECT message_type, data FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+            delete: `DELETE FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+        };
+        this.tableExists = true;
+    }
+}
+exports.CassandraChatMessageHistory = CassandraChatMessageHistory;

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

@@ -0,0 +1,44 @@
+import { DseClientOptions } from "cassandra-driver";
+import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+export interface CassandraChatMessageHistoryOptions extends DseClientOptions {
+    keyspace: string;
+    table: string;
+    sessionId: string;
+}
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+export declare class CassandraChatMessageHistory extends BaseListChatMessageHistory {
+    lc_namespace: string[];
+    private keyspace;
+    private table;
+    private client;
+    private sessionId;
+    private tableExists;
+    private options;
+    private queries;
+    constructor(options: CassandraChatMessageHistoryOptions);
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    getMessages(): Promise<BaseMessage[]>;
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    addMessage(message: BaseMessage): Promise<void>;
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    clear(): Promise<void>;
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    private ensureTable;
+}

package/dist/stores/message/cassandra.js

@@ -0,0 +1,131 @@
+import { Client } from "cassandra-driver";
+import { BaseListChatMessageHistory, } from "../../schema/index.js";
+import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
+/**
+ * Class for storing chat message history within Cassandra. It extends the
+ * BaseListChatMessageHistory class and provides methods to get, add, and
+ * clear messages.
+ */
+export class CassandraChatMessageHistory extends BaseListChatMessageHistory {
+    constructor(options) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "stores", "message", "cassandra"]
+        });
+        Object.defineProperty(this, "keyspace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "table", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            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, "tableExists", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queries", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = new Client(options);
+        this.keyspace = options.keyspace;
+        this.table = options.table;
+        this.sessionId = options.sessionId;
+        this.tableExists = false;
+        this.options = options;
+    }
+    /**
+     * Method to get all the messages stored in the Cassandra database.
+     * @returns Array of stored BaseMessage instances.
+     */
+    async getMessages() {
+        await this.ensureTable();
+        const resultSet = await this.client.execute(this.queries.select, [this.sessionId], { prepare: true });
+        const storedMessages = resultSet.rows.map((row) => ({
+            type: row.message_type,
+            data: JSON.parse(row.data),
+        }));
+        const baseMessages = mapStoredMessagesToChatMessages(storedMessages);
+        return baseMessages;
+    }
+    /**
+     * Method to add a new message to the Cassandra database.
+     * @param message The BaseMessage instance to add.
+     * @returns A promise that resolves when the message has been added.
+     */
+    async addMessage(message) {
+        await this.ensureTable();
+        const messages = mapChatMessagesToStoredMessages([message]);
+        const { type, data } = messages[0];
+        return this.client
+            .execute(this.queries.insert, [this.sessionId, type, JSON.stringify(data)], { prepare: true, ...this.options })
+            .then(() => { });
+    }
+    /**
+     * Method to clear all the messages from the Cassandra database.
+     * @returns A promise that resolves when all messages have been cleared.
+     */
+    async clear() {
+        await this.ensureTable();
+        return this.client
+            .execute(this.queries.delete, [this.sessionId], {
+            prepare: true,
+            ...this.options,
+        })
+            .then(() => { });
+    }
+    /**
+     * Method to initialize the Cassandra database.
+     * @returns Promise that resolves when the database has been initialized.
+     */
+    async ensureTable() {
+        if (this.tableExists) {
+            return;
+        }
+        await this.client.execute(`
+    CREATE TABLE IF NOT EXISTS ${this.keyspace}.${this.table} (
+        session_id text,
+        message_ts timestamp,
+        message_type text,
+        data text,
+        PRIMARY KEY ((session_id), message_ts)
+      );
+    `);
+        this.queries = {
+            insert: `INSERT INTO ${this.keyspace}.${this.table} (session_id, message_ts, message_type, data) VALUES (?, toTimestamp(now()), ?, ?);`,
+            select: `SELECT message_type, data FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+            delete: `DELETE FROM ${this.keyspace}.${this.table} WHERE session_id = ?;`,
+        };
+        this.tableExists = true;
+    }
+}

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

@@ -21,12 +21,33 @@ export type ConvexChatMessageHistoryInput<DataModel extends GenericDataModel, Ta
 }> = any> = {
     readonly ctx: GenericActionCtx<DataModel>;
     readonly sessionId: DocumentByName<DataModel, TableName>[SessionIdFieldName];
+    /**
+     * Defaults to "messages"
+     */
     readonly table?: TableName;
+    /**
+     * Defaults to "bySessionId"
+     */
     readonly index?: IndexName;
+    /**
+     * Defaults to "sessionId"
+     */
     readonly sessionIdField?: SessionIdFieldName;
+    /**
+     * Defaults to "message"
+     */
     readonly messageTextFieldName?: MessageTextFieldName;
+    /**
+     * Defaults to `internal.langchain.db.insert`
+     */
     readonly insert?: InsertMutation;
+    /**
+     * Defaults to `internal.langchain.db.lookup`
+     */
     readonly lookup?: LookupQuery;
+    /**
+     * Defaults to `internal.langchain.db.deleteMany`
+     */
     readonly deleteMany?: DeleteManyMutation;
 };
 export declare class ConvexChatMessageHistory<DataModel extends GenericDataModel, SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends VectorIndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {

package/dist/util/bedrock.cjs

@@ -11,7 +11,7 @@ class BedrockLLMInputOutputAdapter {
     /** Adapter class to prepare the inputs from Langchain to a format
     that LLM model expects. Also, provides a helper function to extract
     the generated text from the model response. */
-    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}) {
+    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}, bedrockMethod = "invoke") {
         const inputBody = {};
         if (provider === "anthropic") {
             inputBody.prompt = prompt;
@@ -32,6 +32,15 @@ class BedrockLLMInputOutputAdapter {
                 temperature,
             };
         }
+        else if (provider === "cohere") {
+            inputBody.prompt = prompt;
+            inputBody.max_tokens = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stop_sequences = stopSequences;
+            if (bedrockMethod === "invoke-with-response-stream") {
+                inputBody.stream = true;
+            }
+        }
         return { ...inputBody, ...modelKwargs };
     }
     /**
@@ -48,6 +57,9 @@ class BedrockLLMInputOutputAdapter {
         else if (provider === "ai21") {
             return responseBody?.completions?.[0]?.data?.text ?? "";
         }
+        else if (provider === "cohere") {
+            return responseBody?.generations?.[0]?.text ?? responseBody?.text ?? "";
+        }
         // I haven't been able to get a response with more than one result in it.
         return responseBody.results?.[0]?.outputText;
     }

package/dist/util/bedrock.d.ts

@@ -29,7 +29,10 @@ export interface BaseBedrockInput {
     endpointUrl?: string;
     /** Override the default endpoint hostname. */
     endpointHost?: string;
-    /** Optional additional stop sequences to pass to the model. Currently only supported for Anthropic and AI21. */
+    /**
+     * Optional additional stop sequences to pass to the model. Currently only supported for Anthropic and AI21.
+     * @deprecated Use .bind({ "stop": [...] }) instead
+     * */
     stopSequences?: string[];
     /** Additional kwargs to pass to the model. */
     modelKwargs?: Record<string, unknown>;
@@ -49,7 +52,7 @@ export declare class BedrockLLMInputOutputAdapter {
     /** Adapter class to prepare the inputs from Langchain to a format
     that LLM model expects. Also, provides a helper function to extract
     the generated text from the model response. */
-    static prepareInput(provider: string, prompt: string, maxTokens?: number, temperature?: number, stopSequences?: string[] | undefined, modelKwargs?: Record<string, unknown>): Dict;
+    static prepareInput(provider: string, prompt: string, maxTokens?: number, temperature?: number, stopSequences?: string[] | undefined, modelKwargs?: Record<string, unknown>, bedrockMethod?: "invoke" | "invoke-with-response-stream"): Dict;
     /**
      * Extracts the generated text from the service response.
      * @param provider The provider name.

package/dist/util/bedrock.js

@@ -8,7 +8,7 @@ export class BedrockLLMInputOutputAdapter {
     /** Adapter class to prepare the inputs from Langchain to a format
     that LLM model expects. Also, provides a helper function to extract
     the generated text from the model response. */
-    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}) {
+    static prepareInput(provider, prompt, maxTokens = 50, temperature = 0, stopSequences = undefined, modelKwargs = {}, bedrockMethod = "invoke") {
         const inputBody = {};
         if (provider === "anthropic") {
             inputBody.prompt = prompt;
@@ -29,6 +29,15 @@ export class BedrockLLMInputOutputAdapter {
                 temperature,
             };
         }
+        else if (provider === "cohere") {
+            inputBody.prompt = prompt;
+            inputBody.max_tokens = maxTokens;
+            inputBody.temperature = temperature;
+            inputBody.stop_sequences = stopSequences;
+            if (bedrockMethod === "invoke-with-response-stream") {
+                inputBody.stream = true;
+            }
+        }
         return { ...inputBody, ...modelKwargs };
     }
     /**
@@ -45,6 +54,9 @@ export class BedrockLLMInputOutputAdapter {
         else if (provider === "ai21") {
             return responseBody?.completions?.[0]?.data?.text ?? "";
         }
+        else if (provider === "cohere") {
+            return responseBody?.generations?.[0]?.text ?? responseBody?.text ?? "";
+        }
         // I haven't been able to get a response with more than one result in it.
         return responseBody.results?.[0]?.outputText;
     }

package/dist/util/fast-json-patch/index.cjs

@@ -28,6 +28,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.unescapePathComponent = exports.escapePathComponent = exports.deepClone = exports.JsonPatchError = void 0;
 __exportStar(require("./src/core.cjs"), exports);
+__exportStar(require("./src/duplex.cjs"), exports);
 var helpers_js_1 = require("./src/helpers.cjs");
 Object.defineProperty(exports, "JsonPatchError", { enumerable: true, get: function () { return helpers_js_1.PatchError; } });
 Object.defineProperty(exports, "deepClone", { enumerable: true, get: function () { return helpers_js_1._deepClone; } });

package/dist/util/fast-json-patch/index.d.ts

@@ -1,4 +1,5 @@
 export * from "./src/core.js";
+export * from "./src/duplex.js";
 export { PatchError as JsonPatchError, _deepClone as deepClone, escapePathComponent, unescapePathComponent, } from "./src/helpers.js";
 /**
  * Default export for backwards compat

package/dist/util/fast-json-patch/index.js

@@ -1,4 +1,5 @@
 export * from "./src/core.js";
+export * from "./src/duplex.js";
 export { PatchError as JsonPatchError, _deepClone as deepClone, escapePathComponent, unescapePathComponent, } from "./src/helpers.js";
 /**
  * Default export for backwards compat