halo-agent 0.1.0

Halo Agent and Workflow execution kernel
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

use super::*;

#[test]
fn planned_task_instruction_is_merged_into_the_existing_system_message() {
    let mut system = Message::new(Role::System);
    system.content.push(Part {
        r#type: "text".into(),
        data: part_data(json!({"value": "parent instructions"})),
    });
    let user = Message::new(Role::User);
    let input = Message::new(Role::User);

    let messages = task_messages(Messages::new([system, user]), input);

    assert_eq!(
        messages
            .0
            .iter()
            .map(|message| message.role.clone())
            .collect::<Vec<_>>(),
        [Role::System, Role::User, Role::User]
    );
    assert_eq!(messages.0[0].content.len(), 2);
    assert_eq!(
        messages.0[0].content[0].data["value"],
        "parent instructions"
    );
    assert!(
        messages.0[0].content[1].data["value"]
            .as_str()
            .unwrap()
            .contains("current planned task objective")
    );
}