phi-kernel-tools 0.1.1

Kernel tools for phi-agent — file system, multi-agent, and shell tools
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

//! Benchmarks: kernel file tools (read_file, write_file, list_files).

use agent_base::{Language, SessionId, Tool, ToolContext, UserEvent};
use criterion::{Criterion, black_box, criterion_group, criterion_main};
use phi_kernel_tools::file::{ListFilesTool, ReadFileTool, WriteFileTool};
use tempfile::TempDir;

fn make_ctx() -> ToolContext {
    let (tx, _rx) = tokio::sync::mpsc::unbounded_channel::<UserEvent>();
    ToolContext {
        session_id: SessionId::new(1),
        user_event_tx: tx,
        llm_client: None,
        session_store: None,
        language: Language::En,
        cancel_token: tokio_util::sync::CancellationToken::new(),
    }
}

fn bench_read_file(c: &mut Criterion) {
    let rt = tokio::runtime::Runtime::new().unwrap();
    let temp = TempDir::new().unwrap();
    let file_path = temp.path().join("test.txt");
    let content = "This is a test file.\n".repeat(1000);
    std::fs::write(&file_path, &content).unwrap();

    let tool = ReadFileTool::new(temp.path().to_path_buf());
    let ctx = make_ctx();

    c.bench_function("file/read_1000_lines", |b| {
        b.iter(|| {
            let args = serde_json::json!({"path": "test.txt"});
            let _ = black_box(rt.block_on(tool.call(&args, &ctx)));
        });
    });
}

fn bench_write_file(c: &mut Criterion) {
    let rt = tokio::runtime::Runtime::new().unwrap();
    let temp = TempDir::new().unwrap();
    let tool = WriteFileTool::new(temp.path().to_path_buf());
    let ctx = make_ctx();
    let content = "Hello, world!\n".repeat(100);

    c.bench_function("file/write_100_lines", |b| {
        b.iter(|| {
            let args = serde_json::json!({"path": "bench_output.txt", "content": content});
            let _ = black_box(rt.block_on(tool.call(&args, &ctx)));
        });
    });
}

fn bench_list_files(c: &mut Criterion) {
    let rt = tokio::runtime::Runtime::new().unwrap();
    let temp = TempDir::new().unwrap();
    // Create 100 files
    for i in 0..100 {
        std::fs::write(temp.path().join(format!("file_{:03}.txt", i)), "data").unwrap();
    }
    let sub = temp.path().join("sub");
    std::fs::create_dir(&sub).unwrap();
    for i in 0..20 {
        std::fs::write(sub.join(format!("sub_{:03}.txt", i)), "data").unwrap();
    }

    let tool = ListFilesTool::new(temp.path().to_path_buf());
    let ctx = make_ctx();

    c.bench_function("file/list_100_flat", |b| {
        b.iter(|| {
            let args = serde_json::json!({"path": "."});
            let _ = black_box(rt.block_on(tool.call(&args, &ctx)));
        });
    });
}

criterion_group! {
    name = file_tools_benches;
    config = Criterion::default().sample_size(200);
    targets = bench_read_file, bench_write_file, bench_list_files
}
criterion_main!(file_tools_benches);