aipack 0.8.22

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

//! Defines the `run` helpers for Lua scripts.
//!
//! ---
//!
//! ## Lua documentation
//!
//! This module adds helper functions for updating run properties at runtime.
//!
//! ### Functions
//!
//! - `aip.run.set_label(label: string)`  
//! - `aip.run.pin(iden: string, content: string | {label?: string, content: string})`  
//! - `aip.run.pin(iden: string, priority: number, content: string | {label?: string, content: string})`  
//!   

use crate::Result;
use crate::model::{RunBmc, RunForUpdate, RuntimeCtx};
use crate::runtime::Runtime;
use crate::script::LuaValueExt;
use crate::script::support::create_pin;
use mlua::{Lua, Table, Value, Variadic};

/// Registers the `run.set_label` and `run.pin` helpers in Lua.
pub fn init_module(lua: &Lua, runtime: &Runtime) -> Result<Table> {
	let table = lua.create_table()?;

	// -- run.set_label
	{
		let rt = runtime.clone();
		let set_label_fn = lua
			.create_function(move |lua, label: Value| set_run_label(lua, &rt, label).map_err(mlua::Error::external))?;
		table.set("set_label", set_label_fn)?;
	}

	// -- run.pin
	{
		let rt = runtime.clone();
		let run_pin_fn = lua.create_function(move |lua, args: Variadic<Value>| {
			create_pin(lua, &rt, /*for_task*/ false, args).map_err(mlua::Error::external)
		})?;
		table.set("pin", run_pin_fn)?;
	}

	Ok(table)
}

fn set_run_label(lua: &Lua, runtime: &Runtime, label: Value) -> Result<()> {
	let label = label
		.x_as_lua_str()
		.ok_or("aip.run.set_label(label) – expected <string> for parameter `label`.")?
		.to_string();

	let ctx = RuntimeCtx::extract_from_global(lua)?;
	let mm = runtime.mm();

	let run_id = ctx
		.get_run_id(mm)?
		.ok_or("Cannot call 'aip.run.set_label(...)' outside of a run context.")?;

	let run_u = RunForUpdate {
		label: Some(label),
		..Default::default()
	};

	RunBmc::update(mm, run_id, run_u)?;

	Ok(())
}

// region:    --- Tests

#[cfg(test)]
mod tests {
	type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>;

	use crate::_test_support::run_reflective_agent_with_runtime;
	use crate::model::{PinBmc, RunBmc};
	use crate::runtime::Runtime;

	#[tokio::test(flavor = "multi_thread")]
	async fn test_lua_run_set_label_simple() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01().await?;
		let fx_code = r#"
aip.run.set_label("My Run Label")
return "OK"
		"#;

		// -- Exec
		let res = run_reflective_agent_with_runtime(fx_code, None, runtime.clone()).await?;

		// -- Check
		assert_eq!(res.as_str().unwrap_or_default(), "OK");
		// check run label was updated
		let runs = RunBmc::list(runtime.mm(), None)?;
		assert_eq!(runs.len(), 1);
		assert_eq!(runs[0].label, Some("My Run Label".to_string()));

		Ok(())
	}

	#[tokio::test(flavor = "multi_thread")]
	async fn test_lua_run_pin_simple() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01().await?;
		let fx_code = r#"
aip.run.pin("some-iden", "Some pin content")		
return "OK"
			"#;

		// -- Exec
		let res = run_reflective_agent_with_runtime(fx_code, None, runtime.clone()).await?;

		// -- Check
		assert_eq!(res.as_str().unwrap_or_default(), "OK");
		// check pins
		let pins = PinBmc::list_for_run(runtime.mm(), 0.into())?;
		assert_eq!(pins.len(), 1);
		assert_eq!(pins[0].content.as_deref().unwrap_or_default(), "\"Some pin content\"");

		Ok(())
	}

	#[tokio::test(flavor = "multi_thread")]
	async fn test_lua_run_pin_with_priority() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01().await?;
		let fx_code = r#"
aip.run.pin("some-iden", 0.7, "Other content")
return "OK"
		"#;

		// -- Exec
		let res = run_reflective_agent_with_runtime(fx_code, None, runtime.clone()).await?;

		// -- Check
		assert_eq!(res.as_str().unwrap_or_default(), "OK");
		let pins = PinBmc::list_for_run(runtime.mm(), 0.into())?;
		assert_eq!(pins.len(), 1);
		assert_eq!(pins[0].priority, Some(0.7));

		Ok(())
	}
}

// endregion: --- Tests