update

backend/controller/agent_cli_api.go

@@ -26,7 +26,7 @@ func (h *AgentCLIHandler) ListSessions(c *gin.Context) {
 		pageSize = 20
 	}
 
-	q := h.db.Model(&model.AgentCLISession{}).Where("user_id = ?", uid)
+	q := h.db.Model(&model.AIRun{}).Where("user_id = ? AND feature = ?", uid, "agent_cli")
 	if mode := c.Query("mode"); mode != "" {
 		mode = agentcli.NormalizeRunMode(mode)
 		if mode == agentcli.RunModeKubernetes {
@@ -36,18 +36,18 @@ func (h *AgentCLIHandler) ListSessions(c *gin.Context) {
 		}
 	}
 	if sid := c.Query("server_id"); sid != "" {
-		q = q.Where("server_id = ?", sid)
+		q = q.Where("subject_type = ? AND subject_id = ?", "host", sid)
 	}
 	if cid := c.Query("cluster_id"); cid != "" {
-		q = q.Where("cluster_id = ?", cid)
+		q = q.Where("subject_id = ?", cid)
 	}
 	if ns := c.Query("namespace"); ns != "" {
-		q = q.Where("namespace = ?", ns)
+		q = q.Where("subject_scope = ?", ns)
 	}
 
 	var total int64
 	q.Count(&total)
-	var sessions []model.AgentCLISession
+	var sessions []model.AIRun
 	if err := q.Order("created_at DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&sessions).Error; err != nil {
 		c.JSON(http.StatusOK, model.Response{Code: 500, Message: "查询失败: " + err.Error()})
 		return
@@ -60,13 +60,13 @@ func (h *AgentCLIHandler) ListSessions(c *gin.Context) {
 func (h *AgentCLIHandler) GetSession(c *gin.Context) {
 	uid := middleware.GetUserID(c)
 	id := c.Param("id")
-	var sess model.AgentCLISession
+	var sess model.AIRun
 	if err := h.db.Where("user_id = ?", uid).First(&sess, id).Error; err != nil {
 		c.JSON(http.StatusOK, model.Response{Code: 404, Message: "会话不存在"})
 		return
 	}
-	var msgs []model.AgentCLIMessage
-	h.db.Where("session_id = ?", id).Order("id ASC").Find(&msgs)
+	var msgs []model.AIRunMessage
+	h.db.Where("run_id = ?", sess.RunID).Order("seq ASC, id ASC").Find(&msgs)
 	c.JSON(http.StatusOK, model.Response{Code: 200, Data: gin.H{"session": sess, "messages": msgs}})
 }
 
@@ -75,16 +75,18 @@ func (h *AgentCLIHandler) GetSession(c *gin.Context) {
 func (h *AgentCLIHandler) DeleteSession(c *gin.Context) {
 	uid := middleware.GetUserID(c)
 	id := c.Param("id")
-	var sess model.AgentCLISession
+	var sess model.AIRun
 	if err := h.db.Where("user_id = ?", uid).First(&sess, id).Error; err != nil {
 		c.JSON(http.StatusOK, model.Response{Code: 404, Message: "会话不存在"})
 		return
 	}
-	if err := h.db.Where("session_id = ?", id).Delete(&model.AgentCLIMessage{}).Error; err != nil {
+	if err := h.db.Where("run_id = ?", sess.RunID).Delete(&model.AIRunMessage{}).Error; err != nil {
 		c.JSON(http.StatusOK, model.Response{Code: 500, Message: "删除消息失败: " + err.Error()})
 		return
 	}
-	if err := h.db.Delete(&model.AgentCLISession{}, id).Error; err != nil {
+	_ = h.db.Where("run_id = ?", sess.RunID).Delete(&model.AIRunSpan{}).Error
+	_ = h.db.Where("run_id = ?", sess.RunID).Delete(&model.OpsCommand{}).Error
+	if err := h.db.Delete(&model.AIRun{}, id).Error; err != nil {
 		c.JSON(http.StatusOK, model.Response{Code: 500, Message: "删除会话失败: " + err.Error()})
 		return
 	}