From 3a5004b0acdac4d6f0e341edc2157bd04ed6d2a4 Mon Sep 17 00:00:00 2001
From: Pavel Porvatov <Pavel.Porvatov@jetbrains.com>
Date: Fri, 27 Oct 2023 11:01:59 +0200
Subject: [PATCH] LLM-979 Implement AI Assistant activation workflow

- Implemented settings states

GitOrigin-RevId: d9e2653f445fc146521f42c300b7ce1a83e1de2c
---
 .../src/com/intellij/ui/layout/ComponentPredicate.kt          | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/platform/platform-api/src/com/intellij/ui/layout/ComponentPredicate.kt b/platform/platform-api/src/com/intellij/ui/layout/ComponentPredicate.kt
index 68b38374da56..715d8fdbeaa7 100644
--- a/platform/platform-api/src/com/intellij/ui/layout/ComponentPredicate.kt
+++ b/platform/platform-api/src/com/intellij/ui/layout/ComponentPredicate.kt
@@ -6,6 +6,8 @@ import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.observable.properties.ObservableProperty
 import com.intellij.openapi.observable.properties.whenPropertyChanged
 import com.intellij.openapi.application.EDT
+import com.intellij.openapi.application.ModalityState
+import com.intellij.openapi.application.asContextElement
 import com.intellij.openapi.options.advanced.AdvancedSettings
 import com.intellij.openapi.options.advanced.AdvancedSettingsChangeListener
 import com.intellij.ui.DocumentAdapter
@@ -113,7 +115,7 @@ fun <T> StateFlow<T>.predicate(scope: CoroutineScope, predicate: (T) -> Boolean)
     override fun addListener(listener: (Boolean) -> Unit) {
       scope.launch {
         collect { value ->
-          withContext(Dispatchers.EDT) {
+          withContext(Dispatchers.EDT + ModalityState.any().asContextElement()) {
             listener(predicate(value))
           }
         }