wuyang/zk-data-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enforce session workspace boundaries

Browse Source
This commit is contained in:
武阳
2026-05-08 17:09:07 +08:00
parent 5b14f55736
commit 0bbba2b936
13 changed files with 509 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_agent_prompting.py
+14 -11
View File
@@ -34,6 +34,9 @@ class AgentPromptingTests(unittest.TestCase):
prompt = render_system_prompt(parts)
self.assertIn('# 系统规则', prompt)
self.assertIn('# 处理任务', prompt)
self.assertIn('# 工作空间边界', prompt)
self.assertIn('当前 session 目录是本轮任务的默认工作区', prompt)
self.assertIn('平台服务代码目录始终只读', prompt)
self.assertIn('# 使用工具', prompt)
self.assertIn('# Skills', prompt)
self.assertIn('product-data', prompt)
@@ -100,7 +103,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Plugins', prompt)
self.assertIn('# 插件', prompt)
def test_prompt_builder_mentions_hook_policy_when_manifest_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -119,7 +122,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Hook Policy', prompt)
self.assertIn('# Hook 策略', prompt)
def test_prompt_builder_mentions_mcp_when_manifest_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -162,7 +165,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Search', prompt)
self.assertIn('# 搜索', prompt)
self.assertIn('web_search', prompt)
def test_prompt_builder_mentions_remote_when_manifest_is_loaded(self) -> None:
@@ -185,7 +188,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Remote', prompt)
self.assertIn('# 远程环境', prompt)
def test_prompt_builder_mentions_account_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -204,7 +207,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Account', prompt)
self.assertIn('# 账号', prompt)
def test_prompt_builder_mentions_ask_user_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -223,7 +226,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Ask User', prompt)
self.assertIn('# 询问用户', prompt)
def test_prompt_builder_mentions_config_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -244,7 +247,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Config', prompt)
self.assertIn('# 配置', prompt)
def test_prompt_builder_mentions_lsp_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -261,7 +264,7 @@ class AgentPromptingTests(unittest.TestCase):
prompt = render_system_prompt(parts)
self.assertIn('# LSP', prompt)
self.assertIn('Use the LSP tool', prompt)
self.assertIn('使用 LSP 工具', prompt)
def test_prompt_builder_mentions_tasks_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -278,7 +281,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Tasks', prompt)
self.assertIn('# 任务', prompt)
def test_prompt_builder_mentions_teams_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -297,7 +300,7 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Teams', prompt)
self.assertIn('# 团队', prompt)
def test_prompt_builder_mentions_planning_when_runtime_is_loaded(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
@@ -317,4 +320,4 @@ class AgentPromptingTests(unittest.TestCase):
)
prompt = render_system_prompt(parts)
self.assertIn('# Planning', prompt)
self.assertIn('# 计划', prompt)