agentq 0.1.0

An embeddable job queue with idempotency keys, per-lane backpressure and bounded concurrency, for agent tool calls.
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

use std::{
    pin::Pin,
    task::{Context, Poll},
};

use tokio::sync::oneshot;

use crate::error::JobLost;

#[derive(Debug, Clone)]
pub enum Outcome {
    Completed { output: String },
    Failed { reason: String },
}

#[derive(Debug)]
pub struct JobHandle {
    receiver: oneshot::Receiver<Outcome>,
}

impl JobHandle {
    pub(crate) fn new(receiver: oneshot::Receiver<Outcome>) -> Self {
        JobHandle { receiver }
    }
}

impl Future for JobHandle {
    type Output = Result<Outcome, JobLost>;

    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        Pin::new(&mut self.receiver)
            .poll(cx)
            .map(|result| result.map_err(|_| JobLost))
    }
}