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
68
69
70

use crate::agent::find_agent;
use crate::run::{RunBaseOptions, RunSubAgentParams, run_agent};
use crate::types::RunAgentResponse;
use crate::{Error, Result};

pub async fn exec_run_sub_agent(params: RunSubAgentParams) -> Result<()> {
	// Normalize inputs to JsonValue format
	let RunSubAgentParams {
		runtime,
		parent_uid,
		agent_dir,
		agent_name,
		inputs,
		agent_options,
		response_shot,
	} = params;

	// -- Find agent and build run base options
	// NOTE: For now, do not pass through the caller baseOptions.
	// NOTE: Might need to find a way to pass it through (perhaps via CTX, or a _aipack_.run_base_options)

	let result: Result<RunAgentResponse> = (async || {
		// find the agent
		let agent = find_agent(&agent_name, &runtime, agent_dir.as_ref())
			.map_err(|e| Error::custom(format!("Failed to find agent '{agent_name}': {e}")))?;

		// -- If we had a agent options, need to overrid the agent options.
		let agent = match agent_options {
			Some(agent_options) => agent.new_merge(agent_options)?,
			None => agent,
		};

		// -- Build the environment
		// NOTE: For now, do not inherit the parent run, But eventually mgith be past in the RunAgentParams
		let run_base_options = RunBaseOptions::default();

		let res = run_agent(&runtime, Some(parent_uid), agent, inputs, &run_base_options, true)
			.await
			.map_err(|e| Error::custom(format!("Failed to run agent '{agent_name}': {e}")))?;
		Ok(res)
	})()
	.await;

	match response_shot {
		Some(response_shot) => {
			match result {
				Ok(result) => {
					if let Err(err) = response_shot.send(Ok(result)).await {
						return Err(Error::custom(format!(
							"Failed to send response to agent '{agent_name}': {err}"
						)));
					}
				}
				Err(err) => {
					// Handle the error case
					if let Err(err) = response_shot.send(Err(Error::custom(err.to_string()))).await {
						return Err(Error::custom(format!(
							"Failed to send response to agent '{agent_name}': {err}"
						)));
					}
				}
			}
		}
		None => {
			result?;
		}
	}

	Ok(())
}