SqliteTraceConsumer

SqliteTraceConsumer exists in src/agent/consumers/sqlite_trace.rs and implements StreamConsumer.

Definition

pub struct SqliteTraceConsumer {
    pool: Arc<SqlitePool>,
    session_id: String,
    agent_name: String,
    model: String,
    job_type: String,
    job_version: i64,
    start_time: Instant,
    sequence_counter: Arc<Mutex<i64>>,
    events: Arc<Mutex<Vec<serde_json::Value>>>,
    pending_event: Arc<tokio::sync::Mutex<Option<PendingEvent>>>,
}

Constructor

pub fn new(
    pool: Arc<SqlitePool>,
    session_id: String,
    agent_name: String,
    model: String,
    job_type: String,
    job_version: i64,
) -> Self

Behavior that matches the current code

• Consecutive content and reasoning events are accumulated and flushed as a single row when the event type changes or the stream ends.
• Non-streaming events are written as their own rows.
• The consumer also stores JSON event objects in memory for later retrieval through get_events().
• Database writes happen from the consumer's async flush path; this page should not assume a separate background worker API beyond what the current implementation provides.

Public helper

pub fn get_events(&self) -> Vec<serde_json::Value>

This returns the in-memory copy of the events seen so far.