UNPKG

ai21

Version:

AI21 TypeScript SDK

38 lines (37 loc) 891 B
import { Logprobs } from './LogProbs'; import { AssistantMessage } from './ChatMessage'; export interface ChatCompletionResponse { id: string; choices: ChatCompletionResponseChoice[]; usage: { prompt_tokens: number; completion_tokens: number; total_tokens: number; }; } export interface ChatCompletionResponseChoice { index: number; message: AssistantMessage; logprobs?: Logprobs; finish_reason?: string; } export interface ChatCompletionChunk { id: string; choices: ChoicesChunk[]; usage?: UsageInfo; } export interface ChoiceDelta { content?: string; role?: string; } export interface ChoicesChunk { index: number; delta: ChoiceDelta; logprobs?: Logprobs; finish_reason?: string; } export interface UsageInfo { prompt_tokens: number; completion_tokens: number; total_tokens: number; }