oxi-sdk 0.25.3

oxi AI agent SDK — build isolated, multi-agent AI systems
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

//! Observability layer — tracing, audit, cost tracking, and event sourcing.

mod audit;
mod cost;
mod event_store;
mod trace;

// Re-exports
pub use audit::{AuditEntry, AuditFilter, AuditLog};
pub use cost::{
    CostBreakdown, CostSnapshot, CostTracker, CostTrackerConfig, GlobalCostSnapshot, TokenUsage,
};
pub use event_store::{EventQuery, EventStore, EventStoreConfig, StoredEvent};
pub use trace::{Span, SpanContext, SpanGuard, SpanId, SpanKind, SpanStatus, TraceId, Tracer};

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn smoke() {
        // Tracer smoke test
        let tracer = Tracer::new();
        let _guard = tracer.start("test", SpanKind::Agent);
        drop(_guard);
        // AuditLog smoke test
        let log = AuditLog::new(64);
        log.log(AuditEntry::lifecycle("test".into(), "smoke".into()));
        // CostTracker smoke test
        let registry = std::sync::Arc::new(oxi_ai::ModelRegistry::new());
        let tracker = CostTracker::new(registry, CostTrackerConfig::default());
        let model = oxi_ai::Model::new(
            "dummy/model",
            "Dummy",
            oxi_ai::Api::AnthropicMessages,
            "dummy",
            "https://dummy.com",
        );
        tracker.record("agent-1", &model, TokenUsage::default());
        let snap = tracker.snapshot("agent-1");
        assert!(snap.is_some());
        // EventStore smoke test
        let store = EventStore::default();
        let seq = store.append("s1", "test", serde_json::json!({"x": 1}));
        assert_eq!(seq, 1);
    }
}