crabmate 0.4.0

Rust AI agent: OpenAI-compatible chat/completions, function calling, HTTP serve, ops CLI
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! AG-UI 标准事件枚举（CrabMate 使用的子集）。
//!
//! 遵循 [AG-UI 协议](https://docs.ag-ui.com/concepts/events) 标准事件类型。
//! CrabMate 特有的非标准事件通过 `Custom` 变体承载。

use serde::Serialize;

/// AG-UI 标准事件（仅覆盖 CrabMate 需要的子集）。
///
/// 部分变体（RunStarted、TextMessage*、ReasoningMessage*、State*）暂未在 V2Encoder
/// conversion 中构造，将在后续 Phase（消息生命周期、状态同步）接入。
#[allow(dead_code)]
#[derive(Debug, Clone, Serialize)]
#[serde(tag = "type", rename_all = "SCREAMING_SNAKE_CASE")]
pub(crate) enum AgUiEvent {
    // ── 生命周期 ──
    RunStarted {
        #[serde(rename = "threadId")]
        thread_id: String,
        #[serde(rename = "runId")]
        run_id: String,
    },
    RunFinished {
        #[serde(rename = "threadId")]
        thread_id: String,
        #[serde(rename = "runId")]
        run_id: String,
        /// CrabMate 扩展：与 v1 `stream_ended.tiktoken_prompt_tokens` 同源；成功路径通常已先发
        /// `CUSTOM conversation_saved`，本字段作后备。
        #[serde(
            rename = "tiktokenPromptTokens",
            skip_serializing_if = "Option::is_none"
        )]
        tiktoken_prompt_tokens: Option<crate::cm_types::TiktokenPromptTokensSnapshot>,
    },
    RunError {
        #[serde(rename = "threadId")]
        thread_id: String,
        #[serde(rename = "runId")]
        run_id: String,
        error: AgUiErrorBody,
    },

    // ── 文本消息 ──
    TextMessageStart {
        #[serde(rename = "messageId")]
        message_id: String,
        role: String,
        #[serde(skip_serializing_if = "Option::is_none")]
        metadata: Option<serde_json::Value>,
    },
    TextMessageContent {
        #[serde(rename = "messageId")]
        message_id: String,
        delta: String,
    },
    TextMessageEnd {
        #[serde(rename = "messageId")]
        message_id: String,
    },

    // ── 工具调用 ──
    ToolCallStart {
        #[serde(rename = "toolCallId")]
        tool_call_id: String,
        name: String,
        #[serde(rename = "parentMessageId")]
        parent_message_id: String,
    },
    ToolCallArgs {
        #[serde(rename = "toolCallId")]
        tool_call_id: String,
        args: String,
    },
    ToolCallEnd {
        #[serde(rename = "toolCallId")]
        tool_call_id: String,
    },
    ToolCallResult {
        #[serde(rename = "toolCallId")]
        tool_call_id: String,
        content: String,
        #[serde(skip_serializing_if = "Option::is_none")]
        metadata: Option<serde_json::Value>,
    },

    // ── Reasoning（思维链） ──
    ReasoningMessageStart {
        #[serde(rename = "messageId")]
        message_id: String,
    },
    ReasoningMessageContent {
        #[serde(rename = "messageId")]
        message_id: String,
        delta: String,
    },
    ReasoningMessageEnd {
        #[serde(rename = "messageId")]
        message_id: String,
    },

    // ── 状态同步 ──
    StateSnapshot {
        state: serde_json::Value,
    },
    StateDelta {
        state: serde_json::Value,
    },

    // ── CrabMate 扩展 ──
    Custom {
        #[serde(rename = "customType")]
        custom_type: String,
        data: serde_json::Value,
    },
}

/// AG-UI 错误体。
#[derive(Debug, Clone, Serialize)]
pub(crate) struct AgUiErrorBody {
    pub message: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub code: Option<String>,
    /// 与 HTTP `x-request-id` 对齐；旧客户端忽略。
    #[serde(rename = "requestId", skip_serializing_if = "Option::is_none")]
    pub request_id: Option<String>,
}