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

use super::{
	aip_truncate,
	ensure,
	ensure_single_trailing_newline,
	// text_common.rs
	escape_decode,
	escape_decode_if_needed,
	extract_line_blocks,
	// text_formatter.rs
	format_size,
	remove_first_line,
	remove_first_lines,
	remove_last_line,
	remove_last_lines,
	replace_markers_with_default_parkers,
	// text_split.rs
	split_first,
	// text_split_lines.rs
	split_first_line,
	split_last,
	split_last_line,
	// text_trim.rs
	trim,
	trim_end,
	trim_start,
};
use crate::Result;
use crate::runtime::Runtime;
use mlua::{Lua, Table};

pub fn init_module(lua: &Lua, _runtime: &Runtime) -> Result<Table> {
	let table = lua.create_table()?;

	// --- Functions from text_common.rs
	table.set("escape_decode", lua.create_function(escape_decode)?)?;
	table.set("escape_decode_if_needed", lua.create_function(escape_decode_if_needed)?)?;
	table.set("remove_first_line", lua.create_function(remove_first_line)?)?;
	table.set("remove_first_lines", lua.create_function(remove_first_lines)?)?;
	table.set("remove_last_lines", lua.create_function(remove_last_lines)?)?;
	table.set("remove_last_line", lua.create_function(remove_last_line)?)?;
	table.set("truncate", lua.create_function(aip_truncate)?)?;
	table.set(
		"replace_markers",
		lua.create_function(replace_markers_with_default_parkers)?,
	)?;
	table.set("extract_line_blocks", lua.create_function(extract_line_blocks)?)?;
	table.set("ensure", lua.create_function(ensure)?)?;

	table.set(
		"ensure_single_trailing_newline",
		lua.create_function(ensure_single_trailing_newline)?,
	)?;
	// deprecated (use ensure_single_trailing_newline)
	table.set(
		"ensure_single_ending_newline",
		lua.create_function(ensure_single_trailing_newline)?,
	)?;

	// --- Functions from text_split.rs
	table.set("split_first", lua.create_function(split_first)?)?;
	table.set("split_last", lua.create_function(split_last)?)?;

	// --- Functions from text_formatter.rs
	table.set("format_size", lua.create_function(format_size)?)?;

	// --- Functions from text_split_lines.rs
	table.set("split_first_line", lua.create_function(split_first_line)?)?;
	table.set("split_last_line", lua.create_function(split_last_line)?)?;

	// --- Functions from text_trim.rs
	table.set("trim", lua.create_function(trim)?)?;
	table.set("trim_start", lua.create_function(trim_start)?)?;
	table.set("trim_end", lua.create_function(trim_end)?)?;

	Ok(table)
}