langchain 0.0.203 → 0.0.204-rc.0

package/dist/stores/message/cloudflare_d1.cjs

@@ -1,152 +1,17 @@
 "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.
- * @example
- * ```typescript
- * const memory = new BufferMemory({
- *   returnMessages: true,
- *   chatHistory: new CloudflareD1MessageHistory({
- *     tableName: "stored_message",
- *     sessionId: "example",
- *     database: env.DB,
- *   }),
- * });
- *
- * const chainInput = { input };
- *
- * const res = await memory.chatHistory.invoke(chainInput);
- * await memory.saveContext(chainInput, {
- *   output: res,
- * });
- * ```
- */
-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);
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
     }
-    /**
-     * 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;
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("@langchain/community/stores/message/cloudflare_d1"), exports);

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

@@ -1,67 +1 @@
-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.
- * @example
- * ```typescript
- * const memory = new BufferMemory({
- *   returnMessages: true,
- *   chatHistory: new CloudflareD1MessageHistory({
- *     tableName: "stored_message",
- *     sessionId: "example",
- *     database: env.DB,
- *   }),
- * });
- *
- * const chainInput = { input };
- *
- * const res = await memory.chatHistory.invoke(chainInput);
- * await memory.saveContext(chainInput, {
- *   output: res,
- * });
- * ```
- */
-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>;
-}
+export * from "@langchain/community/stores/message/cloudflare_d1";

package/dist/stores/message/cloudflare_d1.js

@@ -1,148 +1 @@
-import { v4 } from "uuid";
-import { BaseListChatMessageHistory, } from "../../schema/index.js";
-import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
-/**
- * Class for storing and retrieving chat message history from a
- * Cloudflare D1 database. Extends the BaseListChatMessageHistory class.
- * @example
- * ```typescript
- * const memory = new BufferMemory({
- *   returnMessages: true,
- *   chatHistory: new CloudflareD1MessageHistory({
- *     tableName: "stored_message",
- *     sessionId: "example",
- *     database: env.DB,
- *   }),
- * });
- *
- * const chainInput = { input };
- *
- * const res = await memory.chatHistory.invoke(chainInput);
- * await memory.saveContext(chainInput, {
- *   output: res,
- * });
- * ```
- */
-export class CloudflareD1MessageHistory extends 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 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 = mapChatMessagesToStoredMessages([message]);
-        const query = `INSERT INTO ${this.tableName} (id, session_id, type, content, role, name, additional_kwargs) VALUES(?, ?, ?, ?, ?, ?, ?)`;
-        const id = 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();
-    }
-}
+export * from "@langchain/community/stores/message/cloudflare_d1";

package/dist/stores/message/convex.cjs

@@ -1,120 +1,17 @@
 "use strict";
-/* eslint-disable @typescript-eslint/no-explicit-any */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ConvexChatMessageHistory = void 0;
-// eslint-disable-next-line import/no-extraneous-dependencies
-const server_1 = require("convex/server");
-const index_js_1 = require("../../schema/index.cjs");
-const utils_js_1 = require("./utils.cjs");
-class ConvexChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
-    constructor(config) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "convex"]
-        });
-        Object.defineProperty(this, "ctx", {
-            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, "table", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "index", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "sessionIdField", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "messageTextFieldName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "insert", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "lookup", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "deleteMany", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.ctx = config.ctx;
-        this.sessionId = config.sessionId;
-        this.table = config.table ?? "messages";
-        this.index = config.index ?? "bySessionId";
-        this.sessionIdField =
-            config.sessionIdField ?? "sessionId";
-        this.messageTextFieldName =
-            config.messageTextFieldName ?? "message";
-        this.insert =
-            config.insert ?? (0, server_1.makeFunctionReference)("langchain/db:insert");
-        this.lookup =
-            config.lookup ?? (0, server_1.makeFunctionReference)("langchain/db:lookup");
-        this.deleteMany =
-            config.deleteMany ??
-                (0, server_1.makeFunctionReference)("langchain/db:deleteMany");
-    }
-    async getMessages() {
-        const convexDocuments = await this.ctx.runQuery(this.lookup, {
-            table: this.table,
-            index: this.index,
-            keyField: this.sessionIdField,
-            key: this.sessionId,
-        });
-        return (0, utils_js_1.mapStoredMessagesToChatMessages)(convexDocuments.map((doc) => doc[this.messageTextFieldName]));
-    }
-    async addMessage(message) {
-        const messages = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
-        // TODO: Remove chunking when Convex handles the concurrent requests correctly
-        const PAGE_SIZE = 16;
-        for (let i = 0; i < messages.length; i += PAGE_SIZE) {
-            await Promise.all(messages.slice(i, i + PAGE_SIZE).map((message) => this.ctx.runMutation(this.insert, {
-                table: this.table,
-                document: {
-                    [this.sessionIdField]: this.sessionId,
-                    [this.messageTextFieldName]: message,
-                },
-            })));
-        }
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
     }
-    async clear() {
-        await this.ctx.runMutation(this.deleteMany, {
-            table: this.table,
-            index: this.index,
-            keyField: this.sessionIdField,
-            key: this.sessionId,
-        });
-    }
-}
-exports.ConvexChatMessageHistory = ConvexChatMessageHistory;
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("@langchain/community/stores/message/convex"), exports);

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

@@ -1,81 +1 @@
-import { DocumentByName, FieldPaths, FunctionReference, GenericActionCtx, GenericDataModel, NamedTableInfo, TableNamesInDataModel, IndexNames } from "convex/server";
-import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
-/**
- * Type that defines the config required to initialize the
- * ConvexChatMessageHistory class. At minimum it needs a sessionId
- * and an ActionCtx.
- */
-export type ConvexChatMessageHistoryInput<DataModel extends GenericDataModel, TableName extends TableNamesInDataModel<DataModel> = "messages", IndexName extends IndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", SessionIdFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "sessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
-    table: string;
-    document: object;
-}> = any, LookupQuery extends FunctionReference<"query", "internal", {
-    table: string;
-    index: string;
-    keyField: string;
-    key: string;
-}, object[]> = any, DeleteManyMutation extends FunctionReference<"mutation", "internal", {
-    table: string;
-    index: string;
-    keyField: string;
-    key: string;
-}> = 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 IndexNames<NamedTableInfo<DataModel, TableName>> = "bySessionId", MessageTextFieldName extends FieldPaths<NamedTableInfo<DataModel, TableName>> = "message", InsertMutation extends FunctionReference<"mutation", "internal", {
-    table: string;
-    document: object;
-}> = any, LookupQuery extends FunctionReference<"query", "internal", {
-    table: string;
-    index: string;
-    keyField: string;
-    key: string;
-}, object[]> = any, DeleteManyMutation extends FunctionReference<"mutation", "internal", {
-    table: string;
-    index: string;
-    keyField: string;
-    key: string;
-}> = any> extends BaseListChatMessageHistory {
-    lc_namespace: string[];
-    private readonly ctx;
-    private readonly sessionId;
-    private readonly table;
-    private readonly index;
-    private readonly sessionIdField;
-    private readonly messageTextFieldName;
-    private readonly insert;
-    private readonly lookup;
-    private readonly deleteMany;
-    constructor(config: ConvexChatMessageHistoryInput<DataModel, TableName, IndexName, SessionIdFieldName, MessageTextFieldName, InsertMutation, LookupQuery, DeleteManyMutation>);
-    getMessages(): Promise<BaseMessage[]>;
-    addMessage(message: BaseMessage): Promise<void>;
-    clear(): Promise<void>;
-}
+export * from "@langchain/community/stores/message/convex";