agent-stream-kit 0.19.0

Agent Stream Kit
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

use crate::askit::ASKit;
use crate::context::AgentContext;
use crate::error::AgentError;
use crate::value::AgentValue;

#[derive(Clone, Debug)]
pub enum AgentEventMessage {
    AgentOut {
        agent: String,
        ctx: AgentContext,
        pin: String,
        value: AgentValue,
    },
    BoardOut {
        name: String,
        ctx: AgentContext,
        value: AgentValue,
    },
}

pub async fn send_agent_out(
    askit: &ASKit,
    agent: String,
    ctx: AgentContext,
    pin: String,
    value: AgentValue,
) -> Result<(), AgentError> {
    askit
        .tx()?
        .send(AgentEventMessage::AgentOut {
            agent,
            ctx,
            pin,
            value,
        })
        .await
        .map_err(|_| AgentError::SendMessageFailed("Failed to send AgentOut message".to_string()))
}

pub fn try_send_agent_out(
    askit: &ASKit,
    agent: String,
    ctx: AgentContext,
    pin: String,
    value: AgentValue,
) -> Result<(), AgentError> {
    askit
        .tx()?
        .try_send(AgentEventMessage::AgentOut {
            agent,
            ctx,
            pin,
            value,
        })
        .map_err(|_| {
            AgentError::SendMessageFailed("Failed to try_send AgentOut message".to_string())
        })
}

pub async fn send_board_out(
    askit: &ASKit,
    name: String,
    ctx: AgentContext,
    value: AgentValue,
) -> Result<(), AgentError> {
    askit
        .tx()?
        .send(AgentEventMessage::BoardOut { name, ctx, value })
        .await
        .map_err(|_| {
            AgentError::SendMessageFailed("Failed to try_send BoardOut message".to_string())
        })
}

// Processing AgentOut message
pub async fn agent_out(
    askit: &ASKit,
    source_agent: String,
    ctx: AgentContext,
    pin: String,
    value: AgentValue,
) {
    let targets;
    {
        let env_edges = askit.channels.lock().unwrap();
        targets = env_edges.get(&source_agent).cloned();
    }

    if targets.is_none() {
        return;
    }

    for target in targets.unwrap() {
        let (target_agent, source_pin, target_pin) = target;

        if source_pin != pin {
            // Skip if source_handle does not match with the given port.
            continue;
        }

        {
            let env_agents = askit.agents.lock().unwrap();
            if !env_agents.contains_key(&target_agent) {
                continue;
            }
        }

        askit
            .agent_input(target_agent.clone(), ctx.clone(), target_pin, value.clone())
            .await
            .unwrap_or_else(|e| {
                log::error!("Failed to send message to {}: {}", target_agent, e);
            });
    }
}

pub async fn board_out(askit: &ASKit, name: String, ctx: AgentContext, value: AgentValue) {
    {
        let mut board_value = askit.board_value.lock().unwrap();
        board_value.insert(name.clone(), value.clone());
    }
    let board_nodes;
    {
        let env_board_nodes = askit.board_out_agents.lock().unwrap();
        board_nodes = env_board_nodes.get(&name).cloned();
    }
    if let Some(board_nodes) = board_nodes {
        for node in board_nodes {
            // Perhaps we could process this by send_message_to BoardOutAgent

            let edges;
            {
                let env_edges = askit.channels.lock().unwrap();
                edges = env_edges.get(&node).cloned();
            }
            let Some(edges) = edges else {
                // edges not found
                continue;
            };
            for (target_agent, _source_pin, target_pin) in edges {
                askit
                    .agent_input(target_agent.clone(), ctx.clone(), target_pin, value.clone())
                    .await
                    .unwrap_or_else(|e| {
                        log::error!("Failed to send message to {}: {}", target_agent, e);
                    });
            }
        }
    }

    askit.emit_board(name, value);
}