act-sdk 0.6.0

Rust SDK for building ACT (Agent Component Tools) WebAssembly components
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

/// Raw tool event before conversion to WIT types.
/// Used internally by ActContext; the generated code converts to WIT ToolEvent.
pub enum RawToolEvent {
    Content {
        data: Vec<u8>,
        mime_type: Option<String>,
        metadata: Vec<(String, Vec<u8>)>,
    },
    Error {
        kind: String,
        message: String,
        default_language: String,
    },
}

/// Context passed to tool functions. Provides metadata access and stream writing.
///
/// Events are buffered in memory via `send_text()`/`send_content()` and
/// written to the WIT stream after the tool function returns.
pub struct ActContext<C = ()> {
    metadata: C,
    events: Vec<RawToolEvent>,
}

impl<C> ActContext<C> {
    #[doc(hidden)]
    pub fn __new(metadata: C) -> Self {
        Self {
            metadata,
            events: Vec::new(),
        }
    }

    /// Access the deserialized metadata.
    pub fn metadata(&self) -> &C {
        &self.metadata
    }

    /// Send a text content event (buffered).
    pub fn send_text(&mut self, text: impl Into<String>) {
        self.send_content(
            text.into().into_bytes(),
            Some(crate::constants::MIME_TEXT.to_string()),
            vec![],
        );
    }

    /// Send a CBOR-encoded content event (buffered).
    pub fn send_cbor<T: serde::Serialize>(&mut self, value: &T) {
        let mut buf = Vec::new();
        ciborium::into_writer(value, &mut buf).expect("CBOR serialization should not fail");
        self.send_content(buf, Some(crate::constants::MIME_CBOR.to_string()), vec![]);
    }

    /// Send a JSON-encoded content event (buffered).
    pub fn send_json<T: serde::Serialize>(&mut self, value: &T) {
        let data = serde_json::to_vec(value).unwrap_or_default();
        self.send_content(data, Some(crate::constants::MIME_JSON.to_string()), vec![]);
    }

    /// Send a content event with explicit data, MIME type, and metadata (buffered).
    pub fn send_content(
        &mut self,
        data: Vec<u8>,
        mime_type: Option<String>,
        metadata: Vec<(String, Vec<u8>)>,
    ) {
        self.events.push(RawToolEvent::Content {
            data,
            mime_type,
            metadata,
        });
    }

    /// Drain all buffered events. Called by generated code.
    #[doc(hidden)]
    pub fn __take_events(&mut self) -> Vec<RawToolEvent> {
        std::mem::take(&mut self.events)
    }
}