crb-superagent 0.0.30

CRB | Composable Runtime Blocks | Agent Extensions
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
42
43
44
45
46
47
48
49
50

use super::runtime::RunMission;
use super::{Mission, Observer};
use anyhow::Result;
use async_trait::async_trait;
use crb_agent::{Address, Agent, Context, MessageFor, ToAddress};

impl<M: Mission> RunMission<M> {
    pub fn report_to<S>(&mut self, address: impl ToAddress<S>)
    where
        S: Finished<M>,
        M::Goal: Clone,
    {
        let address = address.to_address();
        let observer = Box::new(address);
        self.observers.push(observer);
    }
}

impl<S, M> Observer<M> for Address<S>
where
    S: Finished<M>,
    M: Mission,
    M::Goal: Clone,
{
    fn check(&mut self, output: &M::Goal) -> Result<()> {
        let output = output.clone();
        let event = FinishedEvent { output };
        self.send(event)
    }
}

#[async_trait]
pub trait Finished<M: Mission>: Agent {
    async fn handle(&mut self, output: M::Goal, ctx: &mut Context<Self>) -> Result<()>;
}

struct FinishedEvent<M: Mission> {
    output: M::Goal,
}

#[async_trait]
impl<S, M> MessageFor<S> for FinishedEvent<M>
where
    S: Finished<M>,
    M: Mission,
{
    async fn handle(self: Box<Self>, agent: &mut S, ctx: &mut Context<S>) -> Result<()> {
        agent.handle(self.output, ctx).await
    }
}