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

use crate::Result;
use crate::event::{OneShotRx, OneShotTx, new_one_shot_channel};
use tokio::io::{self, AsyncBufReadExt, AsyncWriteExt, BufReader};

// region:    --- Types

#[derive(Debug)]
pub struct PromptParams {
	pub message: String,
	pub one_shot_res: OneShotTx<String>,
}

impl PromptParams {
	pub fn new(message: impl Into<String>) -> (Self, OneShotRx<String>) {
		let message = message.into();
		let (tx, rx) = new_one_shot_channel::<String>("prompt-param-one-shot");
		(
			Self {
				message,
				one_shot_res: tx,
			},
			rx,
		)
	}
}

// endregion: --- Types

pub async fn prompt(param: PromptParams) -> Result<()> {
	let PromptParams { message, one_shot_res } = param;

	let mut stdout = io::stdout();
	let mut stdin = BufReader::new(io::stdin());
	let mut input = String::new();

	stdout.write_all(message.as_bytes()).await?;
	stdout.flush().await?;

	stdin.read_line(&mut input).await?;

	one_shot_res.send(input).await?;

	Ok(())
}