langchain 0.0.133 → 0.0.135

package/dist/llms/writer.d.ts

@@ -0,0 +1,60 @@
+import { BaseLLMParams, LLM } from "./base.js";
+/**
+ * Interface for the input parameters specific to the Writer model.
+ */
+export interface WriterInput extends BaseLLMParams {
+    /** Writer API key */
+    apiKey?: string;
+    /** Writer organization ID */
+    orgId?: string | number;
+    /** Model to use */
+    model?: string;
+    /** Sampling temperature to use */
+    temperature?: number;
+    /** Minimum number of tokens to generate. */
+    minTokens?: number;
+    /** Maximum number of tokens to generate in the completion. */
+    maxTokens?: number;
+    /** Generates this many completions server-side and returns the "best"." */
+    bestOf?: number;
+    /** Penalizes repeated tokens according to frequency. */
+    frequencyPenalty?: number;
+    /** Whether to return log probabilities. */
+    logprobs?: number;
+    /** Number of completions to generate. */
+    n?: number;
+    /** Penalizes repeated tokens regardless of frequency. */
+    presencePenalty?: number;
+    /** Total probability mass of tokens to consider at each step. */
+    topP?: number;
+}
+/**
+ * Class representing a Writer Large Language Model (LLM). It interacts
+ * with the Writer API to generate text completions.
+ */
+export declare class Writer extends LLM implements WriterInput {
+    static lc_name(): string;
+    get lc_secrets(): {
+        [key: string]: string;
+    } | undefined;
+    get lc_aliases(): {
+        [key: string]: string;
+    } | undefined;
+    lc_serializable: boolean;
+    apiKey: string;
+    orgId: number;
+    model: string;
+    temperature?: number;
+    minTokens?: number;
+    maxTokens?: number;
+    bestOf?: number;
+    frequencyPenalty?: number;
+    logprobs?: number;
+    n?: number;
+    presencePenalty?: number;
+    topP?: number;
+    constructor(fields?: WriterInput);
+    _llmType(): string;
+    /** @ignore */
+    _call(prompt: string, options: this["ParsedCallOptions"]): Promise<string>;
+}

package/dist/llms/writer.js

@@ -0,0 +1,163 @@
+import { Writer as WriterClient } from "@writerai/writer-sdk";
+import { LLM } from "./base.js";
+import { getEnvironmentVariable } from "../util/env.js";
+/**
+ * Class representing a Writer Large Language Model (LLM). It interacts
+ * with the Writer API to generate text completions.
+ */
+export class Writer extends LLM {
+    static lc_name() {
+        return "Writer";
+    }
+    get lc_secrets() {
+        return {
+            apiKey: "WRITER_API_KEY",
+            orgId: "WRITER_ORG_ID",
+        };
+    }
+    get lc_aliases() {
+        return {
+            apiKey: "writer_api_key",
+            orgId: "writer_org_id",
+        };
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "orgId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "model", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "palmyra-instruct"
+        });
+        Object.defineProperty(this, "temperature", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "minTokens", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "maxTokens", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "bestOf", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "frequencyPenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "logprobs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "n", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "presencePenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topP", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const apiKey = fields?.apiKey ?? getEnvironmentVariable("WRITER_API_KEY");
+        const orgId = fields?.orgId ?? getEnvironmentVariable("WRITER_ORG_ID");
+        if (!apiKey) {
+            throw new Error("Please set the WRITER_API_KEY environment variable or pass it to the constructor as the apiKey field.");
+        }
+        if (!orgId) {
+            throw new Error("Please set the WRITER_ORG_ID environment variable or pass it to the constructor as the orgId field.");
+        }
+        this.apiKey = apiKey;
+        this.orgId = typeof orgId === "string" ? parseInt(orgId, 10) : orgId;
+        this.model = fields?.model ?? this.model;
+        this.temperature = fields?.temperature ?? this.temperature;
+        this.minTokens = fields?.minTokens ?? this.minTokens;
+        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
+        this.bestOf = fields?.bestOf ?? this.bestOf;
+        this.frequencyPenalty = fields?.frequencyPenalty ?? this.frequencyPenalty;
+        this.logprobs = fields?.logprobs ?? this.logprobs;
+        this.n = fields?.n ?? this.n;
+        this.presencePenalty = fields?.presencePenalty ?? this.presencePenalty;
+        this.topP = fields?.topP ?? this.topP;
+    }
+    _llmType() {
+        return "writer";
+    }
+    /** @ignore */
+    async _call(prompt, options) {
+        const sdk = new WriterClient({
+            security: {
+                apiKey: this.apiKey,
+            },
+            organizationId: this.orgId,
+        });
+        return this.caller.callWithOptions({ signal: options.signal }, async () => {
+            try {
+                const res = await sdk.completions.create({
+                    completionRequest: {
+                        prompt,
+                        stop: options.stop,
+                        temperature: this.temperature,
+                        minTokens: this.minTokens,
+                        maxTokens: this.maxTokens,
+                        bestOf: this.bestOf,
+                        n: this.n,
+                        frequencyPenalty: this.frequencyPenalty,
+                        logprobs: this.logprobs,
+                        presencePenalty: this.presencePenalty,
+                        topP: this.topP,
+                    },
+                    modelId: this.model,
+                });
+                return (res.completionResponse?.choices?.[0].text ?? "No completion found.");
+            }
+            catch (e) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                e.response = e.rawResponse;
+                throw e;
+            }
+        });
+    }
+}

package/dist/load/import_constants.cjs

@@ -29,6 +29,7 @@ exports.optionalImportEntrypoints = [
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
+    "langchain/llms/writer",
     "langchain/prompts/load",
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
@@ -69,6 +70,7 @@ exports.optionalImportEntrypoints = [
     "langchain/document_loaders/web/github",
     "langchain/document_loaders/web/notiondb",
     "langchain/document_loaders/web/notionapi",
+    "langchain/document_loaders/web/recursive_url",
     "langchain/document_loaders/web/s3",
     "langchain/document_loaders/web/sonix_audio",
     "langchain/document_loaders/web/confluence",
@@ -109,11 +111,13 @@ exports.optionalImportEntrypoints = [
     "langchain/stores/message/dynamodb",
     "langchain/stores/message/firestore",
     "langchain/stores/message/momento",
+    "langchain/stores/message/mongodb",
     "langchain/stores/message/redis",
     "langchain/stores/message/ioredis",
     "langchain/stores/message/upstash_redis",
     "langchain/stores/message/planetscale",
     "langchain/stores/message/xata",
     "langchain/storage/ioredis",
+    "langchain/hub",
     "langchain/experimental/multimodal_embeddings/googlevertexai",
 ];

package/dist/load/import_constants.js

@@ -26,6 +26,7 @@ export const optionalImportEntrypoints = [
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
+    "langchain/llms/writer",
     "langchain/prompts/load",
     "langchain/vectorstores/analyticdb",
     "langchain/vectorstores/elasticsearch",
@@ -66,6 +67,7 @@ export const optionalImportEntrypoints = [
     "langchain/document_loaders/web/github",
     "langchain/document_loaders/web/notiondb",
     "langchain/document_loaders/web/notionapi",
+    "langchain/document_loaders/web/recursive_url",
     "langchain/document_loaders/web/s3",
     "langchain/document_loaders/web/sonix_audio",
     "langchain/document_loaders/web/confluence",
@@ -106,11 +108,13 @@ export const optionalImportEntrypoints = [
     "langchain/stores/message/dynamodb",
     "langchain/stores/message/firestore",
     "langchain/stores/message/momento",
+    "langchain/stores/message/mongodb",
     "langchain/stores/message/redis",
     "langchain/stores/message/ioredis",
     "langchain/stores/message/upstash_redis",
     "langchain/stores/message/planetscale",
     "langchain/stores/message/xata",
     "langchain/storage/ioredis",
+    "langchain/hub",
     "langchain/experimental/multimodal_embeddings/googlevertexai",
 ];

package/dist/load/import_map.cjs

@@ -25,7 +25,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__openai = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = void 0;
+exports.evaluation = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -76,6 +76,7 @@ exports.retrievers__parent_document = __importStar(require("../retrievers/parent
 exports.retrievers__time_weighted = __importStar(require("../retrievers/time_weighted.cjs"));
 exports.retrievers__document_compressors__chain_extract = __importStar(require("../retrievers/document_compressors/chain_extract.cjs"));
 exports.retrievers__hyde = __importStar(require("../retrievers/hyde.cjs"));
+exports.retrievers__score_threshold = __importStar(require("../retrievers/score_threshold.cjs"));
 exports.retrievers__vespa = __importStar(require("../retrievers/vespa.cjs"));
 exports.cache = __importStar(require("../cache/index.cjs"));
 exports.stores__doc__in_memory = __importStar(require("../stores/doc/in_memory.cjs"));

package/dist/load/import_map.d.ts

@@ -48,6 +48,7 @@ export * as retrievers__parent_document from "../retrievers/parent_document.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";
 export * as retrievers__document_compressors__chain_extract from "../retrievers/document_compressors/chain_extract.js";
 export * as retrievers__hyde from "../retrievers/hyde.js";
+export * as retrievers__score_threshold from "../retrievers/score_threshold.js";
 export * as retrievers__vespa from "../retrievers/vespa.js";
 export * as cache from "../cache/index.js";
 export * as stores__doc__in_memory from "../stores/doc/in_memory.js";

package/dist/load/import_map.js

@@ -49,6 +49,7 @@ export * as retrievers__parent_document from "../retrievers/parent_document.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";
 export * as retrievers__document_compressors__chain_extract from "../retrievers/document_compressors/chain_extract.js";
 export * as retrievers__hyde from "../retrievers/hyde.js";
+export * as retrievers__score_threshold from "../retrievers/score_threshold.js";
 export * as retrievers__vespa from "../retrievers/vespa.js";
 export * as cache from "../cache/index.js";
 export * as stores__doc__in_memory from "../stores/doc/in_memory.js";

package/dist/memory/summary_buffer.d.ts

@@ -26,7 +26,7 @@ export declare class ConversationSummaryBufferMemory extends BaseConversationSum
      * @param _ InputValues object, not used in this method.
      * @returns Promise that resolves with MemoryVariables object containing the loaded chat messages.
      */
-    loadMemoryVariables(_: InputValues): Promise<MemoryVariables>;
+    loadMemoryVariables(_?: InputValues): Promise<MemoryVariables>;
     /**
      * Method that saves the context of the conversation, including the input
      * and output values, and prunes the memory if it exceeds the maximum

package/dist/retrievers/score_threshold.cjs

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ScoreThresholdRetriever = void 0;
+const base_js_1 = require("../vectorstores/base.cjs");
+class ScoreThresholdRetriever extends base_js_1.VectorStoreRetriever {
+    constructor(input) {
+        super(input);
+        Object.defineProperty(this, "minSimilarityScore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "kIncrement", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 10
+        });
+        Object.defineProperty(this, "maxK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 100
+        });
+        this.maxK = input.maxK ?? this.maxK;
+        this.minSimilarityScore =
+            input.minSimilarityScore ?? this.minSimilarityScore;
+        this.kIncrement = input.kIncrement ?? this.kIncrement;
+    }
+    async getRelevantDocuments(query) {
+        let currentK = 0;
+        let filteredResults = [];
+        do {
+            currentK += this.kIncrement;
+            const results = await this.vectorStore.similaritySearchWithScore(query, currentK, this.filter);
+            filteredResults = results.filter(([, score]) => score >= this.minSimilarityScore);
+        } while (filteredResults.length >= currentK && currentK < this.maxK);
+        return filteredResults.map((documents) => documents[0]).slice(0, this.maxK);
+    }
+    static fromVectorStore(vectorStore, options) {
+        return new this({ ...options, vectorStore });
+    }
+}
+exports.ScoreThresholdRetriever = ScoreThresholdRetriever;

package/dist/retrievers/score_threshold.d.ts

@@ -0,0 +1,15 @@
+import { VectorStore, VectorStoreRetriever, VectorStoreRetrieverInput } from "../vectorstores/base.js";
+import { Document } from "../document.js";
+export type ScoreThresholdRetrieverInput<V extends VectorStore> = Omit<VectorStoreRetrieverInput<V>, "k"> & {
+    maxK?: number;
+    kIncrement?: number;
+    minSimilarityScore: number;
+};
+export declare class ScoreThresholdRetriever<V extends VectorStore> extends VectorStoreRetriever<V> {
+    minSimilarityScore: number;
+    kIncrement: number;
+    maxK: number;
+    constructor(input: ScoreThresholdRetrieverInput<V>);
+    getRelevantDocuments(query: string): Promise<Document[]>;
+    static fromVectorStore<V extends VectorStore>(vectorStore: V, options: Omit<ScoreThresholdRetrieverInput<V>, "vectorStore">): ScoreThresholdRetriever<V>;
+}

package/dist/retrievers/score_threshold.js

@@ -0,0 +1,41 @@
+import { VectorStoreRetriever, } from "../vectorstores/base.js";
+export class ScoreThresholdRetriever extends VectorStoreRetriever {
+    constructor(input) {
+        super(input);
+        Object.defineProperty(this, "minSimilarityScore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "kIncrement", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 10
+        });
+        Object.defineProperty(this, "maxK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 100
+        });
+        this.maxK = input.maxK ?? this.maxK;
+        this.minSimilarityScore =
+            input.minSimilarityScore ?? this.minSimilarityScore;
+        this.kIncrement = input.kIncrement ?? this.kIncrement;
+    }
+    async getRelevantDocuments(query) {
+        let currentK = 0;
+        let filteredResults = [];
+        do {
+            currentK += this.kIncrement;
+            const results = await this.vectorStore.similaritySearchWithScore(query, currentK, this.filter);
+            filteredResults = results.filter(([, score]) => score >= this.minSimilarityScore);
+        } while (filteredResults.length >= currentK && currentK < this.maxK);
+        return filteredResults.map((documents) => documents[0]).slice(0, this.maxK);
+    }
+    static fromVectorStore(vectorStore, options) {
+        return new this({ ...options, vectorStore });
+    }
+}

package/dist/sql_db.cjs

@@ -51,6 +51,12 @@ class SqlDatabase extends serializable_js_1.Serializable {
             writable: true,
             value: 3
         });
+        Object.defineProperty(this, "customDescription", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.appDataSource = fields.appDataSource;
         this.appDataSourceOptions = fields.appDataSource.options;
         if (fields?.includesTables && fields?.ignoreTables) {
@@ -60,6 +66,7 @@ class SqlDatabase extends serializable_js_1.Serializable {
         this.ignoreTables = fields?.ignoreTables ?? [];
         this.sampleRowsInTableInfo =
             fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;
+        this.customDescription = Object.fromEntries(Object.entries(fields?.customDescription ?? {}).filter(([key, _]) => this.allTables.map((table) => table.tableName).includes(key)));
     }
     static async fromDataSourceParams(fields) {
         const sqlDatabase = new SqlDatabase(fields);
@@ -100,7 +107,7 @@ class SqlDatabase extends serializable_js_1.Serializable {
             (0, sql_utils_js_1.verifyListTablesExistInDatabase)(this.allTables, targetTables, "Wrong target table name:");
             selectedTables = this.allTables.filter((currentTable) => targetTables.includes(currentTable.tableName));
         }
-        return (0, sql_utils_js_1.generateTableInfoFromTables)(selectedTables, this.appDataSource, this.sampleRowsInTableInfo);
+        return (0, sql_utils_js_1.generateTableInfoFromTables)(selectedTables, this.appDataSource, this.sampleRowsInTableInfo, this.customDescription);
     }
     /**
      * Execute a SQL command and return a string representing the results.

package/dist/sql_db.d.ts

@@ -11,6 +11,7 @@ export declare class SqlDatabase extends Serializable implements SqlDatabaseOpti
     includesTables: Array<string>;
     ignoreTables: Array<string>;
     sampleRowsInTableInfo: number;
+    customDescription?: Record<string, string>;
     protected constructor(fields: SqlDatabaseDataSourceParams);
     static fromDataSourceParams(fields: SqlDatabaseDataSourceParams): Promise<SqlDatabase>;
     static fromOptionsParams(fields: SqlDatabaseOptionsParams): Promise<SqlDatabase>;

package/dist/sql_db.js

@@ -48,6 +48,12 @@ export class SqlDatabase extends Serializable {
             writable: true,
             value: 3
         });
+        Object.defineProperty(this, "customDescription", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.appDataSource = fields.appDataSource;
         this.appDataSourceOptions = fields.appDataSource.options;
         if (fields?.includesTables && fields?.ignoreTables) {
@@ -57,6 +63,7 @@ export class SqlDatabase extends Serializable {
         this.ignoreTables = fields?.ignoreTables ?? [];
         this.sampleRowsInTableInfo =
             fields?.sampleRowsInTableInfo ?? this.sampleRowsInTableInfo;
+        this.customDescription = Object.fromEntries(Object.entries(fields?.customDescription ?? {}).filter(([key, _]) => this.allTables.map((table) => table.tableName).includes(key)));
     }
     static async fromDataSourceParams(fields) {
         const sqlDatabase = new SqlDatabase(fields);
@@ -97,7 +104,7 @@ export class SqlDatabase extends Serializable {
             verifyListTablesExistInDatabase(this.allTables, targetTables, "Wrong target table name:");
             selectedTables = this.allTables.filter((currentTable) => targetTables.includes(currentTable.tableName));
         }
-        return generateTableInfoFromTables(selectedTables, this.appDataSource, this.sampleRowsInTableInfo);
+        return generateTableInfoFromTables(selectedTables, this.appDataSource, this.sampleRowsInTableInfo, this.customDescription);
     }
     /**
      * Execute a SQL command and return a string representing the results.

package/dist/stores/message/mongodb.cjs

@@ -0,0 +1,48 @@
+"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");
+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 });
+    }
+    async clear() {
+        await this.collection.deleteOne({ _id: new mongodb_1.ObjectId(this.sessionId) });
+    }
+}
+exports.MongoDBChatMessageHistory = MongoDBChatMessageHistory;

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

@@ -0,0 +1,15 @@
+import { Collection, Document as MongoDBDocument } from "mongodb";
+import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+export interface MongoDBChatMessageHistoryInput {
+    collection: Collection<MongoDBDocument>;
+    sessionId: string;
+}
+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>;
+}

package/dist/stores/message/mongodb.js

@@ -0,0 +1,44 @@
+import { ObjectId } from "mongodb";
+import { BaseListChatMessageHistory } from "../../schema/index.js";
+import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
+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) });
+    }
+}

package/dist/tools/sql.cjs

@@ -126,14 +126,20 @@ class ListTablesSqlTool extends base_js_1.Tool {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: `Input is an empty string, output is a comma separated list of tables in the database.`
+            value: `Input is an empty string, output is a comma-separated list of tables in the database.`
         });
         this.db = db;
     }
-    /** @ignore */
     async _call(_) {
         try {
-            const tables = this.db.allTables.map((table) => table.tableName);
+            let selectedTables = this.db.allTables;
+            if (this.db.includesTables.length > 0) {
+                selectedTables = selectedTables.filter((currentTable) => this.db.includesTables.includes(currentTable.tableName));
+            }
+            if (this.db.ignoreTables.length > 0) {
+                selectedTables = selectedTables.filter((currentTable) => !this.db.ignoreTables.includes(currentTable.tableName));
+            }
+            const tables = selectedTables.map((table) => table.tableName);
             return tables.join(", ");
         }
         catch (error) {

package/dist/tools/sql.d.ts

@@ -51,7 +51,6 @@ export declare class ListTablesSqlTool extends Tool implements SqlTool {
     name: string;
     db: SqlDatabase;
     constructor(db: SqlDatabase);
-    /** @ignore */
     _call(_: string): Promise<string>;
     description: string;
 }