langchain 0.1.9 → 0.1.11

package/dist/agents/toolkits/sql/index.cjs

@@ -1,6 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createSqlAgent = exports.SqlToolkit = void 0;
+exports.SQL_SUFFIX = exports.SQL_PREFIX = exports.createSqlAgent = exports.SqlToolkit = void 0;
 var sql_js_1 = require("./sql.cjs");
 Object.defineProperty(exports, "SqlToolkit", { enumerable: true, get: function () { return sql_js_1.SqlToolkit; } });
 Object.defineProperty(exports, "createSqlAgent", { enumerable: true, get: function () { return sql_js_1.createSqlAgent; } });
+var prompt_js_1 = require("./prompt.cjs");
+Object.defineProperty(exports, "SQL_PREFIX", { enumerable: true, get: function () { return prompt_js_1.SQL_PREFIX; } });
+Object.defineProperty(exports, "SQL_SUFFIX", { enumerable: true, get: function () { return prompt_js_1.SQL_SUFFIX; } });

package/dist/agents/toolkits/sql/index.d.ts

@@ -1 +1,2 @@
 export { SqlCreatePromptArgs, SqlToolkit, createSqlAgent } from "./sql.js";
+export { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";

package/dist/agents/toolkits/sql/index.js

@@ -1 +1,2 @@
 export { SqlToolkit, createSqlAgent } from "./sql.js";
+export { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";

package/dist/chains/combine_documents/base.cjs

@@ -10,7 +10,7 @@ exports.DEFAULT_DOCUMENT_PROMPT =
 async function formatDocuments({ documentPrompt, documentSeparator, documents, config, }) {
     const formattedDocs = await Promise.all(documents.map((document) => documentPrompt
         .withConfig({ runName: "document_formatter" })
-        .invoke({ page_content: document.pageContent }, config)));
+        .invoke({ ...document.metadata, page_content: document.pageContent }, config)));
     return formattedDocs.join(documentSeparator);
 }
 exports.formatDocuments = formatDocuments;

package/dist/chains/combine_documents/base.js

@@ -7,6 +7,6 @@ export const DEFAULT_DOCUMENT_PROMPT =
 export async function formatDocuments({ documentPrompt, documentSeparator, documents, config, }) {
     const formattedDocs = await Promise.all(documents.map((document) => documentPrompt
         .withConfig({ runName: "document_formatter" })
-        .invoke({ page_content: document.pageContent }, config)));
+        .invoke({ ...document.metadata, page_content: document.pageContent }, config)));
     return formattedDocs.join(documentSeparator);
 }

package/dist/chains/history_aware_retriever.d.ts

@@ -1,5 +1,5 @@
 import type { LanguageModelLike } from "@langchain/core/language_models/base";
-import { type RunnableInterface } from "@langchain/core/runnables";
+import { type Runnable, type RunnableInterface } from "@langchain/core/runnables";
 import { type BasePromptTemplate } from "@langchain/core/prompts";
 import type { DocumentInterface } from "@langchain/core/documents";
 import type { BaseMessage } from "@langchain/core/messages";
@@ -49,7 +49,7 @@ export type CreateHistoryAwareRetrieverParams = {
  * const result = await chain.invoke({"input": "...", "chat_history": [] })
  * ```
  */
-export declare function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt, }: CreateHistoryAwareRetrieverParams): Promise<RunnableInterface<{
+export declare function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt, }: CreateHistoryAwareRetrieverParams): Promise<Runnable<{
     input: string;
     chat_history: string | BaseMessage[];
 }, DocumentInterface[]>>;

package/dist/chains/openai_functions/base.d.ts

@@ -2,7 +2,7 @@ import type { z } from "zod";
 import { JsonSchema7Type } from "zod-to-json-schema";
 import type { BaseOutputParser } from "@langchain/core/output_parsers";
 import type { BasePromptTemplate } from "@langchain/core/prompts";
-import type { RunnableInterface } from "@langchain/core/runnables";
+import type { Runnable, RunnableInterface } from "@langchain/core/runnables";
 import type { BaseFunctionCallOptions, BaseLanguageModelInput, FunctionDefinition } from "@langchain/core/language_models/base";
 import type { InputValues } from "@langchain/core/utils/types";
 import type { BaseMessage } from "@langchain/core/messages";
@@ -77,7 +77,7 @@ export type CreateOpenAIFnRunnableConfig<RunInput extends Record<string, any>, R
  * // { name: 'John Doe', age: 30, fav_food: 'chocolate chip cookies' }
  * ```
  */
-export declare function createOpenAIFnRunnable<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> = Record<string, any>>(config: CreateOpenAIFnRunnableConfig<RunInput, RunOutput>): RunnableInterface<RunInput, RunOutput>;
+export declare function createOpenAIFnRunnable<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> = Record<string, any>>(config: CreateOpenAIFnRunnableConfig<RunInput, RunOutput>): Runnable<RunInput, RunOutput>;
 /**
  * Configuration params for the createStructuredOutputRunnable method.
  */
@@ -150,4 +150,4 @@ export type CreateStructuredOutputRunnableConfig<RunInput extends Record<string,
  * // { name: 'John Doe', age: 30, fav_food: 'chocolate chip cookies' }
  * ```
  */
-export declare function createStructuredOutputRunnable<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> = Record<string, any>>(config: CreateStructuredOutputRunnableConfig<RunInput, RunOutput>): RunnableInterface<RunInput, RunOutput>;
+export declare function createStructuredOutputRunnable<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> = Record<string, any>>(config: CreateStructuredOutputRunnableConfig<RunInput, RunOutput>): Runnable<RunInput, RunOutput>;

package/dist/chains/openai_functions/structured_output.cjs

@@ -25,7 +25,7 @@ class FunctionCallStructuredOutputParser extends output_parsers_1.BaseLLMOutputP
             fields = fieldsOrSchema;
         }
         if (fields.jsonSchema === undefined && fields.zodSchema === undefined) {
-            throw new Error(`Must provide one of "jsonSchema" or "zodSchema".`);
+            throw new Error(`Must provide at least one of "jsonSchema" or "zodSchema".`);
         }
         super(fields);
         Object.defineProperty(this, "lc_namespace", {
@@ -55,7 +55,7 @@ class FunctionCallStructuredOutputParser extends output_parsers_1.BaseLLMOutputP
         if (fields.jsonSchema !== undefined) {
             this.jsonSchemaValidator = new json_schema_1.Validator(fields.jsonSchema, "7");
         }
-        else {
+        if (fields.zodSchema !== undefined) {
             this.zodSchema = fields.zodSchema;
         }
     }

package/dist/chains/openai_functions/structured_output.js

@@ -22,7 +22,7 @@ export class FunctionCallStructuredOutputParser extends BaseLLMOutputParser {
             fields = fieldsOrSchema;
         }
         if (fields.jsonSchema === undefined && fields.zodSchema === undefined) {
-            throw new Error(`Must provide one of "jsonSchema" or "zodSchema".`);
+            throw new Error(`Must provide at least one of "jsonSchema" or "zodSchema".`);
         }
         super(fields);
         Object.defineProperty(this, "lc_namespace", {
@@ -52,7 +52,7 @@ export class FunctionCallStructuredOutputParser extends BaseLLMOutputParser {
         if (fields.jsonSchema !== undefined) {
             this.jsonSchemaValidator = new Validator(fields.jsonSchema, "7");
         }
-        else {
+        if (fields.zodSchema !== undefined) {
             this.zodSchema = fields.zodSchema;
         }
     }

package/dist/chains/retrieval.d.ts

@@ -1,5 +1,5 @@
 import type { BaseRetrieverInterface } from "@langchain/core/retrievers";
-import { type RunnableInterface } from "@langchain/core/runnables";
+import { type Runnable, type RunnableInterface } from "@langchain/core/runnables";
 import type { BaseMessage } from "@langchain/core/messages";
 import type { DocumentInterface } from "@langchain/core/documents";
 /**
@@ -52,7 +52,7 @@ export type CreateRetrievalChainParams<RunOutput> = {
  * const response = await chain.invoke({ input: "..." });
  * ```
  */
-export declare function createRetrievalChain<RunOutput>({ retriever, combineDocsChain, }: CreateRetrievalChainParams<RunOutput>): Promise<RunnableInterface<{
+export declare function createRetrievalChain<RunOutput>({ retriever, combineDocsChain, }: CreateRetrievalChainParams<RunOutput>): Promise<Runnable<{
     input: string;
     chat_history?: BaseMessage[] | string;
 } & {

package/dist/chains/sql_db/index.cjs

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SQL_SAP_HANA_PROMPT = exports.SQL_MYSQL_PROMPT = exports.SQL_MSSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = exports.SqlDatabaseChain = void 0;
+exports.SQL_PROMPTS_MAP = exports.SQL_SAP_HANA_PROMPT = exports.SQL_MYSQL_PROMPT = exports.SQL_MSSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = exports.createSqlQueryChain = exports.SqlDatabaseChain = void 0;
 var sql_db_chain_js_1 = require("./sql_db_chain.cjs");
 Object.defineProperty(exports, "SqlDatabaseChain", { enumerable: true, get: function () { return sql_db_chain_js_1.SqlDatabaseChain; } });
+Object.defineProperty(exports, "createSqlQueryChain", { enumerable: true, get: function () { return sql_db_chain_js_1.createSqlQueryChain; } });
 var sql_db_prompt_js_1 = require("./sql_db_prompt.cjs");
 Object.defineProperty(exports, "DEFAULT_SQL_DATABASE_PROMPT", { enumerable: true, get: function () { return sql_db_prompt_js_1.DEFAULT_SQL_DATABASE_PROMPT; } });
 Object.defineProperty(exports, "SQL_POSTGRES_PROMPT", { enumerable: true, get: function () { return sql_db_prompt_js_1.SQL_POSTGRES_PROMPT; } });
@@ -10,3 +11,4 @@ Object.defineProperty(exports, "SQL_SQLITE_PROMPT", { enumerable: true, get: fun
 Object.defineProperty(exports, "SQL_MSSQL_PROMPT", { enumerable: true, get: function () { return sql_db_prompt_js_1.SQL_MSSQL_PROMPT; } });
 Object.defineProperty(exports, "SQL_MYSQL_PROMPT", { enumerable: true, get: function () { return sql_db_prompt_js_1.SQL_MYSQL_PROMPT; } });
 Object.defineProperty(exports, "SQL_SAP_HANA_PROMPT", { enumerable: true, get: function () { return sql_db_prompt_js_1.SQL_SAP_HANA_PROMPT; } });
+Object.defineProperty(exports, "SQL_PROMPTS_MAP", { enumerable: true, get: function () { return sql_db_prompt_js_1.SQL_PROMPTS_MAP; } });

package/dist/chains/sql_db/index.d.ts

@@ -1,2 +1,2 @@
-export { SqlDatabaseChain, SqlDatabaseChainInput } from "./sql_db_chain.js";
-export { DEFAULT_SQL_DATABASE_PROMPT, SQL_POSTGRES_PROMPT, SQL_SQLITE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_SAP_HANA_PROMPT, } from "./sql_db_prompt.js";
+export { SqlDatabaseChain, type SqlDatabaseChainInput, type CreateSqlQueryChainFields, createSqlQueryChain, } from "./sql_db_chain.js";
+export { DEFAULT_SQL_DATABASE_PROMPT, SQL_POSTGRES_PROMPT, SQL_SQLITE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_SAP_HANA_PROMPT, SQL_PROMPTS_MAP, } from "./sql_db_prompt.js";

package/dist/chains/sql_db/index.js

@@ -1,2 +1,2 @@
-export { SqlDatabaseChain } from "./sql_db_chain.js";
-export { DEFAULT_SQL_DATABASE_PROMPT, SQL_POSTGRES_PROMPT, SQL_SQLITE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_SAP_HANA_PROMPT, } from "./sql_db_prompt.js";
+export { SqlDatabaseChain, createSqlQueryChain, } from "./sql_db_chain.js";
+export { DEFAULT_SQL_DATABASE_PROMPT, SQL_POSTGRES_PROMPT, SQL_SQLITE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_SAP_HANA_PROMPT, SQL_PROMPTS_MAP, } from "./sql_db_prompt.js";

package/dist/chains/sql_db/sql_db_chain.cjs

@@ -1,7 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SqlDatabaseChain = void 0;
+exports.createSqlQueryChain = exports.SqlDatabaseChain = void 0;
 const base_1 = require("@langchain/core/language_models/base");
+const runnables_1 = require("@langchain/core/runnables");
+const output_parsers_1 = require("@langchain/core/output_parsers");
 const sql_db_prompt_js_1 = require("./sql_db_prompt.cjs");
 const base_js_1 = require("../base.cjs");
 const llm_chain_js_1 = require("../llm_chain.cjs");
@@ -187,3 +189,82 @@ class SqlDatabaseChain extends base_js_1.BaseChain {
     }
 }
 exports.SqlDatabaseChain = SqlDatabaseChain;
+const strip = (text) => {
+    // Replace escaped quotes with actual quotes
+    let newText = text.replace(/\\"/g, '"').trim();
+    // Remove wrapping quotes if the entire string is wrapped in quotes
+    if (newText.startsWith('"') && newText.endsWith('"')) {
+        newText = newText.substring(1, newText.length - 1);
+    }
+    return newText;
+};
+const difference = (setA, setB) => new Set([...setA].filter((x) => !setB.has(x)));
+/**
+ * Create a SQL query chain that can create SQL queries for the given database.
+ * Returns a Runnable.
+ *
+ * @param {BaseLanguageModel} llm The language model to use in the chain.
+ * @param {SqlDatabase} db The database to use in the chain.
+ * @param {BasePromptTemplate | undefined} prompt The prompt to use in the chain.
+ * @param {BaseLanguageModel | undefined} k The amount of docs/results to return. Passed through the prompt input value `top_k`.
+ * @param {SqlDialect} dialect The SQL dialect to use in the chain.
+ * @returns {Promise<RunnableSequence<Record<string, unknown>, string>>} A runnable sequence representing the chain.
+ * @example ```typescript
+ * const datasource = new DataSource({
+ *   type: "sqlite",
+ *   database: "../../../../Chinook.db",
+ * });
+ * const db = await SqlDatabase.fromDataSourceParams({
+ *   appDataSource: datasource,
+ * });
+ * const llm = new ChatOpenAI({ temperature: 0 });
+ * const chain = await createSqlQueryChain({
+ *   llm,
+ *   db,
+ *   dialect: "sqlite",
+ * });
+ * ```
+ */
+async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect, }) {
+    let promptToUse;
+    if (prompt) {
+        promptToUse = prompt;
+    }
+    else if (sql_db_prompt_js_1.SQL_PROMPTS_MAP[dialect]) {
+        promptToUse = sql_db_prompt_js_1.SQL_PROMPTS_MAP[dialect];
+    }
+    else {
+        promptToUse = sql_db_prompt_js_1.DEFAULT_SQL_DATABASE_PROMPT;
+    }
+    if (difference(new Set(["input", "top_k", "table_info"]), new Set(promptToUse.inputVariables)).size > 0) {
+        throw new Error(`Prompt must have input variables: 'input', 'top_k', 'table_info'. Received prompt with input variables: ` +
+            `${promptToUse.inputVariables}. Full prompt:\n\n${promptToUse}`);
+    }
+    if (promptToUse.inputVariables.includes("dialect")) {
+        promptToUse = await promptToUse.partial({ dialect });
+    }
+    promptToUse = await promptToUse.partial({ top_k: k.toString() });
+    const inputs = {
+        input: (x) => {
+            if ("question" in x) {
+                return `${x.question}\nSQLQuery: `;
+            }
+            throw new Error("Input must include a question property.");
+        },
+        table_info: async (x) => db.getTableInfo(x.tableNamesToUse),
+    };
+    return runnables_1.RunnableSequence.from([
+        runnables_1.RunnablePassthrough.assign(inputs),
+        (x) => {
+            const newInputs = { ...x };
+            delete newInputs.question;
+            delete newInputs.tableNamesToUse;
+            return newInputs;
+        },
+        promptToUse,
+        llm.bind({ stop: ["\nSQLResult:"] }),
+        new output_parsers_1.StringOutputParser(),
+        strip,
+    ]);
+}
+exports.createSqlQueryChain = createSqlQueryChain;

package/dist/chains/sql_db/sql_db_chain.d.ts

@@ -1,7 +1,9 @@
-import type { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
+import type { BaseLanguageModel, BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { ChainValues } from "@langchain/core/utils/types";
-import { PromptTemplate } from "@langchain/core/prompts";
+import { BasePromptTemplate, PromptTemplate } from "@langchain/core/prompts";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
+import { RunnableSequence } from "@langchain/core/runnables";
+import { SqlDialect } from "./sql_db_prompt.js";
 import { BaseChain, ChainInputs } from "../base.js";
 import type { SqlDatabase } from "../../sql_db.js";
 /**
@@ -74,3 +76,40 @@ export declare class SqlDatabaseChain extends BaseChain {
      */
     private verifyNumberOfTokens;
 }
+export interface CreateSqlQueryChainFields {
+    llm: BaseLanguageModel;
+    db: SqlDatabase;
+    prompt?: BasePromptTemplate;
+    /**
+     * @default 5
+     */
+    k?: number;
+    dialect: SqlDialect;
+}
+/**
+ * Create a SQL query chain that can create SQL queries for the given database.
+ * Returns a Runnable.
+ *
+ * @param {BaseLanguageModel} llm The language model to use in the chain.
+ * @param {SqlDatabase} db The database to use in the chain.
+ * @param {BasePromptTemplate | undefined} prompt The prompt to use in the chain.
+ * @param {BaseLanguageModel | undefined} k The amount of docs/results to return. Passed through the prompt input value `top_k`.
+ * @param {SqlDialect} dialect The SQL dialect to use in the chain.
+ * @returns {Promise<RunnableSequence<Record<string, unknown>, string>>} A runnable sequence representing the chain.
+ * @example ```typescript
+ * const datasource = new DataSource({
+ *   type: "sqlite",
+ *   database: "../../../../Chinook.db",
+ * });
+ * const db = await SqlDatabase.fromDataSourceParams({
+ *   appDataSource: datasource,
+ * });
+ * const llm = new ChatOpenAI({ temperature: 0 });
+ * const chain = await createSqlQueryChain({
+ *   llm,
+ *   db,
+ *   dialect: "sqlite",
+ * });
+ * ```
+ */
+export declare function createSqlQueryChain({ llm, db, prompt, k, dialect, }: CreateSqlQueryChainFields): Promise<RunnableSequence<Record<string, unknown>, string>>;

package/dist/chains/sql_db/sql_db_chain.js

@@ -1,5 +1,7 @@
 import { calculateMaxTokens, getModelContextSize, } from "@langchain/core/language_models/base";
-import { DEFAULT_SQL_DATABASE_PROMPT } from "./sql_db_prompt.js";
+import { RunnablePassthrough, RunnableSequence, } from "@langchain/core/runnables";
+import { StringOutputParser } from "@langchain/core/output_parsers";
+import { DEFAULT_SQL_DATABASE_PROMPT, SQL_PROMPTS_MAP, } from "./sql_db_prompt.js";
 import { BaseChain } from "../base.js";
 import { LLMChain } from "../llm_chain.js";
 import { getPromptTemplateFromDataSource } from "../../util/sql_utils.js";
@@ -183,3 +185,81 @@ export class SqlDatabaseChain extends BaseChain {
         }
     }
 }
+const strip = (text) => {
+    // Replace escaped quotes with actual quotes
+    let newText = text.replace(/\\"/g, '"').trim();
+    // Remove wrapping quotes if the entire string is wrapped in quotes
+    if (newText.startsWith('"') && newText.endsWith('"')) {
+        newText = newText.substring(1, newText.length - 1);
+    }
+    return newText;
+};
+const difference = (setA, setB) => new Set([...setA].filter((x) => !setB.has(x)));
+/**
+ * Create a SQL query chain that can create SQL queries for the given database.
+ * Returns a Runnable.
+ *
+ * @param {BaseLanguageModel} llm The language model to use in the chain.
+ * @param {SqlDatabase} db The database to use in the chain.
+ * @param {BasePromptTemplate | undefined} prompt The prompt to use in the chain.
+ * @param {BaseLanguageModel | undefined} k The amount of docs/results to return. Passed through the prompt input value `top_k`.
+ * @param {SqlDialect} dialect The SQL dialect to use in the chain.
+ * @returns {Promise<RunnableSequence<Record<string, unknown>, string>>} A runnable sequence representing the chain.
+ * @example ```typescript
+ * const datasource = new DataSource({
+ *   type: "sqlite",
+ *   database: "../../../../Chinook.db",
+ * });
+ * const db = await SqlDatabase.fromDataSourceParams({
+ *   appDataSource: datasource,
+ * });
+ * const llm = new ChatOpenAI({ temperature: 0 });
+ * const chain = await createSqlQueryChain({
+ *   llm,
+ *   db,
+ *   dialect: "sqlite",
+ * });
+ * ```
+ */
+export async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect, }) {
+    let promptToUse;
+    if (prompt) {
+        promptToUse = prompt;
+    }
+    else if (SQL_PROMPTS_MAP[dialect]) {
+        promptToUse = SQL_PROMPTS_MAP[dialect];
+    }
+    else {
+        promptToUse = DEFAULT_SQL_DATABASE_PROMPT;
+    }
+    if (difference(new Set(["input", "top_k", "table_info"]), new Set(promptToUse.inputVariables)).size > 0) {
+        throw new Error(`Prompt must have input variables: 'input', 'top_k', 'table_info'. Received prompt with input variables: ` +
+            `${promptToUse.inputVariables}. Full prompt:\n\n${promptToUse}`);
+    }
+    if (promptToUse.inputVariables.includes("dialect")) {
+        promptToUse = await promptToUse.partial({ dialect });
+    }
+    promptToUse = await promptToUse.partial({ top_k: k.toString() });
+    const inputs = {
+        input: (x) => {
+            if ("question" in x) {
+                return `${x.question}\nSQLQuery: `;
+            }
+            throw new Error("Input must include a question property.");
+        },
+        table_info: async (x) => db.getTableInfo(x.tableNamesToUse),
+    };
+    return RunnableSequence.from([
+        RunnablePassthrough.assign(inputs),
+        (x) => {
+            const newInputs = { ...x };
+            delete newInputs.question;
+            delete newInputs.tableNamesToUse;
+            return newInputs;
+        },
+        promptToUse,
+        llm.bind({ stop: ["\nSQLResult:"] }),
+        new StringOutputParser(),
+        strip,
+    ]);
+}

package/dist/chains/sql_db/sql_db_prompt.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SQL_ORACLE_PROMPT = exports.SQL_SAP_HANA_PROMPT = exports.SQL_MSSQL_PROMPT = exports.SQL_MYSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = void 0;
+exports.SQL_PROMPTS_MAP = exports.SQL_ORACLE_PROMPT = exports.SQL_SAP_HANA_PROMPT = exports.SQL_MSSQL_PROMPT = exports.SQL_MYSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = void 0;
 /* eslint-disable spaced-comment */
 const prompts_1 = require("@langchain/core/prompts");
 exports.DEFAULT_SQL_DATABASE_PROMPT = new prompts_1.PromptTemplate({
@@ -139,3 +139,11 @@ Only use the following tables:
 Question: {input}`,
     inputVariables: ["dialect", "table_info", "input", "top_k"],
 });
+exports.SQL_PROMPTS_MAP = {
+    oracle: exports.SQL_ORACLE_PROMPT,
+    postgres: exports.SQL_POSTGRES_PROMPT,
+    sqlite: exports.SQL_SQLITE_PROMPT,
+    mysql: exports.SQL_MYSQL_PROMPT,
+    mssql: exports.SQL_MSSQL_PROMPT,
+    "sap hana": exports.SQL_SAP_HANA_PROMPT,
+};

package/dist/chains/sql_db/sql_db_prompt.d.ts

@@ -1,4 +1,4 @@
-import { PromptTemplate } from "@langchain/core/prompts";
+import { BasePromptTemplate, PromptTemplate } from "@langchain/core/prompts";
 export declare const DEFAULT_SQL_DATABASE_PROMPT: PromptTemplate<{
     input: any;
     top_k: any;
@@ -41,3 +41,5 @@ export declare const SQL_ORACLE_PROMPT: PromptTemplate<{
     dialect: any;
     table_info: any;
 }, any>;
+export type SqlDialect = "oracle" | "postgres" | "sqlite" | "mysql" | "mssql" | "sap hana";
+export declare const SQL_PROMPTS_MAP: Record<SqlDialect, BasePromptTemplate>;

package/dist/chains/sql_db/sql_db_prompt.js

@@ -136,3 +136,11 @@ Only use the following tables:
 Question: {input}`,
     inputVariables: ["dialect", "table_info", "input", "top_k"],
 });
+export const SQL_PROMPTS_MAP = {
+    oracle: SQL_ORACLE_PROMPT,
+    postgres: SQL_POSTGRES_PROMPT,
+    sqlite: SQL_SQLITE_PROMPT,
+    mysql: SQL_MYSQL_PROMPT,
+    mssql: SQL_MSSQL_PROMPT,
+    "sap hana": SQL_SAP_HANA_PROMPT,
+};

package/dist/document_loaders/web/confluence.cjs

@@ -19,7 +19,7 @@ const base_js_1 = require("../base.cjs");
  * ```
  */
 class ConfluencePagesLoader extends base_js_1.BaseDocumentLoader {
-    constructor({ baseUrl, spaceKey, username, accessToken, limit = 25, personalAccessToken, }) {
+    constructor({ baseUrl, spaceKey, username, accessToken, limit = 25, expand = "body.storage", personalAccessToken, }) {
         super();
         Object.defineProperty(this, "baseUrl", {
             enumerable: true,
@@ -51,6 +51,16 @@ class ConfluencePagesLoader extends base_js_1.BaseDocumentLoader {
             writable: true,
             value: void 0
         });
+        /**
+         * expand parameter for confluence rest api
+         * description can be found at https://developer.atlassian.com/server/confluence/expansions-in-the-rest-api/
+         */
+        Object.defineProperty(this, "expand", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "personalAccessToken", {
             enumerable: true,
             configurable: true,
@@ -62,6 +72,7 @@ class ConfluencePagesLoader extends base_js_1.BaseDocumentLoader {
         this.username = username;
         this.accessToken = accessToken;
         this.limit = limit;
+        this.expand = expand;
         this.personalAccessToken = personalAccessToken;
     }
     /**
@@ -126,7 +137,7 @@ class ConfluencePagesLoader extends base_js_1.BaseDocumentLoader {
      * @returns Promise resolving to an array of ConfluencePage objects.
      */
     async fetchAllPagesInSpace(start = 0) {
-        const url = `${this.baseUrl}/rest/api/content?spaceKey=${this.spaceKey}&limit=${this.limit}&start=${start}&expand=body.storage`;
+        const url = `${this.baseUrl}/rest/api/content?spaceKey=${this.spaceKey}&limit=${this.limit}&start=${start}&expand=${this.expand}`;
         const data = await this.fetchConfluenceData(url);
         if (data.size === 0) {
             return [];

package/dist/document_loaders/web/confluence.d.ts

@@ -11,6 +11,7 @@ export interface ConfluencePagesLoaderParams {
     accessToken?: string;
     personalAccessToken?: string;
     limit?: number;
+    expand?: string;
 }
 /**
  * Interface representing a Confluence page.
@@ -51,8 +52,13 @@ export declare class ConfluencePagesLoader extends BaseDocumentLoader {
     readonly username?: string;
     readonly accessToken?: string;
     readonly limit: number;
+    /**
+     * expand parameter for confluence rest api
+     * description can be found at https://developer.atlassian.com/server/confluence/expansions-in-the-rest-api/
+     */
+    readonly expand?: string;
     readonly personalAccessToken?: string;
-    constructor({ baseUrl, spaceKey, username, accessToken, limit, personalAccessToken, }: ConfluencePagesLoaderParams);
+    constructor({ baseUrl, spaceKey, username, accessToken, limit, expand, personalAccessToken, }: ConfluencePagesLoaderParams);
     /**
      * Returns the authorization header for the request.
      * @returns The authorization header as a string, or undefined if no credentials were provided.

package/dist/document_loaders/web/confluence.js

@@ -16,7 +16,7 @@ import { BaseDocumentLoader } from "../base.js";
  * ```
  */
 export class ConfluencePagesLoader extends BaseDocumentLoader {
-    constructor({ baseUrl, spaceKey, username, accessToken, limit = 25, personalAccessToken, }) {
+    constructor({ baseUrl, spaceKey, username, accessToken, limit = 25, expand = "body.storage", personalAccessToken, }) {
         super();
         Object.defineProperty(this, "baseUrl", {
             enumerable: true,
@@ -48,6 +48,16 @@ export class ConfluencePagesLoader extends BaseDocumentLoader {
             writable: true,
             value: void 0
         });
+        /**
+         * expand parameter for confluence rest api
+         * description can be found at https://developer.atlassian.com/server/confluence/expansions-in-the-rest-api/
+         */
+        Object.defineProperty(this, "expand", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "personalAccessToken", {
             enumerable: true,
             configurable: true,
@@ -59,6 +69,7 @@ export class ConfluencePagesLoader extends BaseDocumentLoader {
         this.username = username;
         this.accessToken = accessToken;
         this.limit = limit;
+        this.expand = expand;
         this.personalAccessToken = personalAccessToken;
     }
     /**
@@ -123,7 +134,7 @@ export class ConfluencePagesLoader extends BaseDocumentLoader {
      * @returns Promise resolving to an array of ConfluencePage objects.
      */
     async fetchAllPagesInSpace(start = 0) {
-        const url = `${this.baseUrl}/rest/api/content?spaceKey=${this.spaceKey}&limit=${this.limit}&start=${start}&expand=body.storage`;
+        const url = `${this.baseUrl}/rest/api/content?spaceKey=${this.spaceKey}&limit=${this.limit}&start=${start}&expand=${this.expand}`;
         const data = await this.fetchConfluenceData(url);
         if (data.size === 0) {
             return [];

package/dist/util/entrypoint_deprecation.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.logVersion010MigrationWarning = void 0;
+const env_js_1 = require("./env.cjs");
 function logVersion010MigrationWarning({ oldEntrypointName, newEntrypointName, newPackageName = "@langchain/community", }) {
     let finalEntrypointName = "";
     if (newEntrypointName === undefined) {
@@ -29,6 +30,8 @@ function logVersion010MigrationWarning({ oldEntrypointName, newEntrypointName, n
             `This will be mandatory after the next "langchain" minor version bump to 0.2.`,
         ].join("\n");
     }
-    console.warn(warningText);
+    if ((0, env_js_1.getEnvironmentVariable)("LANGCHAIN_SUPPRESS_MIGRATION_WARNINGS") !== "true") {
+        console.warn(warningText);
+    }
 }
 exports.logVersion010MigrationWarning = logVersion010MigrationWarning;

package/dist/util/entrypoint_deprecation.js

@@ -1,3 +1,4 @@
+import { getEnvironmentVariable } from "./env.js";
 export function logVersion010MigrationWarning({ oldEntrypointName, newEntrypointName, newPackageName = "@langchain/community", }) {
     let finalEntrypointName = "";
     if (newEntrypointName === undefined) {
@@ -26,5 +27,7 @@ export function logVersion010MigrationWarning({ oldEntrypointName, newEntrypoint
             `This will be mandatory after the next "langchain" minor version bump to 0.2.`,
         ].join("\n");
     }
-    console.warn(warningText);
+    if (getEnvironmentVariable("LANGCHAIN_SUPPRESS_MIGRATION_WARNINGS") !== "true") {
+        console.warn(warningText);
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.1.9",
+  "version": "0.1.11",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {