langchain 0.0.57 → 0.0.58

package/dist/agents/chat/index.d.ts

@@ -14,7 +14,7 @@ export type CreatePromptArgs = {
     /** List of input variables the final prompt will expect. */
     inputVariables?: string[];
 };
-type ChatAgentInput = Optional<AgentInput, "outputParser">;
+export type ChatAgentInput = Optional<AgentInput, "outputParser">;
 /**
  * Agent for the MRKL chain.
  * @augments Agent
@@ -39,4 +39,3 @@ export declare class ChatAgent extends Agent {
     static createPrompt(tools: Tool[], args?: CreatePromptArgs): ChatPromptTemplate;
     static fromLLMAndTools(llm: BaseLanguageModel, tools: Tool[], args?: CreatePromptArgs & AgentArgs): ChatAgent;
 }
-export {};

package/dist/agents/executor.d.ts

@@ -4,7 +4,7 @@ import { Tool } from "../tools/base.js";
 import { StoppingMethod } from "./types.js";
 import { SerializedLLMChain } from "../chains/serde.js";
 import { ChainValues } from "../schema/index.js";
-interface AgentExecutorInput extends ChainInputs {
+export interface AgentExecutorInput extends ChainInputs {
     agent: BaseSingleActionAgent | BaseMultiActionAgent;
     tools: Tool[];
     returnIntermediateSteps?: boolean;
@@ -31,4 +31,3 @@ export declare class AgentExecutor extends BaseChain {
     _chainType(): "agent_executor";
     serialize(): SerializedLLMChain;
 }
-export {};

package/dist/agents/index.cjs

@@ -1,8 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.createVectorStoreAgent = exports.createSqlAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.SqlToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = exports.Tool = void 0;
-var base_js_1 = require("../tools/base.cjs");
-Object.defineProperty(exports, "Tool", { enumerable: true, get: function () { return base_js_1.Tool; } });
+exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.createVectorStoreAgent = exports.createSqlAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.SqlToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = void 0;
 var agent_js_1 = require("./agent.cjs");
 Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_js_1.Agent; } });
 Object.defineProperty(exports, "BaseSingleActionAgent", { enumerable: true, get: function () { return agent_js_1.BaseSingleActionAgent; } });

package/dist/agents/index.d.ts

@@ -1,12 +1,11 @@
-export { Tool } from "../tools/base.js";
-export { Agent, BaseSingleActionAgent, LLMSingleActionAgent } from "./agent.js";
+export { Agent, AgentArgs, BaseSingleActionAgent, LLMSingleActionAgent, LLMSingleActionAgentInput, } from "./agent.js";
 export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreInfo, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, createVectorStoreAgent, } from "./agent_toolkits/index.js";
-export { ChatAgent } from "./chat/index.js";
+export { ChatAgent, ChatAgentInput } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";
 export { ChatConversationalAgent, ChatConversationalAgentInput, } from "./chat_convo/index.js";
 export { ChatConversationalAgentOutputParser } from "./chat_convo/outputParser.js";
-export { AgentExecutor } from "./executor.js";
-export { initializeAgentExecutor, initializeAgentExecutorWithOptions, } from "./initialize.js";
-export { ZeroShotAgent } from "./mrkl/index.js";
+export { AgentExecutor, AgentExecutorInput } from "./executor.js";
+export { initializeAgentExecutor, initializeAgentExecutorWithOptions, InitializeAgentExecutorOptions, } from "./initialize.js";
+export { ZeroShotAgent, ZeroShotAgentInput } from "./mrkl/index.js";
 export { ZeroShotAgentOutputParser } from "./mrkl/outputParser.js";
 export { AgentActionOutputParser, AgentInput, SerializedAgent, SerializedAgentT, SerializedZeroShotAgent, StoppingMethod, } from "./types.js";

package/dist/agents/index.js

@@ -1,5 +1,4 @@
-export { Tool } from "../tools/base.js";
-export { Agent, BaseSingleActionAgent, LLMSingleActionAgent } from "./agent.js";
+export { Agent, BaseSingleActionAgent, LLMSingleActionAgent, } from "./agent.js";
 export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, createVectorStoreAgent, } from "./agent_toolkits/index.js";
 export { ChatAgent } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";

package/dist/agents/initialize.cjs

@@ -40,6 +40,13 @@ const initializeAgentExecutor = async (tools, llm, _agentType, _verbose, _callba
     }
 };
 exports.initializeAgentExecutor = initializeAgentExecutor;
+/**
+ * Initialize an agent executor with options
+ * @param tools Array of tools to use in the agent
+ * @param llm LLM or ChatModel to use in the agent
+ * @param options Options for the agent, including agentType, agentArgs, and other options for AgentExecutor.fromAgentAndTools
+ * @returns AgentExecutor
+ */
 const initializeAgentExecutorWithOptions = async (tools, llm, options = {
     agentType: llm._modelType() === "base_chat_model"
         ? "chat-zero-shot-react-description"

package/dist/agents/initialize.d.ts

@@ -7,7 +7,7 @@ import { AgentExecutor } from "./executor.js";
 import { ZeroShotAgent } from "./mrkl/index.js";
 type AgentType = "zero-shot-react-description" | "chat-zero-shot-react-description" | "chat-conversational-react-description";
 export declare const initializeAgentExecutor: (tools: Tool[], llm: BaseLanguageModel, _agentType?: AgentType, _verbose?: boolean, _callbackManager?: CallbackManager) => Promise<AgentExecutor>;
-type AgentExecutorOptions = ({
+export type InitializeAgentExecutorOptions = ({
     agentType: "zero-shot-react-description";
     agentArgs?: Parameters<typeof ZeroShotAgent.fromLLMAndTools>[2];
 } & Omit<Parameters<typeof AgentExecutor.fromAgentAndTools>[0], "agent" | "tools" | "memory">) | ({
@@ -17,5 +17,12 @@ type AgentExecutorOptions = ({
     agentType: "chat-conversational-react-description";
     agentArgs?: Parameters<typeof ChatConversationalAgent.fromLLMAndTools>[2];
 } & Omit<Parameters<typeof AgentExecutor.fromAgentAndTools>[0], "agent" | "tools">);
-export declare const initializeAgentExecutorWithOptions: (tools: Tool[], llm: BaseLanguageModel, options?: AgentExecutorOptions) => Promise<AgentExecutor>;
+/**
+ * Initialize an agent executor with options
+ * @param tools Array of tools to use in the agent
+ * @param llm LLM or ChatModel to use in the agent
+ * @param options Options for the agent, including agentType, agentArgs, and other options for AgentExecutor.fromAgentAndTools
+ * @returns AgentExecutor
+ */
+export declare const initializeAgentExecutorWithOptions: (tools: Tool[], llm: BaseLanguageModel, options?: InitializeAgentExecutorOptions) => Promise<AgentExecutor>;
 export {};

package/dist/agents/initialize.js

@@ -36,6 +36,13 @@ export const initializeAgentExecutor = async (tools, llm, _agentType, _verbose,
             throw new Error("Unknown agent type");
     }
 };
+/**
+ * Initialize an agent executor with options
+ * @param tools Array of tools to use in the agent
+ * @param llm LLM or ChatModel to use in the agent
+ * @param options Options for the agent, including agentType, agentArgs, and other options for AgentExecutor.fromAgentAndTools
+ * @returns AgentExecutor
+ */
 export const initializeAgentExecutorWithOptions = async (tools, llm, options = {
     agentType: llm._modelType() === "base_chat_model"
         ? "chat-zero-shot-react-description"

package/dist/chat_models/anthropic.cjs

@@ -29,8 +29,6 @@ const DEFAULT_STOP_SEQUENCES = [sdk_1.HUMAN_PROMPT];
  * `anthropic.complete`} can be passed through {@link invocationKwargs},
  * even if not explicitly available on this class.
  *
- * @augments BaseLLM
- * @augments AnthropicInput
  */
 class ChatAnthropic extends base_js_1.BaseChatModel {
     constructor(fields) {
@@ -137,6 +135,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             ...this.invocationKwargs,
         };
     }
+    /** @ignore */
     _identifyingParams() {
         return {
             model_name: this.modelName,
@@ -160,21 +159,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
         })
             .join("") + sdk_1.AI_PROMPT);
     }
-    /**
-     * Call out to Anthropic's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stopSequences] - Optional list of stop sequences to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { ChatAnthropic } from "langchain/chat_models/openai";
-     * const anthropic = new ChatAnthropic();
-     * const response = await anthropic.generate(new HumanChatMessage(["Tell me a joke."]));
-     * ```
-     */
+    /** @ignore */
     async _generate(messages, stopSequences) {
         if (this.stopSequences && stopSequences) {
             throw new Error(`"stopSequence" parameter found in input and default params`);
@@ -236,6 +221,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
     _llmType() {
         return "anthropic";
     }
+    /** @ignore */
     _combineLLMOutput() {
         return [];
     }

package/dist/chat_models/anthropic.d.ts

@@ -1,7 +1,10 @@
 import { CompletionResponse, SamplingParameters } from "@anthropic-ai/sdk";
 import { BaseChatModel, BaseChatModelParams } from "./base.js";
 import { BaseChatMessage, ChatResult } from "../schema/index.js";
-interface ModelParams {
+/**
+ * Input to AnthropicChat class.
+ */
+export interface AnthropicInput {
     /** Amount of randomness injected into the response. Ranges
      * from 0 to 1. Use temp closer to 0 for analytical /
      * multiple choice, and temp closer to 1 for creative
@@ -25,18 +28,12 @@ interface ModelParams {
     /** A maximum number of tokens to generate before stopping. */
     maxTokensToSample: number;
     /** A list of strings upon which to stop generating.
-     * You probably want ["\n\nHuman:"], as that's the cue for
+     * You probably want `["\n\nHuman:"]`, as that's the cue for
      * the next turn in the dialog agent.
      */
     stopSequences?: string[];
     /** Whether to stream the results or not */
     streaming?: boolean;
-}
-/**
- * Input to AnthropicChat class.
- * @augments ModelParams
- */
-interface AnthropicInput extends ModelParams {
     /** Anthropic API key */
     apiKey?: string;
     /** Model name to use */
@@ -60,8 +57,6 @@ type Kwargs = Record<string, any>;
  * `anthropic.complete`} can be passed through {@link invocationKwargs},
  * even if not explicitly available on this class.
  *
- * @augments BaseLLM
- * @augments AnthropicInput
  */
 export declare class ChatAnthropic extends BaseChatModel implements AnthropicInput {
     apiKey?: string;
@@ -82,6 +77,7 @@ export declare class ChatAnthropic extends BaseChatModel implements AnthropicInp
      * Get the parameters used to invoke the model
      */
     invocationParams(): Omit<SamplingParameters, "prompt"> & Kwargs;
+    /** @ignore */
     _identifyingParams(): {
         model: string;
         temperature?: number | undefined;
@@ -110,25 +106,12 @@ export declare class ChatAnthropic extends BaseChatModel implements AnthropicInp
         model_name: string;
     };
     private formatMessagesAsPrompt;
-    /**
-     * Call out to Anthropic's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stopSequences] - Optional list of stop sequences to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { ChatAnthropic } from "langchain/chat_models/openai";
-     * const anthropic = new ChatAnthropic();
-     * const response = await anthropic.generate(new HumanChatMessage(["Tell me a joke."]));
-     * ```
-     */
+    /** @ignore */
     _generate(messages: BaseChatMessage[], stopSequences?: string[]): Promise<ChatResult>;
     /** @ignore */
     completionWithRetry(request: SamplingParameters & Kwargs): Promise<CompletionResponse>;
     _llmType(): string;
+    /** @ignore */
     _combineLLMOutput(): never[];
 }
 export {};

package/dist/chat_models/anthropic.js

@@ -26,8 +26,6 @@ const DEFAULT_STOP_SEQUENCES = [HUMAN_PROMPT];
  * `anthropic.complete`} can be passed through {@link invocationKwargs},
  * even if not explicitly available on this class.
  *
- * @augments BaseLLM
- * @augments AnthropicInput
  */
 export class ChatAnthropic extends BaseChatModel {
     constructor(fields) {
@@ -134,6 +132,7 @@ export class ChatAnthropic extends BaseChatModel {
             ...this.invocationKwargs,
         };
     }
+    /** @ignore */
     _identifyingParams() {
         return {
             model_name: this.modelName,
@@ -157,21 +156,7 @@ export class ChatAnthropic extends BaseChatModel {
         })
             .join("") + AI_PROMPT);
     }
-    /**
-     * Call out to Anthropic's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stopSequences] - Optional list of stop sequences to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { ChatAnthropic } from "langchain/chat_models/openai";
-     * const anthropic = new ChatAnthropic();
-     * const response = await anthropic.generate(new HumanChatMessage(["Tell me a joke."]));
-     * ```
-     */
+    /** @ignore */
     async _generate(messages, stopSequences) {
         if (this.stopSequences && stopSequences) {
             throw new Error(`"stopSequence" parameter found in input and default params`);
@@ -233,6 +218,7 @@ export class ChatAnthropic extends BaseChatModel {
     _llmType() {
         return "anthropic";
     }
+    /** @ignore */
     _combineLLMOutput() {
         return [];
     }

package/dist/chat_models/openai.cjs

@@ -179,6 +179,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
             ...this.modelKwargs,
         };
     }
+    /** @ignore */
     _identifyingParams() {
         return {
             model_name: this.modelName,
@@ -192,21 +193,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
     identifyingParams() {
         return this._identifyingParams();
     }
-    /**
-     * Call out to OpenAI's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stop] - Optional list of stop words to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { OpenAI } from "langchain/llms/openai";
-     * const openai = new OpenAI();
-     * const response = await openai.generate(["Tell me a joke."]);
-     * ```
-     */
+    /** @ignore */
     async _generate(messages, stop) {
         const tokenUsage = {};
         if (this.stop && stop) {
@@ -345,6 +332,7 @@ class ChatOpenAI extends base_js_1.BaseChatModel {
     _llmType() {
         return "openai";
     }
+    /** @ignore */
     _combineLLMOutput(...llmOutputs) {
         return llmOutputs.reduce((acc, llmOutput) => {
             if (llmOutput && llmOutput.tokenUsage) {

package/dist/chat_models/openai.d.ts

@@ -10,7 +10,7 @@ interface TokenUsage {
 interface OpenAILLMOutput {
     tokenUsage: TokenUsage;
 }
-interface ModelParams {
+export interface OpenAIInput {
     /** Sampling temperature to use, between 0 and 2, defaults to 1 */
     temperature: number;
     /** Total probability mass of tokens to consider at each step, between 0 and 1, defaults to 1 */
@@ -30,12 +30,6 @@ interface ModelParams {
      * defaults to the maximum number of tokens allowed by the model.
      */
     maxTokens?: number;
-}
-/**
- * Input to OpenAI class.
- * @augments ModelParams
- */
-interface OpenAIInput extends ModelParams {
     /** Model name to use */
     modelName: string;
     /** Holds any additional parameters that are valid to pass to {@link
@@ -90,6 +84,7 @@ export declare class ChatOpenAI extends BaseChatModel implements OpenAIInput {
      * Get the parameters used to invoke the model
      */
     invocationParams(): Omit<CreateChatCompletionRequest, "messages"> & Kwargs;
+    /** @ignore */
     _identifyingParams(): {
         apiKey?: string | Promise<string> | ((name: string) => string) | ((name: string) => Promise<string>) | undefined;
         organization?: string | undefined;
@@ -137,21 +132,7 @@ export declare class ChatOpenAI extends BaseChatModel implements OpenAIInput {
         logit_bias?: object | null | undefined;
         model_name: string;
     };
-    /**
-     * Call out to OpenAI's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stop] - Optional list of stop words to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { OpenAI } from "langchain/llms/openai";
-     * const openai = new OpenAI();
-     * const response = await openai.generate(["Tell me a joke."]);
-     * ```
-     */
+    /** @ignore */
     _generate(messages: BaseChatMessage[], stop?: string[]): Promise<ChatResult>;
     getNumTokensFromMessages(messages: BaseChatMessage[]): Promise<{
         totalCount: number;
@@ -160,6 +141,7 @@ export declare class ChatOpenAI extends BaseChatModel implements OpenAIInput {
     /** @ignore */
     completionWithRetry(request: CreateChatCompletionRequest, options?: StreamingAxiosConfiguration): Promise<CreateChatCompletionResponse>;
     _llmType(): string;
+    /** @ignore */
     _combineLLMOutput(...llmOutputs: OpenAILLMOutput[]): OpenAILLMOutput;
 }
 export {};

package/dist/chat_models/openai.js

@@ -173,6 +173,7 @@ export class ChatOpenAI extends BaseChatModel {
             ...this.modelKwargs,
         };
     }
+    /** @ignore */
     _identifyingParams() {
         return {
             model_name: this.modelName,
@@ -186,21 +187,7 @@ export class ChatOpenAI extends BaseChatModel {
     identifyingParams() {
         return this._identifyingParams();
     }
-    /**
-     * Call out to OpenAI's endpoint with k unique prompts
-     *
-     * @param messages - The messages to pass into the model.
-     * @param [stop] - Optional list of stop words to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { OpenAI } from "langchain/llms/openai";
-     * const openai = new OpenAI();
-     * const response = await openai.generate(["Tell me a joke."]);
-     * ```
-     */
+    /** @ignore */
     async _generate(messages, stop) {
         const tokenUsage = {};
         if (this.stop && stop) {
@@ -339,6 +326,7 @@ export class ChatOpenAI extends BaseChatModel {
     _llmType() {
         return "openai";
     }
+    /** @ignore */
     _combineLLMOutput(...llmOutputs) {
         return llmOutputs.reduce((acc, llmOutput) => {
             if (llmOutput && llmOutput.tokenUsage) {

package/dist/document_loaders/fs/pdf.cjs

@@ -81,12 +81,12 @@ class PDFLoader extends buffer_js_1.BufferLoader {
 exports.PDFLoader = PDFLoader;
 async function PDFLoaderImports() {
     try {
-        const { default: mod } = await import("pdfjs-dist");
+        const { default: mod } = await import("pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js");
         const { getDocument, version } = mod;
         return { getDocument, version };
     }
     catch (e) {
         console.error(e);
-        throw new Error("Failed to load pdfjs-dist. Please install it with eg. `npm install pdfjs-dist`.");
+        throw new Error("Failed to load pdf-parse. Please install it with eg. `npm install pdf-parse`.");
     }
 }

package/dist/document_loaders/fs/pdf.d.ts

@@ -1,4 +1,5 @@
 /// <reference types="node" resolution-mode="require"/>
+/// <reference types="types/pdf-parse.js" />
 import { Document } from "../../document.js";
 import { BufferLoader } from "./buffer.js";
 export declare class PDFLoader extends BufferLoader {
@@ -11,7 +12,7 @@ export declare class PDFLoader extends BufferLoader {
     parse(raw: Buffer, metadata: Document["metadata"]): Promise<Document[]>;
 }
 declare function PDFLoaderImports(): Promise<{
-    getDocument: typeof import("pdfjs-dist").getDocument;
+    getDocument: typeof import("pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js").getDocument;
     version: string;
 }>;
 export {};

package/dist/document_loaders/fs/pdf.js

@@ -77,12 +77,12 @@ export class PDFLoader extends BufferLoader {
 }
 async function PDFLoaderImports() {
     try {
-        const { default: mod } = await import("pdfjs-dist");
+        const { default: mod } = await import("pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js");
         const { getDocument, version } = mod;
         return { getDocument, version };
     }
     catch (e) {
         console.error(e);
-        throw new Error("Failed to load pdfjs-dist. Please install it with eg. `npm install pdfjs-dist`.");
+        throw new Error("Failed to load pdf-parse. Please install it with eg. `npm install pdf-parse`.");
     }
 }

package/dist/embeddings/cohere.cjs

@@ -19,10 +19,6 @@ class CohereEmbeddings extends base_js_1.Embeddings {
             writable: true,
             value: "small"
         });
-        /**
-         * The maximum number of documents to embed in a single request. This is
-         * limited by the Cohere API to a maximum of 96.
-         */
         Object.defineProperty(this, "batchSize", {
             enumerable: true,
             configurable: true,
@@ -104,11 +100,7 @@ class CohereEmbeddings extends base_js_1.Embeddings {
             this.client.init(this.apiKey);
         }
     }
-    /**
-     * Dynamically imports the required dependencies for the CohereEmbeddings class.
-     * @returns An object containing the imported cohere-ai module.
-     * @throws An error if the cohere-ai dependency is not installed.
-     */
+    /** @ignore */
     static async imports() {
         try {
             const { default: cohere } = await import("cohere-ai");

package/dist/embeddings/cohere.d.ts

@@ -1,16 +1,17 @@
 import { Embeddings, EmbeddingsParams } from "./base.js";
-interface ModelParams {
+export interface CohereEmbeddingsParams extends EmbeddingsParams {
     modelName: string;
+    /**
+     * The maximum number of documents to embed in a single request. This is
+     * limited by the Cohere API to a maximum of 96.
+     */
+    batchSize?: number;
 }
 /**
  * A class for generating embeddings using the Cohere API.
  */
-export declare class CohereEmbeddings extends Embeddings implements ModelParams {
+export declare class CohereEmbeddings extends Embeddings implements CohereEmbeddingsParams {
     modelName: string;
-    /**
-     * The maximum number of documents to embed in a single request. This is
-     * limited by the Cohere API to a maximum of 96.
-     */
     batchSize: number;
     private apiKey;
     private client;
@@ -18,9 +19,8 @@ export declare class CohereEmbeddings extends Embeddings implements ModelParams
      * Constructor for the CohereEmbeddings class.
      * @param fields - An optional object with properties to configure the instance.
      */
-    constructor(fields?: EmbeddingsParams & Partial<ModelParams> & {
+    constructor(fields?: Partial<CohereEmbeddingsParams> & {
         verbose?: boolean;
-        batchSize?: number;
         apiKey?: string;
     });
     /**
@@ -45,13 +45,8 @@ export declare class CohereEmbeddings extends Embeddings implements ModelParams
      * Initializes the Cohere client if it hasn't been initialized already.
      */
     private maybeInitClient;
-    /**
-     * Dynamically imports the required dependencies for the CohereEmbeddings class.
-     * @returns An object containing the imported cohere-ai module.
-     * @throws An error if the cohere-ai dependency is not installed.
-     */
+    /** @ignore */
     static imports(): Promise<{
         cohere: typeof import("cohere-ai");
     }>;
 }
-export {};

package/dist/embeddings/cohere.js

@@ -16,10 +16,6 @@ export class CohereEmbeddings extends Embeddings {
             writable: true,
             value: "small"
         });
-        /**
-         * The maximum number of documents to embed in a single request. This is
-         * limited by the Cohere API to a maximum of 96.
-         */
         Object.defineProperty(this, "batchSize", {
             enumerable: true,
             configurable: true,
@@ -101,11 +97,7 @@ export class CohereEmbeddings extends Embeddings {
             this.client.init(this.apiKey);
         }
     }
-    /**
-     * Dynamically imports the required dependencies for the CohereEmbeddings class.
-     * @returns An object containing the imported cohere-ai module.
-     * @throws An error if the cohere-ai dependency is not installed.
-     */
+    /** @ignore */
     static async imports() {
         try {
             const { default: cohere } = await import("cohere-ai");

package/dist/embeddings/openai.cjs

@@ -17,20 +17,12 @@ class OpenAIEmbeddings extends base_js_1.Embeddings {
             writable: true,
             value: "text-embedding-ada-002"
         });
-        /**
-         * The maximum number of documents to embed in a single request. This is
-         * limited by the OpenAI API to a maximum of 2048.
-         */
         Object.defineProperty(this, "batchSize", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: 512
         });
-        /**
-         * Whether to strip new lines from the input text. This is recommended by
-         * OpenAI, but may not be suitable for all use cases.
-         */
         Object.defineProperty(this, "stripNewLines", {
             enumerable: true,
             configurable: true,

package/dist/embeddings/openai.d.ts

@@ -1,36 +1,35 @@
 import { ConfigurationParameters } from "openai";
 import { Embeddings, EmbeddingsParams } from "./base.js";
-interface ModelParams {
+export interface OpenAIEmbeddingsParams extends EmbeddingsParams {
     /** Model name to use */
     modelName: string;
     /**
      * Timeout to use when making requests to OpenAI.
      */
     timeout?: number;
-}
-export declare class OpenAIEmbeddings extends Embeddings implements ModelParams {
-    modelName: string;
     /**
      * The maximum number of documents to embed in a single request. This is
      * limited by the OpenAI API to a maximum of 2048.
      */
-    batchSize: number;
+    batchSize?: number;
     /**
      * Whether to strip new lines from the input text. This is recommended by
      * OpenAI, but may not be suitable for all use cases.
      */
+    stripNewLines?: boolean;
+}
+export declare class OpenAIEmbeddings extends Embeddings implements OpenAIEmbeddingsParams {
+    modelName: string;
+    batchSize: number;
     stripNewLines: boolean;
     timeout?: number;
     private client;
     private clientConfig;
-    constructor(fields?: Partial<ModelParams> & EmbeddingsParams & {
+    constructor(fields?: Partial<OpenAIEmbeddingsParams> & {
         verbose?: boolean;
-        batchSize?: number;
         openAIApiKey?: string;
-        stripNewLines?: boolean;
     }, configuration?: ConfigurationParameters);
     embedDocuments(texts: string[]): Promise<number[][]>;
     embedQuery(text: string): Promise<number[]>;
     private embeddingWithRetry;
 }
-export {};