JBAI-10776: Disabled ML-based quick fix imports ranking

Set the registry key to the “DISABLED” state. We’re doing this due to an unclear issue with model loading that some users have experienced. We encountered this problem specifically with JetEntry ML models in production. For more details, visit https://youtrack.jetbrains.com/issue/LLM-16032.


(cherry picked from commit 45b7cafa8df1199cdfb15a7c69b8ee87cff74f05)

IJ-CR-161529

GitOrigin-RevId: be502b2b59c449011af0a7f8a2bc886ef152263b

Author: ermolenkodev

python/intellij.python.ml.features/resources/intellij.python.ml.features.xml

@@ -3,7 +3,7 @@
     <statistics.counterUsagesCollector implementationClass="com.intellij.python.ml.features.imports.PyCharmImportsRankingLogs"/>
     <registryKey
       key="quickfix.ranking.ml"
-      defaultValue="[IN_EXPERIMENT*|ENABLED|DISABLED]"
+      defaultValue="[IN_EXPERIMENT|ENABLED|DISABLED*]"
       description="Enable ML ranking in quick fix for missing imports"/>
     <postStartupActivity implementation="com.intellij.python.ml.features.imports.QuickfixRankingModelLoading"/>
     <applicationService serviceInterface="com.jetbrains.python.codeInsight.imports.ImportChooser"

python/intellij.python.ml.features/src/com/intellij/python/ml/features/imports/mlModel.kt

@@ -13,6 +13,7 @@ import com.jetbrains.ml.api.model.MLModel
 import com.jetbrains.ml.api.model.MLModelLoader
 import com.jetbrains.ml.models.PythonImportsRankingModelHolder
 import com.jetbrains.ml.tools.model.MLModelLoaders
+import com.intellij.openapi.diagnostic.thisLogger
 import com.jetbrains.ml.tools.model.ModelDistributionReaders
 import com.jetbrains.ml.tools.model.catboost.CatBoostDistributionFormat
 import com.jetbrains.ml.tools.model.suspendable.MLModelSuspendableService
@@ -30,7 +31,13 @@ class ImportsRankingModelService : MLModelSuspendableService<MLModel<Double>, Do
 
 private class QuickfixRankingModelLoading : ProjectActivity {
   override suspend fun execute(project: Project) {
-    service<ImportsRankingModelService>().loadModel()
+    val rankingStatus = service<FinalImportRankingStatusService>().status
+    if (rankingStatus is FinalImportRankingStatus.Disabled) return
+    try {
+      service<ImportsRankingModelService>().loadModel()
+    } catch (e: RuntimeException) {
+      thisLogger().error("Failed to load python imports ranking model", e)
+    }
   }
 }