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

fix: custom prompts (#1328)

Browse Source
This commit is contained in:
yetone
2025-02-21 00:45:41 +08:00
committed by GitHub
parent ac86a3371e
commit fa89b147db
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lua/avante/path.lua
View File

@@ -107,14 +107,19 @@ Prompt.get = function(project_root)
for _, entry in ipairs(scanner) do for _, entry in ipairs(scanner) do
local file = Path:new(entry) local file = Path:new(entry)
if file:is_file() then if file:is_file() then
if not entry:match("%.avanterules$") then goto continue end
if entry:find("planning") and Prompt.templates.planning == nil then if entry:find("planning") and Prompt.templates.planning == nil then
Utils.info(string.format("Using %s as planning system prompt", entry))
Prompt.templates.planning = file:read() Prompt.templates.planning = file:read()
elseif entry:find("editing") and Prompt.templates.editing == nil then elseif entry:find("editing") and Prompt.templates.editing == nil then
Utils.info(string.format("Using %s as editing system prompt", entry))
Prompt.templates.editing = file:read() Prompt.templates.editing = file:read()
elseif entry:find("suggesting") and Prompt.templates.suggesting == nil then elseif entry:find("suggesting") and Prompt.templates.suggesting == nil then
Utils.info(string.format("Using %s as suggesting system prompt", entry))
Prompt.templates.suggesting = file:read() Prompt.templates.suggesting = file:read()
end end
end end
::continue::
end end
Path:new(debug.getinfo(1).source:match("@?(.*/)"):gsub("/lua/avante/path.lua$", "") .. "templates") Path:new(debug.getinfo(1).source:match("@?(.*/)"):gsub("/lua/avante/path.lua$", "") .. "templates")