LLM_check/__init__.py at main · williamli-15/LLM_check · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
"""
恶意编辑检测系统

基于LangGraph多Agent协作的智能模型恶意编辑检测系统
集成DEED方法和TaskTracker技术，提供全面的模型安全检测能力
"""

__version__ = "1.0.0"
__author__ = "Malicious Edit Detection Team"
__description__ = "LangGraph-based Multi-Agent Collaborative Intelligent Model Malicious Editing Detection System"

# 核心组件导入
from .core.state import DetectionState, EditDetectionFeatures, HarmfulnessMetrics, IntentAnalysisResult, SecurityAssessment
from .core.deed_detector import DEEDDetector
from .core.langgraph_workflow import LangGraphMaliciousEditDetectionWorkflow

# Agent导入
from .agents.base_agent import BaseAgent
from .agents.edit_detection_agent import EditDetectionAgent
from .agents.harmfulness_analysis_agent import HarmfulnessAnalysisAgent
from .agents.intent_analysis_agent import IntentAnalysisAgent
from .agents.security_assessment_agent import SecurityAssessmentAgent
from .agents.summarizer_agent import SummarizerAgent
from .agents.orchestrator_agent import OrchestratorAgent, DetectionPipeline

# 验证框架
from .validation.keti_validation import KETIValidationFramework

# 配置管理
from .config.config_manager import ConfigManager, get_config

# 便捷接口
__all__ = [
    # 版本信息
    "__version__",
    "__author__",
    "__description__",

    # 核心组件
    "DetectionState",
    "EditDetectionFeatures",
    "HarmfulnessMetrics",
    "IntentAnalysisResult",
    "SecurityAssessment",
    "DEEDDetector",
    "LangGraphMaliciousEditDetectionWorkflow",

    # Agent组件
    "BaseAgent",
    "EditDetectionAgent",
    "HarmfulnessAnalysisAgent",
    "IntentAnalysisAgent",
    "SecurityAssessmentAgent",
    "SummarizerAgent",
    "OrchestratorAgent",
    "DetectionPipeline",

    # 验证框架
    "KETIValidationFramework",

    # 配置管理
    "ConfigManager",
    "get_config",
]