deepagents----/tests/test_subagents.py

"""
SubAgent配置测试

测试所有SubAgent配置是否符合DeepAgents框架规范
"""

import sys
import os

# 添加src目录到Python路径
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))

import pytest
from src.agents.subagents import (
    get_subagent_configs,
    validate_subagent_config,
    get_validated_subagent_configs
)


class TestSubAgentConfigs:
    """SubAgent配置测试类"""

    def test_subagent_count(self):
        """测试SubAgent数量"""
        configs = get_subagent_configs()
        assert len(configs) == 6, f"应该有6个SubAgent，实际有{len(configs)}个"

    def test_required_fields(self):
        """测试所有必需字段是否存在"""
        configs = get_subagent_configs()
        required_fields = ["name", "description", "system_prompt"]

        for config in configs:
            for field in required_fields:
                assert field in config, f"SubAgent {config.get('name', 'unknown')} 缺少必需字段: {field}"

    def test_name_format(self):
        """测试name是否使用kebab-case格式"""
        configs = get_subagent_configs()

        for config in configs:
            name = config["name"]
            # 检查是否只包含小写字母和连字符
            assert all(c.islower() or c == '-' for c in name), \
                f"SubAgent name必须使用kebab-case格式: {name}"
            # 不应该以连字符开始或结束
            assert not name.startswith('-') and not name.endswith('-'), \
                f"SubAgent name不应该以连字符开始或结束: {name}"

    def test_system_prompt_not_empty(self):
        """测试system_prompt不为空"""
        configs = get_subagent_configs()

        for config in configs:
            system_prompt = config.get("system_prompt", "")
            assert system_prompt.strip(), \
                f"SubAgent {config['name']} 的system_prompt不能为空"
            # 检查system_prompt应该相当详细（至少500字符）
            assert len(system_prompt) > 500, \
                f"SubAgent {config['name']} 的system_prompt过短（应该>500字符）"

    def test_no_prompt_field(self):
        """测试配置中不应该使用'prompt'字段（常见错误）"""
        configs = get_subagent_configs()

        for config in configs:
            assert "prompt" not in config, \
                f"SubAgent {config['name']} 使用了错误的字段'prompt'，应该使用'system_prompt'"

    def test_description_present(self):
        """测试description字段存在且有意义"""
        configs = get_subagent_configs()

        for config in configs:
            description = config.get("description", "")
            assert description.strip(), \
                f"SubAgent {config['name']} 的description不能为空"
            # 描述应该简洁（10-100字符）
            assert 10 <= len(description) <= 200, \
                f"SubAgent {config['name']} 的description长度不合适（应该10-200字符）"

    def test_tools_field_type(self):
        """测试tools字段类型正确"""
        configs = get_subagent_configs()

        for config in configs:
            if "tools" in config:
                assert isinstance(config["tools"], list), \
                    f"SubAgent {config['name']} 的tools字段应该是列表"

    def test_specific_subagent_names(self):
        """测试6个SubAgent的具体名称"""
        configs = get_subagent_configs()
        expected_names = {
            "intent-analyzer",
            "search-orchestrator",
            "source-validator",
            "content-analyzer",
            "confidence-evaluator",
            "report-generator"
        }

        actual_names = {config["name"] for config in configs}
        assert actual_names == expected_names, \
            f"SubAgent名称不匹配。期望: {expected_names}, 实际: {actual_names}"

    def test_system_prompt_mentions_files(self):
        """测试system_prompt是否提到虚拟文件系统路径"""
        configs = get_subagent_configs()

        # 某些SubAgent应该在system_prompt中提到文件路径
        file_related_agents = [
            "intent-analyzer",
            "search-orchestrator",
            "source-validator",
            "content-analyzer",
            "confidence-evaluator",
            "report-generator"
        ]

        for config in configs:
            if config["name"] in file_related_agents:
                system_prompt = config["system_prompt"]
                # 检查是否提到虚拟文件系统（以/开头的路径）
                assert "/" in system_prompt, \
                    f"SubAgent {config['name']} 的system_prompt应该提到虚拟文件系统路径"

    def test_search_orchestrator_has_tools(self):
        """测试search-orchestrator应该有搜索工具"""
        configs = get_subagent_configs()

        search_orchestrator = next(
            (c for c in configs if c["name"] == "search-orchestrator"),
            None
        )

        assert search_orchestrator is not None, "未找到search-orchestrator"
        assert "tools" in search_orchestrator, "search-orchestrator应该有tools字段"
        assert len(search_orchestrator["tools"]) > 0, \
            "search-orchestrator应该至少有一个工具"

    def test_validate_function(self):
        """测试validate_subagent_config函数"""
        # 有效配置
        valid_config = {
            "name": "test-agent",
            "description": "测试agent",
            "system_prompt": "这是一个测试prompt"
        }
        assert validate_subagent_config(valid_config) == True

        # 缺少必需字段
        invalid_config = {
            "name": "test-agent",
            "description": "测试agent"
            # 缺少system_prompt
        }
        with pytest.raises(ValueError, match="缺少必需字段"):
            validate_subagent_config(invalid_config)

        # 错误的name格式
        invalid_name_config = {
            "name": "TestAgent",  # 应该是kebab-case
            "description": "测试agent",
            "system_prompt": "测试"
        }
        with pytest.raises(ValueError, match="kebab-case"):
            validate_subagent_config(invalid_name_config)

    def test_get_validated_configs(self):
        """测试get_validated_subagent_configs函数"""
        configs = get_validated_subagent_configs()
        assert len(configs) == 6, "应该返回6个经过验证的SubAgent配置"

    def test_system_prompt_structure(self):
        """测试system_prompt是否有良好的结构"""
        configs = get_subagent_configs()

        for config in configs:
            system_prompt = config["system_prompt"]

            # 应该有清晰的任务说明
            assert any(keyword in system_prompt for keyword in ["任务", "流程", "步骤"]), \
                f"SubAgent {config['name']} 的system_prompt应该包含任务说明"

            # 应该有输入输出说明
            assert any(keyword in system_prompt for keyword in ["输入", "输出", "读取", "写入"]), \
                f"SubAgent {config['name']} 的system_prompt应该包含输入输出说明"

    def test_confidence_evaluator_mentions_formula(self):
        """测试confidence-evaluator是否提到置信度计算公式"""
        configs = get_subagent_configs()

        confidence_evaluator = next(
            (c for c in configs if c["name"] == "confidence-evaluator"),
            None
        )

        assert confidence_evaluator is not None
        system_prompt = confidence_evaluator["system_prompt"]

        # 应该提到公式和百分比
        assert "50%" in system_prompt and "30%" in system_prompt and "20%" in system_prompt, \
            "confidence-evaluator应该包含置信度计算公式（50%+30%+20%）"

    def test_source_validator_mentions_tiers(self):
        """测试source-validator是否提到Tier分级"""
        configs = get_subagent_configs()

        source_validator = next(
            (c for c in configs if c["name"] == "source-validator"),
            None
        )

        assert source_validator is not None
        system_prompt = source_validator["system_prompt"]

        # 应该提到Tier 1-4
        for tier in ["Tier 1", "Tier 2", "Tier 3", "Tier 4"]:
            assert tier in system_prompt or tier.replace(" ", "") in system_prompt, \
                f"source-validator应该包含{tier}分级说明"


def print_subagent_summary():
    """打印SubAgent配置摘要"""
    print("\n" + "=" * 60)
    print("SubAgent配置摘要")
    print("=" * 60)

    configs = get_subagent_configs()

    for i, config in enumerate(configs, 1):
        print(f"\n{i}. {config['name']}")
        print(f"   描述: {config['description']}")
        print(f"   System Prompt长度: {len(config['system_prompt'])} 字符")
        if "tools" in config:
            print(f"   工具数量: {len(config['tools'])}")
        else:
            print(f"   工具数量: 0")

    print("\n" + "=" * 60)


if __name__ == "__main__":
    # 运行测试
    print("运行SubAgent配置测试...\n")

    # 打印摘要
    print_subagent_summary()

    # 使用pytest运行测试
    pytest.main([__file__, "-v", "--tb=short"])