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
55
56

//! MCP toolset — discovers tools from an MCP server.

use std::sync::Arc;

use async_trait::async_trait;

use crate::tool::ToolFunction;
use crate::toolset::Toolset;

use super::session_manager::McpSessionManager;

/// A toolset that discovers tools from an MCP server.
pub struct McpToolset {
    session_manager: Arc<McpSessionManager>,
    filter: Option<Vec<String>>,
}

impl McpToolset {
    /// Create a new MCP toolset connected to the given session manager.
    pub fn new(session_manager: Arc<McpSessionManager>) -> Self {
        Self {
            session_manager,
            filter: None,
        }
    }

    /// Only expose tools whose names match the filter.
    pub fn with_filter(mut self, names: Vec<String>) -> Self {
        self.filter = Some(names);
        self
    }

    /// Returns the current filter, if any.
    pub fn filter(&self) -> Option<&[String]> {
        self.filter.as_deref()
    }

    /// Returns a reference to the underlying session manager.
    pub fn session_manager(&self) -> &Arc<McpSessionManager> {
        &self.session_manager
    }
}

#[async_trait]
impl Toolset for McpToolset {
    fn get_tools(&self) -> Vec<Arc<dyn ToolFunction>> {
        // Note: This is synchronous in the trait, but tool discovery is async.
        // In practice, tools should be pre-loaded. For now, return empty.
        // A more complete implementation would cache tools after async initialization.
        vec![]
    }

    async fn close(&self) {
        // Close MCP session if connected
    }
}