langchain 0.0.69 → 0.0.71

package/dist/llms/openai.js

@@ -1,3 +1,4 @@
+import { isNode } from "browser-or-node";
 import { Configuration, OpenAIApi, } from "openai";
 import fetchAdapter from "../util/axios-fetch-adapter.js";
 import { chunkArray } from "../util/chunk.js";
@@ -10,6 +11,12 @@ import { OpenAIChat } from "./openai-chat.js";
  * To use you should have the `openai` package installed, with the
  * `OPENAI_API_KEY` environment variable set.
  *
+ * To use with Azure you should have the `openai` package installed, with the
+ * `AZURE_OPENAI_API_KEY`,
+ * `AZURE_OPENAI_API_INSTANCE_NAME`,
+ * `AZURE_OPENAI_API_DEPLOYMENT_NAME`
+ * and `AZURE_OPENAI_API_VERSION` environment variable set.
+ *
  * @remarks
  * Any parameters that are valid to be passed to {@link
  * https://platform.openai.com/docs/api-reference/completions/create |
@@ -19,7 +26,8 @@ import { OpenAIChat } from "./openai-chat.js";
 export class OpenAI extends BaseLLM {
     constructor(fields, configuration) {
         if (fields?.modelName?.startsWith("gpt-3.5-turbo") ||
-            fields?.modelName?.startsWith("gpt-4")) {
+            fields?.modelName?.startsWith("gpt-4") ||
+            fields?.modelName?.startsWith("gpt-4-32k")) {
             // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
             return new OpenAIChat(fields, configuration);
         }
@@ -108,6 +116,30 @@ export class OpenAI extends BaseLLM {
             writable: true,
             value: false
         });
+        Object.defineProperty(this, "azureOpenAIApiVersion", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "azureOpenAIApiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "azureOpenAIApiInstanceName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "azureOpenAIApiDeploymentName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
@@ -125,9 +157,32 @@ export class OpenAI extends BaseLLM {
                 ? // eslint-disable-next-line no-process-env
                     process.env?.OPENAI_API_KEY
                 : undefined);
-        if (!apiKey) {
-            throw new Error("OpenAI API key not found");
+        const azureApiKey = fields?.azureOpenAIApiKey ??
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.AZURE_OPENAI_API_KEY
+                : undefined);
+        if (!azureApiKey && !apiKey) {
+            throw new Error("(Azure) OpenAI API key not found");
         }
+        const azureApiInstanceName = fields?.azureOpenAIApiInstanceName ??
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.AZURE_OPENAI_API_INSTANCE_NAME
+                : undefined);
+        const azureApiDeploymentName = (fields?.azureOpenAIApiCompletionsDeploymentName ||
+            fields?.azureOpenAIApiDeploymentName) ??
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.AZURE_OPENAI_API_COMPLETIONS_DEPLOYMENT_NAME ||
+                        // eslint-disable-next-line no-process-env
+                        process.env?.AZURE_OPENAI_API_DEPLOYMENT_NAME
+                : undefined);
+        const azureApiVersion = fields?.azureOpenAIApiVersion ??
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.AZURE_OPENAI_API_VERSION
+                : undefined);
         this.modelName = fields?.modelName ?? this.modelName;
         this.modelKwargs = fields?.modelKwargs ?? {};
         this.batchSize = fields?.batchSize ?? this.batchSize;
@@ -142,12 +197,27 @@ export class OpenAI extends BaseLLM {
         this.logitBias = fields?.logitBias;
         this.stop = fields?.stop;
         this.streaming = fields?.streaming ?? false;
+        this.azureOpenAIApiVersion = azureApiVersion;
+        this.azureOpenAIApiKey = azureApiKey;
+        this.azureOpenAIApiInstanceName = azureApiInstanceName;
+        this.azureOpenAIApiDeploymentName = azureApiDeploymentName;
         if (this.streaming && this.n > 1) {
             throw new Error("Cannot stream results when n > 1");
         }
         if (this.streaming && this.bestOf > 1) {
             throw new Error("Cannot stream results when bestOf > 1");
         }
+        if (this.azureOpenAIApiKey) {
+            if (!this.azureOpenAIApiInstanceName) {
+                throw new Error("Azure OpenAI API instance name not found");
+            }
+            if (!this.azureOpenAIApiDeploymentName) {
+                throw new Error("Azure OpenAI API deployment name not found");
+            }
+            if (!this.azureOpenAIApiVersion) {
+                throw new Error("Azure OpenAI API version not found");
+            }
+        }
         this.clientConfig = {
             apiKey,
             ...configuration,
@@ -237,6 +307,7 @@ export class OpenAI extends BaseLLM {
                         prompt: subPrompts[i],
                     }, {
                         ...options,
+                        adapter: fetchAdapter,
                         responseType: "stream",
                         onmessage: (event) => {
                             if (event.data?.trim?.() === "[DONE]") {
@@ -306,18 +377,36 @@ export class OpenAI extends BaseLLM {
     /** @ignore */
     async completionWithRetry(request, options) {
         if (!this.client) {
+            const endpoint = this.azureOpenAIApiKey
+                ? `https://${this.azureOpenAIApiInstanceName}.openai.azure.com/openai/deployments/${this.azureOpenAIApiDeploymentName}`
+                : this.clientConfig.basePath;
             const clientConfig = new Configuration({
                 ...this.clientConfig,
+                basePath: endpoint,
                 baseOptions: {
                     timeout: this.timeout,
-                    adapter: fetchAdapter,
                     ...this.clientConfig.baseOptions,
                 },
             });
             this.client = new OpenAIApi(clientConfig);
         }
+        const axiosOptions = {
+            adapter: isNode ? undefined : fetchAdapter,
+            ...this.clientConfig.baseOptions,
+            ...options,
+        };
+        if (this.azureOpenAIApiKey) {
+            axiosOptions.headers = {
+                "api-key": this.azureOpenAIApiKey,
+                ...axiosOptions.headers,
+            };
+            axiosOptions.params = {
+                "api-version": this.azureOpenAIApiVersion,
+                ...axiosOptions.params,
+            };
+        }
         return this.caller
-            .call(this.client.createCompletion.bind(this.client), request, options)
+            .call(this.client.createCompletion.bind(this.client), request, axiosOptions)
             .then((res) => res.data);
     }
     _llmType() {
@@ -382,4 +471,4 @@ export class PromptLayerOpenAI extends OpenAI {
         return response;
     }
 }
-export { OpenAIChat, PromptLayerOpenAIChat, } from "./openai-chat.js";
+export { OpenAIChat, PromptLayerOpenAIChat } from "./openai-chat.js";

package/dist/memory/base.cjs

@@ -17,7 +17,7 @@ const getInputValue = (inputValues, inputKey) => {
     if (keys.length === 1) {
         return inputValues[keys[0]];
     }
-    throw new Error(`input values have multiple keys, memory only supported when one key currently: ${keys}`);
+    throw new Error(`input values have ${keys.length} keys, you must specify an input key or pass only 1 key as input`);
 };
 exports.getInputValue = getInputValue;
 /**

package/dist/memory/base.js

@@ -13,7 +13,7 @@ export const getInputValue = (inputValues, inputKey) => {
     if (keys.length === 1) {
         return inputValues[keys[0]];
     }
-    throw new Error(`input values have multiple keys, memory only supported when one key currently: ${keys}`);
+    throw new Error(`input values have ${keys.length} keys, you must specify an input key or pass only 1 key as input`);
 };
 /**
  * This function is used by memory classes to get a string representation

package/dist/memory/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MotorheadMemory = exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.ConversationSummaryMemory = exports.getBufferString = exports.getInputValue = exports.BaseMemory = exports.BufferMemory = void 0;
+exports.VectorStoreRetrieverMemory = exports.MotorheadMemory = exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.ConversationSummaryMemory = exports.getBufferString = exports.getInputValue = exports.BaseMemory = exports.BufferMemory = void 0;
 var buffer_memory_js_1 = require("./buffer_memory.cjs");
 Object.defineProperty(exports, "BufferMemory", { enumerable: true, get: function () { return buffer_memory_js_1.BufferMemory; } });
 var base_js_1 = require("./base.cjs");
@@ -17,3 +17,5 @@ var in_memory_js_1 = require("../stores/message/in_memory.cjs");
 Object.defineProperty(exports, "ChatMessageHistory", { enumerable: true, get: function () { return in_memory_js_1.ChatMessageHistory; } });
 var motorhead_memory_js_1 = require("./motorhead_memory.cjs");
 Object.defineProperty(exports, "MotorheadMemory", { enumerable: true, get: function () { return motorhead_memory_js_1.MotorheadMemory; } });
+var vector_store_js_1 = require("./vector_store.cjs");
+Object.defineProperty(exports, "VectorStoreRetrieverMemory", { enumerable: true, get: function () { return vector_store_js_1.VectorStoreRetrieverMemory; } });

package/dist/memory/index.d.ts

@@ -5,3 +5,4 @@ export { BufferWindowMemory, BufferWindowMemoryInput, } from "./buffer_window_me
 export { BaseChatMemory, BaseChatMemoryInput } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
 export { MotorheadMemory, MotorheadMemoryInput } from "./motorhead_memory.js";
+export { VectorStoreRetrieverMemory, VectorStoreRetrieverMemoryParams, } from "./vector_store.js";

package/dist/memory/index.js

@@ -5,3 +5,4 @@ export { BufferWindowMemory, } from "./buffer_window_memory.js";
 export { BaseChatMemory } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
 export { MotorheadMemory } from "./motorhead_memory.js";
+export { VectorStoreRetrieverMemory, } from "./vector_store.js";

package/dist/memory/vector_store.cjs

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorStoreRetrieverMemory = void 0;
+const document_js_1 = require("../document.cjs");
+const base_js_1 = require("./base.cjs");
+class VectorStoreRetrieverMemory extends base_js_1.BaseMemory {
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "vectorStoreRetriever", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "memoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "returnDocs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.vectorStoreRetriever = fields.vectorStoreRetriever;
+        this.inputKey = fields.inputKey;
+        this.memoryKey = fields.memoryKey ?? "memory";
+        this.returnDocs = fields.returnDocs ?? false;
+    }
+    get memoryKeys() {
+        return [this.memoryKey];
+    }
+    async loadMemoryVariables(values) {
+        const query = (0, base_js_1.getInputValue)(values, this.inputKey);
+        const results = await this.vectorStoreRetriever.getRelevantDocuments(query);
+        return {
+            [this.memoryKey]: this.returnDocs
+                ? results
+                : results.map((r) => r.pageContent).join("\n"),
+        };
+    }
+    async saveContext(inputValues, outputValues) {
+        const text = Object.entries(inputValues)
+            .filter(([k]) => k !== this.memoryKey)
+            .concat(Object.entries(outputValues))
+            .map(([k, v]) => `${k}: ${v}`)
+            .join("\n");
+        await this.vectorStoreRetriever.addDocuments([
+            new document_js_1.Document({ pageContent: text }),
+        ]);
+    }
+}
+exports.VectorStoreRetrieverMemory = VectorStoreRetrieverMemory;

package/dist/memory/vector_store.d.ts

@@ -0,0 +1,19 @@
+import { VectorStoreRetriever } from "../vectorstores/base.js";
+import { BaseMemory, InputValues, MemoryVariables, OutputValues } from "./base.js";
+export interface VectorStoreRetrieverMemoryParams {
+    vectorStoreRetriever: VectorStoreRetriever;
+    inputKey?: string;
+    outputKey?: string;
+    memoryKey?: string;
+    returnDocs?: boolean;
+}
+export declare class VectorStoreRetrieverMemory extends BaseMemory implements VectorStoreRetrieverMemoryParams {
+    vectorStoreRetriever: VectorStoreRetriever;
+    inputKey?: string;
+    memoryKey: string;
+    returnDocs: boolean;
+    constructor(fields: VectorStoreRetrieverMemoryParams);
+    get memoryKeys(): string[];
+    loadMemoryVariables(values: InputValues): Promise<MemoryVariables>;
+    saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
+}

package/dist/memory/vector_store.js

@@ -0,0 +1,57 @@
+import { Document } from "../document.js";
+import { BaseMemory, getInputValue, } from "./base.js";
+export class VectorStoreRetrieverMemory extends BaseMemory {
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "vectorStoreRetriever", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "memoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "returnDocs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.vectorStoreRetriever = fields.vectorStoreRetriever;
+        this.inputKey = fields.inputKey;
+        this.memoryKey = fields.memoryKey ?? "memory";
+        this.returnDocs = fields.returnDocs ?? false;
+    }
+    get memoryKeys() {
+        return [this.memoryKey];
+    }
+    async loadMemoryVariables(values) {
+        const query = getInputValue(values, this.inputKey);
+        const results = await this.vectorStoreRetriever.getRelevantDocuments(query);
+        return {
+            [this.memoryKey]: this.returnDocs
+                ? results
+                : results.map((r) => r.pageContent).join("\n"),
+        };
+    }
+    async saveContext(inputValues, outputValues) {
+        const text = Object.entries(inputValues)
+            .filter(([k]) => k !== this.memoryKey)
+            .concat(Object.entries(outputValues))
+            .map(([k, v]) => `${k}: ${v}`)
+            .join("\n");
+        await this.vectorStoreRetriever.addDocuments([
+            new Document({ pageContent: text }),
+        ]);
+    }
+}

package/dist/output_parsers/combining.cjs

@@ -15,20 +15,29 @@ class CombiningOutputParser extends output_parser_js_1.BaseOutputParser {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "outputDelimiter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "-----"
+        });
         this.parsers = parsers;
     }
     async parse(input, callbacks) {
         const inputs = input
             .trim()
-            .split(/Output \d+:/)
+            .split(new RegExp(`${this.outputDelimiter}Output \\d+${this.outputDelimiter}`))
             .slice(1);
         const ret = {};
         for (const [i, p] of this.parsers.entries()) {
             let parsed;
             try {
-                const extracted = inputs[i].includes("```")
+                let extracted = inputs[i].includes("```")
                     ? inputs[i].trim().split(/```/)[1]
                     : inputs[i].trim();
+                if (extracted.endsWith(this.outputDelimiter)) {
+                    extracted = extracted.slice(0, -this.outputDelimiter.length);
+                }
                 parsed = await p.parse(extracted, callbacks);
             }
             catch (e) {
@@ -40,8 +49,10 @@ class CombiningOutputParser extends output_parser_js_1.BaseOutputParser {
     }
     getFormatInstructions() {
         return `${[
-            `Return the following ${this.parsers.length} outputs, each formatted as described below:`,
-            ...this.parsers.map((p, i) => `Output ${i + 1}:\n${p.getFormatInstructions().trim()}`),
+            `Return the following ${this.parsers.length} outputs, each formatted as described below. Include the delimiter characters "${this.outputDelimiter}" in your response:`,
+            ...this.parsers.map((p, i) => `${this.outputDelimiter}Output ${i + 1}${this.outputDelimiter}\n${p
+                .getFormatInstructions()
+                .trim()}\n${this.outputDelimiter}`),
         ].join("\n\n")}\n`;
     }
 }

package/dist/output_parsers/combining.d.ts

@@ -7,6 +7,7 @@ export type CombinedOutput = Record<string, any>;
  */
 export declare class CombiningOutputParser extends BaseOutputParser {
     parsers: BaseOutputParser[];
+    outputDelimiter: string;
     constructor(...parsers: BaseOutputParser[]);
     parse(input: string, callbacks?: Callbacks): Promise<CombinedOutput>;
     getFormatInstructions(): string;

package/dist/output_parsers/combining.js

@@ -12,20 +12,29 @@ export class CombiningOutputParser extends BaseOutputParser {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "outputDelimiter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "-----"
+        });
         this.parsers = parsers;
     }
     async parse(input, callbacks) {
         const inputs = input
             .trim()
-            .split(/Output \d+:/)
+            .split(new RegExp(`${this.outputDelimiter}Output \\d+${this.outputDelimiter}`))
             .slice(1);
         const ret = {};
         for (const [i, p] of this.parsers.entries()) {
             let parsed;
             try {
-                const extracted = inputs[i].includes("```")
+                let extracted = inputs[i].includes("```")
                     ? inputs[i].trim().split(/```/)[1]
                     : inputs[i].trim();
+                if (extracted.endsWith(this.outputDelimiter)) {
+                    extracted = extracted.slice(0, -this.outputDelimiter.length);
+                }
                 parsed = await p.parse(extracted, callbacks);
             }
             catch (e) {
@@ -37,8 +46,10 @@ export class CombiningOutputParser extends BaseOutputParser {
     }
     getFormatInstructions() {
         return `${[
-            `Return the following ${this.parsers.length} outputs, each formatted as described below:`,
-            ...this.parsers.map((p, i) => `Output ${i + 1}:\n${p.getFormatInstructions().trim()}`),
+            `Return the following ${this.parsers.length} outputs, each formatted as described below. Include the delimiter characters "${this.outputDelimiter}" in your response:`,
+            ...this.parsers.map((p, i) => `${this.outputDelimiter}Output ${i + 1}${this.outputDelimiter}\n${p
+                .getFormatInstructions()
+                .trim()}\n${this.outputDelimiter}`),
         ].join("\n\n")}\n`;
     }
 }

package/dist/output_parsers/structured.cjs

@@ -22,13 +22,18 @@ class StructuredOutputParser extends output_parser_js_1.BaseOutputParser {
         return new this(zodSchema);
     }
     getFormatInstructions() {
-        return `The output should be formatted as a JSON instance that conforms to the JSON schema below.
+        return `You must format your output as a JSON value that adheres to a given "JSON Schema" instance.
 
-As an example, for the schema {{"properties": {{"foo": {{"title": "Foo", "description": "a list of strings", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
-the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of the schema. The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+"JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
 
-Here is the output schema:
-\`\`\`
+For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
+would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
+Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+
+Your output will be parsed and type-checked according to the provided schema instance, so make sure all fields in your output match exactly!
+
+Here is the JSON Schema instance your output must adhere to:
+\`\`\`json
 ${JSON.stringify((0, zod_to_json_schema_1.zodToJsonSchema)(this.schema))}
 \`\`\`
 `;

package/dist/output_parsers/structured.js

@@ -19,13 +19,18 @@ export class StructuredOutputParser extends BaseOutputParser {
         return new this(zodSchema);
     }
     getFormatInstructions() {
-        return `The output should be formatted as a JSON instance that conforms to the JSON schema below.
+        return `You must format your output as a JSON value that adheres to a given "JSON Schema" instance.
 
-As an example, for the schema {{"properties": {{"foo": {{"title": "Foo", "description": "a list of strings", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
-the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of the schema. The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+"JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
 
-Here is the output schema:
-\`\`\`
+For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
+would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
+Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
+
+Your output will be parsed and type-checked according to the provided schema instance, so make sure all fields in your output match exactly!
+
+Here is the JSON Schema instance your output must adhere to:
+\`\`\`json
 ${JSON.stringify(zodToJsonSchema(this.schema))}
 \`\`\`
 `;

package/dist/retrievers/time_weighted.d.ts

@@ -1,4 +1,4 @@
-import { VectorStore } from "vectorstores/base.js";
+import { VectorStore } from "../vectorstores/base.js";
 import { Document } from "../document.js";
 import { BaseRetriever } from "../schema/index.js";
 export interface TimeWeightedVectorStoreRetrieverFields {

package/dist/tools/webbrowser.cjs

@@ -171,7 +171,7 @@ class WebBrowser extends base_js_2.Tool {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: `useful for when you need to find something on or summarize a webpage. input should be a comma seperated list of "ONE valid http URL including protocol","what you want to find on the page or empty string for a summary".`
+            value: `useful for when you need to find something on or summarize a webpage. input should be a comma separated list of "ONE valid http URL including protocol","what you want to find on the page or empty string for a summary".`
         });
         this.model = model;
         this.embeddings = embeddings;

package/dist/tools/webbrowser.js

@@ -140,7 +140,7 @@ export class WebBrowser extends Tool {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: `useful for when you need to find something on or summarize a webpage. input should be a comma seperated list of "ONE valid http URL including protocol","what you want to find on the page or empty string for a summary".`
+            value: `useful for when you need to find something on or summarize a webpage. input should be a comma separated list of "ONE valid http URL including protocol","what you want to find on the page or empty string for a summary".`
         });
         this.model = model;
         this.embeddings = embeddings;

package/dist/util/sql_utils.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateTableInfoFromTables = exports.getTableAndColumnsName = exports.verifyIgnoreTablesExistInDatabase = exports.verifyIncludeTablesExistInDatabase = exports.verifyListTablesExistInDatabase = void 0;
+exports.getPromptTemplateFromDataSource = exports.generateTableInfoFromTables = exports.getTableAndColumnsName = exports.verifyIgnoreTablesExistInDatabase = exports.verifyIncludeTablesExistInDatabase = exports.verifyListTablesExistInDatabase = void 0;
+const sql_db_prompt_js_1 = require("../chains/sql_db/sql_db_prompt.cjs");
 const verifyListTablesExistInDatabase = (tablesFromDatabase, listTables, errorPrefixMsg) => {
     const onlyTableNames = tablesFromDatabase.map((table) => table.tableName);
     if (listTables.length > 0) {
@@ -159,3 +160,16 @@ const generateTableInfoFromTables = async (tables, appDataSource, nbSampleRow) =
     return globalString;
 };
 exports.generateTableInfoFromTables = generateTableInfoFromTables;
+const getPromptTemplateFromDataSource = (appDataSource) => {
+    if (appDataSource.options.type === "postgres") {
+        return sql_db_prompt_js_1.SQL_POSTGRES_PROMPT;
+    }
+    if (appDataSource.options.type === "sqlite") {
+        return sql_db_prompt_js_1.SQL_SQLITE_PROMPT;
+    }
+    if (appDataSource.options.type === "mysql") {
+        return sql_db_prompt_js_1.SQL_MYSQL_PROMPT;
+    }
+    return sql_db_prompt_js_1.DEFAULT_SQL_DATABASE_PROMPT;
+};
+exports.getPromptTemplateFromDataSource = getPromptTemplateFromDataSource;

package/dist/util/sql_utils.d.ts

@@ -1,4 +1,5 @@
 import type { DataSource, DataSourceOptions } from "typeorm";
+import { PromptTemplate } from "../prompts/index.js";
 export interface SqlDatabaseParams {
     includesTables?: Array<string>;
     ignoreTables?: Array<string>;
@@ -27,3 +28,4 @@ export declare const verifyIncludeTablesExistInDatabase: (tablesFromDatabase: Ar
 export declare const verifyIgnoreTablesExistInDatabase: (tablesFromDatabase: Array<SqlTable>, ignoreTables: Array<string>) => void;
 export declare const getTableAndColumnsName: (appDataSource: DataSource) => Promise<Array<SqlTable>>;
 export declare const generateTableInfoFromTables: (tables: Array<SqlTable> | undefined, appDataSource: DataSource, nbSampleRow: number) => Promise<string>;
+export declare const getPromptTemplateFromDataSource: (appDataSource: DataSource) => PromptTemplate;

package/dist/util/sql_utils.js

@@ -1,3 +1,4 @@
+import { DEFAULT_SQL_DATABASE_PROMPT, SQL_MYSQL_PROMPT, SQL_POSTGRES_PROMPT, SQL_SQLITE_PROMPT, } from "../chains/sql_db/sql_db_prompt.js";
 export const verifyListTablesExistInDatabase = (tablesFromDatabase, listTables, errorPrefixMsg) => {
     const onlyTableNames = tablesFromDatabase.map((table) => table.tableName);
     if (listTables.length > 0) {
@@ -151,3 +152,15 @@ export const generateTableInfoFromTables = async (tables, appDataSource, nbSampl
     }
     return globalString;
 };
+export const getPromptTemplateFromDataSource = (appDataSource) => {
+    if (appDataSource.options.type === "postgres") {
+        return SQL_POSTGRES_PROMPT;
+    }
+    if (appDataSource.options.type === "sqlite") {
+        return SQL_SQLITE_PROMPT;
+    }
+    if (appDataSource.options.type === "mysql") {
+        return SQL_MYSQL_PROMPT;
+    }
+    return DEFAULT_SQL_DATABASE_PROMPT;
+};

package/dist/vectorstores/chroma.cjs

@@ -1,7 +1,30 @@
 "use strict";
+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]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Chroma = void 0;
-const uuid_1 = require("uuid");
+const uuid = __importStar(require("uuid"));
 const base_js_1 = require("./base.cjs");
 const document_js_1 = require("../document.cjs");
 class Chroma extends base_js_1.VectorStore {
@@ -139,7 +162,7 @@ class Chroma extends base_js_1.VectorStore {
 exports.Chroma = Chroma;
 function ensureCollectionName(collectionName) {
     if (!collectionName) {
-        return `langchain-${(0, uuid_1.v4)()}`;
+        return `langchain-${uuid.v4()}`;
     }
     return collectionName;
 }