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

//! Test-only collection of the new run stream followed by reading saved results.

use std::sync::Arc;

use futures_util::StreamExt;
use halo::agent::{Cancellation, Error, Messages};
use halo::runtime::workflow::{Runtime, graph, node};

pub async fn run_workflow(
    runtime: &mut Runtime,
    initial: Messages,
    next_id: Arc<dyn Fn() -> i64 + Send + Sync>,
    cancellation: Cancellation,
) -> Result<Messages, Error> {
    {
        let mut events = runtime.run(initial, next_id, cancellation).await?;
        while let Some(event) = events.next().await {
            event?;
        }
    }
    Ok(Messages::new(
        runtime
            .state()
            .nodes
            .values()
            .flat_map(|node| node.executions.values())
            .filter(|execution| {
                execution.status == node::ExecutionStatus::Succeeded
                    && execution.to.iter().any(|to| to == graph::END)
            })
            .flat_map(|execution| execution.messages.as_slice().iter().cloned()),
    ))
}