enki-next 0.5.81

Enki's Rust agent runtime, workflow engine, and shared core abstractions.
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

use crate::registry::types::{AgentCard, AgentStatus, DiscoverQuery};
use std::collections::HashMap;
use tokio::sync::RwLock;

pub struct AgentRegistry {
    agents: RwLock<HashMap<String, AgentCard>>,
}

impl Default for AgentRegistry {
    fn default() -> Self {
        Self::new()
    }
}

impl AgentRegistry {
    pub fn new() -> Self {
        Self {
            agents: RwLock::new(HashMap::new()),
        }
    }

    pub async fn register(&self, card: AgentCard) {
        let mut agents = self.agents.write().await;
        agents.insert(card.agent_id.clone(), card);
    }

    pub async fn deregister(&self, agent_id: &str) -> bool {
        let mut agents = self.agents.write().await;
        agents.remove(agent_id).is_some()
    }

    pub async fn update_status(&self, agent_id: &str, status: AgentStatus) -> bool {
        let mut agents = self.agents.write().await;
        if let Some(card) = agents.get_mut(agent_id) {
            card.status = status;
            true
        } else {
            false
        }
    }

    pub async fn get(&self, agent_id: &str) -> Option<AgentCard> {
        let agents = self.agents.read().await;
        agents.get(agent_id).cloned()
    }

    pub async fn discover(&self, query: &DiscoverQuery) -> Vec<AgentCard> {
        let agents = self.agents.read().await;
        agents
            .values()
            .filter(|card| query.matches(card))
            .cloned()
            .collect()
    }

    pub async fn list_all(&self) -> Vec<AgentCard> {
        let agents = self.agents.read().await;
        agents.values().cloned().collect()
    }
}