Class ChatBaiduWenxin

Wrapper around Baidu ERNIE large language models that use the Chat endpoint.

To use you should have the BAIDU_API_KEY and BAIDU_SECRET_KEY environment variable set.

Example

const ernieTurbo = new ChatBaiduWenxin({
  baiduApiKey: "YOUR-API-KEY",
  baiduSecretKey: "YOUR-SECRET-KEY",
});

const ernie = new ChatBaiduWenxin({
  modelName: "ERNIE-Bot",
  temperature: 1,
  baiduApiKey: "YOUR-API-KEY",
  baiduSecretKey: "YOUR-SECRET-KEY",
});

const messages = [new HumanMessage("Hello")];

let res = await ernieTurbo.call(messages);

res = await ernie.call(messages);

Hierarchy

  • BaseChatModel
    • ChatBaiduWenxin

Implements

  • BaiduWenxinChatInput

Constructors

constructor

Properties

accessToken apiUrl modelName streaming baiduApiKey? baiduSecretKey? penaltyScore? prefixMessages? temperature? topP? userId?

Accessors

callKeys

Methods

getAccessToken identifyingParams invocationParams

Constructors

constructor

  • Parameters

    • Optional fields: Partial<BaiduWenxinChatInput> & BaseLanguageModelParams

    Returns ChatBaiduWenxin

Properties

accessToken

accessToken: string

apiUrl

apiUrl: string

modelName

modelName: string = "ERNIE-Bot-turbo"

streaming

streaming: boolean = false

Optional baiduApiKey

baiduApiKey?: string

Optional baiduSecretKey

baiduSecretKey?: string

Optional penaltyScore

penaltyScore?: number

Optional prefixMessages

prefixMessages?: WenxinMessage[]

Optional temperature

temperature?: number

Optional topP

topP?: number

Optional userId

userId?: string

Accessors

callKeys

  • get callKeys(): string[]

  • Returns string[]

Methods

getAccessToken

  • Method that retrieves the access token for making requests to the Baidu API.

    Parameters

    • Optional options: Omit<BaseLanguageModelCallOptions, never>

      Optional parsed call options.

    Returns Promise<any>

    The access token for making requests to the Baidu API.

identifyingParams

  • Get the identifying parameters for the model

    Returns {
        model_name: string;
        penalty_score?: number;
        stream?: boolean;
        system?: string;
        temperature?: number;
        top_p?: number;
        user_id?: string;
    }

    • model_name: string
    • Optional penalty_score?: number
    • Optional stream?: boolean
    • Optional system?: string
    • Optional temperature?: number
    • Optional top_p?: number
    • Optional user_id?: string

invocationParams

  • Get the parameters used to invoke the model

    Returns Omit<ChatCompletionRequest, "messages">

Settings

Member Visibility

Theme

On This Page

Generated using TypeDoc