sniff-cli 0.1.3

An exhaustive LLM-backed slop finder for codebases
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use crate::config::ResolvedConfig;
use crate::report_types::RunStats;
use std::env;

fn configured_rate(name: &str, default: f64) -> f64 {
    env::var(name)
        .ok()
        .and_then(|value| value.trim().parse::<f64>().ok())
        .filter(|value| value.is_finite() && *value >= 0.0)
        .unwrap_or(default)
}

pub(super) fn calculate_cost(s: &RunStats, _config: &ResolvedConfig) -> String {
    let input_per_m = configured_rate("SNIFF_INPUT_COST_PER_MILLION", 0.14);
    let output_per_m = configured_rate("SNIFF_OUTPUT_COST_PER_MILLION", 0.28);
    let rate_suffix = format!("; rates ${input_per_m:.2}/M in / ${output_per_m:.2}/M out");

    let cost =
        (s.input_tokens as f64 / 1e6) * input_per_m + (s.output_tokens as f64 / 1e6) * output_per_m;
    if cost < 0.0001 && (s.input_tokens > 0 || s.output_tokens > 0) {
        format!(
            "<$0.0001 (Tokens: {} in / {} out{})",
            s.input_tokens, s.output_tokens, rate_suffix
        )
    } else {
        format!(
            "${:.4} (Tokens: {} in / {} out{})",
            cost, s.input_tokens, s.output_tokens, rate_suffix
        )
    }
}

#[cfg(test)]
mod tests {
    use super::calculate_cost;
    use crate::config::ResolvedConfig;
    use crate::report_types::RunStats;

    #[test]
    fn cost_summary_discloses_the_rates_used() {
        let stats = RunStats {
            input_tokens: 1_000_000,
            output_tokens: 1_000_000,
            ..RunStats::default()
        };

        let cost = calculate_cost(&stats, &ResolvedConfig::default());

        assert!(cost.contains("rates $0.14/M in / $0.28/M out"));
    }
}