From a57cb89b3546845e9db2fcd45e4a8f16ade09b31 Mon Sep 17 00:00:00 2001 From: Adam Malczewski Date: Wed, 1 Apr 2026 23:14:45 +0900 Subject: minor update --- lib/dispatch/tools/registry.rb | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'lib/dispatch/tools/registry.rb') diff --git a/lib/dispatch/tools/registry.rb b/lib/dispatch/tools/registry.rb index e4a5d3e..571ca1b 100644 --- a/lib/dispatch/tools/registry.rb +++ b/lib/dispatch/tools/registry.rb @@ -10,15 +10,17 @@ module Dispatch def register(tool_definition) name = tool_definition.name - if @tools.key?(name) - raise DuplicateToolError, "Tool '#{name}' is already registered" - end + raise DuplicateToolError, "Tool '#{name}' is already registered" if @tools.key?(name) @tools[name] = tool_definition self end def get(name) + @tools[name] + end + + def fetch(name) @tools.fetch(name) do raise ToolNotFoundError, "Tool '#{name}' not found" end @@ -44,7 +46,7 @@ module Dispatch new_registry = self.class.new names.each do |name| - new_registry.register(get(name)) + new_registry.register(fetch(name)) end new_registry -- cgit v1.2.3