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

use crate::Result;
use crate::event::{OneShotRx, OneShotTx, Tx, new_one_shot_channel};
use crate::run::{RunSubAgentParams, RunTopAgentParams};
use derive_more::From;

#[derive(Debug)]
pub struct RunQueueMessage {
	pub done_tx: RunDoneTx,
	pub action: RunQueueAction,
}

impl RunQueueMessage {
	pub fn new_and_rx(event: impl Into<RunQueueAction>) -> (Self, OneShotRx<()>) {
		let (tx, rx) = new_one_shot_channel::<()>("run_one_shot");
		let evt = Self {
			done_tx: tx.into(),
			action: event.into(),
		};
		(evt, rx)
	}
}

#[derive(Debug, From)]
#[allow(clippy::large_enum_variant)]
pub enum RunQueueAction {
	RunTopAgent(RunTopAgentParams),
	RunSubAgent(RunSubAgentParams),
}

// region:    --- QueueTrx

#[derive(Debug, Clone, From)]
pub struct RunDoneTx(OneShotTx<()>);

impl RunDoneTx {
	pub async fn send(self, t: ()) -> Result<()> {
		self.0.send(()).await
	}
}

// endregion: --- QueueTrx