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

use crate::error::Result;
use crate::manifest::common::{AssetKind, list_manifests, verify_manifest};
use crate::manifest::config::ManifestCreationConfig;
use crate::storage::traits::StorageBackend;

pub fn create_manifest(
    mut config: ManifestCreationConfig,
    software_type: String,
    version: Option<String>,
) -> Result<()> {
    config.software_type = Some(software_type.clone());
    config.version = version.clone();

    // Combine software_type and version into description or metadata
    let enhanced_description = match (&config.description, &version) {
        (Some(desc), Some(ver)) => Some(format!("{desc} (Type: {software_type}, Version: {ver})")),
        (Some(desc), None) => Some(format!("{desc} (Type: {software_type})")),
        (None, Some(ver)) => Some(format!("Type: {software_type}, Version: {ver}")),
        (None, None) => Some(format!("Type: {software_type}")),
    };

    // Update the description in the config
    config.description = enhanced_description;

    // Call the common implementation with AssetKind::Software
    crate::manifest::common::create_manifest(config, AssetKind::Software)
}

/// List software manifests
pub fn list_software_manifests(storage: &dyn StorageBackend) -> Result<()> {
    // Call the unified implementation with AssetKind::Software
    list_manifests(storage, Some(AssetKind::Software))
}

/// Verify a software manifest
pub fn verify_software_manifest(id: &str, storage: &dyn StorageBackend) -> Result<()> {
    // Call the unified implementation
    verify_manifest(id, storage)
}