fix task article link drawer

2026-05-14 21:14:27 +08:00
parent 34dda524d7
commit 765dae4bf1
15 changed files with 735 additions and 104 deletions
@@ -62,9 +62,9 @@ func (q *Queries) CreateAuditLog(ctx context.Context, arg CreateAuditLogParams)
 }

 const createGenerationTask = `-- name: CreateGenerationTask :one
-INSERT INTO generation_tasks (tenant_id, operator_id, article_id, quota_reservation_id, task_type, request_hash, input_params_json, status)
+INSERT INTO generation_tasks (tenant_id, operator_id, article_id, task_batch_id, quota_reservation_id, task_type, request_hash, input_params_json, status)
 VALUES ($1, $2, $3, $4,
-        $5, $6, $7, 'queued')
+        $5, $6, $7, $8, 'queued')
 RETURNING id
 `

@@ -72,6 +72,7 @@ type CreateGenerationTaskParams struct {
 	TenantID           int64       `json:"tenant_id"`
 	OperatorID         pgtype.Int8 `json:"operator_id"`
 	ArticleID          pgtype.Int8 `json:"article_id"`
+	TaskBatchID        pgtype.Text `json:"task_batch_id"`
 	QuotaReservationID pgtype.Int8 `json:"quota_reservation_id"`
 	TaskType           string      `json:"task_type"`
 	RequestHash        pgtype.Text `json:"request_hash"`
@@ -83,6 +84,7 @@ func (q *Queries) CreateGenerationTask(ctx context.Context, arg CreateGeneration
 		arg.TenantID,
 		arg.OperatorID,
 		arg.ArticleID,
+		arg.TaskBatchID,
 		arg.QuotaReservationID,
 		arg.TaskType,
 		arg.RequestHash,