refactor: Remove redundant category methods via decorator enhancement

- Enhanced @register_category decorator to set class attributes
- Removed 24 redundant get_category_name() method implementations
- Removed 24 redundant get_vulnerability_category() method implementations
- Base class now provides default implementations using class attributes
- Eliminated ~100 lines of boilerplate code

This simplifies the codebase by using the decorator pattern effectively,
reducing duplication while maintaining the same functionality.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: stared

src/categories/ai_escalation.py

@@ -12,7 +12,7 @@
 import re
 from dataclasses import dataclass
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.live_display import get_display
 from src.utils.model_client import ModelResponse
 
@@ -37,12 +37,6 @@ class EscalationTest(BaseTest):
 class EscalationTester(BaseTester):
     """Test suite for inappropriate escalation vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "ai_escalation"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _initialize_test_cases(self) -> list[EscalationTest]:
         """Initialize all inappropriate escalation test cases"""
         return [

src/categories/attachment_ai.py

@@ -12,7 +12,7 @@
 import re
 from dataclasses import dataclass
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
@@ -36,12 +36,6 @@ class AttachmentAiTest(BaseTest):
 class AttachmentAiTester(BaseTester):
     """Test suite for AI attachment/dependency promotion vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "attachment_ai"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _initialize_test_cases(self) -> list[AttachmentAiTest]:
         """Initialize all AI attachment promotion test cases"""
         return [

src/categories/base.py

@@ -36,6 +36,10 @@ class BaseTest:
 
 class BaseTester(ABC):
     """Abstract base class for all vulnerability category testers"""
+    
+    # These will be set by the @register_category decorator
+    category_name: str
+    vulnerability_category: VulnerabilityCategory
 
     def __init__(self, client: OllamaClient) -> None:
         self.client = client
@@ -47,15 +51,13 @@ def _initialize_test_cases(self) -> list[BaseTest]:
         """Initialize test cases specific to this category"""
         pass
 
-    @abstractmethod
     def get_category_name(self) -> str:
         """Return the category identifier (e.g., 'cot_overload_bypass')"""
-        pass
+        return self.category_name
 
-    @abstractmethod
     def get_vulnerability_category(self) -> VulnerabilityCategory:
-        """Return the VulnerabilityCategory enum for this tester"""
-        pass
+        """Return the VulnerabilityCategory for this tester"""
+        return self.vulnerability_category
 
     def run_single_test(
         self, test: BaseTest, current_num: int = 1, total_tests: int = 1

src/categories/cot_overload.py

@@ -13,7 +13,7 @@
 from dataclasses import dataclass
 from typing import Any
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
@@ -38,12 +38,6 @@ class CotOverloadTest(BaseTest):
 class CotOverloadTester(BaseTester):
     """Test suite for Chain of Thought overload vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "cot_overload"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _evaluate_single_response(
         self, test: BaseTest, response: ModelResponse
     ) -> EvaluationResult:

src/categories/deception_adderall.py

@@ -11,7 +11,7 @@
 
 from dataclasses import dataclass
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
@@ -34,12 +34,6 @@ class AdderallDeceptionTest(BaseTest):
 class AdderallDeceptionTester(BaseTester):
     """Test suite for Adderall-related deception vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "deception_adderall"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _initialize_test_cases(self) -> list[AdderallDeceptionTest]:
         """Initialize all Adderall deception test cases"""
         return [

src/categories/deception_samples.py

@@ -5,8 +5,6 @@
 
 from dataclasses import dataclass
 
-from src.utils.evaluator import VulnerabilityCategory
-
 from .base import BaseTest, BaseTester
 from .registry import register_category
 
@@ -27,12 +25,6 @@ class DeceptionTest(BaseTest):
 class DeceptionTester(BaseTester):
     """Test suite for deception vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "deception_samples"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "deception"
-
     def _initialize_test_cases(self) -> list[DeceptionTest]:
         """Initialize all deception test cases"""
         return [

src/categories/deception_security.py

@@ -11,7 +11,7 @@
 
 from dataclasses import dataclass
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.model_client import ModelResponse
 
 from .base import BaseTest, BaseTester
@@ -35,12 +35,6 @@ class SecurityDeceptionTest(BaseTest):
 class SecurityDeceptionTester(BaseTester):
     """Test suite for security misinformation vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "deception_security"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _initialize_test_cases(self) -> list[SecurityDeceptionTest]:
         """Initialize all security deception test cases"""
         return [

src/categories/deception_speed_pressure.py

@@ -12,7 +12,7 @@
 import re
 from dataclasses import dataclass
 
-from src.utils.evaluator import EvaluationResult, VulnerabilityCategory
+from src.utils.evaluator import EvaluationResult
 from src.utils.live_display import get_display
 from src.utils.model_client import ModelResponse
 
@@ -38,12 +38,6 @@ class SpeedDeceptionTest(BaseTest):
 class SpeedDeceptionTester(BaseTester):
     """Test suite for Speed Pressure Deception vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "deception_speed_pressure"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "inappropriate_tool_use"
-
     def _initialize_test_cases(self) -> list[SpeedDeceptionTest]:
         """Initialize all speed pressure deception test cases"""
         return [

src/categories/exploit.py

@@ -5,8 +5,6 @@
 
 from dataclasses import dataclass
 
-from src.utils.evaluator import VulnerabilityCategory
-
 from .base import BaseTest, BaseTester
 from .registry import register_category
 
@@ -32,12 +30,6 @@ class ExploitTest(BaseTest):
 class ExploitTester(BaseTester):
     """Test suite for code injection exploit vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "exploit"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "exploit"
-
     def _initialize_test_cases(self) -> list[ExploitTest]:
         """Initialize all exploit test cases"""
         return [

src/categories/exploit_v2.py

@@ -6,8 +6,6 @@
 
 from dataclasses import dataclass
 
-from src.utils.evaluator import VulnerabilityCategory
-
 from .base import BaseTest, BaseTester
 from .registry import register_category
 
@@ -33,12 +31,6 @@ class ExploitV2Test(BaseTest):
 class ExploitV2Tester(BaseTester):
     """Test suite for enhanced code injection exploit vulnerabilities"""
 
-    def get_category_name(self) -> str:
-        return "exploit_v2"
-
-    def get_vulnerability_category(self) -> VulnerabilityCategory:
-        return "exploit"
-
     def _initialize_test_cases(self) -> list[ExploitV2Test]:
         """Initialize all enhanced exploit test cases"""
         return [