langchain 0.0.130 → 0.0.132

package/dist/llms/base.cjs

@@ -34,6 +34,14 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
             this.cache = undefined;
         }
     }
+    /**
+     * This method takes an input and options, and returns a string. It
+     * converts the input to a prompt value and generates a result based on
+     * the prompt.
+     * @param input Input for the LLM.
+     * @param options Options for the LLM call.
+     * @returns A string result based on the prompt.
+     */
     async invoke(input, options) {
         const promptValue = BaseLLM._convertInputToPromptValue(input);
         const result = await this.generatePrompt([promptValue], options, options?.callbacks);
@@ -89,6 +97,14 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
             })));
         }
     }
+    /**
+     * This method takes prompt values, options, and callbacks, and generates
+     * a result based on the prompts.
+     * @param promptValues Prompt values for the LLM.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns An LLMResult based on the prompts.
+     */
     async generatePrompt(promptValues, options, callbacks) {
         const prompts = promptValues.map((promptValue) => promptValue.toString());
         return this.generate(prompts, options, callbacks);
@@ -200,9 +216,25 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
         const { generations } = await this.generate([prompt], options, callbacks);
         return generations[0][0].text;
     }
+    /**
+     * This method is similar to `call`, but it's used for making predictions
+     * based on the input text.
+     * @param text Input text for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A prediction based on the input text.
+     */
     async predict(text, options, callbacks) {
         return this.call(text, options, callbacks);
     }
+    /**
+     * This method takes a list of messages, options, and callbacks, and
+     * returns a predicted message.
+     * @param messages A list of messages for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A predicted message based on the list of messages.
+     */
     async predictMessages(messages, options, callbacks) {
         const text = (0, base_js_1.getBufferString)(messages);
         const prediction = await this.call(text, options, callbacks);

package/dist/llms/base.d.ts

@@ -23,10 +23,26 @@ export declare abstract class BaseLLM<CallOptions extends BaseLLMCallOptions = B
     lc_namespace: string[];
     cache?: BaseCache;
     constructor({ cache, concurrency, ...rest }: BaseLLMParams);
+    /**
+     * This method takes an input and options, and returns a string. It
+     * converts the input to a prompt value and generates a result based on
+     * the prompt.
+     * @param input Input for the LLM.
+     * @param options Options for the LLM call.
+     * @returns A string result based on the prompt.
+     */
     invoke(input: BaseLanguageModelInput, options?: CallOptions): Promise<string>;
     _streamResponseChunks(_input: string, _options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): AsyncGenerator<GenerationChunk>;
     protected _separateRunnableConfigFromCallOptions(options?: Partial<CallOptions>): [RunnableConfig, this["ParsedCallOptions"]];
     _streamIterator(input: BaseLanguageModelInput, options?: CallOptions): AsyncGenerator<string>;
+    /**
+     * This method takes prompt values, options, and callbacks, and generates
+     * a result based on the prompts.
+     * @param promptValues Prompt values for the LLM.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns An LLMResult based on the prompts.
+     */
     generatePrompt(promptValues: BasePromptValue[], options?: string[] | CallOptions, callbacks?: Callbacks): Promise<LLMResult>;
     /**
      * Run the LLM on the given prompts and input.
@@ -47,7 +63,23 @@ export declare abstract class BaseLLM<CallOptions extends BaseLLMCallOptions = B
      * Convenience wrapper for {@link generate} that takes in a single string prompt and returns a single string output.
      */
     call(prompt: string, options?: string[] | CallOptions, callbacks?: Callbacks): Promise<string>;
+    /**
+     * This method is similar to `call`, but it's used for making predictions
+     * based on the input text.
+     * @param text Input text for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A prediction based on the input text.
+     */
     predict(text: string, options?: string[] | CallOptions, callbacks?: Callbacks): Promise<string>;
+    /**
+     * This method takes a list of messages, options, and callbacks, and
+     * returns a predicted message.
+     * @param messages A list of messages for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A predicted message based on the list of messages.
+     */
     predictMessages(messages: BaseMessage[], options?: string[] | CallOptions, callbacks?: Callbacks): Promise<BaseMessage>;
     /**
      * Get the identifying parameters of the LLM.

package/dist/llms/base.js

@@ -31,6 +31,14 @@ export class BaseLLM extends BaseLanguageModel {
             this.cache = undefined;
         }
     }
+    /**
+     * This method takes an input and options, and returns a string. It
+     * converts the input to a prompt value and generates a result based on
+     * the prompt.
+     * @param input Input for the LLM.
+     * @param options Options for the LLM call.
+     * @returns A string result based on the prompt.
+     */
     async invoke(input, options) {
         const promptValue = BaseLLM._convertInputToPromptValue(input);
         const result = await this.generatePrompt([promptValue], options, options?.callbacks);
@@ -86,6 +94,14 @@ export class BaseLLM extends BaseLanguageModel {
             })));
         }
     }
+    /**
+     * This method takes prompt values, options, and callbacks, and generates
+     * a result based on the prompts.
+     * @param promptValues Prompt values for the LLM.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns An LLMResult based on the prompts.
+     */
     async generatePrompt(promptValues, options, callbacks) {
         const prompts = promptValues.map((promptValue) => promptValue.toString());
         return this.generate(prompts, options, callbacks);
@@ -197,9 +213,25 @@ export class BaseLLM extends BaseLanguageModel {
         const { generations } = await this.generate([prompt], options, callbacks);
         return generations[0][0].text;
     }
+    /**
+     * This method is similar to `call`, but it's used for making predictions
+     * based on the input text.
+     * @param text Input text for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A prediction based on the input text.
+     */
     async predict(text, options, callbacks) {
         return this.call(text, options, callbacks);
     }
+    /**
+     * This method takes a list of messages, options, and callbacks, and
+     * returns a predicted message.
+     * @param messages A list of messages for the prediction.
+     * @param options Options for the LLM call.
+     * @param callbacks Callbacks for the LLM call.
+     * @returns A predicted message based on the list of messages.
+     */
     async predictMessages(messages, options, callbacks) {
         const text = getBufferString(messages);
         const prediction = await this.call(text, options, callbacks);

package/dist/llms/bedrock.cjs

@@ -7,6 +7,12 @@ const protocol_http_1 = require("@aws-sdk/protocol-http");
 const sha256_js_1 = require("@aws-crypto/sha256-js");
 const env_js_1 = require("../util/env.cjs");
 const base_js_1 = require("./base.cjs");
+/**
+ * A helper class used within the `Bedrock` class. It is responsible for
+ * preparing the input and output for the Bedrock service. It formats the
+ * input prompt based on the provider (e.g., "anthropic", "ai21",
+ * "amazon") and extracts the generated text from the service response.
+ */
 class BedrockLLMInputOutputAdapter {
     /** Adapter class to prepare the inputs from Langchain to a format
     that LLM model expects. Also, provides a helper function to extract
@@ -28,6 +34,12 @@ class BedrockLLMInputOutputAdapter {
         }
         return inputBody;
     }
+    /**
+     * Extracts the generated text from the service response.
+     * @param provider The provider name.
+     * @param responseBody The response body from the service.
+     * @returns The generated text.
+     */
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static prepareOutput(provider, responseBody) {
         if (provider === "anthropic") {
@@ -39,6 +51,15 @@ class BedrockLLMInputOutputAdapter {
         return responseBody.results[0].outputText;
     }
 }
+/**
+ * A type of Large Language Model (LLM) that interacts with the Bedrock
+ * service. It extends the base `LLM` class and implements the
+ * `BedrockInput` interface. The class is designed to authenticate and
+ * interact with the Bedrock service, which is a part of Amazon Web
+ * Services (AWS). It uses AWS credentials for authentication and can be
+ * configured with various parameters such as the model to use, the AWS
+ * region, and the maximum number of tokens to generate.
+ */
 class Bedrock extends base_js_1.LLM {
     get lc_secrets() {
         return {};

package/dist/llms/bedrock.d.ts

@@ -27,6 +27,15 @@ export interface BedrockInput {
     /** A custom fetch function for low-level access to AWS API. Defaults to fetch() */
     fetchFn?: typeof fetch;
 }
+/**
+ * A type of Large Language Model (LLM) that interacts with the Bedrock
+ * service. It extends the base `LLM` class and implements the
+ * `BedrockInput` interface. The class is designed to authenticate and
+ * interact with the Bedrock service, which is a part of Amazon Web
+ * Services (AWS). It uses AWS credentials for authentication and can be
+ * configured with various parameters such as the model to use, the AWS
+ * region, and the maximum number of tokens to generate.
+ */
 export declare class Bedrock extends LLM implements BedrockInput {
     model: string;
     region: string;

package/dist/llms/bedrock.js

@@ -4,6 +4,12 @@ import { HttpRequest } from "@aws-sdk/protocol-http";
 import { Sha256 } from "@aws-crypto/sha256-js";
 import { getEnvironmentVariable } from "../util/env.js";
 import { LLM } from "./base.js";
+/**
+ * A helper class used within the `Bedrock` class. It is responsible for
+ * preparing the input and output for the Bedrock service. It formats the
+ * input prompt based on the provider (e.g., "anthropic", "ai21",
+ * "amazon") and extracts the generated text from the service response.
+ */
 class BedrockLLMInputOutputAdapter {
     /** Adapter class to prepare the inputs from Langchain to a format
     that LLM model expects. Also, provides a helper function to extract
@@ -25,6 +31,12 @@ class BedrockLLMInputOutputAdapter {
         }
         return inputBody;
     }
+    /**
+     * Extracts the generated text from the service response.
+     * @param provider The provider name.
+     * @param responseBody The response body from the service.
+     * @returns The generated text.
+     */
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static prepareOutput(provider, responseBody) {
         if (provider === "anthropic") {
@@ -36,6 +48,15 @@ class BedrockLLMInputOutputAdapter {
         return responseBody.results[0].outputText;
     }
 }
+/**
+ * A type of Large Language Model (LLM) that interacts with the Bedrock
+ * service. It extends the base `LLM` class and implements the
+ * `BedrockInput` interface. The class is designed to authenticate and
+ * interact with the Bedrock service, which is a part of Amazon Web
+ * Services (AWS). It uses AWS credentials for authentication and can be
+ * configured with various parameters such as the model to use, the AWS
+ * region, and the maximum number of tokens to generate.
+ */
 export class Bedrock extends LLM {
     get lc_secrets() {
         return {};

package/dist/llms/cohere.cjs

@@ -3,7 +3,14 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Cohere = void 0;
 const env_js_1 = require("../util/env.cjs");
 const base_js_1 = require("./base.cjs");
+/**
+ * Class representing a Cohere Large Language Model (LLM). It interacts
+ * with the Cohere API to generate text completions.
+ */
 class Cohere extends base_js_1.LLM {
+    static lc_name() {
+        return "Cohere";
+    }
     get lc_secrets() {
         return {
             apiKey: "COHERE_API_KEY",

package/dist/llms/cohere.d.ts

@@ -1,4 +1,7 @@
 import { LLM, BaseLLMParams } from "./base.js";
+/**
+ * Interface for the input parameters specific to the Cohere model.
+ */
 export interface CohereInput extends BaseLLMParams {
     /** Sampling temperature to use */
     temperature?: number;
@@ -10,7 +13,12 @@ export interface CohereInput extends BaseLLMParams {
     model?: string;
     apiKey?: string;
 }
+/**
+ * Class representing a Cohere Large Language Model (LLM). It interacts
+ * with the Cohere API to generate text completions.
+ */
 export declare class Cohere extends LLM implements CohereInput {
+    static lc_name(): string;
     get lc_secrets(): {
         [key: string]: string;
     } | undefined;

package/dist/llms/cohere.js

@@ -1,6 +1,13 @@
 import { getEnvironmentVariable } from "../util/env.js";
 import { LLM } from "./base.js";
+/**
+ * Class representing a Cohere Large Language Model (LLM). It interacts
+ * with the Cohere API to generate text completions.
+ */
 export class Cohere extends LLM {
+    static lc_name() {
+        return "Cohere";
+    }
     get lc_secrets() {
         return {
             apiKey: "COHERE_API_KEY",

package/dist/llms/googlevertexai.cjs

@@ -94,17 +94,40 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
             },
         ];
     }
+    /**
+     * Formats the input instance as a text instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a text instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted text instance.
+     */
     formatInstanceText(prompt) {
         return { content: prompt };
     }
+    /**
+     * Formats the input instance as a code instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a code instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted code instance.
+     */
     formatInstanceCode(prompt) {
         return { prefix: prompt };
     }
+    /**
+     * Formats the input instance for the Google Vertex AI model based on the
+     * model type (text or code).
+     * @param prompt Prompt to be formatted as an instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted instance.
+     */
     formatInstance(prompt) {
         return this.model.startsWith("code-")
             ? this.formatInstanceCode(prompt)
             : this.formatInstanceText(prompt);
     }
+    /**
+     * Extracts the prediction from the API response.
+     * @param result The API response from which to extract the prediction.
+     * @returns A TextPrediction object representing the extracted prediction.
+     */
     extractPredictionFromResponse(result) {
         return result?.data?.predictions[0];
     }

package/dist/llms/googlevertexai.d.ts

@@ -1,14 +1,29 @@
 import { BaseLLM } from "./base.js";
 import { Generation, LLMResult } from "../schema/index.js";
 import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAILLMResponse } from "../types/googlevertexai-types.js";
+/**
+ * Interface representing the input to the Google Vertex AI model.
+ */
 export interface GoogleVertexAITextInput extends GoogleVertexAIBaseLLMInput {
 }
+/**
+ * Interface representing the instance of text input to the Google Vertex
+ * AI model.
+ */
 interface GoogleVertexAILLMTextInstance {
     content: string;
 }
+/**
+ * Interface representing the instance of code input to the Google Vertex
+ * AI model.
+ */
 interface GoogleVertexAILLMCodeInstance {
     prefix: string;
 }
+/**
+ * Type representing an instance of either text or code input to the
+ * Google Vertex AI model.
+ */
 type GoogleVertexAILLMInstance = GoogleVertexAILLMTextInstance | GoogleVertexAILLMCodeInstance;
 /**
  * Models the data returned from the API call
@@ -41,9 +56,32 @@ export declare class GoogleVertexAI extends BaseLLM implements GoogleVertexAITex
     _llmType(): string;
     _generate(prompts: string[], options: this["ParsedCallOptions"]): Promise<LLMResult>;
     _generatePrompt(prompt: string, options: this["ParsedCallOptions"]): Promise<Generation[]>;
+    /**
+     * Formats the input instance as a text instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a text instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted text instance.
+     */
     formatInstanceText(prompt: string): GoogleVertexAILLMInstance;
+    /**
+     * Formats the input instance as a code instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a code instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted code instance.
+     */
     formatInstanceCode(prompt: string): GoogleVertexAILLMInstance;
+    /**
+     * Formats the input instance for the Google Vertex AI model based on the
+     * model type (text or code).
+     * @param prompt Prompt to be formatted as an instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted instance.
+     */
     formatInstance(prompt: string): GoogleVertexAILLMInstance;
+    /**
+     * Extracts the prediction from the API response.
+     * @param result The API response from which to extract the prediction.
+     * @returns A TextPrediction object representing the extracted prediction.
+     */
     extractPredictionFromResponse(result: GoogleVertexAILLMResponse<TextPrediction>): TextPrediction;
 }
 export {};

package/dist/llms/googlevertexai.js

@@ -91,17 +91,40 @@ export class GoogleVertexAI extends BaseLLM {
             },
         ];
     }
+    /**
+     * Formats the input instance as a text instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a text instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted text instance.
+     */
     formatInstanceText(prompt) {
         return { content: prompt };
     }
+    /**
+     * Formats the input instance as a code instance for the Google Vertex AI
+     * model.
+     * @param prompt Prompt to be formatted as a code instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted code instance.
+     */
     formatInstanceCode(prompt) {
         return { prefix: prompt };
     }
+    /**
+     * Formats the input instance for the Google Vertex AI model based on the
+     * model type (text or code).
+     * @param prompt Prompt to be formatted as an instance.
+     * @returns A GoogleVertexAILLMInstance object representing the formatted instance.
+     */
     formatInstance(prompt) {
         return this.model.startsWith("code-")
             ? this.formatInstanceCode(prompt)
             : this.formatInstanceText(prompt);
     }
+    /**
+     * Extracts the prediction from the API response.
+     * @param result The API response from which to extract the prediction.
+     * @returns A TextPrediction object representing the extracted prediction.
+     */
     extractPredictionFromResponse(result) {
         return result?.data?.predictions[0];
     }

package/dist/llms/hf.cjs

@@ -3,6 +3,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.HuggingFaceInference = void 0;
 const env_js_1 = require("../util/env.cjs");
 const base_js_1 = require("./base.cjs");
+/**
+ * Class implementing the Large Language Model (LLM) interface using the
+ * Hugging Face Inference API for text generation.
+ */
 class HuggingFaceInference extends base_js_1.LLM {
     get lc_secrets() {
         return {

package/dist/llms/hf.d.ts

@@ -1,4 +1,8 @@
 import { LLM, BaseLLMParams } from "./base.js";
+/**
+ * Interface defining the parameters for configuring the Hugging Face
+ * model for text generation.
+ */
 export interface HFInput {
     /** Model to use */
     model: string;
@@ -17,6 +21,10 @@ export interface HFInput {
     /** API key to use. */
     apiKey?: string;
 }
+/**
+ * Class implementing the Large Language Model (LLM) interface using the
+ * Hugging Face Inference API for text generation.
+ */
 export declare class HuggingFaceInference extends LLM implements HFInput {
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/hf.js

@@ -1,5 +1,9 @@
 import { getEnvironmentVariable } from "../util/env.js";
 import { LLM } from "./base.js";
+/**
+ * Class implementing the Large Language Model (LLM) interface using the
+ * Hugging Face Inference API for text generation.
+ */
 export class HuggingFaceInference extends LLM {
     get lc_secrets() {
         return {

package/dist/llms/ollama.cjs

@@ -4,7 +4,14 @@ exports.Ollama = void 0;
 const base_js_1 = require("./base.cjs");
 const ollama_js_1 = require("../util/ollama.cjs");
 const index_js_1 = require("../schema/index.cjs");
+/**
+ * Class that represents the Ollama language model. It extends the base
+ * LLM class and implements the OllamaInput interface.
+ */
 class Ollama extends base_js_1.LLM {
+    static lc_name() {
+        return "Ollama";
+    }
     constructor(fields) {
         super(fields);
         Object.defineProperty(this, "lc_serializable", {

package/dist/llms/ollama.d.ts

@@ -2,8 +2,13 @@ import { LLM, BaseLLMParams } from "./base.js";
 import { OllamaInput, OllamaCallOptions } from "../util/ollama.js";
 import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
 import { GenerationChunk } from "../schema/index.js";
+/**
+ * Class that represents the Ollama language model. It extends the base
+ * LLM class and implements the OllamaInput interface.
+ */
 export declare class Ollama extends LLM implements OllamaInput {
     CallOptions: OllamaCallOptions;
+    static lc_name(): string;
     lc_serializable: boolean;
     model: string;
     baseUrl: string;

package/dist/llms/ollama.js

@@ -1,7 +1,14 @@
 import { LLM } from "./base.js";
 import { createOllamaStream, } from "../util/ollama.js";
 import { GenerationChunk } from "../schema/index.js";
+/**
+ * Class that represents the Ollama language model. It extends the base
+ * LLM class and implements the OllamaInput interface.
+ */
 export class Ollama extends LLM {
+    static lc_name() {
+        return "Ollama";
+    }
     constructor(fields) {
         super(fields);
         Object.defineProperty(this, "lc_serializable", {

package/dist/llms/openai-chat.cjs

@@ -5,13 +5,13 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PromptLayerOpenAIChat = exports.OpenAIChat = void 0;
 const openai_1 = require("openai");
-const env_js_1 = require("../util/env.cjs");
-const axios_fetch_adapter_js_1 = __importDefault(require("../util/axios-fetch-adapter.cjs"));
-const base_js_1 = require("./base.cjs");
 const index_js_1 = require("../schema/index.cjs");
-const prompt_layer_js_1 = require("../util/prompt-layer.cjs");
+const axios_fetch_adapter_js_1 = __importDefault(require("../util/axios-fetch-adapter.cjs"));
 const azure_js_1 = require("../util/azure.cjs");
+const env_js_1 = require("../util/env.cjs");
+const prompt_layer_js_1 = require("../util/prompt-layer.cjs");
 const stream_js_1 = require("../util/stream.cjs");
+const base_js_1 = require("./base.cjs");
 /**
  * Wrapper around OpenAI large language models that use the Chat endpoint.
  *
@@ -35,6 +35,9 @@ const stream_js_1 = require("../util/stream.cjs");
  * @augments AzureOpenAIChatInput
  */
 class OpenAIChat extends base_js_1.LLM {
+    static lc_name() {
+        return "OpenAIChat";
+    }
     get callKeys() {
         return [
             ...super.callKeys,
@@ -46,6 +49,7 @@ class OpenAIChat extends base_js_1.LLM {
         return {
             openAIApiKey: "OPENAI_API_KEY",
             azureOpenAIApiKey: "AZURE_OPENAI_API_KEY",
+            organization: "OPENAI_ORGANIZATION",
         };
     }
     get lc_aliases() {
@@ -188,6 +192,12 @@ class OpenAIChat extends base_js_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "organization", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
@@ -222,6 +232,9 @@ class OpenAIChat extends base_js_1.LLM {
         this.azureOpenAIBasePath =
             fields?.azureOpenAIBasePath ??
                 (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_BASE_PATH");
+        this.organization =
+            fields?.configuration?.organization ??
+                (0, env_js_1.getEnvironmentVariable)("OPENAI_ORGANIZATION");
         this.modelName = fields?.modelName ?? this.modelName;
         this.prefixMessages = fields?.prefixMessages ?? this.prefixMessages;
         this.modelKwargs = fields?.modelKwargs ?? {};
@@ -252,6 +265,7 @@ class OpenAIChat extends base_js_1.LLM {
         }
         this.clientConfig = {
             apiKey: this.openAIApiKey,
+            organization: this.organization,
             ...configuration,
             ...fields?.configuration,
         };
@@ -293,6 +307,11 @@ class OpenAIChat extends base_js_1.LLM {
             ...this.clientConfig,
         };
     }
+    /**
+     * Formats the messages for the OpenAI API.
+     * @param prompt The prompt to be formatted.
+     * @returns Array of formatted messages.
+     */
     formatMessages(prompt) {
         const message = {
             role: "user",
@@ -324,6 +343,12 @@ class OpenAIChat extends base_js_1.LLM {
             void runManager?.handleLLMNewToken(generationChunk.text ?? "");
         }
     }
+    /**
+     * Starts a stream of responses from the OpenAI API.
+     * @param request The request to be sent to the OpenAI API.
+     * @param options Optional configuration for the Axios request.
+     * @returns An iterable object that can be used to iterate over the response chunks.
+     */
     startStream(request, options) {
         let done = false;
         const stream = new TransformStream();
@@ -559,6 +584,12 @@ class PromptLayerOpenAIChat extends OpenAIChat {
             throw new Error("Missing PromptLayer API key");
         }
     }
+    /**
+     * Makes a call to the OpenAI API with retry logic in case of failures.
+     * @param request The request to be sent to the OpenAI API.
+     * @param options Optional configuration for the Axios request.
+     * @returns The response from the OpenAI API.
+     */
     async completionWithRetry(request, options) {
         if (request.stream) {
             return super.completionWithRetry(request, options);