Skip to main content

ToolRegistry

aster::tools::registry

Struct ToolRegistry 

Source 
pub struct ToolRegistry { /* private fields */ }
Expand description

Tool Registry

Manages all available tools in the system, including both native tools and MCP tools. Native tools have higher priority than MCP tools with the same name.

Requirements: 2.1, 2.2, 2.3

Implementations§

Source§

impl ToolRegistry

Source

pub fn new() -> Self

Create a new empty tool registry

Source

pub fn with_managers( permission_manager: Arc<ToolPermissionManager>, audit_logger: Arc<AuditLogger>, ) -> Self

Create a new tool registry with permission manager and audit logger

Source

pub fn set_permission_manager(&mut self, manager: Arc<ToolPermissionManager>)

Set the permission manager

Source

pub fn set_audit_logger(&mut self, logger: Arc<AuditLogger>)

Set the audit logger

Source

pub fn permission_manager(&self) -> Option<&Arc<ToolPermissionManager>>

Get the permission manager

Source

pub fn audit_logger(&self) -> Option<&Arc<AuditLogger>>

Get the audit logger

Source§

impl ToolRegistry

Source

pub fn register(&mut self, tool: Box<dyn Tool>)

Register a native tool

Native tools have higher priority than MCP tools with the same name. If a native tool with the same name already exists, it will be replaced.

§Arguments
  • tool - The tool to register

Requirements: 2.1

Source

pub fn register_mcp(&mut self, name: String, tool: McpToolWrapper)

Register an MCP tool

MCP tools have lower priority than native tools. If a native tool with the same name exists, the MCP tool will be shadowed.

§Arguments
  • name - The tool name
  • tool - The MCP tool wrapper

Requirements: 11.4

Source

pub fn unregister(&mut self, name: &str) -> Option<Box<dyn Tool>>

Unregister a native tool

§Arguments
  • name - The name of the tool to unregister
§Returns

The unregistered tool if it existed

Source

pub fn unregister_mcp(&mut self, name: &str) -> Option<McpToolWrapper>

Unregister an MCP tool

§Arguments
  • name - The name of the tool to unregister
§Returns

The unregistered MCP tool wrapper if it existed

Source

pub fn contains(&self, name: &str) -> bool

Check if a tool is registered (native or MCP)

§Arguments
  • name - The tool name to check
§Returns

true if the tool is registered

Source

pub fn contains_native(&self, name: &str) -> bool

Check if a native tool is registered

Source

pub fn contains_mcp(&self, name: &str) -> bool

Check if an MCP tool is registered

Source

pub fn native_tool_count(&self) -> usize

Get the number of registered native tools

Source

pub fn mcp_tool_count(&self) -> usize

Get the number of registered MCP tools

Source

pub fn tool_count(&self) -> usize

Get the total number of registered tools

Source§

impl ToolRegistry

Source

pub fn get(&self, name: &str) -> Option<&dyn Tool>

Get a tool by name (native tools have priority)

§Arguments
  • name - The tool name to look up
§Returns

A reference to the tool if found, with native tools taking priority

Requirements: 2.2

Source

pub fn get_all(&self) -> Vec<&dyn Tool>

Get all registered tools

Returns all tools with native tools taking priority over MCP tools with the same name.

§Returns

A vector of references to all registered tools

Requirements: 2.3

Source

pub fn get_definitions(&self) -> Vec<ToolDefinition>

Get all tool definitions for LLM consumption

Returns definitions for all tools, with native tools taking priority over MCP tools with the same name.

§Returns

A vector of tool definitions

Requirements: 2.4

Source

pub fn native_tool_names(&self) -> Vec<&str>

Get all native tool names

Source

pub fn mcp_tool_names(&self) -> Vec<&str>

Get all MCP tool names

Source

pub fn tool_names(&self) -> Vec<&str>

Get all tool names (unique, native tools shadow MCP tools)

Source

pub fn is_native(&self, name: &str) -> bool

Check if a tool is a native tool

Source

pub fn is_mcp(&self, name: &str) -> bool

Check if a tool is an MCP tool (and not shadowed by a native tool)

Source§

impl ToolRegistry

Source

pub async fn execute( &self, name: &str, params: Value, context: &ToolContext, on_permission_request: Option<PermissionRequestCallback>, ) -> Result<ToolResult, ToolError>

Execute a tool by name with permission checking and audit logging

This method:

  1. Looks up the tool by name
  2. Performs permission check (if permission manager is configured)
  3. Handles permission request callback for ‘Ask’ behavior
  4. Executes the tool
  5. Records audit log (if audit logger is configured)
§Arguments
  • name - The tool name to execute
  • params - The tool parameters
  • context - The execution context
  • on_permission_request - Optional callback for permission requests
§Returns
  • Ok(ToolResult) - The execution result
  • Err(ToolError) - If the tool is not found, permission denied, or execution fails

Requirements: 2.5, 2.6, 8.1, 8.2

Trait Implementations§

Source§

impl Default for ToolRegistry

Source§

fn default() -> Self

Returns the “default value” for a type. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> Conv for T

Source§

fn conv<T>(self) -> T
where Self: Into<T>,

Converts self into T using Into<T>. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> FutureExt for T

Source§

fn with_context(self, otel_cx: Context) -> WithContext<Self>

Attaches the provided Context to this type, returning a WithContext wrapper. Read more
Source§

fn with_current_context(self) -> WithContext<Self>

Attaches the current Context to this type, returning a WithContext wrapper. Read more
Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> IntoRequest<T> for T

Source§

fn into_request(self) -> Request<T>

Wrap the input message T in a tonic::Request
Source§

impl<Unshared, Shared> IntoShared<Shared> for Unshared
where Shared: FromUnshared<Unshared>,

Source§

fn into_shared(self) -> Shared

Creates a shared type from an unshared type.
Source§

impl<T> Pipe for T
where T: ?Sized,

Source§

fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> R
where Self: Sized,

Pipes by value. This is generally the method you want to use. Read more
Source§

fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> R
where R: 'a,

Borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> R
where R: 'a,

Mutably borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
where Self: Borrow<B>, B: 'a + ?Sized, R: 'a,

Borrows self, then passes self.borrow() into the pipe function. Read more
Source§

fn pipe_borrow_mut<'a, B, R>( &'a mut self, func: impl FnOnce(&'a mut B) -> R, ) -> R
where Self: BorrowMut<B>, B: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.borrow_mut() into the pipe function. Read more
Source§

fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
where Self: AsRef<U>, U: 'a + ?Sized, R: 'a,

Borrows self, then passes self.as_ref() into the pipe function.
Source§

fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
where Self: AsMut<U>, U: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.as_mut() into the pipe function.
Source§

fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
where Self: Deref<Target = T>, T: 'a + ?Sized, R: 'a,

Borrows self, then passes self.deref() into the pipe function.
Source§

fn pipe_deref_mut<'a, T, R>( &'a mut self, func: impl FnOnce(&'a mut T) -> R, ) -> R
where Self: DerefMut<Target = T> + Deref, T: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.deref_mut() into the pipe function.
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T> Tap for T

Source§

fn tap(self, func: impl FnOnce(&Self)) -> Self

Immutable access to a value. Read more
Source§

fn tap_mut(self, func: impl FnOnce(&mut Self)) -> Self

Mutable access to a value. Read more
Source§

fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Immutable access to the Borrow<B> of a value. Read more
Source§

fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Mutable access to the BorrowMut<B> of a value. Read more
Source§

fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Immutable access to the AsRef<R> view of a value. Read more
Source§

fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Mutable access to the AsMut<R> view of a value. Read more
Source§

fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Immutable access to the Deref::Target of a value. Read more
Source§

fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Mutable access to the Deref::Target of a value. Read more
Source§

fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self

Calls .tap() only in debug builds, and is erased in release builds.
Source§

fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self

Calls .tap_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Calls .tap_borrow() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Calls .tap_borrow_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Calls .tap_ref() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Calls .tap_ref_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Calls .tap_deref() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_mut_dbg<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Calls .tap_deref_mut() only in debug builds, and is erased in release builds.
Source§

impl<T> TryConv for T

Source§

fn try_conv<T>(self) -> Result<T, Self::Error>
where Self: TryInto<T>,

Attempts to convert self into T using TryInto<T>. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

impl<T> ErasedDestructor for T
where T: 'static,