15ef2a012c
* fix: never attempt to concatenate nil values, use empty strings (#2692) * [pre-commit.ci lite] apply automatic fixes --------- Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>
79 lines
2.0 KiB
Lua
79 lines
2.0 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 require("avante.config").mode == "agentic" and not require("avante.config").behaviour.enable_fastapply
|
|
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 Utils = require("avante.utils")
|
|
|
|
if input.new_str == nil then input.new_str = "" end
|
|
|
|
if input.old_str == nil then input.old_str = "" end
|
|
|
|
-- Remove trailing spaces from the new string
|
|
input.new_str = Utils.remove_trailing_spaces(input.new_str)
|
|
|
|
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
|