rho-coding-agent 0.28.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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use super::{display_lines, parse_partial_json, StreamedToolCallPreview};
use crate::{
    tool::{ToolContext, ToolRegistry},
    tools::{edit_file::EditFile, read_file::ReadFile},
};

fn context() -> ToolContext {
    ToolContext {
        cwd: std::env::current_dir().unwrap(),
        max_output_bytes: 12_000,
    }
}

#[test]
fn displays_argument_values_before_json_is_complete() {
    let mut tools = ToolRegistry::new();
    tools.register(ReadFile);

    assert_eq!(
        display_lines(
            Some("read_file"),
            r#"{"path":"src/main"#,
            &tools,
            &context(),
        ),
        vec!["read_file", "src/main"]
    );
    assert_eq!(
        display_lines(
            Some("read_file"),
            r#"{"path":"src/main.rs","offset":"#,
            &tools,
            &context(),
        ),
        vec!["read_file", "src/main.rs"]
    );
}

#[test]
fn name_only_previews_do_not_parse_large_streamed_values() {
    let mut tools = ToolRegistry::new();
    tools.register(EditFile);
    let arguments = format!(
        r#"{{"edits":[{{"path":"src/main.rs","new_string":"{}"#,
        "x".repeat(100_000)
    );

    assert_eq!(
        display_lines(Some("edit_file"), &arguments, &tools, &context()),
        vec!["edit_file"]
    );
}

#[test]
fn argument_previews_are_throttled_as_the_stream_grows() {
    let mut tools = ToolRegistry::new();
    tools.register(ReadFile);
    let mut preview = StreamedToolCallPreview::default();

    assert_eq!(
        preview.update(Some("read_file"), r#"{"path":"a"#, &tools, &context(),),
        Some(vec!["read_file".into(), "a".into()])
    );
    assert_eq!(
        preview.update(Some("read_file"), r#"{"path":"abc"#, &tools, &context(),),
        None
    );
    assert_eq!(
        preview.update(
            Some("read_file"),
            r#"{"path":"abcdefghijklmnop"#,
            &tools,
            &context(),
        ),
        Some(vec!["read_file".into(), "abcdefghijklmnop".into()])
    );
}

#[cfg(any(target_os = "linux", target_os = "macos"))]
#[test]
fn command_preview_uses_streamed_arguments() {
    let mut tools = ToolRegistry::new();
    tools.register(crate::tools::bash::Bash::new(/*rtk_enabled*/ false));

    assert_eq!(
        display_lines(Some("bash"), r#"{"command":"cargo te"#, &tools, &context(),),
        vec!["bash cargo te"]
    );
}

#[cfg(windows)]
#[test]
fn command_preview_uses_streamed_arguments() {
    let mut tools = ToolRegistry::new();
    tools.register(crate::tools::powershell::PowerShell::new(
        /*rtk_enabled*/ false,
    ));

    assert_eq!(
        display_lines(
            Some("powershell"),
            r#"{"command":"cargo te"#,
            &tools,
            &context(),
        ),
        vec!["powershell cargo te"]
    );
}

#[test]
fn completes_streaming_strings_and_nested_values() {
    assert_eq!(
        parse_partial_json(r#"{"path":"src/main"#),
        Some(serde_json::json!({"path": "src/main"}))
    );
    assert_eq!(
        parse_partial_json(r#"{"edits":[{"path":"src/main.rs","new_string":"hel"#),
        Some(serde_json::json!({
            "edits": [{"path": "src/main.rs", "new_string": "hel"}]
        }))
    );
}

#[test]
fn keeps_complete_fields_when_the_next_value_has_not_started() {
    assert_eq!(
        parse_partial_json(r#"{"path":"src/main.rs","offset":"#),
        Some(serde_json::json!({"path": "src/main.rs"}))
    );
}

#[test]
fn preserves_partial_escape_sequences() {
    assert_eq!(
        parse_partial_json(r#"{"content":"line\nnext\\"#),
        Some(serde_json::json!({"content": "line\nnext\\"}))
    );
}