Allow multiple LOAD arguments with wildcard LOAD aggregations

redis · Dec 14, 2023 · 6bfd341 · 6bfd341
1 parent f6a4b49
commit 6bfd341
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/redis/commands/search/aggregation.py b/redis/commands/search/aggregation.py
@@ -321,7 +321,10 @@ def build_args(self) -> List[str]:
         if self._loadall:
             ret.append("LOAD")
             ret.append("*")
-        elif self._loadfields:
+
+        # NOTE(imalinovskyi): Users should be able to use @__key attribute 
+        # in wildcard queries     
+        if self._loadfields:
             ret.append("LOAD")
             ret.append(str(len(self._loadfields)))
             ret.extend(self._loadfields)