Aether/crates/aether-usage-runtime/src/lib.rs

mod body_capture;
pub mod config;
pub mod event;
mod executor;
pub mod queue;
pub mod record;
pub mod report;
pub mod report_context;
mod request_metadata;
pub mod runtime;
pub mod settlement;
pub mod standardized_usage;
pub mod usage_mapper;
pub mod worker;
pub mod write;

pub use body_capture::{
    apply_usage_body_capture_policy_to_event, apply_usage_body_capture_policy_to_record,
    UsageBodyCaptureEngine,
};
pub use config::UsageRuntimeConfig;
pub use event::{now_ms, UsageEvent, UsageEventData, UsageEventType, USAGE_EVENT_VERSION};
pub use queue::UsageQueue;
pub use record::build_upsert_usage_record_from_event;
pub use report::{
    extract_gemini_file_mapping_entries, gemini_file_mapping_cache_key,
    infer_internal_finalize_signature, is_local_ai_stream_report_kind,
    is_local_ai_sync_report_kind, normalize_gemini_file_name, report_request_id,
    resolve_internal_finalize_route, should_handle_local_stream_report,
    should_handle_local_sync_report, stream_capture_terminal_state,
    stream_report_missing_terminal_event, stream_report_represents_failure,
    stream_report_requires_observed_terminal_event, sync_report_represents_failure,
    GatewayStreamReportRequest, GatewaySyncReportRequest, GeminiFileMappingEntry,
    InternalFinalizeRoute, StreamCapturedTerminalState, GEMINI_FILE_MAPPING_TTL_SECONDS,
    STREAM_MISSING_TERMINAL_EVENT_CATEGORY, STREAM_MISSING_TERMINAL_EVENT_MESSAGE,
    STREAM_TERMINAL_ERROR_CATEGORY, STREAM_TERMINAL_ERROR_MESSAGE,
};
pub use report_context::{
    build_locally_actionable_report_context_from_request_candidate,
    build_locally_actionable_report_context_from_video_task, report_context_is_locally_actionable,
};
pub use runtime::{
    UsageBillingEventEnricher, UsageBodyCapturePolicy, UsageRequestRecordLevel, UsageRuntime,
    UsageRuntimeAccess, UsageRuntimeMetricsSnapshot,
    DEFAULT_USAGE_REQUEST_BODY_CAPTURE_LIMIT_BYTES,
    DEFAULT_USAGE_RESPONSE_BODY_CAPTURE_LIMIT_BYTES,
};
pub use settlement::{settle_usage_if_needed, UsageSettlementWriter};
pub use standardized_usage::StandardizedUsage;
pub use usage_mapper::{map_usage, map_usage_from_response, UsageMapper};
pub use worker::{
    build_usage_queue_worker, write_event_record, ManualProxyNodeCounter, UsageDataEventRecorder,
    UsageEventRecorder, UsageQueueWorker, UsageRecordWriter,
};
pub use write::{
    build_lifecycle_usage_seed, build_pending_usage_record, build_pending_usage_record_from_seed,
    build_stream_terminal_usage_event, build_stream_terminal_usage_outcome,
    build_stream_terminal_usage_payload_seed, build_stream_terminal_usage_seed,
    build_streaming_usage_record, build_streaming_usage_record_from_seed,
    build_sync_terminal_usage_event, build_sync_terminal_usage_outcome,
    build_sync_terminal_usage_payload_seed, build_sync_terminal_usage_seed,
    build_terminal_usage_context_seed, build_terminal_usage_event_from_outcome,
    build_terminal_usage_event_from_seed, build_usage_event_data_seed, LifecycleUsageSeed,
    StreamTerminalUsagePayloadSeed, SyncTerminalUsagePayloadSeed, TerminalUsageContextSeed,
    TerminalUsageOutcome, TerminalUsageSeed, UsageTerminalState,
};