beet 0.0.8

Malleable Application Framework
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

//! An example demonstrating high level image generation,
//! see the helper methods in `session_ext` for more detailed usage
use beet::prelude::*;

pub fn main() {
	// 1. construct the session
	let agent = GeminiAgent::from_env().with_model(GEMINI_2_5_FLASH_IMAGE);
	let message = session_ext::message(
		"image of game engine editor being used to control fun little robot characters in real life",
	);
	let session = session_ext::user_message_session(agent, message);

	// 2. run and collect output
	for (index, content) in session_ext::run_and_collect_file(session)
		.await
		.into_iter()
		.enumerate()
	{
		// 3. print and save output
		match content {
			ContentEnum::Text(text) => println!("Agent > {text}"),
			ContentEnum::File(file) => {
				let path = AbsPathBuf::new_workspace_rel(format!(
					"target/examples/image/file-{}-{index}.png",
					time_ext::now_millis()
				))
				.unwrap();
				println!("Agent > File: {}", path.display());
				let data = file.data.get().await.unwrap();
				fs_ext::write(&path, data).unwrap();
			}
		}
	}
}