systemprompt-ai 0.56.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

//! `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> {
        vec![
            SchemaDefinition::sql_only(include_str!("../schema/reporting_privacy.sql")),
            SchemaDefinition::sql_only(include_str!("../schema/reporting_capture.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",
                ])),
            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_request_payloads",
                include_str!("../schema/ai_request_payloads.sql"),
            )
            .with_required_columns(columns(&["ai_request_id"])),
            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 migrations(&self) -> Vec<Migration> {
        extension_migrations!()
    }

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

register_extension!(AiExtension);

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