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

use crate::types::MdHeading;
use mlua::IntoLua;
use serde::{Serialize, Serializer};

#[derive(Debug)]
pub struct MdSection {
	pub content: String,
	pub heading: Option<MdHeading>,
}

/// Constructors
/// For now, constructe by
#[allow(unused)]
impl MdSection {
	pub fn from_content(content: impl Into<String>) -> Self {
		Self {
			content: content.into(),
			heading: None,
		}
	}
	pub fn new(content: String, heading: impl Into<Option<MdHeading>>) -> Self {
		Self {
			content,
			heading: heading.into(),
		}
	}
}

/// Getters
impl MdSection {
	/// The content of the section
	/// NOTE: The eventual end newline for this section in the markdown is not included in the content
	#[allow(unused)]
	pub fn content(&self) -> &str {
		&self.content
	}

	#[allow(unused)]
	pub fn heading(&self) -> Option<&MdHeading> {
		self.heading.as_ref()
	}

	// Convenient function that returnes the heading content
	// - "" empty string if not present
	// - "{content}\n" if present (adding the new line)
	pub fn heading_raw(&self) -> String {
		self.heading
			.as_ref()
			.map_or_else(|| "".to_string(), |h| h.content().to_string() + "\n")
	}
}

// region:    --- Serde Serializer

impl Serialize for MdSection {
	fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer,
	{
		use serde::ser::SerializeStruct;
		// Max 3 fields (content, heading, _type)
		let mut state = serializer.serialize_struct("MdSection", 3)?;
		state.serialize_field("_type", "MdSection")?;

		state.serialize_field("content", &self.content)?;
		if let Some(heading) = &self.heading {
			state.serialize_field("heading", heading)?;
		}

		state.end()
	}
}

// endregion: --- Serde Serializer

// region:    --- Lua

impl IntoLua for MdSection {
	fn into_lua(self, lua: &mlua::Lua) -> mlua::Result<mlua::Value> {
		let table = lua.create_table()?;
		table.set("_type", "MdSection")?;

		let heading_raw = self.heading_raw();

		table.set("content", self.content)?;
		table.set("heading_raw", heading_raw)?;
		if let Some(heading) = self.heading {
			table.set("heading_content", heading.content())?;
			table.set("heading_level", heading.level())?;
			table.set("heading_name", heading.name())?;
		} else {
			table.set("heading_content", "")?;
			table.set("heading_level", 0)?;
			table.set("heading_name", "")?;
		}
		Ok(mlua::Value::Table(table))
	}
}

// endregion: --- Lua