wesichain-core 0.3.0

Rust-native LLM agents & chains with resumable ReAct 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
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

use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(tag = "type", content = "data")]
pub enum AgentEvent {
    Status {
        stage: String,
        message: String,
        step: usize,
        thread_id: String,
    },
    Thought {
        content: String,
        step: usize,
        metadata: Option<serde_json::Value>,
    },
    ToolCall {
        id: String,
        tool_name: String,
        input: serde_json::Value,
        step: usize,
    },
    Observation {
        id: String,
        tool_name: String,
        output: serde_json::Value,
        step: usize,
    },
    Final {
        content: String,
        step: usize,
    },
    Error {
        message: String,
        step: usize,
        recoverable: bool,
        source: Option<String>,
    },
    Metadata {
        key: String,
        value: serde_json::Value,
    },
}

impl AgentEvent {
    pub fn step(&self) -> Option<usize> {
        match self {
            Self::Status { step, .. }
            | Self::Thought { step, .. }
            | Self::ToolCall { step, .. }
            | Self::Observation { step, .. }
            | Self::Final { step, .. }
            | Self::Error { step, .. } => Some(*step),
            Self::Metadata { .. } => None,
        }
    }

    pub fn thread_id(&self) -> Option<&str> {
        match self {
            Self::Status { thread_id, .. } => Some(thread_id.as_str()),
            _ => None,
        }
    }
}