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

//! Repository layer for AI domain persistence.
//!
//! Every type here owns `SQLx` queries against the AI domain tables
//! (`ai_requests`, `ai_request_messages`, `ai_tool_calls`,
//! `ai_request_payloads`, `ai_quota_buckets`, `ai_safety_findings`,
//! `ai_gateway_policies`).
//!
//! All repositories return [`crate::error::RepositoryError`]. Services are
//! the only callers — repositories never execute application logic.
//!
//! Copyright (c) systemprompt.io — Business Source License 1.1.
//! See <https://systemprompt.io> for licensing details.

pub mod ai_gateway_policies;
pub mod ai_quota_buckets;
pub mod ai_request_payloads;
pub mod ai_requests;
pub mod ai_safety_findings;

use crate::error::RepositoryError;
use systemprompt_database::DbPool;

pub use ai_gateway_policies::{AiGatewayPolicyRepository, GatewayPolicyRow};
pub use ai_quota_buckets::{
    AiQuotaBucketRepository, IncrementParams, QuotaBucketDelta, QuotaBucketState,
};
pub use ai_request_payloads::{AiRequestPayload, AiRequestPayloadRepository, UpsertPayloadParams};
pub use ai_requests::{AiRequestRepository, InsertToolCallParams};
pub use ai_safety_findings::{AiSafetyFindingRepository, InsertSafetyFinding};

#[derive(Debug, Clone)]
pub struct AiRepositories {
    pub requests: AiRequestRepository,
    pub payloads: AiRequestPayloadRepository,
    pub gateway_policies: AiGatewayPolicyRepository,
    pub quota_buckets: AiQuotaBucketRepository,
    pub safety_findings: AiSafetyFindingRepository,
}

impl AiRepositories {
    pub fn new(db: &DbPool) -> Result<Self, RepositoryError> {
        Ok(Self {
            requests: AiRequestRepository::new(db)?,
            payloads: AiRequestPayloadRepository::new(db)?,
            gateway_policies: AiGatewayPolicyRepository::new(db)?,
            quota_buckets: AiQuotaBucketRepository::new(db)?,
            safety_findings: AiSafetyFindingRepository::new(db)?,
        })
    }
}