adk-tool 0.6.0

Tool system for Rust Agent Development Kit (ADK-Rust) agents (FunctionTool, MCP, Google Search)
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

//! Lightweight [`ToolContext`] implementation for use outside the agent loop.
//!
//! [`SimpleToolContext`] provides sensible defaults for all trait methods so
//! that callers in MCP server mode, testing, or sub-agent delegation can
//! invoke tools without constructing a full invocation context.
//!
//! # Example
//!
//! ```rust,no_run
//! use adk_tool::SimpleToolContext;
//! use std::sync::Arc;
//!
//! let ctx = SimpleToolContext::new("my-caller");
//! let ctx: Arc<dyn adk_core::ToolContext> = Arc::new(ctx);
//! ```

use adk_core::context::{Artifacts, CallbackContext, MemoryEntry, ReadonlyContext};
use adk_core::types::Content;
use adk_core::{EventActions, Result, ToolContext};
use async_trait::async_trait;
use std::sync::{Arc, Mutex};

/// A lightweight [`ToolContext`] with sensible defaults for non-agent callers.
///
/// Implements [`ReadonlyContext`], [`CallbackContext`], and [`ToolContext`]
/// with minimal configuration. Construct via [`SimpleToolContext::new`] with
/// a caller name; all other fields use safe defaults.
pub struct SimpleToolContext {
    caller_name: String,
    invocation_id: String,
    function_call_id: String,
    user_content: Content,
    actions: Mutex<EventActions>,
}

impl SimpleToolContext {
    /// Create a new context with the given caller name.
    ///
    /// Generates unique UUIDs for `invocation_id` and `function_call_id`.
    /// The caller name is returned by both [`agent_name()`](ReadonlyContext::agent_name)
    /// and [`app_name()`](ReadonlyContext::app_name).
    pub fn new(caller_name: impl Into<String>) -> Self {
        Self {
            caller_name: caller_name.into(),
            invocation_id: uuid::Uuid::new_v4().to_string(),
            function_call_id: uuid::Uuid::new_v4().to_string(),
            user_content: Content::new("user"),
            actions: Mutex::new(EventActions::default()),
        }
    }

    /// Override the default function call ID.
    ///
    /// By default a UUID is generated at construction. Use this builder
    /// method to provide a specific ID instead.
    pub fn with_function_call_id(mut self, id: impl Into<String>) -> Self {
        self.function_call_id = id.into();
        self
    }
}

#[async_trait]
impl ReadonlyContext for SimpleToolContext {
    fn invocation_id(&self) -> &str {
        &self.invocation_id
    }

    fn agent_name(&self) -> &str {
        &self.caller_name
    }

    fn user_id(&self) -> &str {
        "anonymous"
    }

    fn app_name(&self) -> &str {
        &self.caller_name
    }

    fn session_id(&self) -> &str {
        ""
    }

    fn branch(&self) -> &str {
        ""
    }

    fn user_content(&self) -> &Content {
        &self.user_content
    }
}

#[async_trait]
impl CallbackContext for SimpleToolContext {
    fn artifacts(&self) -> Option<Arc<dyn Artifacts>> {
        None
    }
}

#[async_trait]
impl ToolContext for SimpleToolContext {
    fn function_call_id(&self) -> &str {
        &self.function_call_id
    }

    fn actions(&self) -> EventActions {
        self.actions.lock().unwrap().clone()
    }

    fn set_actions(&self, actions: EventActions) {
        *self.actions.lock().unwrap() = actions;
    }

    async fn search_memory(&self, _query: &str) -> Result<Vec<MemoryEntry>> {
        Ok(vec![])
    }
}