vtcode 0.136.3

A Rust-based terminal coding agent with modular architecture supporting multiple LLM providers
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

#![allow(missing_docs)]
use assert_fs::TempDir;
use vtcode_core::config::ToolDocumentationMode;
use vtcode_core::config::constants::tools;
use vtcode_core::config::types::CapabilityLevel;
use vtcode_core::tools::ToolRegistry;
use vtcode_core::tools::handlers::{SessionSurface, SessionToolsConfig, ToolModelCapabilities, ToolProfile};

fn default_public_tool_names() -> Vec<String> {
    vec![
        tools::EXEC_COMMAND.to_string(),
        tools::WRITE_STDIN.to_string(),
        tools::APPLY_PATCH.to_string(),
    ]
}

/// The public catalog order derives from rig's `ToolSet` (HashMap-backed), so
/// it is not a stable contract. Compare the tool set order-insensitively.
fn assert_same_tool_set(actual: &[String], expected: &[String]) {
    let mut actual_sorted = actual.to_vec();
    actual_sorted.sort();
    let mut expected_sorted = expected.to_vec();
    expected_sorted.sort();
    assert_eq!(actual_sorted, expected_sorted);
}

#[tokio::test]
async fn audit_public_catalog_is_runtime_driven() {
    let temp = TempDir::new().expect("tempdir");
    let registry = ToolRegistry::new(temp.path().to_path_buf()).await;

    let public_names = registry.available_tools().await;
    let schema_names: Vec<_> = registry
        .schema_entries(SessionToolsConfig::full_public(
            SessionSurface::Interactive,
            CapabilityLevel::CodeSearch,
            ToolDocumentationMode::Full,
            ToolModelCapabilities::default(),
        ))
        .await
        .into_iter()
        .map(|entry| entry.name)
        .collect();

    assert_eq!(public_names, schema_names);
    assert_same_tool_set(&public_names, &default_public_tool_names());
    assert!(!public_names.contains(&tools::CODE_SEARCH.to_string()));
    assert!(!public_names.contains(&tools::UNIFIED_SEARCH.to_string()));
    assert!(!public_names.contains(&tools::UNIFIED_FILE.to_string()));
    assert!(!public_names.contains(&tools::UNIFIED_EXEC.to_string()));
}

#[tokio::test]
async fn audit_code_search_is_advanced_only() {
    let temp = TempDir::new().expect("tempdir");
    let registry = ToolRegistry::new(temp.path().to_path_buf()).await;

    let advanced_schema_names: Vec<_> = registry
        .schema_entries(
            SessionToolsConfig::full_public(
                SessionSurface::Interactive,
                CapabilityLevel::CodeSearch,
                ToolDocumentationMode::Full,
                ToolModelCapabilities::default(),
            )
            .with_tool_profile(ToolProfile::AdvancedVtCode),
        )
        .await
        .into_iter()
        .map(|entry| entry.name)
        .collect();

    assert!(advanced_schema_names.contains(&tools::CODE_SEARCH.to_string()));
}

#[tokio::test]
async fn audit_public_catalog_hides_legacy_aliases() {
    let temp = TempDir::new().expect("tempdir");
    let registry = ToolRegistry::new(temp.path().to_path_buf()).await;

    let public_names = registry.available_tools().await;

    for legacy_name in [
        tools::READ_FILE,
        tools::WRITE_FILE,
        tools::EDIT_FILE,
        tools::DELETE_FILE,
        tools::RUN_PTY_CMD,
        tools::SEND_PTY_INPUT,
        tools::LIST_PTY_SESSIONS,
        tools::UNIFIED_SEARCH,
        tools::UNIFIED_FILE,
        tools::UNIFIED_EXEC,
    ] {
        assert!(
            !public_names.iter().any(|name| name == legacy_name),
            "legacy alias leaked into public catalog: {legacy_name}"
        );
    }
}

#[tokio::test]
async fn audit_acp_subset_comes_from_same_catalog() {
    let temp = TempDir::new().expect("tempdir");
    let registry = ToolRegistry::new(temp.path().to_path_buf()).await;

    let acp_names = registry
        .public_tool_names(SessionSurface::Acp, CapabilityLevel::CodeSearch)
        .await;

    assert_same_tool_set(&acp_names, &default_public_tool_names());
    assert!(!acp_names.contains(&tools::CODE_SEARCH.to_string()));
}