feat: Update GetLLMStormTaskResult to use GetItem instead of LPopKeys…

…Blocking for improved task result retrieval
Laisky · Feb 5, 2025 · a5d694e · a5d694e
1 parent 934837c
commit a5d694e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/library/db/redis/tasks.go b/library/db/redis/tasks.go
@@ -28,8 +28,7 @@ func (db *DB) AddLLMStormTask(ctx context.Context,
 // 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
-	// LPopKeysBlocking returns (usedKey, poppedValue, error)
-	_, val, err := db.db.LPopKeysBlocking(ctx, key)
+	val, err := db.db.GetItem(ctx, key)
 	if err != nil {
 		return nil, errors.Wrapf(err, "failed to get task result by key `%s`", key)
 	}