Interface for AI21 input parameters.

interface AI21Input {
    ai21ApiKey?: string;
    baseUrl?: string;
    cache?: boolean | BaseCache<Generation[]>;
    callbackManager?: CallbackManager;
    callbacks?: Callbacks;
    concurrency?: number;
    countPenalty?: AI21PenaltyData;
    frequencyPenalty?: AI21PenaltyData;
    logitBias?: Record<string, number>;
    maxConcurrency?: number;
    maxRetries?: number;
    maxTokens?: number;
    metadata?: Record<string, unknown>;
    minTokens?: number;
    model?: string;
    numResults?: number;
    onFailedAttempt?: FailedAttemptHandler;
    presencePenalty?: AI21PenaltyData;
    stop?: string[];
    tags?: string[];
    temperature?: number;
    topP?: number;
    verbose?: boolean;
}

Hierarchy

  • BaseLLMParams
    • AI21Input

Implemented by

Properties

ai21ApiKey?: string
baseUrl?: string
cache?: boolean | BaseCache<Generation[]>
callbackManager?: CallbackManager

Use callbacks instead

callbacks?: Callbacks
concurrency?: number

Use maxConcurrency instead

countPenalty?: AI21PenaltyData
frequencyPenalty?: AI21PenaltyData
logitBias?: Record<string, number>
maxConcurrency?: number

The maximum number of concurrent calls that can be made. Defaults to Infinity, which means no limit.

maxRetries?: number

The maximum number of retries that can be made for a single call, with an exponential backoff between each attempt. Defaults to 6.

maxTokens?: number
metadata?: Record<string, unknown>
minTokens?: number
model?: string
numResults?: number
onFailedAttempt?: FailedAttemptHandler

Custom handler to handle failed attempts. Takes the originally thrown error object as input, and should itself throw an error if the input error is not retryable.

presencePenalty?: AI21PenaltyData
stop?: string[]
tags?: string[]
temperature?: number
topP?: number
verbose?: boolean