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

#[cfg(feature = "candle")]
mod bert;
#[cfg(feature = "candle")]
pub use self::bert::*;
mod trigger_with_user_sentence;
pub use self::trigger_with_user_sentence::*;

use beet_core::prelude::*;
use beet_flow::prelude::*;
use std::borrow::Cow;

#[derive(Default)]
pub struct LanguagePlugin;

impl Plugin for LanguagePlugin {
	fn build(&self, app: &mut App) {
		app.register_type::<Sentence>()
			.add_observer(trigger_with_user_sentence::<GetOutcome>);

		#[cfg(feature = "candle")]
		app.init_asset::<Bert>().init_asset_loader::<BertLoader>();

		let world = app.world_mut();
		world.register_component::<Sentence>();
	}
}


/// This component is for use with [`SentenceFlow`]. Add to either the agent or a child behavior.
#[derive(Debug, Clone, Component, PartialEq, Reflect)]
#[reflect(Component)]
pub struct Sentence(pub Cow<'static, str>);
impl Sentence {
	pub fn new(s: impl Into<Cow<'static, str>>) -> Self { Self(s.into()) }
}

impl Default for Sentence {
	fn default() -> Self { Self::new("placeholder") }
}