langchain 0.0.134 → 0.0.136

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",
@@ -110,6 +111,7 @@ 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",

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",
@@ -107,6 +108,7 @@ 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",

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/output_parsers/prompts.d.ts

@@ -1,3 +1,3 @@
 import { PromptTemplate } from "../prompts/prompt.js";
 export declare const NAIVE_FIX_TEMPLATE = "Instructions:\n--------------\n{instructions}\n--------------\nCompletion:\n--------------\n{completion}\n--------------\n\nAbove, the Completion did not satisfy the constraints given in the Instructions.\nError:\n--------------\n{error}\n--------------\n\nPlease try again. Please only respond with an answer that satisfies the constraints laid out in the Instructions:";
-export declare const NAIVE_FIX_PROMPT: PromptTemplate<any, any>;
+export declare const NAIVE_FIX_PROMPT: PromptTemplate<import("../prompts/prompt.js").ParamsFromFString<"Instructions:\n--------------\n{instructions}\n--------------\nCompletion:\n--------------\n{completion}\n--------------\n\nAbove, the Completion did not satisfy the constraints given in the Instructions.\nError:\n--------------\n{error}\n--------------\n\nPlease try again. Please only respond with an answer that satisfies the constraints laid out in the Instructions:">, any>;

package/dist/prompts/prompt.cjs

@@ -88,8 +88,10 @@ class PromptTemplate extends base_js_1.BaseStringPromptTemplate {
     /**
      * Load prompt template from a template f-string
      */
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static fromTemplate(template, { templateFormat = "f-string", ...rest } = {}) {
+        if (templateFormat === "jinja2") {
+            throw new Error("jinja2 templates are not currently supported.");
+        }
         const names = new Set();
         (0, template_js_1.parseTemplate)(template, templateFormat).forEach((node) => {
             if (node.type === "variable") {

package/dist/prompts/prompt.d.ts

@@ -24,6 +24,16 @@ export interface PromptTemplateInput<RunInput extends InputValues = any, Partial
      */
     validateTemplate?: boolean;
 }
+type NonAlphanumeric = " " | "\t" | "\n" | "\r" | '"' | "'" | "{" | "[" | "(" | "`" | ":" | ";";
+/**
+ * Recursive type to extract template parameters from a string.
+ * @template T - The input string.
+ * @template Result - The resulting array of extracted template parameters.
+ */
+type ExtractTemplateParamsRecursive<T extends string, Result extends string[] = []> = T extends `${string}{${infer Param}}${infer Rest}` ? Param extends `${NonAlphanumeric}${string}` ? ExtractTemplateParamsRecursive<Rest, Result> : ExtractTemplateParamsRecursive<Rest, [...Result, Param]> : Result;
+export type ParamsFromFString<T extends string> = {
+    [Key in ExtractTemplateParamsRecursive<T>[number]]: string;
+};
 /**
  * Schema to represent a basic prompt for an LLM.
  * @augments BasePromptTemplate
@@ -69,7 +79,7 @@ export declare class PromptTemplate<RunInput extends InputValues = any, PartialV
     /**
      * Load prompt template from a template f-string
      */
-    static fromTemplate<RunInput extends InputValues = any>(template: string, { templateFormat, ...rest }?: Omit<PromptTemplateInput, "template" | "inputVariables">): PromptTemplate<RunInput, any>;
+    static fromTemplate<RunInput extends InputValues = Symbol, T extends string = string>(template: T, { templateFormat, ...rest }?: Omit<PromptTemplateInput<RunInput, string>, "template" | "inputVariables">): PromptTemplate<RunInput extends Symbol ? ParamsFromFString<T> : RunInput, any>;
     /**
      * Partially applies values to the prompt template.
      * @param values The values to be partially applied to the prompt template.
@@ -79,3 +89,4 @@ export declare class PromptTemplate<RunInput extends InputValues = any, PartialV
     serialize(): SerializedPromptTemplate;
     static deserialize(data: SerializedPromptTemplate): Promise<PromptTemplate>;
 }
+export {};

package/dist/prompts/prompt.js

@@ -85,8 +85,10 @@ export class PromptTemplate extends BaseStringPromptTemplate {
     /**
      * Load prompt template from a template f-string
      */
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static fromTemplate(template, { templateFormat = "f-string", ...rest } = {}) {
+        if (templateFormat === "jinja2") {
+            throw new Error("jinja2 templates are not currently supported.");
+        }
         const names = new Set();
         parseTemplate(template, templateFormat).forEach((node) => {
             if (node.type === "variable") {

package/dist/retrievers/parent_document.cjs

@@ -84,12 +84,19 @@ class ParentDocumentRetriever extends retriever_js_1.BaseRetriever {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "parentK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.vectorstore = fields.vectorstore;
         this.docstore = fields.docstore;
         this.childSplitter = fields.childSplitter;
         this.parentSplitter = fields.parentSplitter;
         this.idKey = fields.idKey ?? this.idKey;
         this.childK = fields.childK;
+        this.parentK = fields.parentK;
     }
     async _getRelevantDocuments(query) {
         const subDocs = await this.vectorstore.similaritySearch(query, this.childK);
@@ -107,7 +114,7 @@ class ParentDocumentRetriever extends retriever_js_1.BaseRetriever {
                 parentDocs.push(parentDoc);
             }
         }
-        return parentDocs;
+        return parentDocs.slice(0, this.parentK);
     }
     /**
      * Adds documents to the docstore and vectorstores.

package/dist/retrievers/parent_document.d.ts

@@ -14,6 +14,7 @@ export interface ParentDocumentRetrieverFields extends BaseRetrieverInput {
     parentSplitter?: TextSplitter;
     idKey?: string;
     childK?: number;
+    parentK?: number;
 }
 /**
  * A type of document retriever that splits input documents into smaller chunks
@@ -33,6 +34,7 @@ export declare class ParentDocumentRetriever extends BaseRetriever {
     protected parentSplitter?: TextSplitter;
     protected idKey: string;
     protected childK?: number;
+    protected parentK?: number;
     constructor(fields: ParentDocumentRetrieverFields);
     _getRelevantDocuments(query: string): Promise<Document[]>;
     /**

package/dist/retrievers/parent_document.js

@@ -58,12 +58,19 @@ export class ParentDocumentRetriever extends BaseRetriever {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "parentK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.vectorstore = fields.vectorstore;
         this.docstore = fields.docstore;
         this.childSplitter = fields.childSplitter;
         this.parentSplitter = fields.parentSplitter;
         this.idKey = fields.idKey ?? this.idKey;
         this.childK = fields.childK;
+        this.parentK = fields.parentK;
     }
     async _getRelevantDocuments(query) {
         const subDocs = await this.vectorstore.similaritySearch(query, this.childK);
@@ -81,7 +88,7 @@ export class ParentDocumentRetriever extends BaseRetriever {
                 parentDocs.push(parentDoc);
             }
         }
-        return parentDocs;
+        return parentDocs.slice(0, this.parentK);
     }
     /**
      * Adds documents to the docstore and vectorstores.

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>;
+}