langchain 0.0.129 → 0.0.131

package/dist/document_loaders/web/youtube.cjs

@@ -0,0 +1,77 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.YoutubeLoader = void 0;
+const youtube_transcript_1 = require("youtube-transcript");
+const youtubei_js_1 = require("youtubei.js");
+const document_js_1 = require("../../document.cjs");
+const base_js_1 = require("../base.cjs");
+class YoutubeLoader extends base_js_1.BaseDocumentLoader {
+    constructor(config) {
+        super();
+        Object.defineProperty(this, "videoId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "language", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "addVideoInfo", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.videoId = config.videoId;
+        this.language = config?.language;
+        this.addVideoInfo = config?.addVideoInfo ?? false;
+    }
+    static getVideoID(url) {
+        const match = url.match(/.*(?:youtu.be\/|v\/|u\/\w\/|embed\/|watch\?v=)([^#&?]*).*/);
+        if (match !== null && match[1].length === 11) {
+            return match[1];
+        }
+        else {
+            throw new Error("Failed to get youtube video id from the url");
+        }
+    }
+    static createFromUrl(url, config) {
+        const videoId = YoutubeLoader.getVideoID(url);
+        return new YoutubeLoader({ ...config, videoId });
+    }
+    async load() {
+        let transcript;
+        const metadata = {
+            source: this.videoId,
+        };
+        try {
+            transcript = await youtube_transcript_1.YoutubeTranscript.fetchTranscript(this.videoId, {
+                lang: this.language,
+            });
+            if (transcript === undefined) {
+                throw new Error("Transcription not found");
+            }
+            if (this.addVideoInfo) {
+                const youtube = await youtubei_js_1.Innertube.create();
+                const info = (await youtube.getBasicInfo(this.videoId)).basic_info;
+                metadata.description = info.short_description;
+                metadata.title = info.title;
+                metadata.view_count = info.view_count;
+                metadata.author = info.author;
+            }
+        }
+        catch (e) {
+            throw new Error(`Failed to get YouTube video transcription: ${e.message}`);
+        }
+        const document = new document_js_1.Document({
+            pageContent: transcript.map((item) => item.text).join(" "),
+            metadata,
+        });
+        return [document];
+    }
+}
+exports.YoutubeLoader = YoutubeLoader;

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

@@ -0,0 +1,17 @@
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+interface YoutubeConfig {
+    videoId: string;
+    language?: string;
+    addVideoInfo?: boolean;
+}
+export declare class YoutubeLoader extends BaseDocumentLoader {
+    private videoId;
+    private language?;
+    private addVideoInfo;
+    constructor(config: YoutubeConfig);
+    private static getVideoID;
+    static createFromUrl(url: string, config?: Omit<YoutubeConfig, "videoId">): YoutubeLoader;
+    load(): Promise<Document[]>;
+}
+export {};

package/dist/document_loaders/web/youtube.js

@@ -0,0 +1,73 @@
+import { YoutubeTranscript } from "youtube-transcript";
+import { Innertube } from "youtubei.js";
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+export class YoutubeLoader extends BaseDocumentLoader {
+    constructor(config) {
+        super();
+        Object.defineProperty(this, "videoId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "language", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "addVideoInfo", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.videoId = config.videoId;
+        this.language = config?.language;
+        this.addVideoInfo = config?.addVideoInfo ?? false;
+    }
+    static getVideoID(url) {
+        const match = url.match(/.*(?:youtu.be\/|v\/|u\/\w\/|embed\/|watch\?v=)([^#&?]*).*/);
+        if (match !== null && match[1].length === 11) {
+            return match[1];
+        }
+        else {
+            throw new Error("Failed to get youtube video id from the url");
+        }
+    }
+    static createFromUrl(url, config) {
+        const videoId = YoutubeLoader.getVideoID(url);
+        return new YoutubeLoader({ ...config, videoId });
+    }
+    async load() {
+        let transcript;
+        const metadata = {
+            source: this.videoId,
+        };
+        try {
+            transcript = await YoutubeTranscript.fetchTranscript(this.videoId, {
+                lang: this.language,
+            });
+            if (transcript === undefined) {
+                throw new Error("Transcription not found");
+            }
+            if (this.addVideoInfo) {
+                const youtube = await Innertube.create();
+                const info = (await youtube.getBasicInfo(this.videoId)).basic_info;
+                metadata.description = info.short_description;
+                metadata.title = info.title;
+                metadata.view_count = info.view_count;
+                metadata.author = info.author;
+            }
+        }
+        catch (e) {
+            throw new Error(`Failed to get YouTube video transcription: ${e.message}`);
+        }
+        const document = new Document({
+            pageContent: transcript.map((item) => item.text).join(" "),
+            metadata,
+        });
+        return [document];
+    }
+}

package/dist/document_transformers/html_to_text.cjs

@@ -5,6 +5,9 @@ const html_to_text_1 = require("html-to-text");
 const document_js_1 = require("../document.cjs");
 const document_js_2 = require("../schema/document.cjs");
 class HtmlToTextTransformer extends document_js_2.MappingDocumentTransformer {
+    static lc_name() {
+        return "HtmlToTextTransformer";
+    }
     constructor(options = {}) {
         super(options);
         Object.defineProperty(this, "options", {

package/dist/document_transformers/html_to_text.d.ts

@@ -3,6 +3,7 @@ import { Document } from "../document.js";
 import { MappingDocumentTransformer } from "../schema/document.js";
 export declare class HtmlToTextTransformer extends MappingDocumentTransformer {
     protected options: HtmlToTextOptions;
+    static lc_name(): string;
     constructor(options?: HtmlToTextOptions);
     _transformDocument(document: Document): Promise<Document>;
 }

package/dist/document_transformers/html_to_text.js

@@ -2,6 +2,9 @@ import { htmlToText } from "html-to-text";
 import { Document } from "../document.js";
 import { MappingDocumentTransformer } from "../schema/document.js";
 export class HtmlToTextTransformer extends MappingDocumentTransformer {
+    static lc_name() {
+        return "HtmlToTextTransformer";
+    }
     constructor(options = {}) {
         super(options);
         Object.defineProperty(this, "options", {

package/dist/document_transformers/mozilla_readability.cjs

@@ -6,6 +6,9 @@ const jsdom_1 = require("jsdom");
 const document_js_1 = require("../document.cjs");
 const document_js_2 = require("../schema/document.cjs");
 class MozillaReadabilityTransformer extends document_js_2.MappingDocumentTransformer {
+    static lc_name() {
+        return "MozillaReadabilityTransformer";
+    }
     constructor(options = {}) {
         super(options);
         Object.defineProperty(this, "options", {

package/dist/document_transformers/mozilla_readability.d.ts

@@ -3,6 +3,7 @@ import { Document } from "../document.js";
 import { MappingDocumentTransformer } from "../schema/document.js";
 export declare class MozillaReadabilityTransformer extends MappingDocumentTransformer {
     protected options: Options;
+    static lc_name(): string;
     constructor(options?: Options);
     _transformDocument(document: Document): Promise<Document>;
 }

package/dist/document_transformers/mozilla_readability.js

@@ -3,6 +3,9 @@ import { JSDOM } from "jsdom";
 import { Document } from "../document.js";
 import { MappingDocumentTransformer } from "../schema/document.js";
 export class MozillaReadabilityTransformer extends MappingDocumentTransformer {
+    static lc_name() {
+        return "MozillaReadabilityTransformer";
+    }
     constructor(options = {}) {
         super(options);
         Object.defineProperty(this, "options", {

package/dist/document_transformers/openai_functions.cjs

@@ -7,6 +7,9 @@ const document_js_2 = require("../schema/document.cjs");
 const index_js_1 = require("../chains/openai_functions/index.cjs");
 const openai_js_1 = require("../chat_models/openai.cjs");
 class MetadataTagger extends document_js_2.MappingDocumentTransformer {
+    static lc_name() {
+        return "MetadataTagger";
+    }
     constructor(fields) {
         super();
         Object.defineProperty(this, "taggingChain", {

package/dist/document_transformers/openai_functions.d.ts

@@ -6,6 +6,7 @@ import { MappingDocumentTransformer } from "../schema/document.js";
 import { TaggingChainOptions } from "../chains/openai_functions/index.js";
 import { ChatOpenAI } from "../chat_models/openai.js";
 export declare class MetadataTagger extends MappingDocumentTransformer {
+    static lc_name(): string;
     protected taggingChain: BaseChain;
     constructor(fields: {
         taggingChain: BaseChain;

package/dist/document_transformers/openai_functions.js

@@ -4,6 +4,9 @@ import { MappingDocumentTransformer } from "../schema/document.js";
 import { createTaggingChain, } from "../chains/openai_functions/index.js";
 import { ChatOpenAI } from "../chat_models/openai.js";
 export class MetadataTagger extends MappingDocumentTransformer {
+    static lc_name() {
+        return "MetadataTagger";
+    }
     constructor(fields) {
         super();
         Object.defineProperty(this, "taggingChain", {

package/dist/embeddings/cohere.cjs

@@ -13,7 +13,8 @@ class CohereEmbeddings extends base_js_1.Embeddings {
      * @param fields - An optional object with properties to configure the instance.
      */
     constructor(fields) {
-        super(fields ?? {});
+        const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
+        super(fieldsWithDefaults);
         Object.defineProperty(this, "modelName", {
             enumerable: true,
             configurable: true,
@@ -38,12 +39,12 @@ class CohereEmbeddings extends base_js_1.Embeddings {
             writable: true,
             value: void 0
         });
-        const apiKey = fields?.apiKey || (0, env_js_1.getEnvironmentVariable)("COHERE_API_KEY");
+        const apiKey = fieldsWithDefaults?.apiKey || (0, env_js_1.getEnvironmentVariable)("COHERE_API_KEY");
         if (!apiKey) {
             throw new Error("Cohere API key not found");
         }
-        this.modelName = fields?.modelName ?? this.modelName;
-        this.batchSize = fields?.batchSize ?? this.batchSize;
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
         this.apiKey = apiKey;
     }
     /**
@@ -53,16 +54,18 @@ class CohereEmbeddings extends base_js_1.Embeddings {
      */
     async embedDocuments(texts) {
         await this.maybeInitClient();
-        const subPrompts = (0, chunk_js_1.chunkArray)(texts, this.batchSize);
+        const batches = (0, chunk_js_1.chunkArray)(texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            texts: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
-        for (let i = 0; i < subPrompts.length; i += 1) {
-            const input = subPrompts[i];
-            const { body } = await this.embeddingWithRetry({
-                model: this.modelName,
-                texts: input,
-            });
-            for (let j = 0; j < input.length; j += 1) {
-                embeddings.push(body.embeddings[j]);
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { body: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse.embeddings[j]);
             }
         }
         return embeddings;

package/dist/embeddings/cohere.js

@@ -10,7 +10,8 @@ export class CohereEmbeddings extends Embeddings {
      * @param fields - An optional object with properties to configure the instance.
      */
     constructor(fields) {
-        super(fields ?? {});
+        const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
+        super(fieldsWithDefaults);
         Object.defineProperty(this, "modelName", {
             enumerable: true,
             configurable: true,
@@ -35,12 +36,12 @@ export class CohereEmbeddings extends Embeddings {
             writable: true,
             value: void 0
         });
-        const apiKey = fields?.apiKey || getEnvironmentVariable("COHERE_API_KEY");
+        const apiKey = fieldsWithDefaults?.apiKey || getEnvironmentVariable("COHERE_API_KEY");
         if (!apiKey) {
             throw new Error("Cohere API key not found");
         }
-        this.modelName = fields?.modelName ?? this.modelName;
-        this.batchSize = fields?.batchSize ?? this.batchSize;
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
         this.apiKey = apiKey;
     }
     /**
@@ -50,16 +51,18 @@ export class CohereEmbeddings extends Embeddings {
      */
     async embedDocuments(texts) {
         await this.maybeInitClient();
-        const subPrompts = chunkArray(texts, this.batchSize);
+        const batches = chunkArray(texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            texts: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
-        for (let i = 0; i < subPrompts.length; i += 1) {
-            const input = subPrompts[i];
-            const { body } = await this.embeddingWithRetry({
-                model: this.modelName,
-                texts: input,
-            });
-            for (let j = 0; j < input.length; j += 1) {
-                embeddings.push(body.embeddings[j]);
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { body: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse.embeddings[j]);
             }
         }
         return embeddings;

package/dist/embeddings/googlevertexai.cjs

@@ -39,7 +39,7 @@ class GoogleVertexAIEmbeddings extends base_js_1.Embeddings {
     async embedDocuments(documents) {
         const instanceChunks = (0, chunk_js_1.chunkArray)(documents.map((document) => ({
             content: document,
-        })), 5); // Vertex AI accepts max 5 instnaces per prediction
+        })), 5); // Vertex AI accepts max 5 instances per prediction
         const parameters = {};
         const options = {};
         const responses = await Promise.all(instanceChunks.map((instances) => this.connection.request(instances, parameters, options)));

package/dist/embeddings/googlevertexai.js

@@ -36,7 +36,7 @@ export class GoogleVertexAIEmbeddings extends Embeddings {
     async embedDocuments(documents) {
         const instanceChunks = chunkArray(documents.map((document) => ({
             content: document,
-        })), 5); // Vertex AI accepts max 5 instnaces per prediction
+        })), 5); // Vertex AI accepts max 5 instances per prediction
         const parameters = {};
         const options = {};
         const responses = await Promise.all(instanceChunks.map((instances) => this.connection.request(instances, parameters, options)));

package/dist/embeddings/openai.cjs

@@ -12,7 +12,8 @@ const base_js_1 = require("./base.cjs");
 const azure_js_1 = require("../util/azure.cjs");
 class OpenAIEmbeddings extends base_js_1.Embeddings {
     constructor(fields, configuration) {
-        super(fields ?? {});
+        const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
+        super(fieldsWithDefaults);
         Object.defineProperty(this, "modelName", {
             enumerable: true,
             configurable: true,
@@ -79,27 +80,30 @@ class OpenAIEmbeddings extends base_js_1.Embeddings {
             writable: true,
             value: void 0
         });
-        const apiKey = fields?.openAIApiKey ?? (0, env_js_1.getEnvironmentVariable)("OPENAI_API_KEY");
-        const azureApiKey = fields?.azureOpenAIApiKey ??
+        const apiKey = fieldsWithDefaults?.openAIApiKey ??
+            (0, env_js_1.getEnvironmentVariable)("OPENAI_API_KEY");
+        const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
             (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_KEY");
         if (!azureApiKey && !apiKey) {
             throw new Error("OpenAI or Azure OpenAI API key not found");
         }
-        const azureApiInstanceName = fields?.azureOpenAIApiInstanceName ??
+        const azureApiInstanceName = fieldsWithDefaults?.azureOpenAIApiInstanceName ??
             (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_INSTANCE_NAME");
-        const azureApiDeploymentName = (fields?.azureOpenAIApiEmbeddingsDeploymentName ||
-            fields?.azureOpenAIApiDeploymentName) ??
+        const azureApiDeploymentName = (fieldsWithDefaults?.azureOpenAIApiEmbeddingsDeploymentName ||
+            fieldsWithDefaults?.azureOpenAIApiDeploymentName) ??
             ((0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_EMBEDDINGS_DEPLOYMENT_NAME") ||
                 (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_DEPLOYMENT_NAME"));
-        const azureApiVersion = fields?.azureOpenAIApiVersion ??
+        const azureApiVersion = fieldsWithDefaults?.azureOpenAIApiVersion ??
             (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_VERSION");
         this.azureOpenAIBasePath =
-            fields?.azureOpenAIBasePath ??
+            fieldsWithDefaults?.azureOpenAIBasePath ??
                 (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_BASE_PATH");
-        this.modelName = fields?.modelName ?? this.modelName;
-        this.batchSize = fields?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
-        this.stripNewLines = fields?.stripNewLines ?? this.stripNewLines;
-        this.timeout = fields?.timeout;
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize =
+            fieldsWithDefaults?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
+        this.stripNewLines =
+            fieldsWithDefaults?.stripNewLines ?? this.stripNewLines;
+        this.timeout = fieldsWithDefaults?.timeout;
         this.azureOpenAIApiVersion = azureApiVersion;
         this.azureOpenAIApiKey = azureApiKey;
         this.azureOpenAIApiInstanceName = azureApiInstanceName;
@@ -121,16 +125,18 @@ class OpenAIEmbeddings extends base_js_1.Embeddings {
         };
     }
     async embedDocuments(texts) {
-        const subPrompts = (0, chunk_js_1.chunkArray)(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
+        const batches = (0, chunk_js_1.chunkArray)(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            input: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
-        for (let i = 0; i < subPrompts.length; i += 1) {
-            const input = subPrompts[i];
-            const { data } = await this.embeddingWithRetry({
-                model: this.modelName,
-                input,
-            });
-            for (let j = 0; j < input.length; j += 1) {
-                embeddings.push(data.data[j].embedding);
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { data: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse.data[j].embedding);
             }
         }
         return embeddings;

package/dist/embeddings/openai.js

@@ -6,7 +6,8 @@ import { Embeddings } from "./base.js";
 import { getEndpoint } from "../util/azure.js";
 export class OpenAIEmbeddings extends Embeddings {
     constructor(fields, configuration) {
-        super(fields ?? {});
+        const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
+        super(fieldsWithDefaults);
         Object.defineProperty(this, "modelName", {
             enumerable: true,
             configurable: true,
@@ -73,27 +74,30 @@ export class OpenAIEmbeddings extends Embeddings {
             writable: true,
             value: void 0
         });
-        const apiKey = fields?.openAIApiKey ?? getEnvironmentVariable("OPENAI_API_KEY");
-        const azureApiKey = fields?.azureOpenAIApiKey ??
+        const apiKey = fieldsWithDefaults?.openAIApiKey ??
+            getEnvironmentVariable("OPENAI_API_KEY");
+        const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
             getEnvironmentVariable("AZURE_OPENAI_API_KEY");
         if (!azureApiKey && !apiKey) {
             throw new Error("OpenAI or Azure OpenAI API key not found");
         }
-        const azureApiInstanceName = fields?.azureOpenAIApiInstanceName ??
+        const azureApiInstanceName = fieldsWithDefaults?.azureOpenAIApiInstanceName ??
             getEnvironmentVariable("AZURE_OPENAI_API_INSTANCE_NAME");
-        const azureApiDeploymentName = (fields?.azureOpenAIApiEmbeddingsDeploymentName ||
-            fields?.azureOpenAIApiDeploymentName) ??
+        const azureApiDeploymentName = (fieldsWithDefaults?.azureOpenAIApiEmbeddingsDeploymentName ||
+            fieldsWithDefaults?.azureOpenAIApiDeploymentName) ??
             (getEnvironmentVariable("AZURE_OPENAI_API_EMBEDDINGS_DEPLOYMENT_NAME") ||
                 getEnvironmentVariable("AZURE_OPENAI_API_DEPLOYMENT_NAME"));
-        const azureApiVersion = fields?.azureOpenAIApiVersion ??
+        const azureApiVersion = fieldsWithDefaults?.azureOpenAIApiVersion ??
             getEnvironmentVariable("AZURE_OPENAI_API_VERSION");
         this.azureOpenAIBasePath =
-            fields?.azureOpenAIBasePath ??
+            fieldsWithDefaults?.azureOpenAIBasePath ??
                 getEnvironmentVariable("AZURE_OPENAI_BASE_PATH");
-        this.modelName = fields?.modelName ?? this.modelName;
-        this.batchSize = fields?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
-        this.stripNewLines = fields?.stripNewLines ?? this.stripNewLines;
-        this.timeout = fields?.timeout;
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize =
+            fieldsWithDefaults?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
+        this.stripNewLines =
+            fieldsWithDefaults?.stripNewLines ?? this.stripNewLines;
+        this.timeout = fieldsWithDefaults?.timeout;
         this.azureOpenAIApiVersion = azureApiVersion;
         this.azureOpenAIApiKey = azureApiKey;
         this.azureOpenAIApiInstanceName = azureApiInstanceName;
@@ -115,16 +119,18 @@ export class OpenAIEmbeddings extends Embeddings {
         };
     }
     async embedDocuments(texts) {
-        const subPrompts = chunkArray(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
+        const batches = chunkArray(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            input: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
         const embeddings = [];
-        for (let i = 0; i < subPrompts.length; i += 1) {
-            const input = subPrompts[i];
-            const { data } = await this.embeddingWithRetry({
-                model: this.modelName,
-                input,
-            });
-            for (let j = 0; j < input.length; j += 1) {
-                embeddings.push(data.data[j].embedding);
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { data: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse.data[j].embedding);
             }
         }
         return embeddings;

package/dist/evaluation/qa/eval_chain.cjs

@@ -5,6 +5,9 @@ const prompt_js_1 = require("./prompt.cjs");
 const llm_chain_js_1 = require("../../chains/llm_chain.cjs");
 const eqSet = (xs, ys) => xs.size === ys.size && [...xs].every((x) => ys.has(x));
 class QAEvalChain extends llm_chain_js_1.LLMChain {
+    static lc_name() {
+        return "QAEvalChain";
+    }
     static fromLlm(llm, options = {}) {
         const prompt = options.prompt || prompt_js_1.QA_PROMPT;
         const expectedInputVars = new Set([

package/dist/evaluation/qa/eval_chain.d.ts

@@ -8,6 +8,7 @@ export interface EvaluateArgs {
     predictionKey: string;
 }
 export declare class QAEvalChain extends LLMChain {
+    static lc_name(): string;
     static fromLlm(llm: BaseLanguageModel, options?: {
         prompt?: PromptTemplate;
         chainInput?: Omit<LLMChainInput, "llm">;

package/dist/evaluation/qa/eval_chain.js

@@ -2,6 +2,9 @@ import { QA_PROMPT } from "./prompt.js";
 import { LLMChain } from "../../chains/llm_chain.js";
 const eqSet = (xs, ys) => xs.size === ys.size && [...xs].every((x) => ys.has(x));
 export class QAEvalChain extends LLMChain {
+    static lc_name() {
+        return "QAEvalChain";
+    }
     static fromLlm(llm, options = {}) {
         const prompt = options.prompt || QA_PROMPT;
         const expectedInputVars = new Set([

package/dist/experimental/babyagi/agent.cjs

@@ -7,6 +7,9 @@ const task_creation_js_1 = require("./task_creation.cjs");
 const task_execution_js_1 = require("./task_execution.cjs");
 const task_prioritization_js_1 = require("./task_prioritization.cjs");
 class BabyAGI extends base_js_1.BaseChain {
+    static lc_name() {
+        return "BabyAGI";
+    }
     constructor({ creationChain, prioritizationChain, executionChain, vectorstore, maxIterations = 100, verbose, callbacks, }) {
         super(undefined, verbose, callbacks);
         Object.defineProperty(this, "taskList", {

package/dist/experimental/babyagi/agent.d.ts

@@ -17,6 +17,7 @@ export interface BabyAGIInputs extends Omit<ChainInputs, "memory" | "callbackMan
     maxIterations?: number;
 }
 export declare class BabyAGI extends BaseChain implements BabyAGIInputs {
+    static lc_name(): string;
     taskList: Task[];
     creationChain: BaseChain;
     prioritizationChain: BaseChain;

package/dist/experimental/babyagi/agent.js

@@ -4,6 +4,9 @@ import { TaskCreationChain } from "./task_creation.js";
 import { TaskExecutionChain } from "./task_execution.js";
 import { TaskPrioritizationChain } from "./task_prioritization.js";
 export class BabyAGI extends BaseChain {
+    static lc_name() {
+        return "BabyAGI";
+    }
     constructor({ creationChain, prioritizationChain, executionChain, vectorstore, maxIterations = 100, verbose, callbacks, }) {
         super(undefined, verbose, callbacks);
         Object.defineProperty(this, "taskList", {

package/dist/experimental/babyagi/task_creation.cjs

@@ -5,6 +5,9 @@ const llm_chain_js_1 = require("../../chains/llm_chain.cjs");
 const prompt_js_1 = require("../../prompts/prompt.cjs");
 /** Chain to generate tasks. */
 class TaskCreationChain extends llm_chain_js_1.LLMChain {
+    static lc_name() {
+        return "TaskCreationChain";
+    }
     static fromLLM(fields) {
         const taskCreationTemplate = `You are an task creation AI that uses the result of an execution agent` +
             ` to create new tasks with the following objective: {objective},` +

package/dist/experimental/babyagi/task_creation.d.ts

@@ -1,5 +1,6 @@
 import { LLMChain, LLMChainInput } from "../../chains/llm_chain.js";
 /** Chain to generate tasks. */
 export declare class TaskCreationChain extends LLMChain {
+    static lc_name(): string;
     static fromLLM(fields: Omit<LLMChainInput, "prompt">): LLMChain;
 }

package/dist/experimental/babyagi/task_creation.js

@@ -2,6 +2,9 @@ import { LLMChain } from "../../chains/llm_chain.js";
 import { PromptTemplate } from "../../prompts/prompt.js";
 /** Chain to generate tasks. */
 export class TaskCreationChain extends LLMChain {
+    static lc_name() {
+        return "TaskCreationChain";
+    }
     static fromLLM(fields) {
         const taskCreationTemplate = `You are an task creation AI that uses the result of an execution agent` +
             ` to create new tasks with the following objective: {objective},` +