rig-ecs 0.44.0

rig inside a Bevy World: effects as entities, handlers as entities, the driver as a system.
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

//! `examples/agent_with_tools` side by side: an agent with two tools
//! answers "Calculate 2 - 5." — there the builder's `dynamic_tools`, here
//! a `Grant` link per tool entity and a run entity. The model is scripted
//! (`support`), so no key.

#![allow(
    clippy::expect_used,
    clippy::unwrap_used,
    clippy::indexing_slicing,
    clippy::panic,
    reason = "an example: user code, thirty lines, a mock behind it"
)]

mod support;

use bevy_app::Startup;
use bevy_ecs::prelude::*;
use rig_core::message::AssistantContent;
use rig_ecs::{bus::Handlers, prelude::*, systems::RunCommands};

const PREAMBLE: &str = "You are a calculator here to help the user perform arithmetic operations. \
     You must use the provided tools before answering.";

fn main() {
    support::app()
        .add_systems(Startup, ask)
        .add_observer(support::print_the_answer_and_exit)
        .run();
}

fn ask(mut handlers: Handlers, mut commands: Commands) {
    let model = support::Scripted::new(vec![
        vec![support::call(
            "subtract",
            serde_json::json!({"x": 2, "y": 5}),
        )],
        vec![AssistantContent::text("-3")],
    ]);
    let tools = vec![support::add(), support::subtract()];
    let (model, tools) = support::register(&mut handlers, model, tools);
    let agent = support::agent(&mut commands, model, PREAMBLE, 2);
    for tool in tools {
        commands.spawn((Grant(tool), ChildOf(agent)));
    }
    commands.queue(move |world: &mut World| {
        world.spawn_run(agent, &[], "Calculate 2 - 5.", false, None);
    });
}