Add files via upload

internal/database/group.go

@@ -195,10 +195,21 @@ func (db *DB) DeleteGroup(id string) error {
 }
 
 // AddConversationToGroup 将对话添加到分组
+// 注意：一个对话只能属于一个分组，所以在添加新分组之前，会先删除该对话的所有旧分组关联
 func (db *DB) AddConversationToGroup(conversationID, groupID string) error {
-	id := uuid.New().String()
+	// 先删除该对话的所有旧分组关联，确保一个对话只属于一个分组
 	_, err := db.Exec(
-		"INSERT OR REPLACE INTO conversation_group_mappings (id, conversation_id, group_id, created_at) VALUES (?, ?, ?, ?)",
+		"DELETE FROM conversation_group_mappings WHERE conversation_id = ?",
+		conversationID,
+	)
+	if err != nil {
+		return fmt.Errorf("删除对话旧分组关联失败: %w", err)
+	}
+	
+	// 然后插入新的分组关联
+	id := uuid.New().String()
+	_, err = db.Exec(
+		"INSERT INTO conversation_group_mappings (id, conversation_id, group_id, created_at) VALUES (?, ?, ?, ?)",
 		id, conversationID, groupID, time.Now(),
 	)
 	if err != nil {

internal/database/vulnerability.go

@@ -90,7 +90,7 @@ func (db *DB) GetVulnerability(id string) (*Vulnerability, error) {
 }
 
 // ListVulnerabilities 列出漏洞
-func (db *DB) ListVulnerabilities(limit, offset int, conversationID, severity, status string) ([]*Vulnerability, error) {
+func (db *DB) ListVulnerabilities(limit, offset int, id, conversationID, severity, status string) ([]*Vulnerability, error) {
 	query := `
 		SELECT id, conversation_id, title, description, severity, status,
 		       vulnerability_type, target, proof, impact, recommendation,
@@ -100,6 +100,10 @@ func (db *DB) ListVulnerabilities(limit, offset int, conversationID, severity, s
 	`
 	args := []interface{}{}
 
+	if id != "" {
+		query += " AND id = ?"
+		args = append(args, id)
+	}
 	if conversationID != "" {
 		query += " AND conversation_id = ?"
 		args = append(args, conversationID)