atlas-cli 0.2.0

Machine Learning Lifecycle & Transparency Manager - Create and verify manifests for ML models and datasets
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

//! # Atlas CLI
//!
//! Machine Learning (ML) Lifecycle & Transparency Manager
//!
//! A command-line interface tool for creating, managing, and verifying Content Provenance
//! and Authenticity (C2PA) manifests for machine learning models, datasets, and related artifacts.
//!
//! ## Installation
//!
//! ```bash
//! cargo install atlas-cli
//! ```
//!
//! ## Quick Start
//!
//! Create a model manifest:
//! ```bash
//! atlas-cli model create \
//!     --paths=model.onnx \
//!     --ingredient-names="Main Model" \
//!     --name="My Model" \
//!     --author-org="My Organization" \
//!     --author-name="My Name" \
//!     --print
//! ```
//!
//! For more examples and detailed documentation, see:
//! - [User Guide](https://github.com/IntelLabs/atlas-cli/blob/main/docs/USER_GUIDE.md)
//! - [Examples](https://github.com/IntelLabs/atlas-cli/blob/main/docs/EXAMPLES.md)
//! - [Development Guide](https://github.com/IntelLabs/atlas-cli/blob/main/docs/DEVELOPMENT.md)

#![doc(html_root_url = "https://docs.rs/atlas-cli/0.2.0")]

pub mod cc_attestation;
pub mod cli;
pub mod error;
pub mod hash;
pub mod in_toto;
pub mod manifest;
pub mod signing;
pub mod slsa;
pub mod storage;
#[cfg(test)]
mod tests;
pub mod utils;

use std::path::PathBuf;
use storage::config::StorageConfig;

// Re-export error types
pub use error::{Error, Result};

/// CLI configuration options
#[derive(Debug, Clone)]
pub struct Config {
    /// Path to private key for signing
    pub key_path: Option<PathBuf>,
    /// Storage backend configuration
    pub storage_config: StorageConfig,
    /// Whether to show progress bars
    pub show_progress: bool,
}

impl Default for Config {
    fn default() -> Self {
        Self {
            key_path: None,
            storage_config: StorageConfig::default(),
            show_progress: true,
        }
    }
}

/// Initialize logging for the CLI
///
/// # Examples
///
/// ```
/// use atlas_cli::init_logging;
///
/// // Initialize with default settings
/// let result = init_logging();
/// // Note: This might fail if already initialized
/// assert!(result.is_ok() || result.is_err());
/// ```
pub fn init_logging() -> Result<()> {
    env_logger::try_init().map_err(|e| Error::InitializationError(e.to_string()))
}

// Re-export commonly used types and traits
pub use storage::traits::StorageBackend;