langchain 0.0.203 → 0.0.204-rc.1

package/dist/stores/doc/in_memory.d.ts

@@ -1,46 +1 @@
-import { Document } from "../../document.js";
-import { Docstore } from "../../schema/index.js";
-import { BaseStoreInterface } from "../../schema/storage.js";
-/**
- * Class for storing and retrieving documents in memory asynchronously.
- * Extends the Docstore class.
- */
-export declare class InMemoryDocstore extends Docstore implements BaseStoreInterface<string, Document> {
-    _docs: Map<string, Document>;
-    constructor(docs?: Map<string, Document>);
-    /**
-     * Searches for a document in the store based on its ID.
-     * @param search The ID of the document to search for.
-     * @returns The document with the given ID.
-     */
-    search(search: string): Promise<Document>;
-    /**
-     * Adds new documents to the store.
-     * @param texts An object where the keys are document IDs and the values are the documents themselves.
-     * @returns Void
-     */
-    add(texts: Record<string, Document>): Promise<void>;
-    mget(keys: string[]): Promise<Document[]>;
-    mset(keyValuePairs: [string, Document][]): Promise<void>;
-    mdelete(_keys: string[]): Promise<void>;
-    yieldKeys(_prefix?: string): AsyncGenerator<string>;
-}
-/**
- * Class for storing and retrieving documents in memory synchronously.
- */
-export declare class SynchronousInMemoryDocstore {
-    _docs: Map<string, Document>;
-    constructor(docs?: Map<string, Document>);
-    /**
-     * Searches for a document in the store based on its ID.
-     * @param search The ID of the document to search for.
-     * @returns The document with the given ID.
-     */
-    search(search: string): Document;
-    /**
-     * Adds new documents to the store.
-     * @param texts An object where the keys are document IDs and the values are the documents themselves.
-     * @returns Void
-     */
-    add(texts: Record<string, Document>): void;
-}
+export * from "@langchain/community/stores/doc/in_memory";

package/dist/stores/doc/in_memory.js

@@ -1,102 +1 @@
-import { Docstore } from "../../schema/index.js";
-/**
- * Class for storing and retrieving documents in memory asynchronously.
- * Extends the Docstore class.
- */
-export class InMemoryDocstore extends Docstore {
-    constructor(docs) {
-        super();
-        Object.defineProperty(this, "_docs", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this._docs = docs ?? new Map();
-    }
-    /**
-     * Searches for a document in the store based on its ID.
-     * @param search The ID of the document to search for.
-     * @returns The document with the given ID.
-     */
-    async search(search) {
-        const result = this._docs.get(search);
-        if (!result) {
-            throw new Error(`ID ${search} not found.`);
-        }
-        else {
-            return result;
-        }
-    }
-    /**
-     * Adds new documents to the store.
-     * @param texts An object where the keys are document IDs and the values are the documents themselves.
-     * @returns Void
-     */
-    async add(texts) {
-        const keys = [...this._docs.keys()];
-        const overlapping = Object.keys(texts).filter((x) => keys.includes(x));
-        if (overlapping.length > 0) {
-            throw new Error(`Tried to add ids that already exist: ${overlapping}`);
-        }
-        for (const [key, value] of Object.entries(texts)) {
-            this._docs.set(key, value);
-        }
-    }
-    async mget(keys) {
-        return Promise.all(keys.map((key) => this.search(key)));
-    }
-    async mset(keyValuePairs) {
-        await Promise.all(keyValuePairs.map(([key, value]) => this.add({ [key]: value })));
-    }
-    async mdelete(_keys) {
-        throw new Error("Not implemented.");
-    }
-    // eslint-disable-next-line require-yield
-    async *yieldKeys(_prefix) {
-        throw new Error("Not implemented");
-    }
-}
-/**
- * Class for storing and retrieving documents in memory synchronously.
- */
-export class SynchronousInMemoryDocstore {
-    constructor(docs) {
-        Object.defineProperty(this, "_docs", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this._docs = docs ?? new Map();
-    }
-    /**
-     * Searches for a document in the store based on its ID.
-     * @param search The ID of the document to search for.
-     * @returns The document with the given ID.
-     */
-    search(search) {
-        const result = this._docs.get(search);
-        if (!result) {
-            throw new Error(`ID ${search} not found.`);
-        }
-        else {
-            return result;
-        }
-    }
-    /**
-     * Adds new documents to the store.
-     * @param texts An object where the keys are document IDs and the values are the documents themselves.
-     * @returns Void
-     */
-    add(texts) {
-        const keys = [...this._docs.keys()];
-        const overlapping = Object.keys(texts).filter((x) => keys.includes(x));
-        if (overlapping.length > 0) {
-            throw new Error(`Tried to add ids that already exist: ${overlapping}`);
-        }
-        for (const [key, value] of Object.entries(texts)) {
-            this._docs.set(key, value);
-        }
-    }
-}
+export * from "@langchain/community/stores/doc/in_memory";

package/dist/stores/message/cassandra.cjs

@@ -1,160 +1,17 @@
 "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.
- * @example
- * ```typescript
- * const chatHistory = new CassandraChatMessageHistory({
- *   cloud: {
- *     secureConnectBundle: "<path to your secure bundle>",
- *   },
- *   credentials: {
- *     username: "token",
- *     password: "<your Cassandra access token>",
- *   },
- *   keyspace: "langchain",
- *   table: "message_history",
- *   sessionId: "<some unique session identifier>",
- * });
- *
- * 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 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(() => { });
+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 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;
+    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/cassandra"), exports);

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

@@ -1,69 +1 @@
-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.
- * @example
- * ```typescript
- * const chatHistory = new CassandraChatMessageHistory({
- *   cloud: {
- *     secureConnectBundle: "<path to your secure bundle>",
- *   },
- *   credentials: {
- *     username: "token",
- *     password: "<your Cassandra access token>",
- *   },
- *   keyspace: "langchain",
- *   table: "message_history",
- *   sessionId: "<some unique session identifier>",
- * });
- *
- * 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 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;
-}
+export * from "@langchain/community/stores/message/cassandra";

package/dist/stores/message/cassandra.js

@@ -1,156 +1 @@
-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.
- * @example
- * ```typescript
- * const chatHistory = new CassandraChatMessageHistory({
- *   cloud: {
- *     secureConnectBundle: "<path to your secure bundle>",
- *   },
- *   credentials: {
- *     username: "token",
- *     password: "<your Cassandra access token>",
- *   },
- *   keyspace: "langchain",
- *   table: "message_history",
- *   sessionId: "<some unique session identifier>",
- * });
- *
- * 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 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;
-    }
-}
+export * from "@langchain/community/stores/message/cassandra";