beet_agent 0.0.8

ECS agentic workflow patterns
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
90
91
92
93
94
95

//! Helpers for building a session, sessions are expected to
//! have the following hierarchy
//! ```
//! Session
//! 	Actor1
//! 		Message1
//! 			TextContent
//! 			FileContent
//! 		Message2
//! 			TextContent
//! 	Actor2
//! 		Message1
//! 			TextContent
//! 			..
//! ```
//!
use crate::prelude::Message;
use crate::prelude::*;
use beet_core::prelude::*;
use std::path::Path;

pub async fn workspace_file(path: impl AsRef<Path>) -> Result<FileContent> {
	session_ext::file(AbsPathBuf::new_workspace_rel(path)?.to_string()).await
}

pub async fn file(path: impl AsRef<str>) -> Result<FileContent> {
	FileContent::new(path).await
}

/// Add a *completed* piece of text content, where no more
/// text will be added to this piece of content.
pub fn text(text: impl AsRef<str>) -> TextContent { TextContent::new(text) }

/// Create a [`Message`] with the given text and file content as children
pub fn message<M>(content: impl IntoContentVec<M>) -> impl Bundle {
	(Message::default(), content.into_content_vec().into_bundle())
}

/// Create a session, inserting the `agent` and `user_message`,
/// then triggering `MessageRequest` on the agent
#[rustfmt::skip]
pub fn user_message_session(agent:impl Bundle, user_message:impl Bundle)->impl Bundle {
	(
		Session::default(),
		children![
			(
				UserRole,
				children![user_message],
			),
			(
				agent,
				OnSpawn::trigger(MessageRequest)
			)
		]
	)
}


/// Helper to run the provided session, then collects the first
/// emitted file
/// ## Panics
/// Panics if no Agent entity is found
pub async fn run_and_collect_file(session: impl Bundle) -> Vec<ContentEnum> {
	let mut app = App::new();
	app.add_plugins((MinimalPlugins, AgentPlugin));
	app.world_mut().spawn(session);

	todo!("beet_flow");
	// app.world_mut().run_system_cached(collect_output).unwrap()
}

/// Collect all TextContent and FileContent found under the Agent entity
/// ## Panics
/// Panics if no Agent entity is found
pub fn collect_output(
	agents: Query<Entity, With<AgentRole>>,
	children: Query<&Children>,
	text: Query<&TextContent>,
	files: Query<&FileContent>,
) -> Vec<ContentEnum> {
	let agent = agents.single().expect("No agent found in world");

	children
		.iter_descendants_depth_first(agent)
		.filter_map(|entity| {
			if let Some(text) = text.get(entity).ok() {
				Some(ContentEnum::Text(text.clone()))
			} else if let Some(file) = files.get(entity).ok() {
				Some(ContentEnum::File(file.clone()))
			} else {
				None
			}
		})
		.collect()
}