adk-graph 0.6.0

Graph-based workflow orchestration for ADK-Rust agents
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

//! Set action node executor.
//!
//! Supports three modes:
//! - **set**: Insert each variable into state.
//! - **merge**: Deep-merge each variable with existing state values.
//! - **delete**: Signal key removal by setting to null.

use std::collections::HashMap;

use adk_action::{SetMode, SetNodeConfig, interpolate_variables};
use serde_json::Value;

use crate::error::{GraphError, Result};
use crate::node::{NodeContext, NodeOutput};

/// Execute a Set action node.
pub async fn execute_set(config: &SetNodeConfig, ctx: &NodeContext) -> Result<NodeOutput> {
    let state = &ctx.state;
    let mut output = NodeOutput::new();

    // Load environment variables if configured
    if let Some(env_cfg) = &config.env_vars {
        if env_cfg.load_from_env {
            for key in &env_cfg.keys {
                let env_key = if let Some(prefix) = &env_cfg.prefix {
                    format!("{prefix}{key}")
                } else {
                    key.clone()
                };
                if let Ok(val) = std::env::var(&env_key) {
                    output = output.with_update(key, Value::String(val));
                    tracing::debug!(key = %key, "loaded env var");
                }
            }
        }
    }

    match config.mode {
        SetMode::Set => {
            for var in &config.variables {
                let value = resolve_variable_value(&var.value, state);
                if var.is_secret {
                    tracing::debug!(key = %var.key, value = "***", "set secret variable");
                } else {
                    tracing::debug!(key = %var.key, value = %value, "set variable");
                }
                output = output.with_update(&var.key, value);
            }
        }
        SetMode::Merge => {
            for var in &config.variables {
                let new_value = resolve_variable_value(&var.value, state);
                let merged = if let Some(existing) = state.get(&var.key) {
                    deep_merge(existing.clone(), new_value)
                } else {
                    new_value
                };
                if var.is_secret {
                    tracing::debug!(key = %var.key, value = "***", "merged secret variable");
                } else {
                    tracing::debug!(key = %var.key, value = %merged, "merged variable");
                }
                output = output.with_update(&var.key, merged);
            }
        }
        SetMode::Delete => {
            for var in &config.variables {
                tracing::debug!(key = %var.key, "deleting variable");
                output = output.with_update(&var.key, Value::Null);
            }
        }
    }

    Ok(output)
}

/// Resolve a variable value, interpolating string templates.
fn resolve_variable_value(value: &Value, state: &HashMap<String, Value>) -> Value {
    match value {
        Value::String(s) => {
            let interpolated = interpolate_variables(s, state);
            // Try to parse as JSON if the original was a template
            if s.contains("{{") {
                // If the entire string was a single variable reference that resolved to
                // a non-string JSON value, try to parse it back
                if let Ok(parsed) = serde_json::from_str::<Value>(&interpolated) {
                    if !parsed.is_string() {
                        return parsed;
                    }
                }
            }
            Value::String(interpolated)
        }
        _ => value.clone(),
    }
}

/// Deep-merge two JSON values. Objects are merged recursively; other types are overwritten.
fn deep_merge(base: Value, overlay: Value) -> Value {
    match (base, overlay) {
        (Value::Object(mut base_map), Value::Object(overlay_map)) => {
            for (key, overlay_val) in overlay_map {
                let merged = if let Some(base_val) = base_map.remove(&key) {
                    deep_merge(base_val, overlay_val)
                } else {
                    overlay_val
                };
                base_map.insert(key, merged);
            }
            Value::Object(base_map)
        }
        (_, overlay) => overlay,
    }
}

impl From<adk_action::ActionError> for GraphError {
    fn from(err: adk_action::ActionError) -> Self {
        GraphError::NodeExecutionFailed { node: String::new(), message: err.to_string() }
    }
}