rs-adk 0.5.0

Agent runtime for Gemini Live — tools, streaming, agent transfer, middleware
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

//! Individual MCP tool — wraps an MCP session manager to implement ToolFunction.

use std::sync::Arc;

use async_trait::async_trait;

use crate::error::ToolError;
use crate::tool::ToolFunction;

use super::session_manager::McpSessionManager;

/// An individual MCP tool, wrapping an MCP session manager.
pub struct McpTool {
    name: String,
    description: String,
    input_schema: Option<serde_json::Value>,
    session_manager: Arc<McpSessionManager>,
}

impl McpTool {
    /// Create a new MCP tool proxy with the given name, description, and session.
    pub fn new(
        name: impl Into<String>,
        description: impl Into<String>,
        input_schema: Option<serde_json::Value>,
        session_manager: Arc<McpSessionManager>,
    ) -> Self {
        Self {
            name: name.into(),
            description: description.into(),
            input_schema,
            session_manager,
        }
    }
}

#[async_trait]
impl ToolFunction for McpTool {
    fn name(&self) -> &str {
        &self.name
    }
    fn description(&self) -> &str {
        &self.description
    }
    fn parameters(&self) -> Option<serde_json::Value> {
        self.input_schema.clone()
    }
    async fn call(&self, args: serde_json::Value) -> Result<serde_json::Value, ToolError> {
        self.session_manager
            .call_tool(&self.name, args)
            .await
            .map_err(|e| ToolError::ExecutionFailed(e.to_string()))
    }
}