Struct AdaptiveRegistry 

pub struct AdaptiveRegistry { /* private fields */ }

A tool registry that automatically limits the number of visible tools based on the configured model tier.

Small models get fewer tools (to fit smaller context windows), while large models get access to all registered tools.

Tool priority is determined by insertion order — tools registered first are selected first when the limit is applied.

find_tool() searches all tools regardless of tier limit, ensuring tool execution always works even for tools beyond the active limit.

Example

use traitclaw_core::registries::AdaptiveRegistry;
use traitclaw_core::traits::tool_registry::ToolRegistry;
use traitclaw_core::types::model_info::ModelTier;

let registry = AdaptiveRegistry::new(vec![], ModelTier::Medium);
assert!(registry.is_empty());

Implementations

impl AdaptiveRegistry

pub fn new(tools: Vec<Arc<dyn ErasedTool>>, tier: ModelTier) -> AdaptiveRegistry

Create an adaptive registry with default tier limits.

Default limits: Small=5, Medium=15, Large=unlimited.

pub fn with_limits( self, small: usize, medium: usize, large: usize, ) -> AdaptiveRegistry

Override the default tier limits.

Example

use traitclaw_core::registries::AdaptiveRegistry;
use traitclaw_core::types::model_info::ModelTier;

let registry = AdaptiveRegistry::new(vec![], ModelTier::Small)
    .with_limits(3, 10, 50);

pub fn limits(&self) -> TierLimits

Get the current tier limits.

pub fn tier(&self) -> ModelTier

Get the configured model tier.

Trait Implementations

impl ToolRegistry for AdaptiveRegistry

fn get_tools(&self) -> Vec<Arc<dyn ErasedTool>>

Returns at most limit tools based on the configured ModelTier.

Tools are returned in insertion order (first registered = highest priority).

fn find_tool(&self, name: &str) -> Option<Arc<dyn ErasedTool>>

Searches all tools regardless of tier limit.

This ensures tool execution works even for tools beyond the active limit.

fn len(&self) -> usize

Returns the count of tools visible for the current tier.

fn is_empty(&self) -> bool

Check if the registry has no enabled tools.

fn register(&self, _tool: Arc<dyn ErasedTool>) -> bool

Register a new tool. Returns true if the tool was added.

fn unregister(&self, _name: &str) -> bool

Unregister a tool by name. Returns true if the tool was removed.

fn set_enabled(&self, _name: &str, _enabled: bool) -> bool

Enable or disable a tool by name. Returns true if the state changed.

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

Check if a tool is currently enabled.