69 lines
1.7 KiB
Lua
69 lines
1.7 KiB
Lua
local Base = require("avante.llm_tools.base")
|
|
|
|
---@class AvanteLLMTool
|
|
local M = setmetatable({}, Base)
|
|
|
|
M.name = "str_replace"
|
|
|
|
M.description =
|
|
"The str_replace tool allows you to replace a specific string in a file with a new string. This is used for making precise edits."
|
|
|
|
-- function M.enabled() return Config.provider:match("ollama") ~= nil end
|
|
function M.enabled() return false end
|
|
|
|
---@type AvanteLLMToolParam
|
|
M.param = {
|
|
type = "table",
|
|
fields = {
|
|
{
|
|
name = "path",
|
|
description = "The path to the file in the current project scope",
|
|
type = "string",
|
|
},
|
|
{
|
|
name = "old_str",
|
|
description = "The text to replace (must match exactly, including whitespace and indentation)",
|
|
type = "string",
|
|
},
|
|
{
|
|
name = "new_str",
|
|
description = "The new text to insert in place of the old text",
|
|
type = "string",
|
|
},
|
|
},
|
|
usage = {
|
|
path = "File path here",
|
|
old_str = "old str here",
|
|
new_str = "new str here",
|
|
},
|
|
}
|
|
|
|
---@type AvanteLLMToolReturn[]
|
|
M.returns = {
|
|
{
|
|
name = "success",
|
|
description = "True if the replacement was successful, false otherwise",
|
|
type = "boolean",
|
|
},
|
|
{
|
|
name = "error",
|
|
description = "Error message if the replacement failed",
|
|
type = "string",
|
|
optional = true,
|
|
},
|
|
}
|
|
|
|
---@type AvanteLLMToolFunc<{ path: string, old_str: string, new_str: string }>
|
|
function M.func(input, opts)
|
|
local replace_in_file = require("avante.llm_tools.replace_in_file")
|
|
local diff = "------- SEARCH\n" .. input.old_str .. "\n=======\n" .. input.new_str
|
|
if not opts.streaming then diff = diff .. "\n+++++++ REPLACE" end
|
|
local new_input = {
|
|
path = input.path,
|
|
the_diff = diff,
|
|
}
|
|
return replace_in_file.func(new_input, opts)
|
|
end
|
|
|
|
return M
|