feat: add sandbox mode prompt for enhanced user guidance in executing commands

astrbot/core/pipeline/process_stage/method/agent_sub_stages/internal.py

@@ -42,6 +42,7 @@ from ...utils import (
     KNOWLEDGE_BASE_QUERY_TOOL,
     LLM_SAFETY_MODE_SYSTEM_PROMPT,
     PYTHON_TOOL,
+    SANDBOX_MODE_PROMPT,
     decoded_blocked,
     retrieve_knowledge_base,
 )
@@ -481,7 +482,7 @@ class InternalAgentSubStage(Stage):
         req.func_tool.add_tool(PYTHON_TOOL)
         req.func_tool.add_tool(FILE_UPLOAD_TOOL)
         req.func_tool.add_tool(FILE_DOWNLOAD_TOOL)
-        req.system_prompt += "\nYou have access to a sandboxed environment and can execute shell commands and Python code securely.\n"
+        req.system_prompt += f"\n{SANDBOX_MODE_PROMPT}\n"
 
     async def process(
         self, event: AstrMessageEvent, provider_wake_prefix: str

astrbot/core/pipeline/process_stage/utils.py

@@ -27,6 +27,20 @@ Rules:
 - Output same language as the user's input.
 """
 
+SANDBOX_MODE_PROMPT = (
+    "You have access to a sandboxed environment and can execute shell commands and Python code securely."
+    "Your have extended skills library, such as PDF processing, image generation, data analysis, etc. "
+    "Before handling complex tasks, please retrieve and review the documentation in the in /app/skills/ directory. "
+    "If the current task matches the description of a specific skill, prioritize following the workflow defined by that skill."
+    "Use `ls /app/skills/` to list all available skills. "
+    "Use `cat /app/skills/{skill_name}/SKILL.md` to read the documentation of a specific skill."
+    "SKILL.md might be large, you can read the description first, which is located in the YAML frontmatter of the file."
+    "Use shell commands such as grep, sed, awk to extract relevant information from the documentation as needed.\n"
+    "Note:\n"
+    "1. If you use shell, your command will always runs in the /home/<username>/workspace directory.\n"
+    "2. If you use IPython, you would better use absolute paths when dealing with files to avoid confusion.\n"
+)
+
 
 @dataclass
 class KnowledgeBaseQueryTool(FunctionTool[AstrAgentContext]):