16371498ba
This PR adds a tool to search Linear issues. If the integration is enabled for the account, the tool will return results as expected. Also introduces support for an `active?` method, which allows third-party Copilot tools to be conditionally enabled based on the status of the integration on the account.
30 lines
655 B
Ruby
30 lines
655 B
Ruby
class Captain::ToolRegistryService
|
|
attr_reader :registered_tools, :tools
|
|
|
|
def initialize(assistant)
|
|
@assistant = assistant
|
|
@registered_tools = []
|
|
@tools = {}
|
|
end
|
|
|
|
def register_tool(tool_class)
|
|
tool = tool_class.new(@assistant)
|
|
return unless tool.active?
|
|
|
|
@tools[tool.name] = tool
|
|
@registered_tools << tool.to_registry_format
|
|
end
|
|
|
|
def method_missing(method_name, *arguments)
|
|
if @tools.key?(method_name.to_s)
|
|
@tools[method_name.to_s].execute(*arguments)
|
|
else
|
|
super
|
|
end
|
|
end
|
|
|
|
def respond_to_missing?(method_name, include_private = false)
|
|
@tools.key?(method_name.to_s) || super
|
|
end
|
|
end
|