halo-agent 0.1.1

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
35
36
37
38
39
40

//! Shared projection from the Agent input envelope into a user Message.

use serde_json::{Map, Value};

use crate::{
    agent::Input,
    components::message::{Message, Role},
    components::model::content::part::Part,
};

pub(crate) fn user_message(input: Input) -> Message {
    let content = input
        .payload
        .into_iter()
        .map(|payload| {
            let mut data = match payload.data {
                Value::Object(object) => object,
                value => Map::from_iter([(String::from("value"), value)]),
            };
            if let Some(mime_type) = payload.mime_type {
                data.insert("mime_type".into(), Value::String(mime_type));
            }
            Part {
                r#type: payload.r#type,
                data,
            }
        })
        .collect();
    let mut message = Message {
        role: Role::User,
        content,
        values: Map::new(),
        metadata: input.metadata,
        usage: None,
    };
    for key in ["chat_id", "node_id", "exec_id", "reply_id"] {
        message.metadata.remove(key);
    }
    message
}