Struct MemoryEventStore

Source

pub struct MemoryEventStore { /* private fields */ }

Implementations§

Source §

impl MemoryEventStore

Source

pub fn new() -> Self

Examples found in repository ?

examples/policy_approval.rs (line 9)

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    runtime.register_policy(Policy::require_approval(
12        PolicyId::new("policy_create_node_review"),
13        "Creating graph nodes requires review",
14        PolicyAction::CreateNode,
15    ));
16
17    let result = runtime
18        .create_typed_node(
19            ActorRef::agent("demo"),
20            NodeId::new("sensitive_node"),
21            "memory",
22            json!({ "title": "Potentially sensitive memory" }),
23        )
24        .await;
25
26    match result {
27        Err(StateError::PolicyDenied(message)) => println!("blocked: {message}"),
28        other => println!("unexpected: {other:?}"),
29    }
30
31    let approvals = state
32        .graph()
33        .await
34        .nodes
35        .values()
36        .filter(|node| node.kind == "approval_request")
37        .count();
38    println!("approval_requests={approvals}");
39    Ok(())
40}

More examples

Hide additional examples

examples/replay_and_fork.rs (line 8)

7async fn main() -> Result<(), Box<dyn std::error::Error>> {
8    let state = YoAgentState::load(MemoryEventStore::new()).await?;
9    let first = state
10        .apply_ops(
11            ActorRef::agent("demo"),
12            vec![StateOp::CreateNode {
13                id: NodeId::new("goal_1"),
14                kind: "goal".to_string(),
15                props: json!({ "title": "Improve retry reliability" }),
16            }],
17        )
18        .await?;
19    state
20        .apply_ops(
21            ActorRef::agent("demo"),
22            vec![StateOp::CreateNode {
23                id: NodeId::new("task_1"),
24                kind: "task".to_string(),
25                props: json!({ "title": "Investigate timeout" }),
26            }],
27        )
28        .await?;
29
30    let fork = state
31        .fork_at_event(ForkId::new("fork_before_task"), Some(first))
32        .await?;
33    let diff = diff_graphs(&fork.graph, &state.graph().await);
34    println!("fork_nodes={}", fork.graph.nodes.len());
35    println!("current_nodes={}", state.graph().await.nodes.len());
36    println!("added_nodes={:?}", diff.added_nodes);
37    Ok(())
38}

examples/typed_pack.rs (line 9)

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    runtime.register_pack(
12        Pack::new(PackId::new("pack_lineage"), "lineage", "0.1.0")
13            .add_object_type(ObjectType::new("goal").require("title"))
14            .add_object_type(ObjectType::new("task").require("title"))
15            .add_relation_type(
16                RelationType::new("serves")
17                    .from_kind("task")
18                    .to_kind("goal"),
19            ),
20    );
21
22    runtime
23        .create_typed_node(
24            ActorRef::agent("demo"),
25            NodeId::new("goal_1"),
26            "goal",
27            json!({ "title": "Improve retry reliability" }),
28        )
29        .await?;
30    runtime
31        .create_typed_node(
32            ActorRef::agent("demo"),
33            NodeId::new("task_1"),
34            "task",
35            json!({ "title": "Investigate timeout" }),
36        )
37        .await?;
38    runtime
39        .create_typed_relation(
40            ActorRef::agent("demo"),
41            NodeId::new("task_1"),
42            "serves",
43            NodeId::new("goal_1"),
44            json!({}),
45        )
46        .await?;
47
48    println!("{}", serde_json::to_string_pretty(&state.graph().await)?);
49    Ok(())
50}

examples/behavior_subscription.rs (line 9)

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}

examples/basic_lineage.rs (line 6)

5async fn main() -> Result<(), Box<dyn std::error::Error>> {
6    let state = YoAgentState::load(MemoryEventStore::new()).await?;
7    let actor = ActorRef::agent("example");
8
9    let failure = NodeId::new("failure_retry_timeout");
10    let hypothesis = NodeId::new("hypothesis_retry_state_lost");
11
12    state
13        .apply_ops(
14            actor,
15            vec![
16                StateOp::CreateNode {
17                    id: failure.clone(),
18                    kind: "failure".to_string(),
19                    props: json!({
20                        "title": "Retry state is lost after timeout",
21                        "summary": "The next retry starts from attempt zero."
22                    }),
23                },
24                StateOp::CreateNode {
25                    id: hypothesis.clone(),
26                    kind: "hypothesis".to_string(),
27                    props: json!({
28                        "title": "Attempt count is scoped to the cancelled future"
29                    }),
30                },
31                StateOp::CreateRelation {
32                    from: hypothesis.clone(),
33                    rel: "explains".to_string(),
34                    to: failure,
35                    props: json!({}),
36                },
37            ],
38        )
39        .await?;
40
41    print!("{}", state.lineage(hypothesis).await.to_markdown());
42    Ok(())
43}

examples/yoagent_integration.rs (line 10)

9async fn main() -> Result<(), Box<dyn std::error::Error>> {
10    let state = YoAgentState::load(MemoryEventStore::new()).await?;
11    let sink = YoAgentStateAdapter::new(state.clone(), ActorRef::agent("yoagent"));
12    let run_id = RunId::new("run_demo");
13
14    sink.on_run_started(YoAgentRunStarted {
15        run_id: run_id.clone(),
16        task: "Investigate retry timeout".to_string(),
17        metadata: json!({}),
18    })
19    .await?;
20    sink.on_model_called(YoAgentModelCalled {
21        run_id: run_id.clone(),
22        model: "example-model".to_string(),
23        prompt_summary: "Find likely retry failure cause".to_string(),
24    })
25    .await?;
26    sink.on_model_finished(YoAgentModelFinished {
27        run_id: run_id.clone(),
28        model: "example-model".to_string(),
29        output_summary: "Retry state appears scoped too narrowly".to_string(),
30    })
31    .await?;
32    sink.on_tool_called(YoAgentToolCalled {
33        run_id: run_id.clone(),
34        tool: "cargo test".to_string(),
35        input_summary: "tool_retry_survives_timeout".to_string(),
36    })
37    .await?;
38    sink.on_tool_finished(YoAgentToolFinished {
39        run_id: run_id.clone(),
40        tool: "cargo test".to_string(),
41        output_summary: "test passed".to_string(),
42        success: true,
43    })
44    .await?;
45    sink.on_run_finished(YoAgentRunFinished {
46        run_id,
47        outcome: "patch ready for review".to_string(),
48        metadata: json!({}),
49    })
50    .await?;
51
52    println!(
53        "{}",
54        serde_json::to_string_pretty(&state.store().scan().await?)?
55    );
56    Ok(())
57}