bevy_rig 0.1.0

Bevy ECS primitives and systems for modeling providers, agents, tools, sessions, runs, and workflows on top of Rig.
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

use bevy_ecs::prelude::*;

use crate::{
    run::{Run, RunStatus},
    workflow::{WorkflowInvocation, WorkflowRunStatus},
};

#[derive(Resource, Clone, Debug, Default, PartialEq, Eq)]
pub struct RuntimeDiagnostics {
    pub runs_queued: usize,
    pub runs_running: usize,
    pub runs_completed: usize,
    pub runs_failed: usize,
    pub runs_cancelled: usize,
    pub workflows_queued: usize,
    pub workflows_running: usize,
    pub workflows_completed: usize,
    pub workflows_failed: usize,
}

pub fn refresh_runtime_diagnostics(
    mut diagnostics: ResMut<RuntimeDiagnostics>,
    runs: Query<&RunStatus, With<Run>>,
    workflows: Query<&WorkflowRunStatus, With<WorkflowInvocation>>,
) {
    let mut next = RuntimeDiagnostics::default();

    for status in &runs {
        match status {
            RunStatus::Queued => next.runs_queued += 1,
            RunStatus::Running => next.runs_running += 1,
            RunStatus::Completed => next.runs_completed += 1,
            RunStatus::Failed => next.runs_failed += 1,
            RunStatus::Cancelled => next.runs_cancelled += 1,
        }
    }

    for status in &workflows {
        match status {
            WorkflowRunStatus::Queued => next.workflows_queued += 1,
            WorkflowRunStatus::Running => next.workflows_running += 1,
            WorkflowRunStatus::Completed => next.workflows_completed += 1,
            WorkflowRunStatus::Failed => next.workflows_failed += 1,
        }
    }

    *diagnostics = next;
}