fix task article link drawer

server/internal/tenant/repository/audit_repo.go

@@ -25,6 +25,7 @@ type GenerationTaskInput struct {
 	OperatorID         int64
 	TenantID           int64
 	ArticleID          *int64
+	TaskBatchID        *string
 	QuotaReservationID *int64
 	TaskType           string
 	RequestHash        *string
@@ -91,6 +92,7 @@ func (r *auditRepository) CreateGenerationTask(ctx context.Context, input Genera
 		OperatorID:         pgInt8(operatorID),
 		TenantID:           input.TenantID,
 		ArticleID:          pgInt8(input.ArticleID),
+		TaskBatchID:        pgText(input.TaskBatchID),
 		QuotaReservationID: pgInt8(input.QuotaReservationID),
 		TaskType:           input.TaskType,
 		RequestHash:        pgText(input.RequestHash),

server/internal/tenant/repository/generated/audit.sql.go

@@ -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,

server/internal/tenant/repository/queries/audit.sql

@@ -17,9 +17,9 @@ VALUES (sqlc.arg(operator_id), sqlc.arg(tenant_id), sqlc.arg(module), sqlc.arg(a
         sqlc.arg(before_json), sqlc.arg(after_json), sqlc.arg(result), sqlc.arg(error_message));
 
 -- name: CreateGenerationTask :one
-INSERT INTO generation_tasks (tenant_id, operator_id, article_id, quota_reservation_id, task_type, request_hash, input_params_json, status)
-VALUES (sqlc.arg(tenant_id), sqlc.arg(operator_id), sqlc.arg(article_id), sqlc.arg(quota_reservation_id),
-        sqlc.arg(task_type), sqlc.arg(request_hash), sqlc.arg(input_params_json), 'queued')
+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 (sqlc.arg(tenant_id), sqlc.arg(operator_id), sqlc.arg(article_id), sqlc.arg(task_batch_id),
+        sqlc.arg(quota_reservation_id), sqlc.arg(task_type), sqlc.arg(request_hash), sqlc.arg(input_params_json), 'queued')
 RETURNING id;
 
 -- name: CreateTaskRecord :one