agentix 0.21.0

Multi-provider LLM client for Rust — streaming, non-streaming, tool calls, MCP, DeepSeek, OpenAI, Anthropic, Gemini
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

//! Integration tests for the `mcp-server` feature.

#![cfg(feature = "mcp-server")]

use agentix::{McpServer, ToolBundle, tool, tool_trait::Tool};

// ── Test Tools ────────────────────────────────────────────────────────────────

struct CalcTool;

#[tool]
impl Tool for CalcTool {
    /// Add two numbers.
    /// a: first number
    /// b: second number
    async fn add(&self, a: f64, b: f64) -> f64 {
        a + b
    }

    /// Multiply two numbers.
    /// x: first number
    /// y: second number
    async fn multiply(&self, x: f64, y: f64) -> f64 {
        x * y
    }
}

struct EchoTool;

#[tool]
impl Tool for EchoTool {
    /// Echo a message.
    /// message: message to echo
    async fn echo(&self, message: String) -> String {
        format!("Echo: {message}")
    }
}

// ── Tests ───────────────────────────────────────────────────────────────────

#[test]
fn mcp_server_creation_with_name_and_version() {
    let server = McpServer::new(CalcTool)
        .with_name("test-server")
        .with_version("1.0.0");

    let _ = server;
}

#[test]
fn mcp_server_with_tool_bundle() {
    let bundle = ToolBundle::new().with(CalcTool).with(EchoTool);

    let server = McpServer::new(bundle);
    let _ = server;
}

#[tokio::test]
async fn mcp_server_into_axum_router_creates_router() {
    let bundle = ToolBundle::new().with(CalcTool).with(EchoTool);

    let server = McpServer::new(bundle);
    let _router = server.into_axum_router();
}

// Test that multiple tools are correctly registered
#[test]
fn mcp_server_with_multiple_tools() {
    let bundle = ToolBundle::new().with(CalcTool).with(EchoTool);

    let raw_tools = bundle.raw_tools();

    // Should have 3 tools total (add, multiply, echo)
    assert_eq!(raw_tools.len(), 3, "Expected 3 tools: add, multiply, echo");

    let tool_names: Vec<&str> = raw_tools.iter().map(|t| t.function.name.as_str()).collect();

    assert!(tool_names.contains(&"add"), "Should have 'add' tool");
    assert!(
        tool_names.contains(&"multiply"),
        "Should have 'multiply' tool"
    );
    assert!(tool_names.contains(&"echo"), "Should have 'echo' tool");
}

// Test chained with_* methods
#[test]
fn mcp_server_builder_chaining() {
    let server = McpServer::new(CalcTool)
        .with_name("my-mcp-server")
        .with_version("2.0.0-beta");

    let _ = server;
}