rho-coding-agent 0.26.0

A lightweight agent harness inspired by Pi
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111

#[cfg(any(target_os = "linux", target_os = "macos"))]
pub mod bash;
mod diff;
pub mod edit_file;
mod edit_file_args;
pub mod list_dir;
#[cfg(windows)]
pub mod powershell;
mod process;
pub mod read_file;
pub mod rho;
pub mod rtk;
pub mod skill;
pub mod web;
pub mod write_file;

use crate::{config::Config, diagnostics::RuntimeDiagnostics, tool::ToolRegistry};

pub fn registry(config: &Config, diagnostics: RuntimeDiagnostics) -> ToolRegistry {
    let mut r = ToolRegistry::new();
    r.register(list_dir::ListDir);
    r.register(read_file::ReadFile);
    r.register(write_file::WriteFile);
    r.register(edit_file::EditFile);
    let processes = process::ProcessManager::new(process::ProcessLimits::default());
    r.register(process::Process::new(processes.clone()));
    r.set_shutdown(processes);
    let rtk_enabled = config.rtk && rtk::is_available();
    #[cfg(any(target_os = "linux", target_os = "macos"))]
    r.register(bash::Bash::new(rtk_enabled));
    #[cfg(windows)]
    r.register(powershell::PowerShell::new(rtk_enabled));
    r.register(skill::Skill);
    r.register(rho::Rho::new(diagnostics));
    let (web_search, fetch_content) = web::access_tools(config);
    if web_search.is_available() {
        r.register(web_search);
    }
    r.register(fetch_content);
    r.register(web::GetSearchContent);
    r
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn registry_includes_rho_diagnostics_tool() {
        let registry = registry(
            &Config::default(),
            crate::diagnostics::test_diagnostics("openai", "gpt-test"),
        );

        assert!(registry.get("rho").is_some());
    }

    #[test]
    fn registry_includes_available_web_access_tools() {
        let registry = registry(
            &Config::default(),
            crate::diagnostics::test_diagnostics("openai", "gpt-test"),
        );
        let names = registry
            .specs()
            .into_iter()
            .map(|spec| spec.name)
            .collect::<Vec<_>>();

        assert_eq!(
            names.contains(&"web_search".to_string()),
            web::access_tools(&Config::default()).0.is_available()
        );
        assert!(names.contains(&"fetch_content".to_string()));
        assert!(names.contains(&"get_search_content".to_string()));
        assert_eq!(
            names
                .iter()
                .filter(|name| name.as_str() == "process")
                .count(),
            1
        );
    }

    #[test]
    fn registry_omits_web_search_when_disabled() {
        let config = Config {
            web_search_provider: crate::config::SearchProvider::Disabled,
            ..Config::default()
        };
        let names = registry(
            &config,
            crate::diagnostics::test_diagnostics("openai", "gpt-test"),
        )
        .specs()
        .into_iter()
        .map(|spec| spec.name)
        .collect::<Vec<_>>();

        assert!(!names.contains(&"web_search".to_string()));
        assert!(names.contains(&"fetch_content".to_string()));
        assert!(names.contains(&"get_search_content".to_string()));
        assert_eq!(
            names
                .iter()
                .filter(|name| name.as_str() == "process")
                .count(),
            1
        );
    }
}