systemprompt-models 0.14.6

Foundation data models for systemprompt.io AI governance infrastructure. Shared DTOs, config, and domain types consumed by every layer of the MCP governance pipeline.
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

//! Per-provider AI policy and per-model descriptors.
//!
//! [`AiProviderConfig`] is the deployment policy layered on a registry provider
//! (enable flag, default-model overrides, resilience). [`ModelDefinition`] and
//! its [`ModelCapabilities`], [`ModelLimits`], and [`ModelPricing`] are the
//! per-model descriptors shared with `profile.providers`. Connectivity itself
//! is never modelled here — it lives in the provider registry.

use serde::{Deserialize, Serialize};

use super::config::ResilienceSettings;

const fn default_true() -> bool {
    true
}

#[expect(
    clippy::struct_excessive_bools,
    reason = "model capability matrix: each bool is an independent provider feature flag, not \
              state"
)]
#[derive(Debug, Clone, Copy, Default, Serialize, Deserialize, schemars::JsonSchema)]
pub struct ModelCapabilities {
    #[serde(default)]
    pub vision: bool,

    #[serde(default)]
    pub audio_input: bool,

    #[serde(default)]
    pub video_input: bool,

    #[serde(default)]
    pub image_generation: bool,

    #[serde(default)]
    pub audio_generation: bool,

    #[serde(default)]
    pub streaming: bool,

    #[serde(default)]
    pub tools: bool,

    #[serde(default)]
    pub structured_output: bool,

    #[serde(default)]
    pub system_prompts: bool,

    #[serde(default)]
    pub image_resolution_config: bool,
}

#[derive(Debug, Clone, Copy, Default, Serialize, Deserialize, schemars::JsonSchema)]
pub struct ModelLimits {
    #[serde(default)]
    pub context_window: u32,

    #[serde(default)]
    pub max_output_tokens: u32,

    /// Maximum extended-thinking / reasoning budget (in tokens) the upstream
    /// accepts for this model. `None` means the provider validates the budget
    /// itself (Anthropic) or maps it to an effort bucket (`OpenAI`); set it for
    /// providers that reject an out-of-range numeric budget (Gemini: 24576 for
    /// flash, 32768 for pro). The gateway clamps the requested budget to this.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub max_thinking_budget: Option<u32>,
}

#[derive(Debug, Clone, Copy, Default, Serialize, Deserialize, schemars::JsonSchema)]
pub struct ModelPricing {
    #[serde(default)]
    pub input_per_million: f64,

    #[serde(default)]
    pub output_per_million: f64,

    #[serde(default)]
    pub per_image_cents: Option<f64>,
}

#[derive(Debug, Clone, Copy, Default, Serialize, Deserialize)]
pub struct ModelDefinition {
    #[serde(default)]
    pub capabilities: ModelCapabilities,

    #[serde(default)]
    pub limits: ModelLimits,

    #[serde(default)]
    pub pricing: ModelPricing,
}

/// Per-provider AI *policy*, keyed by registry provider name.
///
/// Connectivity (endpoint, credential, model catalog) lives in the profile
/// `providers` registry; this struct carries only the policy a deployment
/// layers on top of an entry: whether the provider is enabled, its agent-side
/// default-model override, image defaults, web-search toggle, and resilience.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AiProviderConfig {
    #[serde(default = "default_true")]
    pub enabled: bool,

    /// Overrides the provider client's built-in default model when non-empty.
    #[serde(default)]
    pub default_model: String,

    #[serde(default)]
    pub default_image_model: String,

    #[serde(default)]
    pub google_search_enabled: bool,

    /// Resilience policy applied to outbound AI provider calls (timeouts,
    /// retry, circuit breaker, bulkhead).
    #[serde(default)]
    pub resilience: ResilienceSettings,
}

impl Default for AiProviderConfig {
    fn default() -> Self {
        Self {
            enabled: true,
            default_model: String::new(),
            default_image_model: String::new(),
            google_search_enabled: false,
            resilience: ResilienceSettings::default(),
        }
    }
}