CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: get custom tools from a function (#1567)

Browse Source
This commit is contained in:
yetone
2025-03-12 17:57:09 +08:00
committed by GitHub
parent e635452389
commit 9e3e8d06a2
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lua/avante/config.lua
View File

@@ -465,7 +465,7 @@ M._defaults = {
throttle = 600, throttle = 600,
}, },
disabled_tools = {}, ---@type string[] disabled_tools = {}, ---@type string[]
---@type AvanteLLMToolPublic[] ---@type AvanteLLMToolPublic[] | fun(): AvanteLLMToolPublic[]
custom_tools = {}, custom_tools = {},
} }

4
lua/avante/llm_tools.lua
View File

@@ -605,8 +605,10 @@ end
---@return AvanteLLMTool[] ---@return AvanteLLMTool[]
function M.get_tools() function M.get_tools()
local custom_tools = Config.custom_tools
if type(custom_tools) == "function" then custom_tools = custom_tools() end
---@type AvanteLLMTool[] ---@type AvanteLLMTool[]
local unfiltered_tools = vim.list_extend(vim.list_extend({}, M._tools), Config.custom_tools) local unfiltered_tools = vim.list_extend(vim.list_extend({}, M._tools), custom_tools)
return vim return vim
.iter(unfiltered_tools) .iter(unfiltered_tools)
:filter(function(tool) ---@param tool AvanteLLMTool :filter(function(tool) ---@param tool AvanteLLMTool