UNPKG

@tanstack/ai

Version:

Type-safe TypeScript AI SDK for streaming chat, tool calling, agents, structured outputs, and multimodal generation.

32 lines (31 loc) 1.06 kB
import { ProviderUsageDetails, TokenUsage } from '../types.js'; /** * Input parameters for building base TokenUsage. * Provider functions should extract these from their SDK's response. */ export interface BaseUsageInput { /** Total input/prompt tokens */ promptTokens: number; /** Total output/completion tokens */ completionTokens: number; /** Total tokens (prompt + completion) */ totalTokens: number; } /** * Builds the base TokenUsage object with core fields. * Provider-specific functions should use this and then add their own details. * * @param input - The base token counts * @returns A TokenUsage object with promptTokens, completionTokens, totalTokens * * @example * ```typescript * const base = buildBaseUsage({ * promptTokens: 100, * completionTokens: 50, * totalTokens: 150 * }); * // Returns: { promptTokens: 100, completionTokens: 50, totalTokens: 150 } * ``` */ export declare function buildBaseUsage<TProviderDetails = ProviderUsageDetails>(input: BaseUsageInput): TokenUsage<TProviderDetails>;