aipack 0.7.7-WIP

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

use crate::_test_support::default_agent_config_for_test;
use crate::Result;
use crate::agent::{Agent, AgentDoc, AgentRef, find_agent};
use crate::runtime::Runtime;

/// Load a Agent form a content.
/// - `path` is just to be used as a path for the agent. Not used to load the content.
pub fn load_inline_agent(path: &str, content: impl Into<String>) -> Result<Agent> {
	let doc = AgentDoc::from_content(path, content)?;
	let agent_ref = AgentRef::LocalPath(path.to_string());
	let agent = doc.into_agent("inline-agent", agent_ref, default_agent_config_for_test())?;
	Ok(agent)
}

pub fn load_test_agent(name: &str, runtime: &Runtime) -> Result<Agent> {
	find_agent(name, runtime.dir_context(), None)
}

/// Will create and agent where the `# Output` refect the return of the `# Data`
/// Used to test Lua module functions
///
/// NOTE: This is a in memory agent, with a fake path
pub fn load_reflective_agent(data_lua_code: &str) -> Result<Agent> {
	load_inline_agent(
		"mock-reflective-agent.aip",
		format!(
			r#"
# Data
```lua
{data_lua_code}
```
# Output
```lua
return data
```
	"#
		),
	)
}