tirea-protocol-ag-ui 0.5.0

AG-UI protocol event encoding and history adapters for tirea
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 super::{AgUiEventContext, Event};
use tirea_contract::{AgentEvent, Transcoder};

pub struct AgUiProtocolEncoder {
    ctx: AgUiEventContext,
}

impl AgUiProtocolEncoder {
    pub fn new() -> Self {
        Self {
            ctx: AgUiEventContext::new(),
        }
    }

    pub fn new_with_frontend_run_id(run_id: impl Into<String>) -> Self {
        Self {
            ctx: AgUiEventContext::new().with_frontend_run_id(run_id),
        }
    }
}

impl Default for AgUiProtocolEncoder {
    fn default() -> Self {
        Self::new()
    }
}

impl Transcoder for AgUiProtocolEncoder {
    type Input = AgentEvent;
    type Output = Event;

    fn transcode(&mut self, item: &AgentEvent) -> Vec<Event> {
        self.ctx.on_agent_event(item)
    }
}