mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-07-12 07:06:38 +02:00
Add files via upload
This commit is contained in:
@@ -98,6 +98,7 @@ func (h *AgentHandler) prepareMultiAgentSession(req *ChatRequest, c *gin.Context
|
|||||||
builtin.ToolListProjectFacts,
|
builtin.ToolListProjectFacts,
|
||||||
builtin.ToolSearchProjectFacts,
|
builtin.ToolSearchProjectFacts,
|
||||||
builtin.ToolDeprecateProjectFact,
|
builtin.ToolDeprecateProjectFact,
|
||||||
|
builtin.ToolRestoreProjectFact,
|
||||||
builtin.ToolListKnowledgeRiskTypes,
|
builtin.ToolListKnowledgeRiskTypes,
|
||||||
builtin.ToolSearchKnowledgeBase,
|
builtin.ToolSearchKnowledgeBase,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -215,7 +215,9 @@ func (h *ProjectHandler) UpdateFact(c *gin.Context) {
|
|||||||
if req.Summary != "" {
|
if req.Summary != "" {
|
||||||
existing.Summary = req.Summary
|
existing.Summary = req.Summary
|
||||||
}
|
}
|
||||||
existing.Body = req.Body
|
if strings.TrimSpace(req.Body) != "" {
|
||||||
|
existing.Body = req.Body
|
||||||
|
}
|
||||||
if req.Confidence != "" {
|
if req.Confidence != "" {
|
||||||
existing.Confidence = req.Confidence
|
existing.Confidence = req.Confidence
|
||||||
}
|
}
|
||||||
@@ -260,3 +262,22 @@ func (h *ProjectHandler) DeprecateFact(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
c.JSON(http.StatusOK, gin.H{"success": true})
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type restoreFactRequest struct {
|
||||||
|
FactKey string `json:"fact_key" binding:"required"`
|
||||||
|
Confidence string `json:"confidence"` // 可选:confirmed | tentative,默认 tentative
|
||||||
|
}
|
||||||
|
|
||||||
|
// RestoreFact POST /api/projects/:id/facts/restore
|
||||||
|
func (h *ProjectHandler) RestoreFact(c *gin.Context) {
|
||||||
|
var req restoreFactRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.db.RestoreProjectFact(c.Param("id"), req.FactKey, req.Confidence); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
||||||
|
}
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ const WebshellSkillHintMultiAgent = "Skills 包请使用 Eino 多代理内置 `s
|
|||||||
|
|
||||||
// webshellAssistantToolList AI 助手在 WebShell 上下文下允许使用的工具清单(展示给模型用)。
|
// webshellAssistantToolList AI 助手在 WebShell 上下文下允许使用的工具清单(展示给模型用)。
|
||||||
// 注意:此处只是展示字符串,真正的权限限制是在调用方设置的 roleTools 切片里。
|
// 注意:此处只是展示字符串,真正的权限限制是在调用方设置的 roleTools 切片里。
|
||||||
const webshellAssistantToolList = "webshell_exec、webshell_file_list、webshell_file_read、webshell_file_write、record_vulnerability、list_vulnerabilities、get_vulnerability、upsert_project_fact、get_project_fact、list_project_facts、search_project_facts、deprecate_project_fact、list_knowledge_risk_types、search_knowledge_base"
|
const webshellAssistantToolList = "webshell_exec、webshell_file_list、webshell_file_read、webshell_file_write、record_vulnerability、list_vulnerabilities、get_vulnerability、upsert_project_fact、get_project_fact、list_project_facts、search_project_facts、deprecate_project_fact、restore_project_fact、list_knowledge_risk_types、search_knowledge_base"
|
||||||
|
|
||||||
// BuildWebshellAssistantContext 根据连接信息与用户原始消息组装 AI 助手的上下文提示词。
|
// BuildWebshellAssistantContext 根据连接信息与用户原始消息组装 AI 助手的上下文提示词。
|
||||||
// 上下文包含:连接 ID、备注、目标系统(及对应命令集建议)、响应编码、可用工具清单、Skills 加载入口、
|
// 上下文包含:连接 ID、备注、目标系统(及对应命令集建议)、响应编码、可用工具清单、Skills 加载入口、
|
||||||
|
|||||||
Reference in New Issue
Block a user