aa-cli 0.0.1-alpha.9

aasm — command-line tool for Agent Assembly
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

//! Cost management subcommands (`aasm cost ...`).

use std::process::ExitCode;

use clap::{Args, Subcommand};

use crate::config::ResolvedContext;
use crate::output::OutputFormat;

pub mod client;
pub mod forecast;
pub mod models;
pub mod summary;

/// Arguments for the `aasm cost` subcommand group.
#[derive(Args)]
pub struct CostArgs {
    #[command(subcommand)]
    pub command: CostCommands,
}

/// Available cost subcommands.
#[derive(Subcommand)]
pub enum CostCommands {
    /// Show cost summary for the current period.
    Summary(summary::SummaryArgs),
    /// Forecast monthly spending based on current daily rate.
    Forecast(forecast::ForecastArgs),
}

/// Dispatch a cost subcommand.
pub fn dispatch(args: CostArgs, ctx: &ResolvedContext, output: OutputFormat) -> ExitCode {
    match args.command {
        CostCommands::Summary(summary_args) => summary::run(summary_args, ctx, output),
        CostCommands::Forecast(forecast_args) => forecast::run(forecast_args, ctx, output),
    }
}