ai/src/generate-text/output-utils.ts

AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript

import { Output } from './output';

/**
 * Infers the complete output type from the output specification.
 */
export type InferCompleteOutput<OUTPUT extends Output> =
  OUTPUT extends Output<infer COMPLETE_OUTPUT, any, any>
    ? COMPLETE_OUTPUT
    : never;

/**
 * Infers the partial output type from the output specification.
 */
export type InferPartialOutput<OUTPUT extends Output> =
  OUTPUT extends Output<any, infer PARTIAL_OUTPUT, any>
    ? PARTIAL_OUTPUT
    : never;

/**
 * Infers the element type from an array output specification.
 */
export type InferElementOutput<OUTPUT extends Output> =
  OUTPUT extends Output<any, any, infer ELEMENT> ? ELEMENT : never;