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",
|
||
]
|