llm 1.3.8

A Rust library unifying multiple LLM backends.
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

use std::sync::Arc;

use chrono::Utc;
use serde_json::Value;

use super::context::ToolContext;
use super::definition::{ToolDefinition, ToolParam};
use super::diff_tracker::DiffTracker;
use super::error::ToolError;
use super::handlers::{
    file_read_tool, ls_tool, patch_tool, plan_tool, rollback_tool, search_tool, shell_tool,
    shell_write_tool,
};
use super::pty::PtySessionManager;

/// Create all built-in tools.
///
/// The `pty_manager` is optional - if None, shell tools are not included.
pub fn builtin_tools() -> Vec<ToolDefinition> {
    vec![echo_tool(), time_tool()]
}

/// Create all built-in tools including shell tools with PTY support.
pub fn builtin_tools_with_pty(
    pty_manager: Arc<PtySessionManager>,
    diff_tracker: Arc<DiffTracker>,
) -> Vec<ToolDefinition> {
    vec![
        // Legacy tools
        echo_tool(),
        time_tool(),
        // New Codex-style tools
        shell_tool(Arc::clone(&pty_manager)),
        shell_write_tool(pty_manager),
        file_read_tool(),
        search_tool(),
        ls_tool(),
        patch_tool(),
        plan_tool(),
        rollback_tool(diff_tracker),
    ]
}

fn echo_tool() -> ToolDefinition {
    ToolDefinition {
        name: "echo",
        description: "Echo back the provided text.",
        params: vec![ToolParam {
            name: "text",
            description: "Text to echo back.",
            param_type: "string",
            items: None,
        }],
        required: vec!["text"],
        executor: Arc::new(exec_echo),
    }
}

fn time_tool() -> ToolDefinition {
    ToolDefinition {
        name: "time_now",
        description: "Return the current UTC time in RFC3339 format.",
        params: Vec::new(),
        required: Vec::new(),
        executor: Arc::new(exec_time),
    }
}

fn exec_echo(_ctx: &ToolContext, args: Value) -> Result<String, ToolError> {
    let text = args
        .get("text")
        .and_then(|v| v.as_str())
        .ok_or_else(|| ToolError::InvalidArgs("missing 'text'".to_string()))?;
    Ok(text.to_string())
}

fn exec_time(_ctx: &ToolContext, _args: Value) -> Result<String, ToolError> {
    Ok(Utc::now().to_rfc3339())
}