vex-llm 1.7.0

LLM provider integrations for VEX
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

//! Built-in tools for VEX agents
//!
//! This module provides a set of ready-to-use tools:
//!
//! - [`CalculatorTool`] - Evaluate mathematical expressions
//! - [`DateTimeTool`] - Get current date and time
//! - [`UuidTool`] - Generate UUIDs
//! - [`HashTool`] - Compute SHA-256/SHA-512 hashes
//! - [`RegexTool`] - Pattern matching and extraction
//! - [`JsonPathTool`] - JSON value extraction
//!
//! All built-in tools are pure computation (no network I/O) and safe for sandboxing.

mod calculator;
mod datetime;
mod hash;
mod json_path;
mod regex;
mod uuid_tool;

pub use calculator::CalculatorTool;
pub use datetime::DateTimeTool;
pub use hash::HashTool;
pub use json_path::JsonPathTool;
pub use regex::RegexTool;
pub use uuid_tool::UuidTool;

use crate::tool::ToolRegistry;
use std::sync::Arc;

/// Create a registry with all built-in tools pre-registered
///
/// # Example
///
/// ```ignore
/// let registry = vex_llm::tools::builtin_registry();
/// assert!(registry.contains("calculator"));
/// assert!(registry.contains("hash"));
/// ```
pub fn builtin_registry() -> ToolRegistry {
    let mut registry = ToolRegistry::new();
    registry.register(Arc::new(CalculatorTool::new()));
    registry.register(Arc::new(DateTimeTool::new()));
    registry.register(Arc::new(UuidTool::new()));
    registry.register(Arc::new(HashTool::new()));
    registry.register(Arc::new(RegexTool::new()));
    registry.register(Arc::new(JsonPathTool::new()));
    registry
}

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

    #[test]
    fn test_builtin_registry() {
        let registry = builtin_registry();

        assert!(registry.contains("calculator"));
        assert!(registry.contains("datetime"));
        assert!(registry.contains("uuid"));
        assert!(registry.contains("hash"));
        assert!(registry.contains("regex"));
        assert!(registry.contains("json_path"));
        assert_eq!(registry.len(), 6);
    }
}