aipack 0.8.23

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

//! The before all processor

use crate::Result;
use crate::agent::Agent;
use crate::model::{Id, RuntimeCtx, Stage};
use crate::run::Literals;
use crate::runtime::Runtime;
use serde_json::Value;

// region:    --- Types

pub struct ProcAfterAllResponse {
	pub after_all: Option<Value>,
	pub outputs: Option<Vec<Value>>,
}

// endregion: --- Types

#[allow(clippy::too_many_arguments)]
pub async fn process_after_all(
	runtime: &Runtime,
	base_rt_ctx: RuntimeCtx,
	run_id: Id,
	agent: &Agent,
	literals: Literals,
	before_all: Value,
	inputs: Option<Vec<Value>>,
	outputs: Option<Vec<Value>>,
) -> Result<ProcAfterAllResponse> {
	let rt_step = runtime.rt_step();

	if let Some(after_all_script) = agent.after_all_script() {
		let outputs_value = if let Some(outputs) = outputs.as_ref() {
			Value::Array(outputs.clone())
		} else {
			Value::Null
		};

		let lua_engine = runtime.new_lua_engine_with_ctx(&literals, base_rt_ctx.with_stage(Stage::AfterAll))?;
		let lua_scope = lua_engine.create_table()?;
		let inputs = inputs.map(Value::Array).unwrap_or(Value::Null);
		lua_scope.set("inputs", lua_engine.serde_to_lua_value(inputs)?)?;
		// Will be Value::Null if outputs were not collected
		lua_scope.set("outputs", lua_engine.serde_to_lua_value(outputs_value)?)?;
		lua_scope.set("before_all", lua_engine.serde_to_lua_value(before_all)?)?;
		lua_scope.set("options", agent.options_as_ref())?;

		// -- Rt Step - After All Start
		rt_step.step_aa_start(run_id).await?;

		let lua_value = lua_engine
			.eval_with_paths(after_all_script, Some(lua_scope), agent.context_dirs())
			.await?;

		// -- Rt Step - After All End
		rt_step.step_aa_end(run_id).await?;

		let after_all = Some(serde_json::to_value(lua_value)?);

		Ok(ProcAfterAllResponse { after_all, outputs })
	} else {
		Ok(ProcAfterAllResponse {
			after_all: None,
			outputs,
		})
	}
}