From 189d8377a403f094160518f4c962c40962ea324d Mon Sep 17 00:00:00 2001
From: Anya Smirnova <anya.smirnova@jetbrains.com>
Date: Thu, 19 Jun 2025 17:26:49 +0200
Subject: [PATCH] ML in SE: Add an experiment group with EC+ML (IJPL-192039)

GitOrigin-RevId: 888db8784ca7dd07d7dc90dccae78010c51d5e38
---
 .../searchEverywhereMl/SearchEverywhereMlExperiment.kt    | 8 +++++++-
 .../intellij/searchEverywhereMl/SearchEverywhereTab.kt    | 7 +++++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereMlExperiment.kt b/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereMlExperiment.kt
index 433dcce72009..0645ea92678e 100644
--- a/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereMlExperiment.kt
+++ b/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereMlExperiment.kt
@@ -118,5 +118,11 @@ object SearchEverywhereMlExperiment {
      * (see, for example, IJPL-171760 or IJPL-55751)
      */
     object ExactMatchManualFix : ActiveExperiment(true)
+
+    /**
+     * An experiment group that combines both EssentialContributorPrediction and ExperimentalModel
+     * functionality for the All tab.
+     */
+    object CombinedExperiment : ActiveExperiment(true)
   }
-}
\ No newline at end of file
+}
diff --git a/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereTab.kt b/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereTab.kt
index fccbe854f230..8d0a749b398e 100644
--- a/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereTab.kt
+++ b/plugins/search-everywhere-ml/src/com/intellij/searchEverywhereMl/SearchEverywhereTab.kt
@@ -141,7 +141,7 @@ sealed interface SearchEverywhereTab {
 
     override val experiments: Map<Int, ExperimentType> = mapOf(
       1 to ExperimentType.EssentialContributorPrediction,
-      2 to ExperimentType.ExperimentalModel,
+      2 to ExperimentType.CombinedExperiment,
     )
 
 
@@ -325,13 +325,16 @@ val SearchEverywhereTab.TabWithExperiments.isSemanticSearchExperiment: Boolean
  *
  * This property checks the current experimental type associated with the tab. If the experiment
  * type corresponds to [ExperimentType.EssentialContributorPrediction], it returns `true`.
+ * It also returns `true` for [ExperimentType.CombinedExperiment], which combines both
+ * EssentialContributorPrediction and ExperimentalModel functionality.
  * Otherwise, it returns `false`.
  *
  * The "Essential Contributor Prediction" experiment is designed to improve search result rankings
  * by prioritizing essential contributors during retrieval.
  */
 val SearchEverywhereTab.All.isEssentialContributorPredictionExperiment: Boolean
-  get() = this.currentExperimentType == ExperimentType.EssentialContributorPrediction
+  get() = this.currentExperimentType == ExperimentType.EssentialContributorPrediction || 
+          this.currentExperimentType == ExperimentType.CombinedExperiment
 
 /**
  * Indicates whether the current tab's active experiment is configured to handle typo-related scenarios.