Function calculate_cost 

pub fn calculate_cost(
    tokens: u32,
    input_cost_per_million: f64,
    output_cost_per_million: f64,
    is_output: bool,
) -> f64