systemprompt-ai 0.64.0

Provider-agnostic LLM integration for systemprompt.io AI governance — Anthropic, OpenAI, Gemini, and local models unified behind one governed pipeline with cost tracking and audit.
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

//! `Extension` registration for the AI domain: schemas, migrations,
//! dependencies.
//!
//! Copyright (c) systemprompt.io — Business Source License 1.1.
//! See <https://systemprompt.io> for licensing details.

use systemprompt_extension::prelude::*;

#[derive(Debug, Clone, Copy, Default)]
pub struct AiExtension;

impl Extension for AiExtension {
    fn metadata(&self) -> ExtensionMetadata {
        ExtensionMetadata {
            id: "ai",
            name: "AI",
            version: env!("CARGO_PKG_VERSION"),
        }
    }

    fn schemas(&self) -> Vec<SchemaDefinition> {
        let mut schemas = request_schemas();
        schemas.extend(gateway_schemas());
        schemas
    }

    fn migrations(&self) -> Vec<Migration> {
        extension_migrations!()
    }

    fn dependencies(&self) -> Vec<&'static str> {
        vec!["users", "mcp"]
    }
}

register_extension!(AiExtension);

fn request_schemas() -> Vec<SchemaDefinition> {
    vec![
        SchemaDefinition::sql_only(include_str!("../schema/ai_request_message_count.sql")),
        SchemaDefinition::new("ai_requests", include_str!("../schema/ai_requests.sql"))
            .with_required_columns(columns(&[
                "id",
                "provider",
                "model",
                "created_at",
                "client_kind",
                "wire_protocol",
                "client_attestation",
                "finish_reason",
                "served_provider",
                "api_key_id",
            ])),
        SchemaDefinition::new(
            "ai_request_attributions",
            include_str!("../schema/ai_request_attributions.sql"),
        )
        .with_required_columns(columns(&["request_id", "dimension", "value", "source"])),
        SchemaDefinition::new(
            "ai_request_client_evidence",
            include_str!("../schema/ai_request_client_evidence.sql"),
        )
        .with_required_columns(columns(&["ai_request_id", "kind_source"])),
        SchemaDefinition::new(
            "ai_request_messages",
            include_str!("../schema/ai_request_messages.sql"),
        )
        .with_required_columns(columns(&["id", "request_id", "role"])),
        SchemaDefinition::new(
            "ai_request_tool_calls",
            include_str!("../schema/ai_request_tool_calls.sql"),
        )
        .with_required_columns(columns(&["id", "request_id", "tool_name"])),
        SchemaDefinition::new(
            "ai_tool_catalogs",
            include_str!("../schema/ai_tool_catalogs.sql"),
        )
        .with_required_columns(columns(&["sha256", "tools"])),
        SchemaDefinition::new(
            "ai_request_payloads",
            include_str!("../schema/ai_request_payloads.sql"),
        )
        .with_required_columns(columns(&["ai_request_id", "offered_tools_sha256"])),
    ]
}

fn gateway_schemas() -> Vec<SchemaDefinition> {
    vec![
        SchemaDefinition::new(
            "ai_safety_findings",
            include_str!("../schema/ai_safety_findings.sql"),
        )
        .with_required_columns(columns(&["id", "ai_request_id", "severity"])),
        SchemaDefinition::new(
            "ai_quota_buckets",
            include_str!("../schema/ai_quota_buckets.sql"),
        )
        .with_required_columns(columns(&[
            "id",
            "subject_kind",
            "subject_id",
            "window_start",
        ])),
        SchemaDefinition::new(
            "ai_gateway_policies",
            include_str!("../schema/ai_gateway_policies.sql"),
        )
        .with_required_columns(columns(&["id", "name", "spec"])),
        SchemaDefinition::new(
            "ai_gateway_thought_signatures",
            include_str!("../schema/ai_gateway_thought_signatures.sql"),
        )
        .with_required_columns(columns(&[
            "conversation_id",
            "tool_use_id",
            "signature",
            "expires_at",
        ])),
        SchemaDefinition::sql_only(include_str!("../schema/tool_call_ledger.sql")),
    ]
}

fn columns(names: &[&str]) -> Vec<String> {
    names.iter().map(|name| (*name).to_owned()).collect()
}