beet_ml 0.0.8

Machine Learning actions built upon beet_flow
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use crate::prelude::*;
use beet_core::prelude::*;
use beet_flow::prelude::*;

/// Runs the child with the [`Sentence`] that is most similar to that of the agent.
/// for use with [`ScoreFlow`]
#[action(nearest_sentence)]
#[derive(Debug, Default, Clone, PartialEq, Component, Reflect)]
#[reflect(Component)]
// #[require(Sentence=||Sentence::new("placeholder"))]
pub struct NearestSentence;

impl NearestSentence {
	pub fn new() -> Self { Self {} }
}

fn nearest_sentence(
	ev: On<GetOutcome>,
	mut commands: Commands,
	mut berts: ResMut<Assets<Bert>>,
	sentences: Query<&Sentence>,
	// TODO double query, ie added running and added asset
	query: Query<(
		&NearestSentence,
		&Sentence,
		&HandleWrapper<Bert>,
		&Children,
	)>,
) -> Result {
	let (_scorer, target_sentence, handle, children) =
		query.get(ev.action())?;
	let bert = berts
		.get_mut(handle)
		.expect(&expect_action::to_have_asset(&ev));
	let entity = bert.closest_sentence_entity(
		target_sentence.0.clone(),
		children.iter().map(|e| e.clone()),
		&sentences,
	)?;
	commands.entity(entity).trigger_target(GetOutcome);
	Ok(())
}

#[cfg(test)]
mod test {
	use crate::prelude::*;
	use beet_core::prelude::*;
	use beet_flow::prelude::*;

	#[test]
	#[ignore = "we need to get sentence_flow back up and running, ie the req/res model like score"]
	fn works() {
		pretty_env_logger::try_init().ok();

		let mut app = App::new();
		app.add_plugins((
			MinimalPlugins,
			workspace_asset_plugin(),
			LanguagePlugin::default(),
			ControlFlowPlugin::default(),
		))
		.finish();
		let on_run =
			observer_ext::observe_trigger_names::<GetOutcome>(app.world_mut());

		let handle =
			block_on_asset_load::<Bert>(&mut app, "ml/default-bert.ron")
				.unwrap();

		app.world_mut()
			.spawn((
				Name::new("root"),
				Sentence::new("destroy"),
				HandleWrapper(handle),
				NearestSentence::default(),
			))
			.with_children(|parent| {
				parent.spawn((Name::new("heal"), Sentence::new("heal")));
				parent.spawn((Name::new("kill"), Sentence::new("kill")));
			})
			.trigger_target(GetOutcome)
			.flush();


		on_run.len().xpect_eq(2);
		on_run.get_index(0).xpect_eq(Some("root".to_string()));
		on_run.get_index(1).xpect_eq(Some("kill".to_string()));
	}
}