124 lines
2.7 KiB
Python
124 lines
2.7 KiB
Python
"""
|
||
DeepAudit Agent 服务模块
|
||
基于 LangGraph 的 AI Agent 代码安全审计
|
||
|
||
架构升级版本 - 支持：
|
||
- 动态Agent树结构
|
||
- 专业知识模块系统
|
||
- Agent间通信机制
|
||
- 完整状态管理
|
||
- Think工具和漏洞报告工具
|
||
|
||
工作流:
|
||
START → Recon → Analysis ⟲ → Verification → Report → END
|
||
|
||
支持动态创建子Agent进行专业化分析
|
||
"""
|
||
|
||
# 从 graph 模块导入主要组件
|
||
from .graph import (
|
||
AgentRunner,
|
||
run_agent_task,
|
||
LLMService,
|
||
AuditState,
|
||
create_audit_graph,
|
||
)
|
||
|
||
# 事件管理
|
||
from .event_manager import EventManager, AgentEventEmitter
|
||
|
||
# Agent 类
|
||
from .agents import (
|
||
BaseAgent, AgentConfig, AgentResult,
|
||
OrchestratorAgent, ReconAgent, AnalysisAgent, VerificationAgent,
|
||
)
|
||
|
||
# 🔥 新增：核心模块（状态管理、注册表、消息）
|
||
from .core import (
|
||
AgentState, AgentStatus,
|
||
AgentRegistry, agent_registry,
|
||
AgentMessage, MessageType, MessagePriority, MessageBus,
|
||
)
|
||
|
||
# 🔥 新增：知识模块系统（基于RAG）
|
||
from .knowledge import (
|
||
KnowledgeLoader, knowledge_loader,
|
||
get_available_modules, get_module_content,
|
||
SecurityKnowledgeRAG, security_knowledge_rag,
|
||
SecurityKnowledgeQueryTool, GetVulnerabilityKnowledgeTool,
|
||
)
|
||
|
||
# 🔥 新增：协作工具
|
||
from .tools import (
|
||
ThinkTool, ReflectTool,
|
||
CreateVulnerabilityReportTool,
|
||
FinishScanTool,
|
||
CreateSubAgentTool, SendMessageTool, ViewAgentGraphTool,
|
||
WaitForMessageTool, AgentFinishTool,
|
||
)
|
||
|
||
# 🔥 新增：遥测模块
|
||
from .telemetry import Tracer, get_global_tracer, set_global_tracer
|
||
|
||
|
||
__all__ = [
|
||
# 核心 Runner
|
||
"AgentRunner",
|
||
"run_agent_task",
|
||
"LLMService",
|
||
|
||
# LangGraph
|
||
"AuditState",
|
||
"create_audit_graph",
|
||
|
||
# 事件管理
|
||
"EventManager",
|
||
"AgentEventEmitter",
|
||
|
||
# Agent 类
|
||
"BaseAgent",
|
||
"AgentConfig",
|
||
"AgentResult",
|
||
"OrchestratorAgent",
|
||
"ReconAgent",
|
||
"AnalysisAgent",
|
||
"VerificationAgent",
|
||
|
||
# 🔥 核心模块
|
||
"AgentState",
|
||
"AgentStatus",
|
||
"AgentRegistry",
|
||
"agent_registry",
|
||
"AgentMessage",
|
||
"MessageType",
|
||
"MessagePriority",
|
||
"MessageBus",
|
||
|
||
# 🔥 知识模块（基于RAG）
|
||
"KnowledgeLoader",
|
||
"knowledge_loader",
|
||
"get_available_modules",
|
||
"get_module_content",
|
||
"SecurityKnowledgeRAG",
|
||
"security_knowledge_rag",
|
||
"SecurityKnowledgeQueryTool",
|
||
"GetVulnerabilityKnowledgeTool",
|
||
|
||
# 🔥 协作工具
|
||
"ThinkTool",
|
||
"ReflectTool",
|
||
"CreateVulnerabilityReportTool",
|
||
"FinishScanTool",
|
||
"CreateSubAgentTool",
|
||
"SendMessageTool",
|
||
"ViewAgentGraphTool",
|
||
"WaitForMessageTool",
|
||
"AgentFinishTool",
|
||
|
||
# 🔥 遥测模块
|
||
"Tracer",
|
||
"get_global_tracer",
|
||
"set_global_tracer",
|
||
]
|
||
|