Class OpenAI<CallOptions>

Wrapper around OpenAI large language models.

To use you should have the openai package installed, with the OPENAI_API_KEY environment variable set.

To use with Azure you should have the openai package installed, with the AZURE_OPENAI_API_KEY, AZURE_OPENAI_API_INSTANCE_NAME, AZURE_OPENAI_API_DEPLOYMENT_NAME and AZURE_OPENAI_API_VERSION environment variable set.

Any parameters that are valid to be passed to openai.createCompletion can be passed through modelKwargs, even if not explicitly available on this class.

const model = new OpenAI({
modelName: "gpt-4",
temperature: 0.7,
maxTokens: 1000,
maxRetries: 5,
});

const res = await model.invoke(
"Question: What would be a good company name for a company that makes colorful socks?\nAnswer:"
);
console.log({ res });

Type Parameters

Hierarchy (view full)

Implements

Constructors

Properties

ParsedCallOptions: Omit<CallOptions,
    | "configurable"
    | "recursionLimit"
    | "runName"
    | "tags"
    | "metadata"
    | "callbacks"
    | "runId">
apiKey?: string

API key to use when making requests to OpenAI. Defaults to the value of OPENAI_API_KEY environment variable.

azureADTokenProvider?: (() => Promise<string>)

A function that returns an access token for Microsoft Entra (formerly known as Azure Active Directory), which will be invoked on every request.

azureOpenAIApiDeploymentName?: string

Azure OpenAI API deployment name to use for completions when making requests to Azure OpenAI. This is the name of the deployment you created in the Azure portal. e.g. "my-openai-deployment" this will be used in the endpoint URL: https://{InstanceName}.openai.azure.com/openai/deployments/my-openai-deployment/

azureOpenAIApiInstanceName?: string

Azure OpenAI API instance name to use when making requests to Azure OpenAI. this is the name of the instance you created in the Azure portal. e.g. "my-openai-instance" this will be used in the endpoint URL: https://my-openai-instance.openai.azure.com/openai/deployments/{DeploymentName}/

azureOpenAIApiKey?: string

API key to use when making requests to Azure OpenAI.

azureOpenAIApiVersion?: string

API version to use when making requests to Azure OpenAI.

azureOpenAIBasePath?: string

Custom base url for Azure OpenAI API. This is useful in case you have a deployment in another region. e.g. setting this value to "https://westeurope.api.cognitive.microsoft.com/openai/deployments" will be result in the endpoint URL: https://westeurope.api.cognitive.microsoft.com/openai/deployments/{DeploymentName}/

batchSize: number = 20

Batch size to use when passing multiple documents to generate

bestOf?: number

Generates bestOf completions server side and returns the "best"

cache?: BaseCache<Generation[]>
callbacks?: Callbacks
caller: AsyncCaller

The async caller should be used by subclasses to make any async calls, which will thus benefit from the concurrency and retry logic.

client: OpenAIClient
clientConfig: ClientOptions
frequencyPenalty: number = 0

Penalizes repeated tokens according to frequency

logitBias?: Record<string, number>

Dictionary used to adjust the probability of specific tokens being generated

maxTokens: number = 256

Maximum number of tokens to generate in the completion. -1 returns as many tokens as possible given the prompt and the model's maximum context size.

metadata?: Record<string, unknown>
model: string = "gpt-3.5-turbo-instruct"

Model name to use

modelKwargs?: Record<string, any>

Holds any additional parameters that are valid to pass to openai.createCompletion that are not explicitly specified on this class.

modelName: string = "gpt-3.5-turbo-instruct"

Model name to use Alias for model

n: number = 1

Number of completions to generate for each prompt

name?: string
openAIApiKey?: string

API key to use when making requests to OpenAI. Defaults to the value of OPENAI_API_KEY environment variable. Alias for apiKey

organization?: string
presencePenalty: number = 0

Penalizes repeated tokens

stop?: string[]

List of stop words to use when generating Alias for stopSequences

stopSequences?: string[]

List of stop words to use when generating

streaming: boolean = false

Whether to stream the results or not. Enabling disables tokenUsage reporting

tags?: string[]
temperature: number = 0.7

Sampling temperature to use

timeout?: number

Timeout to use when making requests to OpenAI.

topP: number = 1

Total probability mass of tokens to consider at each step

user?: string

Unique string identifier representing your end-user, which can help OpenAI to monitor and detect abuse.

verbose: boolean

Whether to print out response text.

Accessors

Methods

  • Convert a runnable to a tool. Return a new instance of RunnableToolLike which contains the runnable, name, description and schema.

    Type Parameters

    • T extends BaseLanguageModelInput = BaseLanguageModelInput

    Parameters

    • fields: {
          description?: string;
          name?: string;
          schema: ZodType<T, ZodTypeDef, T>;
      }
      • Optionaldescription?: string

        The description of the tool. Falls back to the description on the Zod schema if not provided, or undefined if neither are provided.

      • Optionalname?: string

        The name of the tool. If not provided, it will default to the name of the runnable.

      • schema: ZodType<T, ZodTypeDef, T>

        The Zod schema for the input of the tool. Infers the Zod type from the input type of the runnable.

    Returns RunnableToolLike<ZodType<ToolCall | T, ZodTypeDef, ToolCall | T>, string>

    An instance of RunnableToolLike which is a runnable that can be used as a tool.

  • Assigns new fields to the dict output of this runnable. Returns a new runnable.

    Parameters

    • mapping: RunnableMapLike<Record<string, unknown>, Record<string, unknown>>

    Returns Runnable<any, any, RunnableConfig<Record<string, any>>>

  • Default implementation of batch, which calls invoke N times. Subclasses should override this method if they can batch more efficiently.

    Parameters

    • inputs: BaseLanguageModelInput[]

      Array of inputs to each batch call.

    • Optionaloptions: Partial<CallOptions> | Partial<CallOptions>[]

      Either a single call options object to apply to each batch call or an array for each call.

    • OptionalbatchOptions: RunnableBatchOptions & {
          returnExceptions?: false;
      }

    Returns Promise<string[]>

    An array of RunOutputs, or mixed RunOutputs and errors if batchOptions.returnExceptions is set

  • Parameters

    • inputs: BaseLanguageModelInput[]
    • Optionaloptions: Partial<CallOptions> | Partial<CallOptions>[]
    • OptionalbatchOptions: RunnableBatchOptions & {
          returnExceptions: true;
      }

    Returns Promise<(string | Error)[]>

  • Parameters

    • inputs: BaseLanguageModelInput[]
    • Optionaloptions: Partial<CallOptions> | Partial<CallOptions>[]
    • OptionalbatchOptions: RunnableBatchOptions

    Returns Promise<(string | Error)[]>

  • Bind arguments to a Runnable, returning a new Runnable.

    Parameters

    Returns Runnable<BaseLanguageModelInput, string, CallOptions>

    A new RunnableBinding that, when invoked, will apply the bound args.

  • Parameters

    • prompt: string
    • Optionaloptions: string[] | CallOptions
    • Optionalcallbacks: Callbacks

    Returns Promise<string>

    Use .invoke() instead. Will be removed in 0.2.0. Convenience wrapper for generate that takes in a single string prompt and returns a single string output.

  • Run the LLM on the given prompts and input, handling caching.

    Parameters

    • prompts: string[]
    • Optionaloptions: string[] | CallOptions
    • Optionalcallbacks: Callbacks

    Returns Promise<LLMResult>

  • This method takes prompt values, options, and callbacks, and generates a result based on the prompts.

    Parameters

    • promptValues: BasePromptValueInterface[]

      Prompt values for the LLM.

    • Optionaloptions: string[] | CallOptions

      Options for the LLM call.

    • Optionalcallbacks: Callbacks

      Callbacks for the LLM call.

    Returns Promise<LLMResult>

    An LLMResult based on the prompts.

  • Parameters

    • Optional_: RunnableConfig<Record<string, any>>

    Returns Graph

  • Parameters

    • Optionalsuffix: string

    Returns string

  • Parameters

    • content: MessageContent

    Returns Promise<number>

  • 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.

    Parameters

    • input: BaseLanguageModelInput

      Input for the LLM.

    • Optionaloptions: CallOptions

      Options for the LLM call.

    Returns Promise<string>

    A string result based on the prompt.

  • Return a new Runnable that maps a list of inputs to a list of outputs, by calling invoke() with each input.

    Returns Runnable<BaseLanguageModelInput[], string[], CallOptions>

  • Pick keys from the dict output of this runnable. Returns a new runnable.

    Parameters

    • keys: string | string[]

    Returns Runnable<any, any, RunnableConfig<Record<string, any>>>

  • Create a new runnable sequence that runs each individual runnable in series, piping the output of one runnable into another runnable or runnable-like.

    Type Parameters

    • NewRunOutput

    Parameters

    • coerceable: RunnableLike<string, NewRunOutput, RunnableConfig<Record<string, any>>>

      A runnable, function, or object whose values are functions or runnables.

    Returns Runnable<BaseLanguageModelInput, Exclude<NewRunOutput, Error>, RunnableConfig<Record<string, any>>>

    A new runnable sequence.

  • Parameters

    • text: string

      Input text for the prediction.

    • Optionaloptions: string[] | CallOptions

      Options for the LLM call.

    • Optionalcallbacks: Callbacks

      Callbacks for the LLM call.

    Returns Promise<string>

    A prediction based on the input text.

    Use .invoke() instead. Will be removed in 0.2.0.

    This method is similar to call, but it's used for making predictions based on the input text.

  • Parameters

    • messages: BaseMessage[]

      A list of messages for the prediction.

    • Optionaloptions: string[] | CallOptions

      Options for the LLM call.

    • Optionalcallbacks: Callbacks

      Callbacks for the LLM call.

    Returns Promise<BaseMessage>

    A predicted message based on the list of messages.

    Use .invoke() instead. Will be removed in 0.2.0.

    This method takes a list of messages, options, and callbacks, and returns a predicted message.

  • Returns SerializedLLM

    Return a json-like object representing this LLM.

  • Stream output in chunks.

    Parameters

    • input: BaseLanguageModelInput
    • Optionaloptions: Partial<CallOptions>

    Returns Promise<IterableReadableStream<string>>

    A readable stream that is also an iterable.

  • Generate a stream of events emitted by the internal steps of the runnable.

    Use to create an iterator over StreamEvents that provide real-time information about the progress of the runnable, including StreamEvents from intermediate results.

    A StreamEvent is a dictionary with the following schema:

    • event: string - Event names are of the format: on_[runnable_type]_(start|stream|end).
    • name: string - The name of the runnable that generated the event.
    • run_id: string - Randomly generated ID associated with the given execution of the runnable that emitted the event. A child runnable that gets invoked as part of the execution of a parent runnable is assigned its own unique ID.
    • tags: string[] - The tags of the runnable that generated the event.
    • metadata: Record<string, any> - The metadata of the runnable that generated the event.
    • data: Record<string, any>

    Below is a table that illustrates some events that might be emitted by various chains. Metadata fields have been omitted from the table for brevity. Chain definitions have been included after the table.

    ATTENTION This reference table is for the V2 version of the schema.

    +----------------------+-----------------------------+------------------------------------------+
    | event                | input                       | output/chunk                             |
    +======================+=============================+==========================================+
    | on_chat_model_start  | {"messages": BaseMessage[]} |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_chat_model_stream |                             | AIMessageChunk("hello")                  |
    +----------------------+-----------------------------+------------------------------------------+
    | on_chat_model_end    | {"messages": BaseMessage[]} | AIMessageChunk("hello world")            |
    +----------------------+-----------------------------+------------------------------------------+
    | on_llm_start         | {'input': 'hello'}          |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_llm_stream        |                             | 'Hello'                                  |
    +----------------------+-----------------------------+------------------------------------------+
    | on_llm_end           | 'Hello human!'              |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_chain_start       |                             |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_chain_stream      |                             | "hello world!"                           |
    +----------------------+-----------------------------+------------------------------------------+
    | on_chain_end         | [Document(...)]             | "hello world!, goodbye world!"           |
    +----------------------+-----------------------------+------------------------------------------+
    | on_tool_start        | {"x": 1, "y": "2"}          |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_tool_end          |                             | {"x": 1, "y": "2"}                       |
    +----------------------+-----------------------------+------------------------------------------+
    | on_retriever_start   | {"query": "hello"}          |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_retriever_end     | {"query": "hello"}          | [Document(...), ..]                      |
    +----------------------+-----------------------------+------------------------------------------+
    | on_prompt_start      | {"question": "hello"}       |                                          |
    +----------------------+-----------------------------+------------------------------------------+
    | on_prompt_end        | {"question": "hello"}       | ChatPromptValue(messages: BaseMessage[]) |
    +----------------------+-----------------------------+------------------------------------------+
    

    The "on_chain_*" events are the default for Runnables that don't fit one of the above categories.

    In addition to the standard events above, users can also dispatch custom events.

    Custom events will be only be surfaced with in the v2 version of the API!

    A custom event has following format:

    +-----------+------+------------------------------------------------------------+
    | Attribute | Type | Description                                                |
    +===========+======+============================================================+
    | name      | str  | A user defined name for the event.                         |
    +-----------+------+------------------------------------------------------------+
    | data      | Any  | The data associated with the event. This can be anything.  |
    +-----------+------+------------------------------------------------------------+
    

    Here's an example:

    import { RunnableLambda } from "@langchain/core/runnables";
    import { dispatchCustomEvent } from "@langchain/core/callbacks/dispatch";
    // Use this import for web environments that don't support "async_hooks"
    // and manually pass config to child runs.
    // import { dispatchCustomEvent } from "@langchain/core/callbacks/dispatch/web";

    const slowThing = RunnableLambda.from(async (someInput: string) => {
    // Placeholder for some slow operation
    await new Promise((resolve) => setTimeout(resolve, 100));
    await dispatchCustomEvent("progress_event", {
    message: "Finished step 1 of 2",
    });
    await new Promise((resolve) => setTimeout(resolve, 100));
    return "Done";
    });

    const eventStream = await slowThing.streamEvents("hello world", {
    version: "v2",
    });

    for await (const event of eventStream) {
    if (event.event === "on_custom_event") {
    console.log(event);
    }
    }

    Parameters

    • input: BaseLanguageModelInput
    • options: Partial<CallOptions> & {
          version: "v1" | "v2";
      }
    • OptionalstreamOptions: Omit<EventStreamCallbackHandlerInput, "autoClose">

    Returns IterableReadableStream<StreamEvent>

  • Parameters

    • input: BaseLanguageModelInput
    • options: Partial<CallOptions> & {
          encoding: "text/event-stream";
          version: "v1" | "v2";
      }
    • OptionalstreamOptions: Omit<EventStreamCallbackHandlerInput, "autoClose">

    Returns IterableReadableStream<Uint8Array>

  • Stream all output from a runnable, as reported to the callback system. This includes all inner runs of LLMs, Retrievers, Tools, etc. Output is streamed as Log objects, which include a list of jsonpatch ops that describe how the state of the run has changed in each step, and the final state of the run. The jsonpatch ops can be applied in order to construct state.

    Parameters

    • input: BaseLanguageModelInput
    • Optionaloptions: Partial<CallOptions>
    • OptionalstreamOptions: Omit<LogStreamCallbackHandlerInput, "autoClose">

    Returns AsyncGenerator<RunLogPatch, any, unknown>

  • Returns Serialized

  • Default implementation of transform, which buffers input and then calls stream. Subclasses should override this method if they can start producing output while input is still being generated.

    Parameters

    • generator: AsyncGenerator<BaseLanguageModelInput, any, unknown>
    • options: Partial<CallOptions>

    Returns AsyncGenerator<string, any, unknown>

  • Bind config to a Runnable, returning a new Runnable.

    Parameters

    • config: RunnableConfig<Record<string, any>>

      New configuration parameters to attach to the new runnable.

    Returns Runnable<BaseLanguageModelInput, string, CallOptions>

    A new RunnableBinding with a config matching what's passed.

  • Create a new runnable from the current one that will try invoking other passed fallback runnables if the initial invocation fails.

    Parameters

    • fields: {
          fallbacks: Runnable<BaseLanguageModelInput, string, RunnableConfig<Record<string, any>>>[];
      } | Runnable<BaseLanguageModelInput, string, RunnableConfig<Record<string, any>>>[]

    Returns RunnableWithFallbacks<BaseLanguageModelInput, string>

    A new RunnableWithFallbacks.

  • Bind lifecycle listeners to a Runnable, returning a new Runnable. The Run object contains information about the run, including its id, type, input, output, error, startTime, endTime, and any tags or metadata added to the run.

    Parameters

    • params: {
          onEnd?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>);
          onError?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>);
          onStart?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>);
      }

      The object containing the callback functions.

      • OptionalonEnd?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>)

        Called after the runnable finishes running, with the Run object.

          • (run, config?): void | Promise<void>
          • Parameters

            • run: Run
            • Optionalconfig: RunnableConfig<Record<string, any>>

            Returns void | Promise<void>

      • OptionalonError?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>)

        Called if the runnable throws an error, with the Run object.

          • (run, config?): void | Promise<void>
          • Parameters

            • run: Run
            • Optionalconfig: RunnableConfig<Record<string, any>>

            Returns void | Promise<void>

      • OptionalonStart?: ((run: Run, config?: RunnableConfig<Record<string, any>>) => void | Promise<void>)

        Called before the runnable starts running, with the Run object.

          • (run, config?): void | Promise<void>
          • Parameters

            • run: Run
            • Optionalconfig: RunnableConfig<Record<string, any>>

            Returns void | Promise<void>

    Returns Runnable<BaseLanguageModelInput, string, CallOptions>

  • Add retry logic to an existing runnable.

    Parameters

    • Optionalfields: {
          onFailedAttempt?: RunnableRetryFailedAttemptHandler;
          stopAfterAttempt?: number;
      }
      • OptionalonFailedAttempt?: RunnableRetryFailedAttemptHandler
      • OptionalstopAfterAttempt?: number

    Returns RunnableRetry<BaseLanguageModelInput, string, CallOptions>

    A new RunnableRetry that, when invoked, will retry according to the parameters.

  • Type Parameters

    • RunOutput extends Record<string, any> = Record<string, any>

    Parameters

    • schema: Record<string, any> | ZodType<RunOutput, ZodTypeDef, RunOutput>
    • Optionalconfig: StructuredOutputMethodOptions<false>

    Returns Runnable<BaseLanguageModelInput, RunOutput, RunnableConfig<Record<string, any>>>

  • Type Parameters

    • RunOutput extends Record<string, any> = Record<string, any>

    Parameters

    • schema: Record<string, any> | ZodType<RunOutput, ZodTypeDef, RunOutput>
    • Optionalconfig: StructuredOutputMethodOptions<true>

    Returns Runnable<BaseLanguageModelInput, {
        parsed: RunOutput;
        raw: BaseMessage;
    }, RunnableConfig<Record<string, any>>>

  • Model wrapper that returns outputs formatted to match the given schema.

    Type Parameters

    • RunOutput extends Record<string, any> = Record<string, any>

      The output type for the Runnable, expected to be a Zod schema object for structured output validation.

    Parameters

    • schema: Record<string, any> | ZodType<RunOutput, ZodTypeDef, RunOutput>

      The schema for the structured output. Either as a Zod schema or a valid JSON schema object. If a Zod schema is passed, the returned attributes will be validated, whereas with JSON schema they will not be.

    • Optionalconfig: StructuredOutputMethodOptions<boolean>

    Returns Runnable<BaseLanguageModelInput, RunOutput, RunnableConfig<Record<string, any>>> | Runnable<BaseLanguageModelInput, {
        parsed: RunOutput;
        raw: BaseMessage;
    }, RunnableConfig<Record<string, any>>>

    A new runnable that calls the LLM with structured output.

  • Parameters

    • _data: SerializedLLM

    Returns Promise<BaseLanguageModel<any, BaseLanguageModelCallOptions>>

    Load an LLM from a json-like object describing it.

  • Parameters

    • thing: any

    Returns thing is Runnable<any, any, RunnableConfig<Record<string, any>>>