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

use crate::event::{Rx, Tx, new_channel};
use crate::hub::get_hub;
use crate::run::run_executor::RunQueueMessage;
use crate::run::run_executor::run_queue_event::RunQueueAction;
use crate::runtime::Runtime;
use crate::{Error, Result};
use derive_more::{Deref, From};

#[derive(Debug, Clone, From, Deref)]
pub struct RunQueueTx(Tx<RunQueueMessage>);

pub struct RunQueueExecutor {
	rx: Rx<RunQueueMessage>,
	_tx: RunQueueTx,
}

impl RunQueueExecutor {
	pub fn new() -> Self {
		let (tx, rx) = new_channel::<RunQueueMessage>("run_queue_executor");
		Self { rx, _tx: tx.into() }
	}

	/// Consume the key, start it.
	/// NOTE: Make sure to keep at least one ref of the returned tx
	pub fn start(self) -> RunQueueTx {
		tokio::spawn(async move {
			let hub = get_hub();
			loop {
				let mut msg = match self.rx.recv().await {
					Ok(msg) => msg,
					Err(err) => {
						hub.publish(Error::cc("Fail in RunQueueExecutor recv", err));
						continue;
					}
				};

				let tx = msg.done_tx;
				tx.send(()).await;
			}
		});

		self._tx.clone()
	}
}