devai 0.5.12

Command Agent runner to accelerate production coding with genai.
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

//! IMPORTANT: THis file run real agents

type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>; // For tests.

use super::*;
use crate::_test_support::{assert_contains, load_test_agent, run_test_agent, run_test_agent_with_input};
use crate::types::FileMeta;
use simple_fs::SPath;
use value_ext::JsonValueExt;

#[tokio::test]
async fn test_run_agent_llm_c_simple_ok() -> Result<()> {
	// -- Setup & Fixtures
	let runtime = Runtime::new_test_runtime_sandbox_01()?;
	let agent = load_test_agent("./agent-llm/agent-simple.devai", &runtime)?;

	// -- Execute
	let res = run_test_agent(&runtime, &agent).await?;

	// -- Check
	assert_contains(res.as_str().ok_or("Should have output result")?, "sky");

	Ok(())
}

#[tokio::test]
async fn test_run_agent_llm_c_on_file_ok() -> Result<()> {
	// -- Setup & Fixtures
	let runtime = Runtime::new_test_runtime_sandbox_01()?;
	let agent = load_test_agent("./agent-llm/agent-on-file.devai", &runtime)?;

	// -- Execute
	let on_file = SPath::new("./other/hello.txt")?;
	let file_meta = FileMeta::from(on_file);

	let res = run_test_agent_with_input(&runtime, &agent, file_meta).await?;

	// -- Check
	// The output return the {data_path: data.file.path, input_name: input.name}
	assert_eq!(res.x_get_str("data_path")?, "./other/hello.txt");
	assert_eq!(res.x_get_str("input_name")?, "hello.txt");
	let ai_content = res.x_get_str("ai_content")?;
	assert!(ai_content.len() > 10, "The AI response should have some content");
	assert_contains(ai_content, "from the other/hello.txt");

	Ok(())
}

#[tokio::test]
async fn test_run_agent_llm_full_chat_ok() -> Result<()> {
	// -- Setup & Fixtures
	let runtime = Runtime::new_test_runtime_sandbox_01()?;
	let agent = load_test_agent("agent-llm/agent-full-chat.devai", &runtime)?;

	// -- Execute
	let res = run_test_agent(&runtime, &agent).await?;

	// -- Check
	let content = res.as_str().ok_or("Should return a string")?;
	// concatinate the first char of each line
	// Because the `agent-full-chat.devai` system instructs to give only 3 bullet points answer.
	let first_chart_of_each_line = content
		.lines()
		.map(|line| line.chars().next().unwrap_or_default())
		.collect::<String>();
	assert_eq!(first_chart_of_each_line, "---");

	Ok(())
}