takeln 0.11.0

Typed Rust runtime for durable DAG-based agent workflows
Documentation 
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

use crate::emitter::SpanStatus;
use crate::hitl::ResumeRecord;
use std::sync::Arc;

/// Hook for collecting metrics from graph execution.
///
/// Implement this trait to integrate with your preferred metrics system
/// (Prometheus, StatsD, OpenTelemetry Metrics, etc.).
pub trait MetricsHook: Send + Sync {
    /// Called after each node completes (success or failure).
    fn on_node_complete(&self, node_name: &str, duration_ms: u64, status: SpanStatus);
    /// Called when the entire graph execution finishes.
    fn on_graph_complete(&self, thread_id: &str, total_cost: f64, total_duration_ms: u64);
    /// Called after a checkpoint is successfully saved.
    fn on_checkpoint_saved(&self, thread_id: &str, checkpoint_id: &str);
    /// Called when a thread is successfully resumed.
    fn on_resume(&self, _record: &ResumeRecord) {}
}

/// Default no-op metrics hook that discards all metrics.
pub struct NoopMetricsHook;

impl MetricsHook for NoopMetricsHook {
    fn on_node_complete(&self, _: &str, _: u64, _: SpanStatus) {}
    fn on_graph_complete(&self, _: &str, _: f64, _: u64) {}
    fn on_checkpoint_saved(&self, _: &str, _: &str) {}
    fn on_resume(&self, _: &ResumeRecord) {}
}

/// Returns the default no-op metrics hook.
pub(crate) fn default_metrics_hook() -> Arc<dyn MetricsHook> {
    Arc::new(NoopMetricsHook)
}