modular-agent-kit 0.21.0

Modular Agent 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

use std::sync::atomic::AtomicUsize;

use crate::{
    FnvIndexMap,
    spec::{AgentSpec, ConnectionSpec},
};

static ID_COUNTER: AtomicUsize = AtomicUsize::new(1);

pub(crate) fn new_id() -> String {
    return ID_COUNTER
        .fetch_add(1, std::sync::atomic::Ordering::Relaxed)
        .to_string();
}

pub(crate) fn update_ids(
    agents: &Vec<AgentSpec>,
    connections: &Vec<ConnectionSpec>,
) -> (Vec<AgentSpec>, Vec<ConnectionSpec>) {
    let mut new_agents = Vec::new();
    let mut agent_id_map = FnvIndexMap::default();
    for agent in agents {
        let new_id = new_id();
        agent_id_map.insert(agent.id.clone(), new_id.clone());
        let mut new_agent = agent.clone();
        new_agent.id = new_id;
        new_agents.push(new_agent);
    }

    let mut new_connections = Vec::new();
    for connection in connections {
        let Some(source) = agent_id_map.get(&connection.source) else {
            continue;
        };
        let Some(target) = agent_id_map.get(&connection.target) else {
            continue;
        };
        let mut new_connection = connection.clone();
        new_connection.source = source.clone();
        new_connection.target = target.clone();
        new_connections.push(new_connection);
    }

    (new_agents, new_connections)
}