aipack 0.8.24

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

use crate::agent::Agent;
use crate::model::{RuntimeCtx, Stage};
use crate::run::run_agent_task::RunAgentInputResponse;
use crate::run::{AiResponse, Literals};
use crate::runtime::Runtime;
use crate::script::{AipackCustom, FromValue};
use crate::{Error, Result};
use serde_json::Value;

#[allow(clippy::too_many_arguments)]
pub async fn process_output(
	runtime: &Runtime,
	base_rt_ctx: &RuntimeCtx,
	agent: Agent,
	literals: &Literals,
	data: Value,
	before_all: Value,
	input: Value,
	ai_response: Option<AiResponse>,
) -> Result<Option<RunAgentInputResponse>> {
	if let Some(output_script) = agent.output_script() {
		// -- Create the Output Lua Engine
		let lua_engine = runtime.new_lua_engine_with_ctx(literals, base_rt_ctx.with_stage(Stage::Output))?;

		// -- Create the scope
		let lua_scope = lua_engine.create_table()?;
		lua_scope.set("input", lua_engine.serde_to_lua_value(input)?)?;
		lua_scope.set("data", lua_engine.serde_to_lua_value(data)?)?;
		lua_scope.set("before_all", lua_engine.serde_to_lua_value(before_all)?)?;
		lua_scope.set("ai_response", ai_response)?;
		lua_scope.set("options", agent.options_as_ref())?;

		let lua_value = lua_engine
			.eval_with_paths(output_script, Some(lua_scope), agent.context_dirs())
			.await?;
		let output_response = serde_json::to_value(lua_value)?;

		if let Ok(FromValue::AipackCustom(AipackCustom::Redo)) = AipackCustom::from_value(output_response.clone()) {
			return Err(Error::custom(
				"aip.flow.redo_run() can be returned only from # Before All or # After All stages. Returned from # Output stage.",
			));
		}

		Ok(Some(RunAgentInputResponse::OutputResponse(output_response)))
	} else {
		Ok(ai_response.map(RunAgentInputResponse::AiReponse))
	}
}