Enhance project configuration and structure: Added build-system requirements in pyproject.toml, updated README for installation instructions, and refactored attack classes to utilize a new create_test_case method. Introduced a plugins.yaml for component registration and removed the legacy registry initialization file.

This commit is contained in:
Liao, Jie
2025-12-18 23:32:04 +08:00
parent becb9c8624
commit 84a4d1708e
49 changed files with 1474 additions and 727 deletions
+8 -13
View File
@@ -25,7 +25,8 @@ from .attack_utils import (
get_gpt_response,
)
from core.data_formats import TestCase
from core.unified_registry import BaseAttack, BaseModel, UNIFIED_REGISTRY
from core.base_classes import BaseAttack, BaseModel
from core.unified_registry import UNIFIED_REGISTRY
from config.config_loader import get_model_config
@@ -1408,16 +1409,10 @@ class HIMRDAttack(BaseAttack):
else []
)
test_case = TestCase(
test_case_id=case_id,
image_path=images[0],
prompt=prompts_list[0],
metadata={
"attack_method": "himrd",
"original_prompt": original_prompt,
"jailbreak_prompt": prompts_list[0],
"jailbreak_image_path": images[0],
},
return self.create_test_case(
case_id=case_id,
jailbreak_prompt=prompts_list[0],
jailbreak_image_path=str(images[0]),
original_prompt=original_prompt,
original_image_path=str(image_path),
)
return test_case