atomr-agents-core 0.2.1

Core types for the atomr-agents framework: ids, budgets, context, events, errors.
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use serde::{Deserialize, Serialize};

use crate::ids::{AgentId, HarnessId, RunId, ToolId, WorkflowId};
use crate::inference::FinishReason;

/// Structured event emitted by every observable boundary in the
/// framework. Fed to `atomr-telemetry`, used by traces, metrics, and
/// the eval-suite replay path.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "kind", rename_all = "snake_case")]
pub enum Event {
    StrategyResolved {
        strategy: String,
        agent_id: Option<AgentId>,
        elapsed_ms: u64,
        tokens_used: u32,
    },
    ToolInvoked {
        tool_id: ToolId,
        args_hash: u64,
        elapsed_ms: u64,
        ok: bool,
    },
    /// Emitted when an inference turn streams a tool call before it
    /// is dispatched. Lets tracers / UIs surface tool intent in
    /// real-time, distinct from `ToolInvoked` which fires post-call.
    ToolCallStreamed {
        agent_id: AgentId,
        tool_name: String,
        arguments_hash: u64,
        iteration: u32,
    },
    AgentTurn {
        agent_id: AgentId,
        input_tokens: u32,
        output_tokens: u32,
        /// Provider-reported reasoning tokens (e.g. o1-style).
        /// Defaults to 0 for runtimes that don't surface them.
        #[serde(default)]
        reasoning_tokens: u32,
        /// Cached prefix tokens (Anthropic prompt-cache, OpenAI cached
        /// input). Priced lower than `input_tokens` by most providers.
        #[serde(default)]
        cached_tokens: u32,
        finish_reason: Option<FinishReason>,
        elapsed_ms: u64,
    },
    WorkflowStep {
        workflow_id: WorkflowId,
        step_id: String,
        step_kind: String,
        elapsed_ms: u64,
        ok: bool,
    },
    HarnessIteration {
        harness_id: HarnessId,
        iteration: u64,
        outcome: String,
        budget_remaining_tokens: u32,
    },
    Backpressure {
        actor_path: String,
        queued: u32,
        dropped: u32,
    },
}

/// Tagged envelope around an event with timestamp + correlation id.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EventEnvelope {
    pub timestamp_ms: i64,
    pub correlation_id: Option<String>,
    /// LangSmith-style run identification. Optional so existing call
    /// sites still compile; tracers and the run-tree builder require
    /// these to be populated.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub run_id: Option<RunId>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub parent_run_id: Option<RunId>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub tags: Vec<String>,
    pub event: Event,
}

impl EventEnvelope {
    pub fn now(event: Event) -> Self {
        Self {
            timestamp_ms: chrono::Utc::now().timestamp_millis(),
            correlation_id: None,
            run_id: None,
            parent_run_id: None,
            tags: Vec::new(),
            event,
        }
    }

    pub fn with_run(mut self, run_id: RunId, parent: Option<RunId>) -> Self {
        self.run_id = Some(run_id);
        self.parent_run_id = parent;
        self
    }

    pub fn with_tags(mut self, tags: Vec<String>) -> Self {
        self.tags = tags;
        self
    }
}