Struct TelemetryCollector 

pub struct TelemetryCollector { /* private fields */ }

Main telemetry collector for Juncture graph execution.

Creates traces and observations, applies capture configuration, and submits them to the batch writer for async persistence.

Examples

use juncture_telemetry::{TelemetryCollector, SqliteStore};
use std::sync::Arc;

let store = Arc::new(SqliteStore::new("telemetry.db").await?);
let collector = TelemetryCollector::new(store);

let trace = collector.begin_trace("my_graph", Some("thread-1"));
let obs = collector.begin_llm_call(trace.id, "claude-sonnet-4-20250514");
// ... execute LLM call ...
collector.end_llm_call(obs.id, Some(response), usage, cost).await;
collector.end_trace(trace.id, Some(output), total_cost, total_tokens).await;

Implementations

impl TelemetryCollector

pub fn new(store: Arc<dyn TraceStore>) -> Self

Create a new collector with default capture configuration.

pub fn with_capture_config( store: Arc<dyn TraceStore>, config: CaptureConfig, ) -> Self

Create a new collector with custom capture configuration.

pub fn with_langfuse( store: Arc<dyn TraceStore>, config: CaptureConfig, langfuse_config: LangfuseConfig, ) -> Self

Create a new collector with Langfuse cloud export enabled.

When configured, flush() and shutdown() automatically export traces and observations to Langfuse cloud alongside local storage.

pub fn capture_config(&self) -> &CaptureConfig

Get the capture configuration.

pub async fn begin_trace( &self, graph_name: impl Into<String>, session_id: Option<String>, ) -> Result<Trace, StoreError>

Begin a new trace for a graph invocation.

The trace is immediately submitted to the buffer so that observations can reference it without FK constraint violations.

Errors

Returns StoreError::Storage if the submission fails.

pub async fn end_trace( &self, trace: Trace, output: Option<Value>, total_cost: Option<f64>, total_tokens: Option<u64>, ) -> Result<(), StoreError>

End a trace and submit the completed version for async writing.

Errors

Returns StoreError::Storage if the submission fails.

pub async fn track_session( &self, thread_id: impl Into<String>, user_id: Option<String>, ) -> Result<(), StoreError>

Create or update a session.

Errors

Returns StoreError::Storage if the submission fails.

pub fn begin_llm_call( &self, trace_id: Id, parent_id: Option<Id>, model: impl Into<String>, prompt: Option<&Value>, ) -> Observation

Begin an LLM call observation.

pub async fn end_llm_call( &self, obs: Observation, response: Option<&str>, usage: Option<TokenUsage>, cost: Option<f64>, ) -> Result<(), StoreError>

End an LLM call observation and submit it.

Errors

Returns StoreError::Storage if the submission fails.

pub fn begin_tool_call( &self, trace_id: Id, parent_id: Option<Id>, tool_name: impl Into<String>, input: Option<&Value>, ) -> Observation

Begin a tool call observation.

pub async fn end_tool_call( &self, obs: Observation, output: Option<Value>, ) -> Result<(), StoreError>

End a tool call observation and submit it.

Errors

Returns StoreError::Storage if the submission fails.

pub fn begin_span( &self, trace_id: Id, parent_id: Option<Id>, name: impl Into<String>, ) -> Observation

Begin a generic span observation.

pub async fn end_span( &self, obs: Observation, output: Option<Value>, ) -> Result<(), StoreError>

End a span observation and submit it.

Errors

Returns StoreError::Storage if the submission fails.

pub async fn fail_span( &self, obs: Observation, error: impl Into<String>, ) -> Result<(), StoreError>

Record a failed span and submit it.

Errors

Returns StoreError::Storage if the submission fails.

pub async fn flush(&self) -> Result<(), StoreError>

Flush any buffered telemetry items to the store.

Errors

Returns StoreError::Storage if any write fails.

pub async fn shutdown(self) -> Result<(), StoreError>

Shutdown the collector, flushing all remaining items.

Errors

Returns StoreError::Storage if any write fails.

Trait Implementations

impl Clone for TelemetryCollector

fn clone(&self) -> TelemetryCollector

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TelemetryCollector

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.