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

fix: rename llm-tool search to search_keyword (#1364)

Browse Source
This commit is contained in:
Jax Tsai
2025-02-24 22:00:11 +08:00
committed by GitHub
parent b627b335dd
commit 22739f33f2
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lua/avante/llm_tools.lua
View File

@@ -69,7 +69,7 @@ function M.search_files(opts, on_log)
end end
---@type AvanteLLMToolFunc<{ rel_path: string, keyword: string }> ---@type AvanteLLMToolFunc<{ rel_path: string, keyword: string }>
function M.search(opts, on_log) function M.search_keyword(opts, on_log)
local abs_path = get_abs_path(opts.rel_path) local abs_path = get_abs_path(opts.rel_path)
if not has_permission_to_access(abs_path) then return "", "No permission to access path: " .. abs_path end if not has_permission_to_access(abs_path) then return "", "No permission to access path: " .. abs_path end
if not Path:new(abs_path):exists() then return "", "No such file or directory: " .. abs_path end if not Path:new(abs_path):exists() then return "", "No such file or directory: " .. abs_path end
@@ -738,7 +738,7 @@ M._tools = {
}, },
}, },
{ {
name = "search", name = "search_keyword",
description = "Search for a keyword in a directory", description = "Search for a keyword in a directory",
param = { param = {
type = "table", type = "table",

14
tests/llm_tools_spec.lua
View File

@@ -124,7 +124,7 @@ describe("llm_tools", function()
end) end)
end) end)
describe("search", function() describe("search_keyword", function()
local original_exepath = vim.fn.exepath local original_exepath = vim.fn.exepath
after_each(function() vim.fn.exepath = original_exepath end) after_each(function() vim.fn.exepath = original_exepath end)
@@ -147,7 +147,7 @@ describe("llm_tools", function()
file:write("this is nothing") file:write("this is nothing")
file:close() file:close()
local result, err = LlmTools.search({ rel_path = ".", keyword = "searchable" }) local result, err = LlmTools.search_keyword({ rel_path = ".", keyword = "searchable" })
assert.is_nil(err) assert.is_nil(err)
assert.truthy(result:find("searchable.txt")) assert.truthy(result:find("searchable.txt"))
assert.falsy(result:find("nothing.txt")) assert.falsy(result:find("nothing.txt"))
@@ -166,7 +166,7 @@ describe("llm_tools", function()
file:write("content for ag test") file:write("content for ag test")
file:close() file:close()
local result, err = LlmTools.search({ rel_path = ".", keyword = "ag test" }) local result, err = LlmTools.search_keyword({ rel_path = ".", keyword = "ag test" })
assert.is_nil(err) assert.is_nil(err)
assert.is_string(result) assert.is_string(result)
assert.truthy(result:find("ag_test.txt")) assert.truthy(result:find("ag_test.txt"))
@@ -179,7 +179,7 @@ describe("llm_tools", function()
return "" return ""
end end
local result, err = LlmTools.search({ rel_path = ".", keyword = "test" }) local result, err = LlmTools.search_keyword({ rel_path = ".", keyword = "test" })
assert.is_nil(err) assert.is_nil(err)
assert.truthy(result:find("test.txt")) assert.truthy(result:find("test.txt"))
end) end)
@@ -188,18 +188,18 @@ describe("llm_tools", function()
-- Mock exepath to return nothing -- Mock exepath to return nothing
vim.fn.exepath = function() return "" end vim.fn.exepath = function() return "" end
local result, err = LlmTools.search({ rel_path = ".", keyword = "test" }) local result, err = LlmTools.search_keyword({ rel_path = ".", keyword = "test" })
assert.equals("", result) assert.equals("", result)
assert.equals("No search command found", err) assert.equals("No search command found", err)
end) end)
it("should respect path permissions", function() it("should respect path permissions", function()
local result, err = LlmTools.search({ rel_path = "../outside_project", keyword = "test" }) local result, err = LlmTools.search_keyword({ rel_path = "../outside_project", keyword = "test" })
assert.truthy(err:find("No permission to access path")) assert.truthy(err:find("No permission to access path"))
end) end)
it("should handle non-existent paths", function() it("should handle non-existent paths", function()
local result, err = LlmTools.search({ rel_path = "non_existent_dir", keyword = "test" }) local result, err = LlmTools.search_keyword({ rel_path = "non_existent_dir", keyword = "test" })
assert.equals("", result) assert.equals("", result)
assert.truthy(err) assert.truthy(err)
assert.truthy(err:find("No such file or directory")) assert.truthy(err:find("No such file or directory"))