langchain 0.0.203 → 0.0.204-rc.1

package/dist/stores/message/upstash_redis.js

@@ -1,83 +1 @@
-import { Redis } from "@upstash/redis";
-import { BaseListChatMessageHistory, } from "../../schema/index.js";
-import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
-/**
- * Class used to store chat message history in Redis. It provides methods
- * to add, get, and clear messages.
- */
-export class UpstashRedisChatMessageHistory extends BaseListChatMessageHistory {
-    get lc_secrets() {
-        return {
-            "config.url": "UPSTASH_REDIS_REST_URL",
-            "config.token": "UPSTASH_REDIS_REST_TOKEN",
-        };
-    }
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "upstash_redis"]
-        });
-        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, "sessionTTL", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        const { sessionId, sessionTTL, config, client } = fields;
-        if (client) {
-            this.client = client;
-        }
-        else if (config) {
-            this.client = new Redis(config);
-        }
-        else {
-            throw new Error(`Upstash Redis message stores require either a config object or a pre-configured client.`);
-        }
-        this.sessionId = sessionId;
-        this.sessionTTL = sessionTTL;
-    }
-    /**
-     * Retrieves the chat messages from the Redis database.
-     * @returns An array of BaseMessage instances representing the chat history.
-     */
-    async getMessages() {
-        const rawStoredMessages = await this.client.lrange(this.sessionId, 0, -1);
-        const orderedMessages = rawStoredMessages.reverse();
-        const previousMessages = orderedMessages.filter((x) => x.type !== undefined && x.data.content !== undefined);
-        return mapStoredMessagesToChatMessages(previousMessages);
-    }
-    /**
-     * Adds a new message to the chat history in the Redis database.
-     * @param message The message to be added to the chat history.
-     * @returns Promise resolving to void.
-     */
-    async addMessage(message) {
-        const messageToAdd = mapChatMessagesToStoredMessages([message]);
-        await this.client.lpush(this.sessionId, JSON.stringify(messageToAdd[0]));
-        if (this.sessionTTL) {
-            await this.client.expire(this.sessionId, this.sessionTTL);
-        }
-    }
-    /**
-     * Deletes all messages from the chat history in the Redis database.
-     * @returns Promise resolving to void.
-     */
-    async clear() {
-        await this.client.del(this.sessionId);
-    }
-}
+export * from "@langchain/community/stores/message/upstash_redis";

package/dist/stores/message/utils.cjs

@@ -1,26 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = void 0;
-const index_js_1 = require("../../schema/index.cjs");
-/**
- * Transforms an array of `StoredMessage` instances into an array of
- * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
- * function to ensure all messages are in the `StoredMessage` format, then
- * creates new instances of the appropriate `BaseMessage` subclass based
- * on the type of each message. This function is used to prepare stored
- * messages for use in a chat context.
- */
-function mapStoredMessagesToChatMessages(messages) {
-    return messages.map(index_js_1.mapStoredMessageToChatMessage);
-}
-exports.mapStoredMessagesToChatMessages = mapStoredMessagesToChatMessages;
-/**
- * Transforms an array of `BaseMessage` instances into an array of
- * `StoredMessage` instances. It does this by calling the `toDict` method
- * on each `BaseMessage`, which returns a `StoredMessage`. This function
- * is used to prepare chat messages for storage.
- */
-function mapChatMessagesToStoredMessages(messages) {
-    return messages.map((message) => message.toDict());
-}
-exports.mapChatMessagesToStoredMessages = mapChatMessagesToStoredMessages;
+var messages_1 = require("@langchain/core/messages");
+Object.defineProperty(exports, "mapStoredMessagesToChatMessages", { enumerable: true, get: function () { return messages_1.mapStoredMessagesToChatMessages; } });
+Object.defineProperty(exports, "mapChatMessagesToStoredMessages", { enumerable: true, get: function () { return messages_1.mapChatMessagesToStoredMessages; } });

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

@@ -1,17 +1 @@
-import { BaseMessage, StoredMessage } from "../../schema/index.js";
-/**
- * Transforms an array of `StoredMessage` instances into an array of
- * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
- * function to ensure all messages are in the `StoredMessage` format, then
- * creates new instances of the appropriate `BaseMessage` subclass based
- * on the type of each message. This function is used to prepare stored
- * messages for use in a chat context.
- */
-export declare function mapStoredMessagesToChatMessages(messages: StoredMessage[]): BaseMessage[];
-/**
- * Transforms an array of `BaseMessage` instances into an array of
- * `StoredMessage` instances. It does this by calling the `toDict` method
- * on each `BaseMessage`, which returns a `StoredMessage`. This function
- * is used to prepare chat messages for storage.
- */
-export declare function mapChatMessagesToStoredMessages(messages: BaseMessage[]): StoredMessage[];
+export { mapStoredMessagesToChatMessages, mapChatMessagesToStoredMessages, } from "@langchain/core/messages";

package/dist/stores/message/utils.js

@@ -1,21 +1 @@
-import { mapStoredMessageToChatMessage, } from "../../schema/index.js";
-/**
- * Transforms an array of `StoredMessage` instances into an array of
- * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
- * function to ensure all messages are in the `StoredMessage` format, then
- * creates new instances of the appropriate `BaseMessage` subclass based
- * on the type of each message. This function is used to prepare stored
- * messages for use in a chat context.
- */
-export function mapStoredMessagesToChatMessages(messages) {
-    return messages.map(mapStoredMessageToChatMessage);
-}
-/**
- * Transforms an array of `BaseMessage` instances into an array of
- * `StoredMessage` instances. It does this by calling the `toDict` method
- * on each `BaseMessage`, which returns a `StoredMessage`. This function
- * is used to prepare chat messages for storage.
- */
-export function mapChatMessagesToStoredMessages(messages) {
-    return messages.map((message) => message.toDict());
-}
+export { mapStoredMessagesToChatMessages, mapChatMessagesToStoredMessages, } from "@langchain/core/messages";

package/dist/stores/message/xata.cjs

@@ -1,218 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.XataChatMessageHistory = void 0;
-const client_1 = require("@xata.io/client");
-const index_js_1 = require("../../schema/index.cjs");
-const utils_js_1 = require("./utils.cjs");
-const chatMemoryColumns = [
-    { name: "sessionId", type: "string" },
-    { name: "type", type: "string" },
-    { name: "role", type: "string" },
-    { name: "content", type: "text" },
-    { name: "name", type: "string" },
-    { name: "additionalKwargs", type: "text" },
-];
-/**
- * A class for managing chat message history using Xata.io client. It
- * extends the BaseListChatMessageHistory class and provides methods to
- * get, add, and clear messages. It also ensures the existence of a table
- * where the chat messages are stored.
- * @example
- * ```typescript
- * const chatHistory = new XataChatMessageHistory({
- *   table: "messages",
- *   sessionId: new Date().toISOString(),
- *   client: new BaseClient({
- *     databaseURL: process.env.XATA_DB_URL,
- *     apiKey: process.env.XATA_API_KEY,
- *     branch: "main",
- *   }),
- *   apiKey: process.env.XATA_API_KEY,
- * });
- *
- * const chain = new ConversationChain({
- *   llm: new ChatOpenAI(),
- *   memory: new BufferMemory({ chatHistory }),
- * });
- *
- * const response = await chain.invoke({
- *   input: "What did I just say my name was?",
- * });
- * console.log({ response });
- * ```
- */
-class XataChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "xata"]
-        });
-        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, "table", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "tableInitialized", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "createTable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "apiClient", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        const { sessionId, config, client, table } = fields;
-        this.sessionId = sessionId;
-        this.table = table || "memory";
-        if (client) {
-            this.client = client;
-        }
-        else if (config) {
-            this.client = new client_1.BaseClient(config);
-        }
-        else {
-            throw new Error("Either a client or a config must be provided to XataChatMessageHistoryInput");
-        }
-        if (fields.createTable !== false) {
-            this.createTable = true;
-            const apiKey = fields.apiKey || fields.config?.apiKey;
-            if (!apiKey) {
-                throw new Error("If createTable is set, an apiKey must be provided to XataChatMessageHistoryInput, either directly or through the config object");
-            }
-            this.apiClient = new client_1.XataApiClient({ apiKey });
-        }
-        else {
-            this.createTable = false;
-        }
-        this.tableInitialized = false;
-    }
-    /**
-     * Retrieves all messages associated with the session ID, ordered by
-     * creation time.
-     * @returns A promise that resolves to an array of BaseMessage instances.
-     */
-    async getMessages() {
-        await this.ensureTable();
-        const records = await this.client.db[this.table]
-            .filter({ sessionId: this.sessionId })
-            .sort("xata.createdAt", "asc")
-            .getAll();
-        const rawStoredMessages = records;
-        const orderedMessages = rawStoredMessages.map((message) => {
-            const data = {
-                content: message.content,
-                additional_kwargs: JSON.parse(message.additionalKwargs),
-            };
-            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);
-    }
-    /**
-     * Adds a new message to the database.
-     * @param message The BaseMessage instance to be added.
-     * @returns A promise that resolves when the message has been added.
-     */
-    async addMessage(message) {
-        await this.ensureTable();
-        const messageToAdd = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
-        await this.client.db[this.table].create({
-            sessionId: this.sessionId,
-            type: messageToAdd[0].type,
-            content: messageToAdd[0].data.content,
-            role: messageToAdd[0].data.role,
-            name: messageToAdd[0].data.name,
-            additionalKwargs: JSON.stringify(messageToAdd[0].data.additional_kwargs),
-        });
+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]; } };
     }
-    /**
-     * Deletes all messages associated with the session ID.
-     * @returns A promise that resolves when the messages have been deleted.
-     */
-    async clear() {
-        await this.ensureTable();
-        const records = await this.client.db[this.table]
-            .select(["id"])
-            .filter({ sessionId: this.sessionId })
-            .getAll();
-        const ids = records.map((m) => m.id);
-        await this.client.db[this.table].delete(ids);
-    }
-    /**
-     * Checks if the table exists and creates it if it doesn't. This method is
-     * called before any operation on the table.
-     * @returns A promise that resolves when the table has been ensured.
-     */
-    async ensureTable() {
-        if (!this.createTable) {
-            return;
-        }
-        if (this.tableInitialized) {
-            return;
-        }
-        const { databaseURL, branch } = await this.client.getConfig();
-        const [, , host, , database] = databaseURL.split("/");
-        const urlParts = (0, client_1.parseWorkspacesUrlParts)(host);
-        if (urlParts == null) {
-            throw new Error("Invalid databaseURL");
-        }
-        const { workspace, region } = urlParts;
-        const tableParams = {
-            workspace,
-            region,
-            database,
-            branch,
-            table: this.table,
-        };
-        let schema = null;
-        try {
-            schema = await this.apiClient.tables.getTableSchema(tableParams);
-        }
-        catch (e) {
-            // pass
-        }
-        if (schema == null) {
-            await this.apiClient.tables.createTable(tableParams);
-            await this.apiClient.tables.setTableSchema({
-                ...tableParams,
-                schema: {
-                    columns: chatMemoryColumns,
-                },
-            });
-        }
-    }
-}
-exports.XataChatMessageHistory = XataChatMessageHistory;
+    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/xata"), exports);

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

@@ -1,76 +1 @@
-import { BaseClient, BaseClientOptions } from "@xata.io/client";
-import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
-/**
- * An object type that represents the input for the XataChatMessageHistory
- * class.
- */
-export type XataChatMessageHistoryInput<XataClient> = {
-    sessionId: string;
-    config?: BaseClientOptions;
-    client?: XataClient;
-    table?: string;
-    createTable?: boolean;
-    apiKey?: string;
-};
-/**
- * A class for managing chat message history using Xata.io client. It
- * extends the BaseListChatMessageHistory class and provides methods to
- * get, add, and clear messages. It also ensures the existence of a table
- * where the chat messages are stored.
- * @example
- * ```typescript
- * const chatHistory = new XataChatMessageHistory({
- *   table: "messages",
- *   sessionId: new Date().toISOString(),
- *   client: new BaseClient({
- *     databaseURL: process.env.XATA_DB_URL,
- *     apiKey: process.env.XATA_API_KEY,
- *     branch: "main",
- *   }),
- *   apiKey: process.env.XATA_API_KEY,
- * });
- *
- * const chain = new ConversationChain({
- *   llm: new ChatOpenAI(),
- *   memory: new BufferMemory({ chatHistory }),
- * });
- *
- * const response = await chain.invoke({
- *   input: "What did I just say my name was?",
- * });
- * console.log({ response });
- * ```
- */
-export declare class XataChatMessageHistory<XataClient extends BaseClient> extends BaseListChatMessageHistory {
-    lc_namespace: string[];
-    client: XataClient;
-    private sessionId;
-    private table;
-    private tableInitialized;
-    private createTable;
-    private apiClient;
-    constructor(fields: XataChatMessageHistoryInput<XataClient>);
-    /**
-     * Retrieves all messages associated with the session ID, ordered by
-     * creation time.
-     * @returns A promise that resolves to an array of BaseMessage instances.
-     */
-    getMessages(): Promise<BaseMessage[]>;
-    /**
-     * Adds a new message to the database.
-     * @param message The BaseMessage instance to be added.
-     * @returns A promise that resolves when the message has been added.
-     */
-    addMessage(message: BaseMessage): Promise<void>;
-    /**
-     * Deletes all messages associated with the session ID.
-     * @returns A promise that resolves when the messages have been deleted.
-     */
-    clear(): Promise<void>;
-    /**
-     * Checks if the table exists and creates it if it doesn't. This method is
-     * called before any operation on the table.
-     * @returns A promise that resolves when the table has been ensured.
-     */
-    private ensureTable;
-}
+export * from "@langchain/community/stores/message/xata";

package/dist/stores/message/xata.js

@@ -1,214 +1 @@
-import { BaseClient, XataApiClient, parseWorkspacesUrlParts, } from "@xata.io/client";
-import { BaseListChatMessageHistory, } from "../../schema/index.js";
-import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
-const chatMemoryColumns = [
-    { name: "sessionId", type: "string" },
-    { name: "type", type: "string" },
-    { name: "role", type: "string" },
-    { name: "content", type: "text" },
-    { name: "name", type: "string" },
-    { name: "additionalKwargs", type: "text" },
-];
-/**
- * A class for managing chat message history using Xata.io client. It
- * extends the BaseListChatMessageHistory class and provides methods to
- * get, add, and clear messages. It also ensures the existence of a table
- * where the chat messages are stored.
- * @example
- * ```typescript
- * const chatHistory = new XataChatMessageHistory({
- *   table: "messages",
- *   sessionId: new Date().toISOString(),
- *   client: new BaseClient({
- *     databaseURL: process.env.XATA_DB_URL,
- *     apiKey: process.env.XATA_API_KEY,
- *     branch: "main",
- *   }),
- *   apiKey: process.env.XATA_API_KEY,
- * });
- *
- * const chain = new ConversationChain({
- *   llm: new ChatOpenAI(),
- *   memory: new BufferMemory({ chatHistory }),
- * });
- *
- * const response = await chain.invoke({
- *   input: "What did I just say my name was?",
- * });
- * console.log({ response });
- * ```
- */
-export class XataChatMessageHistory extends BaseListChatMessageHistory {
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "stores", "message", "xata"]
-        });
-        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, "table", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "tableInitialized", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "createTable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "apiClient", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        const { sessionId, config, client, table } = fields;
-        this.sessionId = sessionId;
-        this.table = table || "memory";
-        if (client) {
-            this.client = client;
-        }
-        else if (config) {
-            this.client = new BaseClient(config);
-        }
-        else {
-            throw new Error("Either a client or a config must be provided to XataChatMessageHistoryInput");
-        }
-        if (fields.createTable !== false) {
-            this.createTable = true;
-            const apiKey = fields.apiKey || fields.config?.apiKey;
-            if (!apiKey) {
-                throw new Error("If createTable is set, an apiKey must be provided to XataChatMessageHistoryInput, either directly or through the config object");
-            }
-            this.apiClient = new XataApiClient({ apiKey });
-        }
-        else {
-            this.createTable = false;
-        }
-        this.tableInitialized = false;
-    }
-    /**
-     * Retrieves all messages associated with the session ID, ordered by
-     * creation time.
-     * @returns A promise that resolves to an array of BaseMessage instances.
-     */
-    async getMessages() {
-        await this.ensureTable();
-        const records = await this.client.db[this.table]
-            .filter({ sessionId: this.sessionId })
-            .sort("xata.createdAt", "asc")
-            .getAll();
-        const rawStoredMessages = records;
-        const orderedMessages = rawStoredMessages.map((message) => {
-            const data = {
-                content: message.content,
-                additional_kwargs: JSON.parse(message.additionalKwargs),
-            };
-            if (message.role) {
-                data.role = message.role;
-            }
-            if (message.name) {
-                data.name = message.name;
-            }
-            return {
-                type: message.type,
-                data,
-            };
-        });
-        return mapStoredMessagesToChatMessages(orderedMessages);
-    }
-    /**
-     * Adds a new message to the database.
-     * @param message The BaseMessage instance to be added.
-     * @returns A promise that resolves when the message has been added.
-     */
-    async addMessage(message) {
-        await this.ensureTable();
-        const messageToAdd = mapChatMessagesToStoredMessages([message]);
-        await this.client.db[this.table].create({
-            sessionId: this.sessionId,
-            type: messageToAdd[0].type,
-            content: messageToAdd[0].data.content,
-            role: messageToAdd[0].data.role,
-            name: messageToAdd[0].data.name,
-            additionalKwargs: JSON.stringify(messageToAdd[0].data.additional_kwargs),
-        });
-    }
-    /**
-     * Deletes all messages associated with the session ID.
-     * @returns A promise that resolves when the messages have been deleted.
-     */
-    async clear() {
-        await this.ensureTable();
-        const records = await this.client.db[this.table]
-            .select(["id"])
-            .filter({ sessionId: this.sessionId })
-            .getAll();
-        const ids = records.map((m) => m.id);
-        await this.client.db[this.table].delete(ids);
-    }
-    /**
-     * Checks if the table exists and creates it if it doesn't. This method is
-     * called before any operation on the table.
-     * @returns A promise that resolves when the table has been ensured.
-     */
-    async ensureTable() {
-        if (!this.createTable) {
-            return;
-        }
-        if (this.tableInitialized) {
-            return;
-        }
-        const { databaseURL, branch } = await this.client.getConfig();
-        const [, , host, , database] = databaseURL.split("/");
-        const urlParts = parseWorkspacesUrlParts(host);
-        if (urlParts == null) {
-            throw new Error("Invalid databaseURL");
-        }
-        const { workspace, region } = urlParts;
-        const tableParams = {
-            workspace,
-            region,
-            database,
-            branch,
-            table: this.table,
-        };
-        let schema = null;
-        try {
-            schema = await this.apiClient.tables.getTableSchema(tableParams);
-        }
-        catch (e) {
-            // pass
-        }
-        if (schema == null) {
-            await this.apiClient.tables.createTable(tableParams);
-            await this.apiClient.tables.setTableSchema({
-                ...tableParams,
-                schema: {
-                    columns: chatMemoryColumns,
-                },
-            });
-        }
-    }
-}
+export * from "@langchain/community/stores/message/xata";