fix: prompts (#1715)

This commit is contained in:
yetone
2025-03-25 23:16:03 +08:00
committed by GitHub
parent a2aec079c9
commit d82eb8534e
4 changed files with 40 additions and 14 deletions

View File

@@ -10,13 +10,33 @@ local M = setmetatable({}, Base)
M.name = "bash"
local banned_commands = {
"alias",
"curl",
"curlie",
"wget",
"axel",
"aria2c",
"nc",
"telnet",
"lynx",
"w3m",
"links",
"httpie",
"xh",
"http-prompt",
"chrome",
"firefox",
"safari",
}
M.get_description = function()
local provider = Providers[Config.provider]
if Config.provider:match("copilot") and provider.model and provider.model:match("gpt") then
return [[Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures. Do not use bash command to read or modify files, or you will be fired!]]
end
return [[Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures.
local res = ([[Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures.
Do not use bash command to read or modify files, or you will be fired!
@@ -28,7 +48,7 @@ Before executing the command, please follow these steps:
2. Security Check:
- For security and to limit the threat of a prompt injection attack, some commands are limited or banned. If you use a disallowed command, you will receive an error message explaining the restriction. Explain the error to the User.
- Verify that the command is not one of the banned commands: ${BANNED_COMMANDS.join(', ')}.
- Verify that the command is not one of the banned commands: ${BANNED_COMMANDS}.
3. Command Execution:
- After ensuring proper quoting, execute the command.
@@ -155,14 +175,15 @@ gh pr create --title "the pr title" --body "$(cat <<'EOF'
## Test plan
[Checklist of TODOs for testing the pull request...]
🤖 Generated with ${process.env.USER_TYPE === 'ant' ? `[${PRODUCT_NAME}](${PRODUCT_URL})` : PRODUCT_NAME}
🤖 Generated with [avante.nvim](https://github.com/yetone/avante.nvim)
EOF
)"
</example>
Important:
- Return an empty response - the user will see the gh output directly
- Never update git config]]
- Never update git config]]):gsub("${BANNED_COMMANDS}", table.concat(banned_commands, ", "))
return res
end
---@type AvanteLLMToolParam