llm-cost-ops-cli 0.1.0

Command-line interface for LLM Cost Ops
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
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

use clap::{Parser, Subcommand};
use std::path::PathBuf;

// CLI argument definitions for LLM Cost Operations

#[derive(Parser)]
#[command(name = "cost-ops")]
#[command(about = "LLM Cost Operations Platform", long_about = None)]
#[command(version)]
pub struct Cli {
    /// Configuration file path
    #[arg(short, long, value_name = "FILE")]
    pub config: Option<PathBuf>,

    /// Verbosity level (-v, -vv, -vvv)
    #[arg(short, long, action = clap::ArgAction::Count)]
    pub verbose: u8,

    #[command(subcommand)]
    pub command: Commands,
}

#[derive(Subcommand)]
pub enum Commands {
    /// Initialize database and create schema
    Init {
        /// Database URL (e.g., sqlite:cost-ops.db)
        #[arg(long)]
        database_url: Option<String>,
    },

    /// Ingest usage metrics
    Ingest {
        /// Input file path (JSON)
        #[arg(short, long)]
        file: PathBuf,

        /// Provider name
        #[arg(short, long)]
        provider: Option<String>,
    },

    /// Query cost records
    Query {
        /// Time range (e.g., "last-24-hours", "last-7-days")
        #[arg(long, default_value = "last-24-hours")]
        range: String,

        /// Organization ID filter
        #[arg(long)]
        organization: Option<String>,

        /// Group by dimension (provider, model, project)
        #[arg(long)]
        group_by: Option<String>,

        /// Output format (json, table, csv)
        #[arg(long, default_value = "table")]
        output: String,
    },

    /// Generate cost summary
    Summary {
        /// Time period (e.g., "last-30-days")
        #[arg(long, default_value = "last-30-days")]
        period: String,

        /// Organization ID filter
        #[arg(long)]
        organization: Option<String>,
    },

    /// Export cost data
    Export {
        /// Output file path
        #[arg(short, long)]
        output: PathBuf,

        /// Format (json, csv, parquet)
        #[arg(short, long, default_value = "json")]
        format: String,

        /// Time period
        #[arg(long)]
        period: Option<String>,
    },

    /// Manage pricing tables
    Pricing {
        #[command(subcommand)]
        command: PricingCommands,
    },
}

#[derive(Subcommand)]
pub enum PricingCommands {
    /// List all pricing tables
    List,

    /// Add a new pricing table
    Add {
        /// Provider name
        #[arg(long)]
        provider: String,

        /// Model name
        #[arg(long)]
        model: String,

        /// Input price per million tokens
        #[arg(long)]
        input_price: f64,

        /// Output price per million tokens
        #[arg(long)]
        output_price: f64,
    },

    /// Get active pricing for a model
    Get {
        /// Provider name
        #[arg(long)]
        provider: String,

        /// Model name
        #[arg(long)]
        model: String,
    },
}

impl Cli {
    pub fn parse_args() -> Self {
        Self::parse()
    }
}