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

extern crate modular_agent_kit as mak;

use mak::{MAK, AgentContext, AgentStatus, AgentValue};

use crate::common;

const COUNTER_DEF: &str = common::agents::CounterAgent::DEF_NAME;

#[test]
fn test_register_agent_definiton() {
    let mak = MAK::init().unwrap();

    // Check the properties of the counter agent
    let counter_def = mak.get_agent_definition(COUNTER_DEF).unwrap();
    assert_eq!(counter_def.title, Some("Counter".into()));
    assert_eq!(counter_def.inputs, Some(vec!["in".into(), "reset".into()]));
    assert_eq!(counter_def.outputs, Some(vec!["count".into()]));

    mak.quit();
}

#[test]
fn test_agent_new() {
    let mak = MAK::init().unwrap();
    let def = mak.get_agent_definition(COUNTER_DEF).unwrap();
    let spec = def.to_spec();
    let agent = mak::agent_new(mak.clone(), "agent_1".into(), spec).unwrap();
    assert_eq!(agent.def_name(), COUNTER_DEF);
    assert_eq!(agent.id(), "agent_1");
    assert_eq!(agent.status(), &AgentStatus::Init);

    mak.quit();
}

#[tokio::test]
async fn test_agent_start() {
    let mak = MAK::init().unwrap();
    let def = mak.get_agent_definition(COUNTER_DEF).unwrap();
    let spec = def.to_spec();
    let mut agent = mak::agent_new(mak.clone(), "agent_1".into(), spec).unwrap();
    agent.start().await.unwrap();

    assert_eq!(agent.status(), &AgentStatus::Start);

    mak.quit();
}

#[tokio::test]
async fn test_agent_process() {
    let mak = MAK::init().unwrap();
    mak.ready().await.unwrap();

    let counter_def = mak.get_agent_definition(COUNTER_DEF).unwrap();
    let counter_spec = counter_def.to_spec();

    let mut counter_agent =
        mak::agent_new(mak.clone(), "agent_1".into(), counter_spec).unwrap();
    counter_agent.start().await.unwrap();

    let ctx = AgentContext::new();
    counter_agent
        .process(ctx, "in".into(), AgentValue::unit())
        .await
        .unwrap();

    let counter_agent = counter_agent
        .as_any()
        .downcast_ref::<common::agents::CounterAgent>()
        .unwrap();
    assert_eq!(counter_agent.count, 1);

    mak.quit();
}

#[tokio::test]
async fn test_agent_stop() {
    let mak = MAK::init().unwrap();

    mak.ready().await.unwrap();

    let def = mak.get_agent_definition(COUNTER_DEF).unwrap();
    let spec = def.to_spec();
    let mut agent = mak::agent_new(mak.clone(), "agent_1".into(), spec).unwrap();
    agent.start().await.unwrap();

    let ctx = AgentContext::new();
    agent
        .process(ctx, "in".into(), AgentValue::unit())
        .await
        .unwrap();

    agent.stop().await.unwrap();
    assert_eq!(agent.status(), &AgentStatus::Init);

    mak.quit();
}