From 986f1632aa00a23f7f3d1c35f38782951f9c4099 Mon Sep 17 00:00:00 2001
From: Peter Bengtsson <peterbe@github.com>
Date: Tue, 11 Jun 2024 11:05:30 -0400
Subject: [PATCH] Use search aggregate filter on GHES URLs (#51116)

---
 src/search/middleware/contextualize.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/search/middleware/contextualize.js b/src/search/middleware/contextualize.js
index 3d7b6027d971..3428ec9d6c96 100644
--- a/src/search/middleware/contextualize.js
+++ b/src/search/middleware/contextualize.js
@@ -46,8 +46,10 @@ export default async function contextualizeSearch(req, res, next) {
     }
   }
 
-  // Feature flag for now XXX
-  if (req.context.currentVersion === 'enterprise-cloud@latest') {
+  // Feature flag
+  if (
+    ['enterprise-cloud', 'enterprise-server'].includes(req.context.currentVersion.split('@')[0])
+  ) {
     search.aggregate = ['toplevel']
   }