diff --git a/template_manager.py b/template_manager.py index 4f72c8a..6df24dd 100644 --- a/template_manager.py +++ b/template_manager.py @@ -132,7 +132,12 @@ DEFAULT_TEMPLATES = { "image": "agnettaiji.azurecr.io/ai-agents/a2a-litellm-agent:latest", "port": 8000, "agent_framework": "a2a", - "env_requirements": {}, + "env_requirements": { + "optional": { + "AGENT_ACCESS_TOKEN": "客户端访问令牌;若设置,A2A 请求必须携带 X-Agent-Access-Token", + "HEICODE_AGENT_ID": "上层系统分配的 agent_id,用于 health / agent card 暴露与联调排查", + } + }, }, "coding_a2a_agent": { "display_name": "Coding A2A Agent", @@ -153,6 +158,8 @@ DEFAULT_TEMPLATES = { "AGENT_ROLE_NAME": "启动时指定角色名称,例如 backend / reviewer / planner", "AGENT_INSTRUCTION_TEXT": "启动时注入的角色/行为说明文本,支持类似 AGENTS.md / claude.md 内容", "AGENT_INSTRUCTION_FILE": "启动时读取的角色说明文件路径,内容会并入系统提示词", + "AGENT_ACCESS_TOKEN": "客户端访问令牌;若设置,A2A 请求必须携带 X-Agent-Access-Token", + "HEICODE_AGENT_ID": "上层系统分配的 agent_id,用于 health / agent card 暴露与联调排查", "SERVICE_PORT": "服务端口,默认 8000", }, }, diff --git a/tests/test_template_agent_contract.py b/tests/test_template_agent_contract.py index c9afa3f..a5e9bf4 100644 --- a/tests/test_template_agent_contract.py +++ b/tests/test_template_agent_contract.py @@ -106,3 +106,13 @@ def test_dedicated_template_agent_contract_doc_exists(): assert "POST /agents/{agent_name}/stop" in doc_source assert "DELETE /agents/{agent_name}" in doc_source assert "X-Agent-Access-Token" in doc_source + + +def test_template_manager_exposes_agent_access_token_envs(): + """Template definitions should advertise the HM access-token envs explicitly.""" + template_source = _read("template_manager.py") + + assert '"a2a_litellm_agent": {' in template_source + assert '"coding_a2a_agent": {' in template_source + assert '"AGENT_ACCESS_TOKEN": "客户端访问令牌;若设置,A2A 请求必须携带 X-Agent-Access-Token"' in template_source + assert '"HEICODE_AGENT_ID": "上层系统分配的 agent_id,用于 health / agent card 暴露与联调排查"' in template_source