Add files via upload

This commit is contained in:
公明
2025-12-27 20:14:09 +08:00
committed by GitHub
parent 65957b2013
commit 19e493dc8d
3 changed files with 49 additions and 17 deletions
+7
View File
@@ -282,6 +282,13 @@ func New(cfg *config.Config, log *logger.Logger) (*App, error) {
agentHandler: agentHandler,
}
// 设置漏洞工具注册器(内置工具,必须设置)
vulnerabilityRegistrar := func() error {
registerVulnerabilityTool(mcpServer, db, log.Logger)
return nil
}
configHandler.SetVulnerabilityToolRegistrar(vulnerabilityRegistrar)
// 设置知识库初始化器(用于动态初始化,需要在 App 创建后设置)
configHandler.SetKnowledgeInitializer(func() (*handler.KnowledgeHandler, error) {
knowledgeHandler, err := initializeKnowledge(cfg, db, knowledgeDBConn, mcpServer, agentHandler, app, log.Logger)