rho-coding-agent 0.28.1

A lightweight agent harness inspired by Pi
Documentation 
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 tokio::sync::watch;

/// Cooperative cancellation shared by one agent run and its nested operations.
#[derive(Clone, Debug)]
pub struct RunCancellation {
    cancelled: watch::Receiver<bool>,
    cancel: watch::Sender<bool>,
}

impl Default for RunCancellation {
    fn default() -> Self {
        let (cancel, cancelled) = watch::channel(false);
        Self { cancelled, cancel }
    }
}

impl RunCancellation {
    pub fn cancel(&self) {
        self.cancel.send_replace(true);
    }

    pub fn is_cancelled(&self) -> bool {
        *self.cancelled.borrow()
    }

    pub async fn cancelled(&self) {
        let mut cancelled = self.cancelled.clone();
        if *cancelled.borrow_and_update() {
            return;
        }
        while cancelled.changed().await.is_ok() {
            if *cancelled.borrow_and_update() {
                return;
            }
        }
    }
}

#[cfg(test)]
#[path = "cancellation_tests.rs"]
mod tests;