muna 0.0.6

Run prediction functions in your Rust apps.
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

/*
*   Muna
*   Copyright © 2026 NatML Inc. All Rights Reserved.
*/

use std::fmt;
use serde::{Deserialize, Serialize};
use super::{Parameter, User};

/// Predictor access mode.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum PredictorAccess {
    Private,
    Public,
    Unlisted,
}

/// Predictor status.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum PredictorStatus {
    Compiling,
    Active,
    Archived,
}

/// Prediction function.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Predictor {
    /// Predictor tag.
    pub tag: String,
    /// Predictor owner.
    pub owner: User,
    /// Predictor name.
    pub name: String,
    /// Predictor status.
    pub status: PredictorStatus,
    /// Predictor access mode.
    pub access: PredictorAccess,
    /// Predictor signature.
    pub signature: Signature,
    /// Date created.
    pub created: String,
    /// Predictor description.
    #[serde(default)]
    pub description: Option<String>,
    /// Predictor card.
    #[serde(default)]
    pub card: Option<String>,
    /// Predictor media URL.
    #[serde(default)]
    pub media: Option<String>,
    /// Predictor license URL.
    #[serde(default)]
    pub license: Option<String>,
}

/// Prediction signature.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Signature {
    /// Prediction inputs.
    pub inputs: Vec<Parameter>,
    /// Prediction outputs.
    pub outputs: Vec<Parameter>,
}

impl fmt::Display for Predictor {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match serde_json::to_string_pretty(self) {
            Ok(json) => f.write_str(&json),
            Err(_) => write!(f, "{:?}", self),
        }
    }
}