oxi-agent 0.1.4-alpha

Agent runtime with tool-calling loop for AI coding assistants
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

//! Agent event system

use crate::compaction::CompactionEvent;
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "camelCase")]
pub enum AgentEvent {
    // ── Lifecycle events (from pi-mono agent-loop) ──────────────────
    AgentStart { prompts: Vec<oxi_ai::Message> },
    AgentEnd { messages: Vec<oxi_ai::Message>, stop_reason: Option<String> },
    TurnStart { turn_number: u32 },
    TurnEnd { turn_number: u32, assistant_message: oxi_ai::Message, tool_results: Vec<oxi_ai::ToolResultMessage> },
    
    // ── Message events (from pi-mono agent-loop) ────────────────────
    MessageStart { message: oxi_ai::Message },
    MessageUpdate { message: oxi_ai::Message, delta: Option<String> },
    MessageEnd { message: oxi_ai::Message },
    
    // ── Tool execution events (from pi-mono agent-loop) ────────────
    ToolExecutionStart { tool_call_id: String, tool_name: String, args: serde_json::Value },
    ToolExecutionUpdate { tool_call_id: String, tool_name: String, partial_result: String },
    ToolExecutionEnd { tool_call_id: String, tool_name: String, result: oxi_ai::ToolResult, is_error: bool },
    
    // ── Legacy events (kept for backward compatibility) ──────────
    #[serde(rename = "start")]
    Start { prompt: String },
    Thinking,
    ThinkingDelta { text: String },
    TextChunk { text: String },
    ToolCall { tool_call: oxi_ai::ToolCall },
    ToolStart { tool_call_id: String, tool_name: String },
    ToolProgress { tool_call_id: String, message: String },
    ToolComplete { result: oxi_ai::ToolResult },
    ToolError { tool_call_id: String, error: String },
    Complete { content: String, stop_reason: String },
    Error { message: String },
    Iteration { number: usize },
    Usage { input_tokens: usize, output_tokens: usize },
    Compaction { event: CompactionEvent },
    Retry { attempt: usize, max_retries: usize, retry_after_secs: u64, reason: String },
    Fallback { from_model: String, to_model: String },
    Cancelled,
    PartialResponse { content: String },
    
    // ── Loop-specific steering events ─────────────────────────────
    SteeringMessage { message: oxi_ai::Message },
    FollowUpMessage { message: oxi_ai::Message },
}

impl AgentEvent {
    pub fn is_terminal(&self) -> bool {
        matches!(self, AgentEvent::AgentEnd { .. })
    }
    
    pub fn type_name(&self) -> &'static str {
        match self {
            AgentEvent::AgentStart { .. } => "agent_start",
            AgentEvent::AgentEnd { .. } => "agent_end",
            AgentEvent::TurnStart { .. } => "turn_start",
            AgentEvent::TurnEnd { .. } => "turn_end",
            AgentEvent::MessageStart { .. } => "message_start",
            AgentEvent::MessageUpdate { .. } => "message_update",
            AgentEvent::MessageEnd { .. } => "message_end",
            AgentEvent::ToolExecutionStart { .. } => "tool_execution_start",
            AgentEvent::ToolExecutionUpdate { .. } => "tool_execution_update",
            AgentEvent::ToolExecutionEnd { .. } => "tool_execution_end",
            AgentEvent::Start { .. } => "start",
            AgentEvent::Thinking => "thinking",
            AgentEvent::ThinkingDelta { .. } => "thinking_delta",
            AgentEvent::TextChunk { .. } => "text_chunk",
            AgentEvent::ToolCall { .. } => "tool_call",
            AgentEvent::ToolStart { .. } => "tool_start",
            AgentEvent::ToolProgress { .. } => "tool_progress",
            AgentEvent::ToolComplete { .. } => "tool_complete",
            AgentEvent::ToolError { .. } => "tool_error",
            AgentEvent::Complete { .. } => "complete",
            AgentEvent::Error { .. } => "error",
            AgentEvent::Iteration { .. } => "iteration",
            AgentEvent::Usage { .. } => "usage",
            AgentEvent::Compaction { .. } => "compaction",
            AgentEvent::Retry { .. } => "retry",
            AgentEvent::Fallback { .. } => "fallback",
            AgentEvent::Cancelled => "cancelled",
            AgentEvent::PartialResponse { .. } => "partial_response",
            AgentEvent::SteeringMessage { .. } => "steering_message",
            AgentEvent::FollowUpMessage { .. } => "follow_up_message",
        }
    }
}