obol-cli 0.4.0

CLI to estimate the USD cost of an AI-agent transcript (Claude Code, Codex, Pi).
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

use assert_cmd::Command;
use std::fs;

#[test]
fn estimate_json_on_claude_fixture() {
    let tmp = tempfile::tempdir().unwrap();
    let sample = include_str!("../../obol-core/tests/fixtures/litellm-sample.json");
    let claude = include_str!("../../obol-core/tests/fixtures/claude-mini.jsonl");
    // Seed a price snapshot without network: normalize the sample sheet via obol-core.
    let store_json =
        obol_core::pricing::refresh::normalize_litellm(sample.as_bytes(), "2026-06-04").unwrap();
    let dir = tmp.path().join("obol");
    fs::create_dir_all(&dir).unwrap();
    store_json.save(&dir.join("current.json")).unwrap();

    let transcript = tmp.path().join("session.jsonl");
    fs::write(&transcript, claude).unwrap();

    Command::cargo_bin("obol")
        .unwrap()
        .env("OBOL_PRICING_DIR", &dir)
        .args([
            "estimate",
            transcript.to_str().unwrap(),
            "--dialect",
            "claude",
            "--json",
        ])
        .assert()
        .success()
        .stdout(predicates::str::contains("\"total_usd\""));
}

#[test]
fn estimate_reports_bundled_pricing_source() {
    let tmp = tempfile::tempdir().unwrap();
    let claude = include_str!("../../obol-core/tests/fixtures/claude-mini.jsonl");
    let transcript = tmp.path().join("session.jsonl");
    fs::write(&transcript, claude).unwrap();

    // No override + an empty XDG home -> no on-disk snapshot, so the embedded one
    // prices it and the source must be "bundled". Pointing XDG at the (snapshot-free)
    // temp dir keeps this hermetic regardless of the dev's real ~/.local/share/obol.
    Command::cargo_bin("obol")
        .unwrap()
        .env_remove("OBOL_PRICING_DIR")
        .env("XDG_DATA_HOME", tmp.path())
        .args([
            "estimate",
            transcript.to_str().unwrap(),
            "--dialect",
            "claude",
            "--json",
        ])
        .assert()
        .success()
        .stdout(predicates::str::contains("pricing_source"))
        .stdout(predicates::str::contains("bundled"));
}

#[test]
fn estimate_gemini_dialect_string() {
    let tmp = tempfile::tempdir().unwrap();
    let gemini = include_str!("../../obol-core/tests/fixtures/gemini-mini.jsonl");
    let transcript = tmp.path().join("session.jsonl");
    fs::write(&transcript, gemini).unwrap();

    Command::cargo_bin("obol")
        .unwrap()
        .env_remove("OBOL_PRICING_DIR")
        .args([
            "estimate",
            transcript.to_str().unwrap(),
            "--dialect",
            "gemini",
            "--json",
        ])
        .assert()
        .success()
        .stdout(predicates::str::contains("gemini-3-flash-preview"));
}

#[test]
fn estimate_obol_dialect_string() {
    let tmp = tempfile::tempdir().unwrap();
    let obol = include_str!("../../obol-core/tests/fixtures/obol-usage-mini.jsonl");
    let transcript = tmp.path().join("usage.jsonl");
    fs::write(&transcript, obol).unwrap();

    Command::cargo_bin("obol")
        .unwrap()
        .env_remove("OBOL_PRICING_DIR")
        .args([
            "estimate",
            transcript.to_str().unwrap(),
            "--dialect",
            "obol",
            "--json",
        ])
        .assert()
        .success()
        .stdout(predicates::str::contains("claude-opus-4-8"));
}