Struct TraceStore 

pub struct TraceStore<C: ConnectionTrait> { /* private fields */ }

Interface for trace storage

Implementations

impl<C: ConnectionTrait> TraceStore<C>

pub fn new(conn: C) -> Self

Create a new trace store

pub fn with_sampling(conn: C, sample_rate: f64) -> Self

Create with sampling

pub fn start_run( &self, name: impl Into<String>, resource: HashMap<String, String>, ) -> Result<TraceRun>

Start a new trace run

pub fn end_run(&self, trace_id: &TraceId, status: TraceStatus) -> Result<()>

End a trace run

pub fn get_run(&self, trace_id: &TraceId) -> Result<Option<TraceRun>>

Get a trace run

pub fn update_run_metrics( &self, trace_id: &TraceId, tokens: u64, cost_millicents: u64, ) -> Result<()>

Update run metrics

pub fn start_span( &self, trace_id: &TraceId, name: impl Into<String>, parent_span_id: Option<SpanId>, kind: SpanKind, ) -> Result<TraceSpan>

Start a new span

pub fn end_span( &self, trace_id: &TraceId, span_id: &SpanId, status: SpanStatusCode, message: Option<String>, ) -> Result<()>

End a span

pub fn add_span_event( &self, trace_id: &TraceId, span_id: &SpanId, name: impl Into<String>, attributes: HashMap<String, TraceValue>, ) -> Result<()>

Add an event to a span

pub fn set_span_attributes( &self, trace_id: &TraceId, span_id: &SpanId, attributes: HashMap<String, TraceValue>, ) -> Result<()>

Set span attributes

pub fn get_spans(&self, trace_id: &TraceId) -> Result<Vec<TraceSpan>>

Get all spans for a trace

pub fn log_retrieval_hit( &self, trace_id: &TraceId, span_id: &SpanId, hit: RetrievalHitEvent, ) -> Result<()>

Log a retrieval hit

pub fn log_tool_call( &self, trace_id: &TraceId, span_id: &SpanId, call: ToolCallEvent, ) -> Result<()>

Log a tool call

pub fn log_context_packaging( &self, trace_id: &TraceId, span_id: &SpanId, event: ContextPackagingEvent, ) -> Result<()>

Log context packaging

pub fn log_cost(&self, trace_id: &TraceId, event: CostEvent) -> Result<()>

Log a cost event