cargo-ai 0.2.0

Build lightweight AI agents with Cargo. Powered by Rust. Declared in JSON.
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

// src/infra_api/account.rs
//
// Account operations against cargo-ai-infra.
// Keep this module free of CLI concerns so it can be extracted into a library later.

use crate::cargo_ai_metadata;
use crate::config::schema::CargoAiMetadata;
use serde_json::Value;

pub mod agents;
pub mod confirm;
pub mod handle;
pub mod mail_preferences;
pub mod register;
pub mod send_mail;
pub mod status;

pub(crate) fn with_cargo_ai_metadata(body: Value) -> Value {
    with_cargo_ai_metadata_override(body, cargo_ai_metadata::load_request_metadata())
}

pub(crate) fn with_cargo_ai_metadata_override(
    mut body: Value,
    metadata: Option<CargoAiMetadata>,
) -> Value {
    if let Some(metadata) = metadata {
        if let Ok(metadata_value) = serde_json::to_value(metadata) {
            body["cargo_ai_metadata"] = metadata_value;
        }
    }

    body
}

#[cfg(test)]
mod tests {
    use super::{with_cargo_ai_metadata_override, CargoAiMetadata};
    use serde_json::json;

    fn sample_metadata() -> CargoAiMetadata {
        CargoAiMetadata {
            cargo_ai_version: Some(env!("CARGO_PKG_VERSION").to_string()),
            template_schema_version: Some("2026-03-03.r1".to_string()),
            cargo_ai_build_target: Some("aarch64-apple-darwin".to_string()),
            cargo_ai_install_id: Some("install-123".to_string()),
            cargo_ai_binary_sha256: Some("hash-456".to_string()),
        }
    }

    #[test]
    fn with_cargo_ai_metadata_override_adds_top_level_metadata_block() {
        let body = with_cargo_ai_metadata_override(
            json!({
                "action": "status"
            }),
            Some(sample_metadata()),
        );

        assert_eq!(
            body,
            json!({
                "action": "status",
                "cargo_ai_metadata": {
                    "cargo_ai_version": env!("CARGO_PKG_VERSION"),
                    "template_schema_version": "2026-03-03.r1",
                    "cargo_ai_build_target": "aarch64-apple-darwin",
                    "cargo_ai_install_id": "install-123",
                    "cargo_ai_binary_sha256": "hash-456"
                }
            })
        );
    }

    #[test]
    fn with_cargo_ai_metadata_override_keeps_body_unchanged_when_metadata_missing() {
        let body = with_cargo_ai_metadata_override(
            json!({
                "action": "status"
            }),
            None,
        );

        assert_eq!(
            body,
            json!({
                "action": "status"
            })
        );
    }
}