wyrd-spec 0.1.1

Wyrd foundation types: Card envelope, SpecKinds, and pure data contracts.
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

//! Pure PromptCard serde codecs.

use crate::card::prompt::PromptSpec;
use crate::card::prompt::validate::PromptError;
use crate::envelope::Card;
use crate::error::WyrdError;

/// Supported pure PromptCard byte formats.
#[derive(Debug, Clone, Copy, PartialEq, Eq, schemars::JsonSchema)]
#[cfg_attr(feature = "server", derive(utoipa::ToSchema))]
pub enum CardLoadFormat {
    /// JSON card or spec bytes.
    Json,
    /// YAML card or spec bytes.
    Yaml,
}

impl CardLoadFormat {
    /// Detects the card format from a path extension without filesystem IO.
    pub fn from_extension(ext: Option<&str>) -> Result<Self, WyrdError> {
        match ext.map(str::to_ascii_lowercase).as_deref() {
            Some("json") => Ok(Self::Json),
            Some("yaml" | "yml") => Ok(Self::Yaml),
            other => Err(PromptError::LoaderBadExtension {
                extension: other.map(ToOwned::to_owned),
            }
            .into()),
        }
    }
}

/// Parses a full Prompt Card envelope from bytes.
pub fn parse_card_bytes(format: CardLoadFormat, bytes: &[u8]) -> Result<Card, WyrdError> {
    match format {
        CardLoadFormat::Json => {
            serde_json::from_slice(bytes).map_err(|error| WyrdError::Validation {
                message: "failed to parse prompt card JSON".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
        CardLoadFormat::Yaml => {
            serde_yaml::from_slice(bytes).map_err(|error| WyrdError::Validation {
                message: "failed to parse prompt card YAML".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
    }
}

/// Serializes a full Prompt Card envelope to bytes.
pub fn serialize_card(format: CardLoadFormat, card: &Card) -> Result<Vec<u8>, WyrdError> {
    match format {
        CardLoadFormat::Json => {
            serde_json::to_vec_pretty(card).map_err(|error| WyrdError::Internal {
                message: "failed to serialize prompt card JSON".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
        CardLoadFormat::Yaml => {
            serde_yaml::to_string(card)
                .map(String::into_bytes)
                .map_err(|error| WyrdError::Internal {
                    message: "failed to serialize prompt card YAML".to_owned(),
                    details: serde_json::json!({ "source": error.to_string() }),
                })
        }
    }
}

/// Serializes a bare PromptSpec body to bytes.
pub fn serialize_spec_bytes(
    format: CardLoadFormat,
    spec: &PromptSpec,
) -> Result<Vec<u8>, WyrdError> {
    match format {
        CardLoadFormat::Json => {
            serde_json::to_vec_pretty(spec).map_err(|error| WyrdError::Internal {
                message: "failed to serialize prompt spec JSON".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
        CardLoadFormat::Yaml => {
            serde_yaml::to_string(spec)
                .map(String::into_bytes)
                .map_err(|error| WyrdError::Internal {
                    message: "failed to serialize prompt spec YAML".to_owned(),
                    details: serde_json::json!({ "source": error.to_string() }),
                })
        }
    }
}

/// Parses a bare PromptSpec body from bytes.
pub fn parse_spec_bytes(format: CardLoadFormat, bytes: &[u8]) -> Result<PromptSpec, WyrdError> {
    match format {
        CardLoadFormat::Json => {
            serde_json::from_slice(bytes).map_err(|error| WyrdError::Validation {
                message: "failed to parse prompt spec JSON".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
        CardLoadFormat::Yaml => {
            serde_yaml::from_slice(bytes).map_err(|error| WyrdError::Validation {
                message: "failed to parse prompt spec YAML".to_owned(),
                details: serde_json::json!({ "source": error.to_string() }),
            })
        }
    }
}