langchain 0.0.203 → 0.0.204-rc.1

package/dist/stores/message/momento.js

@@ -1,158 +1 @@
-/* eslint-disable no-instanceof/no-instanceof */
-import { CacheDelete, CacheListFetch, CacheListPushBack, InvalidArgumentError, CollectionTtl, } from "@gomomento/sdk-core";
-import { BaseListChatMessageHistory, } from "../../schema/index.js";
-import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
-import { ensureCacheExists } from "../../util/momento.js";
-/**
- * A class that stores chat message history using Momento Cache. It
- * interacts with a Momento cache client to perform operations like
- * fetching, adding, and deleting messages.
- * @example
- * ```typescript
- * const chatHistory = await MomentoChatMessageHistory.fromProps({
- *   client: new CacheClient({
- *     configuration: Configurations.Laptop.v1(),
- *     credentialProvider: CredentialProvider.fromEnvironmentVariable({
- *       environmentVariableName: "MOMENTO_API_KEY",
- *     }),
- *     defaultTtlSeconds: 60 * 60 * 24,
- *   }),
- *   cacheName: "langchain",
- *   sessionId: new Date().toISOString(),
- *   sessionTtl: 300,
- * });
- *
- * const messages = await chatHistory.getMessages();
- * console.log({ messages });
- * ```
- */
-export class MomentoChatMessageHistory extends BaseListChatMessageHistory {
-    constructor(props) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "momento"]
-        });
-        Object.defineProperty(this, "sessionId", {
-            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, "cacheName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "sessionTtl", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.sessionId = props.sessionId;
-        this.client = props.client;
-        this.cacheName = props.cacheName;
-        this.validateTtlSeconds(props.sessionTtl);
-        this.sessionTtl =
-            props.sessionTtl !== undefined
-                ? CollectionTtl.of(props.sessionTtl)
-                : CollectionTtl.fromCacheTtl();
-    }
-    /**
-     * Create a new chat message history backed by Momento.
-     *
-     * @param {MomentoCacheProps} props The settings to instantiate the Momento chat message history.
-     * @param {string} props.sessionId The session ID to use to store the data.
-     * @param {ICacheClient} props.client The Momento cache client.
-     * @param {string} props.cacheName The name of the cache to use to store the data.
-     * @param {number} props.sessionTtl The time to live for the cache items in seconds.
-     * If not specified, the cache client default is used.
-     * @param {boolean} props.ensureCacheExists If true, ensure that the cache exists before returning.
-     * If false, the cache is not checked for existence.
-     * @throws {InvalidArgumentError} If {@link props.sessionTtl} is not strictly positive.
-     * @returns A new chat message history backed by Momento.
-     */
-    static async fromProps(props) {
-        const instance = new MomentoChatMessageHistory(props);
-        if (props.ensureCacheExists || props.ensureCacheExists === undefined) {
-            await ensureCacheExists(props.client, props.cacheName);
-        }
-        return instance;
-    }
-    /**
-     * Validate the user-specified TTL, if provided, is strictly positive.
-     * @param ttlSeconds The TTL to validate.
-     */
-    validateTtlSeconds(ttlSeconds) {
-        if (ttlSeconds !== undefined && ttlSeconds <= 0) {
-            throw new InvalidArgumentError("ttlSeconds must be positive.");
-        }
-    }
-    /**
-     * Fetches messages from the cache.
-     * @returns A Promise that resolves to an array of BaseMessage instances.
-     */
-    async getMessages() {
-        const fetchResponse = await this.client.listFetch(this.cacheName, this.sessionId);
-        let messages = [];
-        if (fetchResponse instanceof CacheListFetch.Hit) {
-            messages = fetchResponse
-                .valueList()
-                .map((serializedStoredMessage) => JSON.parse(serializedStoredMessage));
-        }
-        else if (fetchResponse instanceof CacheListFetch.Miss) {
-            // pass
-        }
-        else if (fetchResponse instanceof CacheListFetch.Error) {
-            throw fetchResponse.innerException();
-        }
-        else {
-            throw new Error(`Unknown response type: ${fetchResponse.toString()}`);
-        }
-        return mapStoredMessagesToChatMessages(messages);
-    }
-    /**
-     * Adds a message to the cache.
-     * @param message The BaseMessage instance to add to the cache.
-     * @returns A Promise that resolves when the message has been added.
-     */
-    async addMessage(message) {
-        const messageToAdd = JSON.stringify(mapChatMessagesToStoredMessages([message])[0]);
-        const pushResponse = await this.client.listPushBack(this.cacheName, this.sessionId, messageToAdd, { ttl: this.sessionTtl });
-        if (pushResponse instanceof CacheListPushBack.Success) {
-            // pass
-        }
-        else if (pushResponse instanceof CacheListPushBack.Error) {
-            throw pushResponse.innerException();
-        }
-        else {
-            throw new Error(`Unknown response type: ${pushResponse.toString()}`);
-        }
-    }
-    /**
-     * Deletes all messages from the cache.
-     * @returns A Promise that resolves when all messages have been deleted.
-     */
-    async clear() {
-        const deleteResponse = await this.client.delete(this.cacheName, this.sessionId);
-        if (deleteResponse instanceof CacheDelete.Success) {
-            // pass
-        }
-        else if (deleteResponse instanceof CacheDelete.Error) {
-            throw deleteResponse.innerException();
-        }
-        else {
-            throw new Error(`Unknown response type: ${deleteResponse.toString()}`);
-        }
-    }
-}
+export * from "@langchain/community/stores/message/momento";

package/dist/stores/message/mongodb.cjs

@@ -1,59 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MongoDBChatMessageHistory = void 0;
-const mongodb_1 = require("mongodb");
-const index_js_1 = require("../../schema/index.cjs");
-const utils_js_1 = require("./utils.cjs");
-/**
- * @example
- * ```typescript
- * const chatHistory = new MongoDBChatMessageHistory({
- *   collection: myCollection,
- *   sessionId: 'unique-session-id',
- * });
- * const messages = await chatHistory.getMessages();
- * await chatHistory.clear();
- * ```
- */
-class MongoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
-    constructor({ collection, sessionId }) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "mongodb"]
-        });
-        Object.defineProperty(this, "collection", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "sessionId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.collection = collection;
-        this.sessionId = sessionId;
-    }
-    async getMessages() {
-        const document = await this.collection.findOne({
-            _id: new mongodb_1.ObjectId(this.sessionId),
-        });
-        const messages = document?.messages || [];
-        return (0, utils_js_1.mapStoredMessagesToChatMessages)(messages);
-    }
-    async addMessage(message) {
-        const messages = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
-        await this.collection.updateOne({ _id: new mongodb_1.ObjectId(this.sessionId) }, {
-            $push: { messages: { $each: messages } },
-        }, { upsert: true });
+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.collection.deleteOne({ _id: new mongodb_1.ObjectId(this.sessionId) });
-    }
-}
-exports.MongoDBChatMessageHistory = MongoDBChatMessageHistory;
+    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/mongodb"), exports);

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

@@ -1,26 +1 @@
-import { Collection, Document as MongoDBDocument } from "mongodb";
-import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
-export interface MongoDBChatMessageHistoryInput {
-    collection: Collection<MongoDBDocument>;
-    sessionId: string;
-}
-/**
- * @example
- * ```typescript
- * const chatHistory = new MongoDBChatMessageHistory({
- *   collection: myCollection,
- *   sessionId: 'unique-session-id',
- * });
- * const messages = await chatHistory.getMessages();
- * await chatHistory.clear();
- * ```
- */
-export declare class MongoDBChatMessageHistory extends BaseListChatMessageHistory {
-    lc_namespace: string[];
-    private collection;
-    private sessionId;
-    constructor({ collection, sessionId }: MongoDBChatMessageHistoryInput);
-    getMessages(): Promise<BaseMessage[]>;
-    addMessage(message: BaseMessage): Promise<void>;
-    clear(): Promise<void>;
-}
+export * from "@langchain/community/stores/message/mongodb";

package/dist/stores/message/mongodb.js

@@ -1,55 +1 @@
-import { ObjectId } from "mongodb";
-import { BaseListChatMessageHistory } from "../../schema/index.js";
-import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
-/**
- * @example
- * ```typescript
- * const chatHistory = new MongoDBChatMessageHistory({
- *   collection: myCollection,
- *   sessionId: 'unique-session-id',
- * });
- * const messages = await chatHistory.getMessages();
- * await chatHistory.clear();
- * ```
- */
-export class MongoDBChatMessageHistory extends BaseListChatMessageHistory {
-    constructor({ collection, sessionId }) {
-        super();
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "mongodb"]
-        });
-        Object.defineProperty(this, "collection", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "sessionId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.collection = collection;
-        this.sessionId = sessionId;
-    }
-    async getMessages() {
-        const document = await this.collection.findOne({
-            _id: new ObjectId(this.sessionId),
-        });
-        const messages = document?.messages || [];
-        return mapStoredMessagesToChatMessages(messages);
-    }
-    async addMessage(message) {
-        const messages = mapChatMessagesToStoredMessages([message]);
-        await this.collection.updateOne({ _id: new ObjectId(this.sessionId) }, {
-            $push: { messages: { $each: messages } },
-        }, { upsert: true });
-    }
-    async clear() {
-        await this.collection.deleteOne({ _id: new ObjectId(this.sessionId) });
-    }
-}
+export * from "@langchain/community/stores/message/mongodb";

package/dist/stores/message/planetscale.cjs

@@ -1,172 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PlanetScaleChatMessageHistory = void 0;
-const database_1 = require("@planetscale/database");
-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
- * PlanetScale database. Extends the BaseListChatMessageHistory class.
- * @example
- * ```typescript
- * const chatHistory = new PlanetScaleChatMessageHistory({
- *   tableName: "stored_message",
- *   sessionId: "lc-example",
- *   config: {
- *     url: "ADD_YOURS_HERE",
- *   },
- * });
- * const chain = new ConversationChain({
- *   llm: new ChatOpenAI(),
- *   memory: chatHistory,
- * });
- * const response = await chain.invoke({
- *   input: "What did I just say my name was?",
- * });
- * console.log({ response });
- * ```
- */
-class PlanetScaleChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
-    get lc_secrets() {
-        return {
-            "config.host": "PLANETSCALE_HOST",
-            "config.username": "PLANETSCALE_USERNAME",
-            "config.password": "PLANETSCALE_PASSWORD",
-            "config.url": "PLANETSCALE_DATABASE_URL",
-        };
-    }
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "planetscale"]
-        });
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "connection", {
-            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, config, client, tableName } = fields;
-        if (client) {
-            this.client = client;
-        }
-        else if (config) {
-            this.client = new database_1.Client(config);
-        }
-        else {
-            throw new Error("Either a client or config must be provided to PlanetScaleChatMessageHistory");
-        }
-        this.connection = this.client.connection();
-        this.tableName = tableName || "langchain_chat_histories";
-        this.tableInitialized = false;
-        this.sessionId = sessionId;
-    }
-    /**
-     * Private method to ensure that the necessary table exists in the
-     * PlanetScale 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 BINARY(16) PRIMARY KEY, session_id VARCHAR(255), type VARCHAR(255), content VARCHAR(255), role VARCHAR(255), name VARCHAR(255), additional_kwargs VARCHAR(255));`;
-        await this.connection.execute(query);
-        const indexQuery = `ALTER TABLE ${this.tableName} MODIFY id BINARY(16) DEFAULT (UUID_TO_BIN(UUID()));`;
-        await this.connection.execute(indexQuery);
-        this.tableInitialized = true;
-    }
-    /**
-     * Method to retrieve all messages from the PlanetScale 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 = :session_id`;
-        const params = {
-            session_id: this.sessionId,
-        };
-        const rawStoredMessages = await this.connection.execute(query, params);
-        const storedMessagesObject = rawStoredMessages.rows;
-        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 PlanetScale 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} (session_id, type, content, role, name, additional_kwargs) VALUES (:session_id, :type, :content, :role, :name, :additional_kwargs)`;
-        const params = {
-            session_id: this.sessionId,
-            type: messageToAdd[0].type,
-            content: messageToAdd[0].data.content,
-            role: messageToAdd[0].data.role,
-            name: messageToAdd[0].data.name,
-            additional_kwargs: JSON.stringify(messageToAdd[0].data.additional_kwargs),
-        };
-        await this.connection.execute(query, params);
-    }
-    /**
-     * Method to delete all messages from the PlanetScale database for the
-     * current session.
-     * @returns Promise that resolves to void.
-     */
-    async clear() {
-        await this.ensureTable();
-        const query = `DELETE FROM ${this.tableName} WHERE session_id = :session_id`;
-        const params = {
-            session_id: this.sessionId,
-        };
-        await this.connection.execute(query, params);
-    }
-}
-exports.PlanetScaleChatMessageHistory = PlanetScaleChatMessageHistory;
+    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/planetscale"), exports);

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

@@ -1,75 +1 @@
-import { Client as PlanetScaleClient, Config as PlanetScaleConfig } from "@planetscale/database";
-import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
-/**
- * Type definition for the input parameters required when instantiating a
- * PlanetScaleChatMessageHistory object.
- */
-export type PlanetScaleChatMessageHistoryInput = {
-    tableName?: string;
-    sessionId: string;
-    config?: PlanetScaleConfig;
-    client?: PlanetScaleClient;
-};
-/**
- * Class for storing and retrieving chat message history from a
- * PlanetScale database. Extends the BaseListChatMessageHistory class.
- * @example
- * ```typescript
- * const chatHistory = new PlanetScaleChatMessageHistory({
- *   tableName: "stored_message",
- *   sessionId: "lc-example",
- *   config: {
- *     url: "ADD_YOURS_HERE",
- *   },
- * });
- * const chain = new ConversationChain({
- *   llm: new ChatOpenAI(),
- *   memory: chatHistory,
- * });
- * const response = await chain.invoke({
- *   input: "What did I just say my name was?",
- * });
- * console.log({ response });
- * ```
- */
-export declare class PlanetScaleChatMessageHistory extends BaseListChatMessageHistory {
-    lc_namespace: string[];
-    get lc_secrets(): {
-        "config.host": string;
-        "config.username": string;
-        "config.password": string;
-        "config.url": string;
-    };
-    client: PlanetScaleClient;
-    private connection;
-    private tableName;
-    private sessionId;
-    private tableInitialized;
-    constructor(fields: PlanetScaleChatMessageHistoryInput);
-    /**
-     * Private method to ensure that the necessary table exists in the
-     * PlanetScale 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 PlanetScale 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 PlanetScale 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 PlanetScale database for the
-     * current session.
-     * @returns Promise that resolves to void.
-     */
-    clear(): Promise<void>;
-}
+export * from "@langchain/community/stores/message/planetscale";