aipack 0.8.24

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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use crate::_test_support::assert_contains;
use crate::agent::Agent;

pub type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>;

#[tokio::test]
async fn test_agent_parse_basic() -> Result<()> {
	// -- Setup & Fixtures
	let content = r#"
# Data 
```lua
-- Some lua data
```
# System

System 1
```
Some system block 1
## Heading 2 System 1
```

# System 
System 2

# User

User 1

# Output

```lua
return "hello"
```
		"#;

	// -- Exec
	let agent = Agent::mock_from_content(content)?;

	// -- Check
	let parts = agent.prompt_parts();
	assert_eq!(parts.len(), 3);
	assert_contains(&parts[0].content, "System 1");
	assert_contains(&parts[0].content, "```");
	assert_contains(&parts[0].content, "Some system block 1");
	assert_contains(&parts[0].content, "## Heading 2 System 1");
	assert_contains(&parts[1].content, "System 2");
	assert_contains(&parts[2].content, "User 1");
	assert_contains(
		agent.data_script().expect("Should have data script"),
		"-- Some lua data",
	);

	Ok(())
}

#[tokio::test]
async fn test_agent_parse_part_block_with_level_1_heading() -> Result<()> {
	// -- Setup & Fixtures
	let content = r#"
# System

System 1
```
Some system block 1
# Heading 1 System 1
```

# Output

```lua
return "hello"
```
		"#;

	// -- Exec
	let agent = Agent::mock_from_content(content)?;

	// -- Check
	let parts = agent.prompt_parts();
	assert_eq!(parts.len(), 1);
	let output_script = agent.output_script().expect("Should have output script");
	assert_contains(&parts[0].content, "System 1");
	assert_contains(&parts[0].content, "```");
	assert_contains(&parts[0].content, "Some system block 1");
	assert_contains(&parts[0].content, "# Heading 1 System 1");
	assert_contains(output_script, "return \"hello\"");

	Ok(())
}