37 lines
794 B
Ruby
37 lines
794 B
Ruby
class Captain::ToolRegistryService
|
|
attr_reader :registered_tools, :tools
|
|
|
|
def initialize(assistant, user: nil)
|
|
@assistant = assistant
|
|
@user = user
|
|
@registered_tools = []
|
|
@tools = {}
|
|
end
|
|
|
|
def register_tool(tool_class)
|
|
tool = tool_class.new(@assistant, user: @user)
|
|
return unless tool.active?
|
|
|
|
@tools[tool.name] = tool
|
|
@registered_tools << tool.to_registry_format
|
|
end
|
|
|
|
def method_missing(method_name, *)
|
|
if @tools.key?(method_name.to_s)
|
|
@tools[method_name.to_s].execute(*)
|
|
else
|
|
super
|
|
end
|
|
end
|
|
|
|
def respond_to_missing?(method_name, include_private = false)
|
|
@tools.key?(method_name.to_s) || super
|
|
end
|
|
|
|
def tools_summary
|
|
@tools.map do |name, tool|
|
|
"- #{name}: #{tool.description}"
|
|
end.join("\n")
|
|
end
|
|
end
|