Function record_tokens 

pub fn record_tokens(direction: &'static str, modality: &str, tokens: u64)

Record tokens a Live session used, by direction (prompt or response) and modality (TEXT, AUDIO, IMAGE, VIDEO).