100 lines
2.3 KiB
Python
100 lines
2.3 KiB
Python
"""
|
||
Agent 工具集
|
||
|
||
提供 Agent 使用的各种工具，包括：
|
||
- 基础工具（文件操作、代码搜索）
|
||
- 分析工具（模式匹配、数据流分析）
|
||
- 外部安全工具（Semgrep、Bandit等）
|
||
- 协作工具（Think、Agent通信）
|
||
- 报告工具（漏洞报告）
|
||
"""
|
||
|
||
from .base import AgentTool, ToolResult
|
||
from .rag_tool import RAGQueryTool, SecurityCodeSearchTool, FunctionContextTool
|
||
from .pattern_tool import PatternMatchTool
|
||
from .code_analysis_tool import CodeAnalysisTool, DataFlowAnalysisTool, VulnerabilityValidationTool
|
||
from .file_tool import FileReadTool, FileSearchTool, ListFilesTool
|
||
from .sandbox_tool import SandboxTool, SandboxHttpTool, VulnerabilityVerifyTool, SandboxManager
|
||
|
||
# 外部安全工具
|
||
from .external_tools import (
|
||
SemgrepTool,
|
||
BanditTool,
|
||
GitleaksTool,
|
||
NpmAuditTool,
|
||
SafetyTool,
|
||
TruffleHogTool,
|
||
OSVScannerTool,
|
||
)
|
||
|
||
# 🔥 新增：思考和推理工具
|
||
from .thinking_tool import ThinkTool, ReflectTool
|
||
|
||
# 🔥 新增：漏洞报告工具
|
||
from .reporting_tool import CreateVulnerabilityReportTool
|
||
|
||
# 🔥 新增：Agent协作工具
|
||
from .agent_tools import (
|
||
CreateSubAgentTool,
|
||
SendMessageTool,
|
||
ViewAgentGraphTool,
|
||
WaitForMessageTool,
|
||
AgentFinishTool,
|
||
RunSubAgentsTool,
|
||
CollectSubAgentResultsTool,
|
||
)
|
||
|
||
__all__ = [
|
||
# 基础
|
||
"AgentTool",
|
||
"ToolResult",
|
||
|
||
# RAG 工具
|
||
"RAGQueryTool",
|
||
"SecurityCodeSearchTool",
|
||
"FunctionContextTool",
|
||
|
||
# 代码分析
|
||
"PatternMatchTool",
|
||
"CodeAnalysisTool",
|
||
"DataFlowAnalysisTool",
|
||
"VulnerabilityValidationTool",
|
||
|
||
# 文件操作
|
||
"FileReadTool",
|
||
"FileSearchTool",
|
||
"ListFilesTool",
|
||
|
||
# 沙箱
|
||
"SandboxTool",
|
||
"SandboxHttpTool",
|
||
"VulnerabilityVerifyTool",
|
||
"SandboxManager",
|
||
|
||
# 外部安全工具
|
||
"SemgrepTool",
|
||
"BanditTool",
|
||
"GitleaksTool",
|
||
"NpmAuditTool",
|
||
"SafetyTool",
|
||
"TruffleHogTool",
|
||
"OSVScannerTool",
|
||
|
||
# 🔥 思考和推理工具
|
||
"ThinkTool",
|
||
"ReflectTool",
|
||
|
||
# 🔥 漏洞报告工具
|
||
"CreateVulnerabilityReportTool",
|
||
|
||
# 🔥 Agent协作工具
|
||
"CreateSubAgentTool",
|
||
"SendMessageTool",
|
||
"ViewAgentGraphTool",
|
||
"WaitForMessageTool",
|
||
"AgentFinishTool",
|
||
"RunSubAgentsTool",
|
||
"CollectSubAgentResultsTool",
|
||
]
|
||
|