vtcode 0.99.1

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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

//! Manual test for PTY functionality

use anyhow::Result;
use assert_fs::TempDir;
use serde_json::json;
use vtcode_core::tools::ToolRegistry;

#[tokio::main]
async fn main() -> Result<()> {
    println!("Testing PTY functionality...");

    // Create a temporary directory for testing
    let temp_dir = TempDir::new()?;
    let workspace = temp_dir.path().to_path_buf();
    let registry = ToolRegistry::new(workspace.clone()).await;

    // Test 1: Basic PTY command
    println!("\n=== Test 1: Basic PTY command ===");
    let args = json!({
        "command": "echo",
        "args": ["Hello, PTY!"]
    });

    match registry.execute_tool("run_pty_cmd", args).await {
        Ok(result) => {
            println!("Success: {:?}", result);
            assert_eq!(result["success"], true);
            assert_eq!(result["code"], 0);
            assert!(result["output"].as_str().unwrap().contains("Hello, PTY!"));
            println!("✓ Test 1 passed");
        }
        Err(e) => {
            println!("Error: {}", e);
            return Err(e);
        }
    }

    // Test 2: PTY command with working directory
    println!("\n=== Test 2: PTY command with working directory ===");

    // Create a test file
    std::fs::write(workspace.join("test.txt"), "Hello, PTY from file!")?;

    let args = json!({
        "command": "cat",
        "args": ["test.txt"]
    });

    match registry.execute_tool("run_pty_cmd", args).await {
        Ok(result) => {
            println!("Success: {:?}", result);
            assert_eq!(result["success"], true);
            assert_eq!(result["code"], 0);
            assert!(
                result["output"]
                    .as_str()
                    .unwrap()
                    .contains("Hello, PTY from file!")
            );
            println!("✓ Test 2 passed");
        }
        Err(e) => {
            println!("Error: {}", e);
            return Err(e);
        }
    }

    // Test 3: PTY session management
    println!("\n=== Test 3: PTY session management ===");

    // Create a PTY session
    let args = json!({
        "session_id": "test_session",
        "command": "bash"
    });

    match registry.execute_tool("create_pty_session", args).await {
        Ok(result) => {
            println!("Create session result: {:?}", result);
            assert_eq!(result["success"], true);
            assert_eq!(result["session_id"], "test_session");
            println!("✓ Session created");
        }
        Err(e) => {
            println!("Error creating session: {}", e);
            return Err(e);
        }
    }

    // List PTY sessions
    let args = json!({});
    match registry.execute_tool("list_pty_sessions", args).await {
        Ok(result) => {
            println!("List sessions result: {:?}", result);
            assert!(
                result["sessions"]
                    .as_array()
                    .unwrap()
                    .contains(&"test_session".into())
            );
            println!("✓ Session listed");
        }
        Err(e) => {
            println!("Error listing sessions: {}", e);
            return Err(e);
        }
    }

    // Close PTY session
    let args = json!({
        "session_id": "test_session"
    });

    match registry.execute_tool("close_pty_session", args).await {
        Ok(result) => {
            println!("Close session result: {:?}", result);
            assert_eq!(result["success"], true);
            assert_eq!(result["session_id"], "test_session");
            println!("✓ Session closed");
        }
        Err(e) => {
            println!("Error closing session: {}", e);
            return Err(e);
        }
    }

    println!("\n=== All tests passed! ===");
    Ok(())
}