feat: Update Redis task keys and refactor AddLLMStormTask to use pend…

…ing task queue

library/db/redis/consts.go

@@ -4,6 +4,10 @@ const (
 	keyPrefix     = "laisky/"
 	keyPrefixTask = keyPrefix + "tasks/"
 
-	// KeyPrefixTaskLLMStorm is the key prefix for LLM storm tasks
-	KeyPrefixTaskLLMStorm = keyPrefixTask + "llm_storm/"
+	// KeyTaskLLMStormPending is the key for LLM storm pending tasks
+	KeyTaskLLMStormPending = keyPrefixTask + "llm_storm/pending"
+	// KeyPrefixTaskLLMStormResult is the key prefix for LLM storm results
+	//
+	//  `KeyPrefixTaskLLMStormResult + <task_id>`
+	KeyPrefixTaskLLMStormResult = keyPrefixTask + "llm_storm/result/"
 )

library/db/redis/tasks.go

@@ -2,28 +2,17 @@ package redis
 
 import (
 	"context"
-	"time"
 
 	"github.com/Laisky/errors/v2"
-	gutils "github.com/Laisky/go-utils/v5"
 )
 
 // AddLLMStormTask adds a new LLMStormTask to the queue
 func (db *DB) AddLLMStormTask(ctx context.Context,
 	prompt string,
 	apiKey string,
 ) (taskID string, err error) {
-	taskID = gutils.UUID7()
-
-	task := &LLMStormTask{
-		TaskID:    taskID,
-		CreatedAt: time.Now(),
-		Prompt:    prompt,
-		APIKey:    apiKey,
-	}
-
-	key := KeyPrefixTaskLLMStorm + task.TaskID
-	if err = db.db.RPush(ctx, key, task); err != nil {
+	task := NewLLMStormTask(prompt, apiKey)
+	if err = db.db.RPush(ctx, KeyTaskLLMStormPending, task); err != nil {
 		return taskID, errors.Wrap(err, "rpush")
 	}