Class: Tools::Registry

Inherits:
Object
  • Object
show all
Defined in:
lib/tools/registry.rb

Overview

Manages tool registration, schema export, and dispatch. Accepts both tool classes (e.g. Base subclasses) and tool instances (e.g. McpTool) via duck typing. Classes are instantiated with the registry’s context on each execution; instances are called directly since they carry their own state.

Examples:

registry = Tools::Registry.new(context: {shell_session: my_shell})
registry.register(Tools::Bash)
registry.execute("bash", {"command" => "ls"})

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(context: {}) ⇒ Registry

Returns a new instance of Registry.

Parameters:

  • context (Hash) (defaults to: {}) —

    keyword arguments forwarded to every tool constructor



21
22
23
24
 
# File 'lib/tools/registry.rb', line 21

def initialize(context: {})
  @tools = {}
  @context = context
end

Instance Attribute Details

#tools ⇒ Hash{String => Class, Object} (readonly)

Returns registered tools keyed by name.

Returns:

  • (Hash{String => Class, Object}) —

    registered tools keyed by name



18
19
20
 
# File 'lib/tools/registry.rb', line 18

def tools
  @tools
end

Instance Method Details

#any? ⇒ Boolean

Returns whether any tools are registered.

Returns:

  • (Boolean) —

    whether any tools are registered



85
86
87
 
# File 'lib/tools/registry.rb', line 85

def any?
  @tools.any?
end

#execute(name, input) ⇒ String, Hash

Execute a tool by name. Classes are instantiated with the registry’s context; instances are called directly.

Parameters:

  • name (String) —

    registered tool name

  • input (Hash) —

    tool input parameters (may include “timeout” for tools that support per-call timeout overrides)

Returns:

  • (String, Hash) —

    tool execution result

Raises:



59
60
61
62
63
 
# File 'lib/tools/registry.rb', line 59

def execute(name, input)
  tool = @tools.fetch(name) { raise UnknownToolError, "Unknown tool: #{name}" }
  instance = tool.is_a?(Class) ? tool.new(**@context) : tool
  instance.execute(input)
end

#register(tool) ⇒ void

This method returns an undefined value.

Register a tool class or instance. Must respond to tool_name and schema.

Parameters:

  • tool (Class<Tools::Base>, #tool_name) —

    tool class or duck-typed instance



29
30
31
 
# File 'lib/tools/registry.rb', line 29

def register(tool)
  @tools[tool.tool_name] = tool
end

#registered?(name) ⇒ Boolean

Returns whether a tool with the given name is registered.

Parameters:

  • name (String) —

    tool name to check

Returns:

  • (Boolean) —

    whether a tool with the given name is registered



80
81
82
 
# File 'lib/tools/registry.rb', line 80

def registered?(name)
  @tools.key?(name)
end

#schemas ⇒ Array<Hash>

Returns tool schemas for the Anthropic API. The last schema is annotated with cache_control so the API caches the entire tools prefix (tools are evaluated first in cache prefix order).

Returns:

  • (Array<Hash>) —

    schema array for the Anthropic tools API parameter. Each schema includes an optional ‘timeout` parameter (seconds) injected by the registry. The agent can override the default per call for long-running operations. Tools with session-dependent schemas (e.g. Think with budget-based maxLength, Bash with CWD in description) are instantiated with context to generate their schema:

    • Think: budget-based maxLength

    • Bash: CWD embedded in description



44
45
46
47
48
49
 
# File 'lib/tools/registry.rb', line 44

def schemas
  default = Anima::Settings.tool_timeout
  result = @tools.values.map { |tool| inject_timeout(resolve_schema(tool), default) }
  result.last[:cache_control] = {type: "ephemeral"}
  result
end

#truncation_threshold(name) ⇒ Integer?

Returns the truncation threshold for a tool, or nil if the tool opts out of truncation (e.g. read_file tool has its own pagination). MCP tools and other duck-typed instances use the default threshold.

Parameters:

  • name (String) —

    registered tool name

Returns:

  • (Integer, nil) —

    character threshold, or nil to skip truncation



71
72
73
74
75
76
 
# File 'lib/tools/registry.rb', line 71

def truncation_threshold(name)
  tool = @tools[name]
  return tool.truncation_threshold if tool&.respond_to?(:truncation_threshold)

  Anima::Settings.max_tool_response_chars
end