Skip to main content

behavior_subscription/
behavior_subscription.rs

1use serde_json::json;
2use yoagent_state::{
3    ActorRef, BehaviorContext, BehaviorId, Event, EventPattern, FnBehavior, MemoryEventStore,
4    NodeId, StateOp, YoAgentRuntime, YoAgentState,
5};
6
7#[tokio::main]
8async fn main() -> Result<(), Box<dyn std::error::Error>> {
9    let state = YoAgentState::load(MemoryEventStore::new()).await?;
10    let mut runtime = YoAgentRuntime::new(state.clone());
11
12    runtime.register_behavior(FnBehavior::new(
13        BehaviorId::new("behavior_failure_to_task"),
14        EventPattern::Kind("failure.observed".to_string()),
15        |_ctx: BehaviorContext, event: Event| async move {
16            let title = event
17                .payload
18                .get("title")
19                .and_then(|value| value.as_str())
20                .unwrap_or("Investigate failure");
21            Ok(vec![StateOp::CreateNode {
22                id: NodeId::new("task_from_failure"),
23                kind: "task".to_string(),
24                props: json!({
25                    "title": format!("Investigate: {title}"),
26                    "status": "Open",
27                    "created_by_behavior": "behavior_failure_to_task",
28                }),
29            }])
30        },
31    ));
32
33    runtime
34        .emit_event(Event::new(
35            ActorRef::agent("demo"),
36            "failure.observed",
37            json!({ "title": "retry timeout loses state" }),
38        ))
39        .await?;
40
41    print!(
42        "{}",
43        state
44            .lineage(NodeId::new("task_from_failure"))
45            .await
46            .to_markdown()
47    );
48    Ok(())
49}