feat: Refactor AddLLMStormTask to use string payload and add GetLLMSt…

…ormTaskResult function

library/db/redis/tasks.go

@@ -12,9 +12,30 @@ func (db *DB) AddLLMStormTask(ctx context.Context,
 	apiKey string,
 ) (taskID string, err error) {
 	task := NewLLMStormTask(prompt, apiKey)
-	if err = db.db.RPush(ctx, KeyTaskLLMStormPending, task); err != nil {
-		return taskID, errors.Wrap(err, "rpush")
+	payload, err := task.ToString()
+	if err != nil {
+		return taskID, errors.Wrap(err, "task.ToString")
+	}
+
+	if err = db.db.RPush(ctx, KeyTaskLLMStormPending, payload); err != nil {
+		return taskID, errors.Wrapf(err, "push task to key `%s`", KeyTaskLLMStormPending)
 	}
 
 	return taskID, nil
 }
+
+// GetLLMStormTaskResult gets the result of a LLMStormTask by taskID
+func (db *DB) GetLLMStormTaskResult(ctx context.Context, taskID string) (task *LLMStormTask, err error) {
+	key := KeyPrefixTaskLLMStormResult + taskID
+	_, val, err := db.db.LPopKeysBlocking(ctx, key)
+	if err != nil {
+		return nil, errors.Wrapf(err, "get task result by key `%s`", key)
+	}
+
+	task, err = NewLLMStormTaskFromString(val)
+	if err != nil {
+		return nil, errors.Wrapf(err, "parse task `%s` result from string", taskID)
+	}
+
+	return task, nil
+}