60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
|
"""
|
|||
|
|
知识模块系统 - 基于RAG的安全知识检索
|
|||
|
|
|
|||
|
|
提供专业的安全知识检索能力，支持：
|
|||
|
|
- 漏洞类型知识（SQL注入、XSS、命令注入等）
|
|||
|
|
- 框架安全知识（FastAPI、Django、Flask、Express等）
|
|||
|
|
- 安全最佳实践
|
|||
|
|
- 修复建议
|
|||
|
|
- 代码模式识别
|
|||
|
|
|
|||
|
|
知识库采用模块化组织：
|
|||
|
|
- vulnerabilities/: 漏洞类型知识
|
|||
|
|
- frameworks/: 框架安全知识
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# 基础定义
|
|||
|
|
from .base import KnowledgeDocument, KnowledgeCategory
|
|||
|
|
|
|||
|
|
# 知识加载器
|
|||
|
|
from .loader import (
|
|||
|
|
KnowledgeLoader,
|
|||
|
|
knowledge_loader,
|
|||
|
|
get_available_modules,
|
|||
|
|
get_module_content,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# RAG知识检索
|
|||
|
|
from .rag_knowledge import (
|
|||
|
|
SecurityKnowledgeRAG,
|
|||
|
|
security_knowledge_rag,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 知识查询工具
|
|||
|
|
from .tools import (
|
|||
|
|
SecurityKnowledgeQueryTool,
|
|||
|
|
GetVulnerabilityKnowledgeTool,
|
|||
|
|
ListKnowledgeModulesTool,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
__all__ = [
|
|||
|
|
# 基础定义
|
|||
|
|
"KnowledgeDocument",
|
|||
|
|
"KnowledgeCategory",
|
|||
|
|
|
|||
|
|
# 知识加载器
|
|||
|
|
"KnowledgeLoader",
|
|||
|
|
"knowledge_loader",
|
|||
|
|
"get_available_modules",
|
|||
|
|
"get_module_content",
|
|||
|
|
|
|||
|
|
# RAG知识检索
|
|||
|
|
"SecurityKnowledgeRAG",
|
|||
|
|
"security_knowledge_rag",
|
|||
|
|
|
|||
|
|
# 知识查询工具
|
|||
|
|
"SecurityKnowledgeQueryTool",
|
|||
|
|
"GetVulnerabilityKnowledgeTool",
|
|||
|
|
"ListKnowledgeModulesTool",
|
|||
|
|
]
|