adk-ui 0.8.0

Dynamic UI generation for ADK-Rust agents - render forms, cards, tables, charts and more
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

use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use serde_json::Value;

use crate::schema::UiEvent;

/// A2UI v0.9 action event payload.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct A2uiActionEvent {
    pub name: String,
    pub surface_id: String,
    pub source_component_id: String,
    pub timestamp: String,
    pub context: Value,
}

#[derive(Debug, Clone)]
pub struct A2uiActionMetadata {
    pub surface_id: String,
    pub source_component_id: String,
    pub timestamp: DateTime<Utc>,
}

impl A2uiActionMetadata {
    pub fn new(surface_id: impl Into<String>, source_component_id: impl Into<String>) -> Self {
        Self {
            surface_id: surface_id.into(),
            source_component_id: source_component_id.into(),
            timestamp: Utc::now(),
        }
    }
}

/// Maps ADK UiEvent into A2UI action events.
#[derive(Debug, Default)]
pub struct UiEventMapper;

impl UiEventMapper {
    pub fn new() -> Self {
        Self
    }

    pub fn to_a2ui_action(&self, event: &UiEvent, meta: &A2uiActionMetadata) -> A2uiActionEvent {
        let (name, context) = match event {
            UiEvent::FormSubmit { action_id, data } => (
                action_id.clone(),
                serde_json::to_value(data).unwrap_or_else(|_| Value::Object(Default::default())),
            ),
            UiEvent::ButtonClick { action_id } => {
                (action_id.clone(), Value::Object(Default::default()))
            }
            UiEvent::InputChange { name, value } => (
                "input_change".to_string(),
                serde_json::json!({ "name": name, "value": value }),
            ),
            UiEvent::TabChange { index } => (
                "tab_change".to_string(),
                serde_json::json!({ "index": index }),
            ),
        };

        A2uiActionEvent {
            name,
            surface_id: meta.surface_id.clone(),
            source_component_id: meta.source_component_id.clone(),
            timestamp: meta.timestamp.to_rfc3339(),
            context,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::schema::UiEvent;
    use serde_json::json;

    #[test]
    fn maps_form_submit_with_context() {
        let mapper = UiEventMapper::new();
        let event = UiEvent::FormSubmit {
            action_id: "submit".to_string(),
            data: [("email".to_string(), json!("a@b.com"))]
                .into_iter()
                .collect(),
        };
        let meta = A2uiActionMetadata::new("main", "form-1");
        let mapped = mapper.to_a2ui_action(&event, &meta);
        assert_eq!(mapped.name, "submit");
        assert_eq!(mapped.surface_id, "main");
        assert_eq!(mapped.source_component_id, "form-1");
        assert!(mapped.context.get("email").is_some());
    }
}