From 2eaaa5873650faf3f297f4642208ded99fcda3c3 Mon Sep 17 00:00:00 2001
From: Shaya Potter <spotter@gmail.com>
Date: Fri, 11 Oct 2024 11:24:27 +0300
Subject: [PATCH] add addscores to aggregate search command (v5) (#2818)

---
 packages/search/lib/commands/AGGREGATE.spec.ts | 7 +++++++
 packages/search/lib/commands/AGGREGATE.ts      | 5 +++++
 2 files changed, 12 insertions(+)

diff --git a/packages/search/lib/commands/AGGREGATE.spec.ts b/packages/search/lib/commands/AGGREGATE.spec.ts
index 1a40f264ff..a5eae0233d 100644
--- a/packages/search/lib/commands/AGGREGATE.spec.ts
+++ b/packages/search/lib/commands/AGGREGATE.spec.ts
@@ -20,6 +20,13 @@ describe('AGGREGATE', () => {
       );
     });
 
+    it('with ADDSCORES', () => {
+      assert.deepEqual(
+        AGGREGATE.transformArguments('index', '*', { ADDSCORES: true }),
+        ['FT.AGGREGATE', 'index', '*', 'ADDSCORES']
+      );
+    });  
+
     describe('with LOAD', () => {
       describe('single', () => {
         describe('without alias', () => {
diff --git a/packages/search/lib/commands/AGGREGATE.ts b/packages/search/lib/commands/AGGREGATE.ts
index a12e455a0c..767705a804 100644
--- a/packages/search/lib/commands/AGGREGATE.ts
+++ b/packages/search/lib/commands/AGGREGATE.ts
@@ -118,6 +118,7 @@ interface FilterStep extends AggregateStep<FT_AGGREGATE_STEPS['FILTER']> {
 
 export interface FtAggregateOptions {
   VERBATIM?: boolean;
+  ADDSCORES?: boolean;
   LOAD?: LoadField | Array<LoadField>;
   TIMEOUT?: number;
   STEPS?: Array<GroupByStep | SortStep | ApplyStep | LimitStep | FilterStep>;
@@ -135,6 +136,10 @@ export default {
       args.push('VERBATIM');
     }
 
+    if (options?.ADDSCORES) {
+      args.push('ADDSCORES');
+    }  
+
     if (options?.LOAD) {
       const length = args.push('LOAD', '');