echo_agent 0.2.0

Production-grade AI Agent framework for Rust — ReAct engine, multi-agent, memory, streaming, MCP, IM channels, workflows
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

//! Criterion benchmarks for echo-agent core operations.
//!
//! Run with: `cargo bench -p echo_agent`

use criterion::{Criterion, black_box, criterion_group, criterion_main};
use echo_agent::prelude::*;
use std::sync::Arc;

/// Benchmark: Agent construction via builder
fn bench_agent_build(c: &mut Criterion) {
    c.bench_function("agent_build", |b| {
        b.iter(|| {
            ReactAgentBuilder::new()
                .model("bench-model")
                .name("bench-agent")
                .system_prompt("You are a benchmark agent")
                .max_iterations(1)
                .build()
                .unwrap()
        })
    });
}

/// Benchmark: Agent construction with tools enabled
fn bench_agent_build_with_tools(c: &mut Criterion) {
    c.bench_function("agent_build_with_tools", |b| {
        b.iter(|| {
            ReactAgentBuilder::new()
                .model("bench-model")
                .name("bench-agent")
                .system_prompt("You are a benchmark agent")
                .max_iterations(1)
                .enable_tools()
                .build()
                .unwrap()
        })
    });
}

/// Benchmark: SharedAgent creation (post-Mutex-removal, lock-free)
fn bench_shared_agent_create(c: &mut Criterion) {
    c.bench_function("shared_agent_create", |b| {
        b.iter(|| {
            let agent = ReactAgentBuilder::new()
                .model("bench")
                .name("bench")
                .system_prompt("p")
                .max_iterations(1)
                .build()
                .unwrap();
            let shared: Arc<dyn echo_agent::agent::Agent> = Arc::new(agent);
            black_box(shared)
        })
    });
}

/// Benchmark: Graph workflow build
fn bench_workflow_build_simple(c: &mut Criterion) {
    c.bench_function("workflow_build_simple", |b| {
        b.iter(|| {
            use echo_orchestration::workflow::GraphBuilder;
            let _graph = GraphBuilder::new("bench_graph")
                .add_function_node("start", |state| {
                    Box::pin(async move {
                        let _ = state.set("key", "value");
                        Ok(())
                    })
                })
                .set_entry("start")
                .set_finish("start")
                .build()
                .unwrap();
            black_box(())
        })
    });
}

/// Benchmark: Token budget allocation calculation (CPU-bound micro-benchmark)
fn bench_token_budget_allocate(c: &mut Criterion) {
    c.bench_function("token_budget_allocate", |b| {
        let budget = echo_core::budget::TokenBudget::new(128_000);
        b.iter(|| {
            let alloc = budget.allocate(5000, 3000, 40000);
            black_box(alloc.ok())
        })
    });
}

criterion_group!(
    benches,
    bench_agent_build,
    bench_agent_build_with_tools,
    bench_shared_agent_create,
    bench_workflow_build_simple,
    bench_token_budget_allocate,
);
criterion_main!(benches);