briefcase-python 2.4.1

//! # Briefcase AI Python Bindings
//!
//! Python FFI bindings for the Briefcase AI core library, providing high-performance
//! AI observability, replay, and decision tracking for Python applications.
//!
//! ## Features
//!
//! - **AI Decision Tracking**: Capture inputs, outputs, and context for every AI decision
//! - **Deterministic Replay**: Reproduce AI decisions exactly with full context preservation
//! - **Cost Management**: Track and optimize AI model usage costs
//! - **Drift Detection**: Monitor model performance and behavior changes
//! - **Data Sanitization**: Built-in privacy controls for sensitive data
//! - **Multiple Storage Backends**: SQLite, cloud storage, and custom backends
//!
//! ## Installation
//!
//! ```bash
//! pip install briefcase-ai
//! ```
//!
//! ## Usage
//!
//! ```python
//! import briefcase_ai
//!
//! # Initialize the library
//! briefcase_ai.init()
//!
//! # Create a decision snapshot
//! decision = briefcase_ai.DecisionSnapshot("ai_function")
//! decision.add_input(briefcase_ai.Input("query", "Hello world", "string"))
//! decision.add_output(briefcase_ai.Output("response", "Hello back!", "string"))
//!
//! # Save to storage
//! storage = briefcase_ai.SqliteBackend.in_memory()
//! decision_id = storage.save_decision(decision)
//! ```

use pyo3::prelude::*;

// Re-export core types with Python wrappers
mod client;
mod cost;
mod drift;
mod models;
mod replay;
mod runtime;
mod sanitization;
mod storage;

use cost::*;
use drift::*;
use models::*;
use replay::*;
use runtime::{init_runtime, RuntimeConfig};
use sanitization::*;
use storage::*;

/// Initialize the Briefcase AI library with default configuration
#[pyfunction]
fn init(py: Python) -> PyResult<()> {
    let config = RuntimeConfig::default();
    init_runtime(config)?;
    let atexit = py.import("atexit")?;
    let core_module = py.import("briefcase_ai._core")?;
    let shutdown = core_module.getattr("_shutdown_runtime")?;
    atexit.call_method1("register", (shutdown,))?;
    Ok(())
}

/// Initialize with custom configuration
#[pyfunction]
#[pyo3(signature = (worker_threads = 2))]
fn init_with_config(worker_threads: usize) -> PyResult<()> {
    let config = RuntimeConfig { worker_threads };
    init_runtime(config)
}

/// Check if the runtime is initialized
#[pyfunction]
fn is_initialized() -> bool {
    runtime::is_initialized()
}

/// Internal function to shutdown the runtime (called by atexit)
#[pyfunction]
fn _shutdown_runtime() -> PyResult<()> {
    runtime::shutdown_runtime()
}

/// Briefcase AI Python module
#[pymodule]
fn _core(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
    // Core data models
    m.add_class::<PyInput>()?;
    m.add_class::<PyOutput>()?;
    m.add_class::<PyModelParameters>()?;
    m.add_class::<PyExecutionContext>()?;
    m.add_class::<PyDecisionSnapshot>()?;
    m.add_class::<PySnapshot>()?;

    // Storage backends
    m.add_class::<PySqliteBackend>()?;
    m.add_class::<PyLakeFSBackend>()?;

    // Core functionality
    m.add_class::<PyDriftCalculator>()?;
    m.add_class::<PyCostCalculator>()?;
    m.add_class::<PySanitizer>()?;
    m.add_class::<PyReplayEngine>()?;

    // Query types
    m.add_class::<PySnapshotQuery>()?;

    // Authenticated client
    m.add_class::<client::PyBriefcaseClient>()?;
    m.add_class::<client::PyValidatedClient>()?;

    // Runtime initialization functions
    m.add_function(wrap_pyfunction!(init, m)?)?;
    m.add_function(wrap_pyfunction!(init_with_config, m)?)?;
    m.add_function(wrap_pyfunction!(is_initialized, m)?)?;
    m.add_function(wrap_pyfunction!(_shutdown_runtime, m)?)?;

    // Add module metadata
    m.add("__version__", "2.4.1")?;
    m.add("__author__", "Briefcase AI Team")?;

    Ok(())
}