fix(chat): add tools from prompt library slash commands (#3012)

Author: petobens

doc/usage/prompt-library.md

@@ -22,3 +22,5 @@ Where `docs` is the `alias` of the prompt.
 
 If your prompt library entries have an `alias` defined then you can invoke them using a slash command. In the cmd line `:CodeCompanion /<alias>` or `/<alias>` if you're in the chat buffer.
 
+When invoked this way, any tools declared on the prompt are added to the current chat buffer before the prompt content is inserted.
+

lua/codecompanion/interactions/chat/slash_commands/init.lua

@@ -5,6 +5,22 @@ local log = require("codecompanion.utils.log")
 
 local api = vim.api
 
+---Add prompt-declared tools to the current chat
+---@param chat CodeCompanion.Chat
+---@param prompt_config table
+local function add_prompt_tools(chat, prompt_config)
+  local prompt_tools = prompt_config.tools or {}
+  local tools_config = chat.tools.tools_config
+
+  vim.iter(prompt_tools):each(function(tool_name)
+    if tools_config.groups and tools_config.groups[tool_name] then
+      chat.tool_registry:add_group(tool_name, { config = tools_config })
+    elseif tools_config[tool_name] then
+      chat.tool_registry:add_single_tool(tool_name, { config = tools_config[tool_name] })
+    end
+  end)
+end
+
 ---Resolve a path to the correct module
 ---@param path string The module or file path
 ---@return table|nil
@@ -176,6 +192,8 @@ end
 ---@return nil
 function SlashCommands.run(selected, chat)
   if selected.from_prompt_library then
+    add_prompt_tools(chat, selected.config)
+
     local context = selected.config.context
     if context then
       interactions.add_context(selected.config, chat)

tests/interactions/chat/test_context.lua

@@ -63,6 +63,43 @@ T["Context"]["Cannot be added twice with the same id"] = function()
   h.eq(1, child.lua_get([[#_G.chat.context_items]]), "Should only have 1 context item")
 end
 
+T["Context"]["Prompt-library slash commands add a single tool"] = function()
+  child.lua([[
+    local SlashCommands = require("codecompanion.interactions.chat.slash_commands")
+
+    SlashCommands.run({
+      from_prompt_library = true,
+      config = {
+        tools = { "weather" },
+        prompts = {},
+      },
+      context = {},
+    }, _G.chat)
+  ]])
+
+  local in_use = child.lua_get([[_G.chat.tool_registry.in_use]])
+  h.expect_tbl_contains("weather", in_use)
+end
+
+T["Context"]["Prompt-library slash commands add a tool group"] = function()
+  child.lua([[
+    local SlashCommands = require("codecompanion.interactions.chat.slash_commands")
+
+    SlashCommands.run({
+      from_prompt_library = true,
+      config = {
+        tools = { "senior_dev" },
+        prompts = {},
+      },
+      context = {},
+    }, _G.chat)
+  ]])
+
+  local in_use = child.lua_get([[_G.chat.tool_registry.in_use]])
+  h.expect_tbl_contains("func", in_use)
+  h.expect_tbl_contains("cmd", in_use)
+end
+
 T["Context"]["Can be deleted"] = function()
   child.lua([[
     -- Add context_items