langchain 0.0.131 → 0.0.132

package/dist/chains/sequential_chain.d.ts

@@ -2,6 +2,9 @@ import { BaseChain, ChainInputs } from "./base.js";
 import { ChainValues } from "../schema/index.js";
 import { SerializedSequentialChain, SerializedSimpleSequentialChain } from "./serde.js";
 import { CallbackManagerForChainRun } from "../callbacks/manager.js";
+/**
+ * Interface for the input parameters of the SequentialChain class.
+ */
 export interface SequentialChainInput extends ChainInputs {
     /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */
     chains: BaseChain[];
@@ -32,6 +35,9 @@ export declare class SequentialChain extends BaseChain implements SequentialChai
     static deserialize(data: SerializedSequentialChain): Promise<SequentialChain>;
     serialize(): SerializedSequentialChain;
 }
+/**
+ * Interface for the input parameters of the SimpleSequentialChain class.
+ */
 export interface SimpleSequentialChainInput extends ChainInputs {
     /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */
     chains: Array<BaseChain>;

package/dist/chains/serde.d.ts

@@ -1,61 +1,115 @@
 import type { SerializedLLM } from "../llms/base.js";
 import type { SerializedBasePromptTemplate } from "../prompts/serde.js";
+/**
+ * Represents the serialized form of an LLMChain. It includes properties
+ * such as `_type`, `llm`, and `prompt`.
+ */
 export type SerializedLLMChain = {
     _type: "llm_chain";
     llm?: SerializedLLM;
     prompt?: SerializedBasePromptTemplate;
 };
+/**
+ * Represents the serialized form of a SequentialChain. It includes
+ * properties such as `_type`, `input_variables`, `output_variables`, and
+ * `chains`.
+ */
 export type SerializedSequentialChain = {
     _type: "sequential_chain";
     input_variables: string[];
     output_variables: string[];
     chains: SerializedBaseChain[];
 };
+/**
+ * Represents the serialized form of a SimpleSequentialChain. It includes
+ * properties such as `_type` and `chains`.
+ */
 export type SerializedSimpleSequentialChain = {
     _type: "simple_sequential_chain";
     chains: Array<SerializedBaseChain>;
 };
+/**
+ * Represents the serialized form of a VectorDBQAChain. It includes
+ * properties such as `_type`, `k`, and `combine_documents_chain`.
+ */
 export type SerializedVectorDBQAChain = {
     _type: "vector_db_qa";
     k: number;
     combine_documents_chain: SerializedBaseChain;
 };
+/**
+ * Represents the serialized form of an APIChain. It includes properties
+ * such as `_type`, `api_request_chain`, `api_answer_chain`, and
+ * `api_docs`.
+ */
 export type SerializedAPIChain = {
     _type: "api_chain";
     api_request_chain: SerializedLLMChain;
     api_answer_chain: SerializedLLMChain;
     api_docs: string;
 };
+/**
+ * Represents the serialized form of a StuffDocumentsChain. It includes
+ * properties such as `_type` and `llm_chain`.
+ */
 export type SerializedStuffDocumentsChain = {
     _type: "stuff_documents_chain";
     llm_chain?: SerializedLLMChain;
 };
+/**
+ * Represents the serialized form of a ChatVectorDBQAChain. It includes
+ * properties such as `_type`, `k`, `combine_documents_chain`, and
+ * `question_generator`.
+ */
 export type SerializedChatVectorDBQAChain = {
     _type: "chat-vector-db";
     k: number;
     combine_documents_chain: SerializedBaseChain;
     question_generator: SerializedLLMChain;
 };
+/**
+ * Represents the serialized form of a MapReduceDocumentsChain. It
+ * includes properties such as `_type`, `llm_chain`, and
+ * `combine_document_chain`.
+ */
 export type SerializedMapReduceDocumentsChain = {
     _type: "map_reduce_documents_chain";
     llm_chain?: SerializedLLMChain;
     combine_document_chain?: SerializedStuffDocumentsChain;
 };
+/**
+ * Represents the serialized form of a RefineDocumentsChain. It includes
+ * properties such as `_type`, `llm_chain`, and `refine_llm_chain`.
+ */
 export type SerializedRefineDocumentsChain = {
     _type: "refine_documents_chain";
     llm_chain?: SerializedLLMChain;
     refine_llm_chain?: SerializedLLMChain;
 };
+/**
+ * Represents the serialized form of an AnalyzeDocumentChain. It includes
+ * properties such as `_type` and `combine_document_chain`.
+ */
 export type SerializedAnalyzeDocumentChain = {
     _type: "analyze_document_chain";
     combine_document_chain?: SerializedBaseChain;
 };
+/**
+ * Represents the serialized form of a ConstitutionalPrinciple. It
+ * includes properties such as `_type`, `critiqueRequest`,
+ * `revisionRequest`, and `name`.
+ */
 export type SerializedConstitutionalPrinciple = {
     _type: "constitutional_principle";
     critiqueRequest: string;
     revisionRequest: string;
     name: string;
 };
+/**
+ * Represents the serialized form of a ConstitutionalChain. It includes
+ * properties such as `_type`, `chain`, `critiqueChain`, `revisionChain`,
+ * and `ConstitutionalPrinciple`.
+ */
 export type SerializedConstitutionalChain = {
     _type: "constitutional_chain";
     chain?: SerializedLLMChain;
@@ -63,4 +117,8 @@ export type SerializedConstitutionalChain = {
     revisionChain?: SerializedBaseChain;
     ConstitutionalPrinciple?: SerializedConstitutionalPrinciple[];
 };
+/**
+ * Represents the serialized form of a BaseChain. It can be one of the
+ * above serialized chain types.
+ */
 export type SerializedBaseChain = SerializedLLMChain | SerializedSequentialChain | SerializedSimpleSequentialChain | SerializedVectorDBQAChain | SerializedAPIChain | SerializedStuffDocumentsChain | SerializedChatVectorDBQAChain | SerializedMapReduceDocumentsChain | SerializedAnalyzeDocumentChain | SerializedRefineDocumentsChain | SerializedConstitutionalChain;

package/dist/chains/sql_db/sql_db_chain.cjs

@@ -6,6 +6,11 @@ const base_js_1 = require("../base.cjs");
 const llm_chain_js_1 = require("../llm_chain.cjs");
 const count_tokens_js_1 = require("../../base_language/count_tokens.cjs");
 const sql_utils_js_1 = require("../../util/sql_utils.cjs");
+/**
+ * Class that represents a SQL database chain in the LangChain framework.
+ * It extends the BaseChain class and implements the functionality
+ * specific to a SQL database chain.
+ */
 class SqlDatabaseChain extends base_js_1.BaseChain {
     static lc_name() {
         return "SqlDatabaseChain";
@@ -134,6 +139,14 @@ class SqlDatabaseChain extends base_js_1.BaseChain {
         }
         return [this.outputKey];
     }
+    /**
+     * Private method that verifies the number of tokens in the input text and
+     * table information. It throws an error if the number of tokens exceeds
+     * the maximum allowed by the language model.
+     * @param inputText The input text.
+     * @param tableinfo The table information.
+     * @returns A promise that resolves when the verification is complete.
+     */
     async verifyNumberOfTokens(inputText, tableinfo) {
         // We verify it only for OpenAI for the moment
         if (this.llm._llmType() !== "openai") {

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

@@ -4,6 +4,11 @@ import { ChainValues } from "../../schema/index.js";
 import { BaseLanguageModel } from "../../base_language/index.js";
 import { CallbackManagerForChainRun } from "../../callbacks/manager.js";
 import { PromptTemplate } from "../../prompts/index.js";
+/**
+ * Interface that extends the ChainInputs interface and defines additional
+ * fields specific to a SQL database chain. It represents the input fields
+ * for a SQL database chain.
+ */
 export interface SqlDatabaseChainInput extends ChainInputs {
     llm: BaseLanguageModel;
     database: SqlDatabase;
@@ -13,6 +18,11 @@ export interface SqlDatabaseChainInput extends ChainInputs {
     sqlOutputKey?: string;
     prompt?: PromptTemplate;
 }
+/**
+ * Class that represents a SQL database chain in the LangChain framework.
+ * It extends the BaseChain class and implements the functionality
+ * specific to a SQL database chain.
+ */
 export declare class SqlDatabaseChain extends BaseChain {
     static lc_name(): string;
     llm: BaseLanguageModel;
@@ -34,5 +44,13 @@ export declare class SqlDatabaseChain extends BaseChain {
     _chainType(): "sql_database_chain";
     get inputKeys(): string[];
     get outputKeys(): string[];
+    /**
+     * Private method that verifies the number of tokens in the input text and
+     * table information. It throws an error if the number of tokens exceeds
+     * the maximum allowed by the language model.
+     * @param inputText The input text.
+     * @param tableinfo The table information.
+     * @returns A promise that resolves when the verification is complete.
+     */
     private verifyNumberOfTokens;
 }

package/dist/chains/sql_db/sql_db_chain.js

@@ -3,6 +3,11 @@ import { BaseChain } from "../base.js";
 import { LLMChain } from "../llm_chain.js";
 import { calculateMaxTokens, getModelContextSize, } from "../../base_language/count_tokens.js";
 import { getPromptTemplateFromDataSource } from "../../util/sql_utils.js";
+/**
+ * Class that represents a SQL database chain in the LangChain framework.
+ * It extends the BaseChain class and implements the functionality
+ * specific to a SQL database chain.
+ */
 export class SqlDatabaseChain extends BaseChain {
     static lc_name() {
         return "SqlDatabaseChain";
@@ -131,6 +136,14 @@ export class SqlDatabaseChain extends BaseChain {
         }
         return [this.outputKey];
     }
+    /**
+     * Private method that verifies the number of tokens in the input text and
+     * table information. It throws an error if the number of tokens exceeds
+     * the maximum allowed by the language model.
+     * @param inputText The input text.
+     * @param tableinfo The table information.
+     * @returns A promise that resolves when the verification is complete.
+     */
     async verifyNumberOfTokens(inputText, tableinfo) {
         // We verify it only for OpenAI for the moment
         if (this.llm._llmType() !== "openai") {

package/dist/chains/summarization/load.d.ts

@@ -1,6 +1,10 @@
 import { BaseLanguageModel } from "../../base_language/index.js";
 import { BasePromptTemplate } from "../../prompts/base.js";
 import { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, MapReduceDocumentsChainInput } from "../combine_docs_chain.js";
+/**
+ * Type for the base parameters that can be used to configure a
+ * summarization chain.
+ */
 type BaseParams = {
     verbose?: boolean;
 };

package/dist/chains/transform.cjs

@@ -2,6 +2,12 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TransformChain = void 0;
 const base_js_1 = require("./base.cjs");
+/**
+ * Class that represents a transform chain. It extends the `BaseChain`
+ * class and implements the `TransformChainFields` interface. It provides
+ * a way to transform input values to output values using a specified
+ * transform function.
+ */
 class TransformChain extends base_js_1.BaseChain {
     static lc_name() {
         return "TransformChain";

package/dist/chains/transform.d.ts

@@ -1,11 +1,22 @@
 import { CallbackManagerForChainRun, Callbacks } from "../callbacks/manager.js";
 import { ChainValues } from "../schema/index.js";
 import { ChainInputs, BaseChain } from "./base.js";
+/**
+ * Interface that extends the `ChainInputs` interface and defines the
+ * fields required for a transform chain. It includes the `transform`
+ * function, `inputVariables`, and `outputVariables` properties.
+ */
 export interface TransformChainFields<I extends ChainValues, O extends ChainValues> extends ChainInputs {
     transform: (values: I, callbacks?: Callbacks) => O | Promise<O>;
     inputVariables: (keyof I extends string ? keyof I : never)[];
     outputVariables: (keyof O extends string ? keyof O : never)[];
 }
+/**
+ * Class that represents a transform chain. It extends the `BaseChain`
+ * class and implements the `TransformChainFields` interface. It provides
+ * a way to transform input values to output values using a specified
+ * transform function.
+ */
 export declare class TransformChain<I extends ChainValues, O extends ChainValues> extends BaseChain {
     static lc_name(): string;
     transformFunc: (values: I, callbacks?: Callbacks) => O | Promise<O>;

package/dist/chains/transform.js

@@ -1,4 +1,10 @@
 import { BaseChain } from "./base.js";
+/**
+ * Class that represents a transform chain. It extends the `BaseChain`
+ * class and implements the `TransformChainFields` interface. It provides
+ * a way to transform input values to output values using a specified
+ * transform function.
+ */
 export class TransformChain extends BaseChain {
     static lc_name() {
         return "TransformChain";

package/dist/chains/vector_db_qa.cjs

@@ -3,6 +3,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.VectorDBQAChain = void 0;
 const base_js_1 = require("./base.cjs");
 const load_js_1 = require("./question_answering/load.cjs");
+/**
+ * Class that represents a VectorDBQAChain. It extends the `BaseChain`
+ * class and implements the `VectorDBQAChainInput` interface. It performs
+ * a similarity search using a vector store and combines the search
+ * results using a specified combine documents chain.
+ */
 class VectorDBQAChain extends base_js_1.BaseChain {
     static lc_name() {
         return "VectorDBQAChain";
@@ -93,6 +99,15 @@ class VectorDBQAChain extends base_js_1.BaseChain {
             k: this.k,
         };
     }
+    /**
+     * Static method that creates a VectorDBQAChain instance from a
+     * BaseLanguageModel and a vector store. It also accepts optional options
+     * to customize the chain.
+     * @param llm The BaseLanguageModel instance.
+     * @param vectorstore The vector store used for similarity search.
+     * @param options Optional options to customize the chain.
+     * @returns A new instance of VectorDBQAChain.
+     */
     static fromLLM(llm, vectorstore, options) {
         const qaChain = (0, load_js_1.loadQAStuffChain)(llm);
         return new this({

package/dist/chains/vector_db_qa.d.ts

@@ -5,6 +5,12 @@ import { BaseLanguageModel } from "../base_language/index.js";
 import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 import { ChainValues } from "../schema/index.js";
 export type LoadValues = Record<string, any>;
+/**
+ * Interface that extends the `ChainInputs` interface and defines the
+ * input fields required for a VectorDBQAChain. It includes properties
+ * such as `vectorstore`, `combineDocumentsChain`,
+ * `returnSourceDocuments`, `k`, and `inputKey`.
+ */
 export interface VectorDBQAChainInput extends Omit<ChainInputs, "memory"> {
     vectorstore: VectorStore;
     combineDocumentsChain: BaseChain;
@@ -12,6 +18,12 @@ export interface VectorDBQAChainInput extends Omit<ChainInputs, "memory"> {
     k?: number;
     inputKey?: string;
 }
+/**
+ * Class that represents a VectorDBQAChain. It extends the `BaseChain`
+ * class and implements the `VectorDBQAChainInput` interface. It performs
+ * a similarity search using a vector store and combines the search
+ * results using a specified combine documents chain.
+ */
 export declare class VectorDBQAChain extends BaseChain implements VectorDBQAChainInput {
     static lc_name(): string;
     k: number;
@@ -27,5 +39,14 @@ export declare class VectorDBQAChain extends BaseChain implements VectorDBQAChai
     _chainType(): "vector_db_qa";
     static deserialize(data: SerializedVectorDBQAChain, values: LoadValues): Promise<VectorDBQAChain>;
     serialize(): SerializedVectorDBQAChain;
+    /**
+     * Static method that creates a VectorDBQAChain instance from a
+     * BaseLanguageModel and a vector store. It also accepts optional options
+     * to customize the chain.
+     * @param llm The BaseLanguageModel instance.
+     * @param vectorstore The vector store used for similarity search.
+     * @param options Optional options to customize the chain.
+     * @returns A new instance of VectorDBQAChain.
+     */
     static fromLLM(llm: BaseLanguageModel, vectorstore: VectorStore, options?: Partial<Omit<VectorDBQAChainInput, "combineDocumentsChain" | "vectorstore">>): VectorDBQAChain;
 }

package/dist/chains/vector_db_qa.js

@@ -1,5 +1,11 @@
 import { BaseChain } from "./base.js";
 import { loadQAStuffChain } from "./question_answering/load.js";
+/**
+ * Class that represents a VectorDBQAChain. It extends the `BaseChain`
+ * class and implements the `VectorDBQAChainInput` interface. It performs
+ * a similarity search using a vector store and combines the search
+ * results using a specified combine documents chain.
+ */
 export class VectorDBQAChain extends BaseChain {
     static lc_name() {
         return "VectorDBQAChain";
@@ -90,6 +96,15 @@ export class VectorDBQAChain extends BaseChain {
             k: this.k,
         };
     }
+    /**
+     * Static method that creates a VectorDBQAChain instance from a
+     * BaseLanguageModel and a vector store. It also accepts optional options
+     * to customize the chain.
+     * @param llm The BaseLanguageModel instance.
+     * @param vectorstore The vector store used for similarity search.
+     * @param options Optional options to customize the chain.
+     * @returns A new instance of VectorDBQAChain.
+     */
     static fromLLM(llm, vectorstore, options) {
         const qaChain = loadQAStuffChain(llm);
         return new this({

package/dist/chat_models/anthropic.cjs

@@ -5,6 +5,11 @@ const sdk_1 = require("@anthropic-ai/sdk");
 const index_js_1 = require("../schema/index.cjs");
 const env_js_1 = require("../util/env.cjs");
 const base_js_1 = require("./base.cjs");
+/**
+ * Extracts the custom role of a generic chat message.
+ * @param message The chat message from which to extract the custom role.
+ * @returns The custom role of the chat message.
+ */
 function extractGenericMessageCustomRole(message) {
     if (message.role !== sdk_1.AI_PROMPT &&
         message.role !== sdk_1.HUMAN_PROMPT &&
@@ -13,6 +18,11 @@ function extractGenericMessageCustomRole(message) {
     }
     return message.role;
 }
+/**
+ * Gets the Anthropic prompt from a base message.
+ * @param message The base message from which to get the Anthropic prompt.
+ * @returns The Anthropic prompt from the base message.
+ */
 function getAnthropicPromptFromMessage(message) {
     const type = message._getType();
     switch (type) {
@@ -234,6 +244,11 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             }
         }
     }
+    /**
+     * Formats messages as a prompt for the model.
+     * @param messages The base messages to format as a prompt.
+     * @returns The formatted prompt.
+     */
     formatMessagesAsPrompt(messages) {
         return (messages
             .map((message) => {
@@ -281,16 +296,21 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             generations,
         };
     }
+    /**
+     * Creates a streaming request with retry.
+     * @param request The parameters for creating a completion.
+     * @returns A streaming request.
+     */
     async createStreamWithRetry(request) {
         if (!this.streamingClient) {
-            const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
+            const options = this.apiUrl ? { baseURL: this.apiUrl } : undefined;
             this.streamingClient = new sdk_1.Anthropic({
                 ...this.clientOptions,
                 ...options,
                 apiKey: this.anthropicApiKey,
             });
         }
-        const makeCompletionRequest = async () => this.streamingClient.completions.create({ ...request, stream: true });
+        const makeCompletionRequest = async () => this.streamingClient.completions.create({ ...request, stream: true }, { headers: request.headers });
         return this.caller.call(makeCompletionRequest);
     }
     /** @ignore */
@@ -299,14 +319,14 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             throw new Error("Missing Anthropic API key.");
         }
         if (!this.batchClient) {
-            const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
+            const options = this.apiUrl ? { baseURL: this.apiUrl } : undefined;
             this.batchClient = new sdk_1.Anthropic({
                 ...this.clientOptions,
                 ...options,
                 apiKey: this.anthropicApiKey,
             });
         }
-        const makeCompletionRequest = async () => this.batchClient.completions.create({ ...request, stream: false });
+        const makeCompletionRequest = async () => this.batchClient.completions.create({ ...request, stream: false }, { headers: request.headers });
         return this.caller.callWithOptions({ signal: options.signal }, makeCompletionRequest);
     }
     _llmType() {

package/dist/chat_models/anthropic.d.ts

@@ -50,6 +50,10 @@ export interface AnthropicInput {
      */
     invocationKwargs?: Kwargs;
 }
+/**
+ * A type representing additional parameters that can be passed to the
+ * Anthropic API.
+ */
 type Kwargs = Record<string, any>;
 /**
  * Wrapper around Anthropic large language models.
@@ -116,9 +120,19 @@ export declare class ChatAnthropic extends BaseChatModel implements AnthropicInp
         model_name: string;
     };
     _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+    /**
+     * Formats messages as a prompt for the model.
+     * @param messages The base messages to format as a prompt.
+     * @returns The formatted prompt.
+     */
     private formatMessagesAsPrompt;
     /** @ignore */
     _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+    /**
+     * Creates a streaming request with retry.
+     * @param request The parameters for creating a completion.
+     * @returns A streaming request.
+     */
     private createStreamWithRetry;
     /** @ignore */
     private completionWithRetry;

package/dist/chat_models/anthropic.js

@@ -2,6 +2,11 @@ import { Anthropic, AI_PROMPT, HUMAN_PROMPT, } from "@anthropic-ai/sdk";
 import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, } from "../schema/index.js";
 import { getEnvironmentVariable } from "../util/env.js";
 import { BaseChatModel } from "./base.js";
+/**
+ * Extracts the custom role of a generic chat message.
+ * @param message The chat message from which to extract the custom role.
+ * @returns The custom role of the chat message.
+ */
 function extractGenericMessageCustomRole(message) {
     if (message.role !== AI_PROMPT &&
         message.role !== HUMAN_PROMPT &&
@@ -10,6 +15,11 @@ function extractGenericMessageCustomRole(message) {
     }
     return message.role;
 }
+/**
+ * Gets the Anthropic prompt from a base message.
+ * @param message The base message from which to get the Anthropic prompt.
+ * @returns The Anthropic prompt from the base message.
+ */
 function getAnthropicPromptFromMessage(message) {
     const type = message._getType();
     switch (type) {
@@ -231,6 +241,11 @@ export class ChatAnthropic extends BaseChatModel {
             }
         }
     }
+    /**
+     * Formats messages as a prompt for the model.
+     * @param messages The base messages to format as a prompt.
+     * @returns The formatted prompt.
+     */
     formatMessagesAsPrompt(messages) {
         return (messages
             .map((message) => {
@@ -278,16 +293,21 @@ export class ChatAnthropic extends BaseChatModel {
             generations,
         };
     }
+    /**
+     * Creates a streaming request with retry.
+     * @param request The parameters for creating a completion.
+     * @returns A streaming request.
+     */
     async createStreamWithRetry(request) {
         if (!this.streamingClient) {
-            const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
+            const options = this.apiUrl ? { baseURL: this.apiUrl } : undefined;
             this.streamingClient = new Anthropic({
                 ...this.clientOptions,
                 ...options,
                 apiKey: this.anthropicApiKey,
             });
         }
-        const makeCompletionRequest = async () => this.streamingClient.completions.create({ ...request, stream: true });
+        const makeCompletionRequest = async () => this.streamingClient.completions.create({ ...request, stream: true }, { headers: request.headers });
         return this.caller.call(makeCompletionRequest);
     }
     /** @ignore */
@@ -296,14 +316,14 @@ export class ChatAnthropic extends BaseChatModel {
             throw new Error("Missing Anthropic API key.");
         }
         if (!this.batchClient) {
-            const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
+            const options = this.apiUrl ? { baseURL: this.apiUrl } : undefined;
             this.batchClient = new Anthropic({
                 ...this.clientOptions,
                 ...options,
                 apiKey: this.anthropicApiKey,
             });
         }
-        const makeCompletionRequest = async () => this.batchClient.completions.create({ ...request, stream: false });
+        const makeCompletionRequest = async () => this.batchClient.completions.create({ ...request, stream: false }, { headers: request.headers });
         return this.caller.callWithOptions({ signal: options.signal }, makeCompletionRequest);
     }
     _llmType() {

package/dist/chat_models/baiduwenxin.cjs

@@ -4,12 +4,22 @@ exports.ChatBaiduWenxin = void 0;
 const base_js_1 = require("./base.cjs");
 const index_js_1 = require("../schema/index.cjs");
 const env_js_1 = require("../util/env.cjs");
+/**
+ * Function that extracts the custom role of a generic chat message.
+ * @param message Chat message from which to extract the custom role.
+ * @returns The custom role of the chat message.
+ */
 function extractGenericMessageCustomRole(message) {
     if (message.role !== "assistant" && message.role !== "user") {
         console.warn(`Unknown message role: ${message.role}`);
     }
     return message.role;
 }
+/**
+ * Function that converts a base message to a Wenxin message role.
+ * @param message Base message to convert.
+ * @returns The Wenxin message role.
+ */
 function messageToWenxinRole(message) {
     const type = message._getType();
     switch (type) {
@@ -158,6 +168,12 @@ class ChatBaiduWenxin extends base_js_1.BaseChatModel {
             throw new Error(`Invalid model name: ${this.modelName}`);
         }
     }
+    /**
+     * Method that retrieves the access token for making requests to the Baidu
+     * API.
+     * @param options Optional parsed call options.
+     * @returns The access token for making requests to the Baidu API.
+     */
     async getAccessToken(options) {
         const url = `https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=${this.baiduApiKey}&client_secret=${this.baiduSecretKey}`;
         const response = await fetch(url, {

package/dist/chat_models/baiduwenxin.d.ts

@@ -1,11 +1,20 @@
 import { BaseChatModel, BaseChatModelParams } from "./base.js";
 import { BaseMessage, ChatResult } from "../schema/index.js";
 import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
+/**
+ * Type representing the role of a message in the Wenxin chat model.
+ */
 export type WenxinMessageRole = "assistant" | "user";
+/**
+ * Interface representing a message in the Wenxin chat model.
+ */
 interface WenxinMessage {
     role: WenxinMessageRole;
     content: string;
 }
+/**
+ * Interface representing a request for a chat completion.
+ */
 interface ChatCompletionRequest {
     messages: WenxinMessage[];
     stream?: boolean;
@@ -14,6 +23,9 @@ interface ChatCompletionRequest {
     top_p?: number;
     penalty_score?: number;
 }
+/**
+ * Interface defining the input to the ChatBaiduWenxin class.
+ */
 declare interface BaiduWenxinChatInput {
     /** Model name to use
      * @default "ERNIE-Bot-turbo"
@@ -83,6 +95,12 @@ export declare class ChatBaiduWenxin extends BaseChatModel implements BaiduWenxi
     topP?: number | undefined;
     penaltyScore?: number | undefined;
     constructor(fields?: Partial<BaiduWenxinChatInput> & BaseChatModelParams);
+    /**
+     * Method that retrieves the access token for making requests to the Baidu
+     * API.
+     * @param options Optional parsed call options.
+     * @returns The access token for making requests to the Baidu API.
+     */
     getAccessToken(options?: this["ParsedCallOptions"]): Promise<any>;
     /**
      * Get the parameters used to invoke the model