diff --git a/plugins/kotlin/code-insight/api/src/org/jetbrains/kotlin/idea/codeinsight/api/applicable/intentions/KotlinPsiUpdateModCommandAction.kt b/plugins/kotlin/code-insight/api/src/org/jetbrains/kotlin/idea/codeinsight/api/applicable/intentions/KotlinPsiUpdateModCommandAction.kt index 0eb7795e0826..b18801686992 100644 --- a/plugins/kotlin/code-insight/api/src/org/jetbrains/kotlin/idea/codeinsight/api/applicable/intentions/KotlinPsiUpdateModCommandAction.kt +++ b/plugins/kotlin/code-insight/api/src/org/jetbrains/kotlin/idea/codeinsight/api/applicable/intentions/KotlinPsiUpdateModCommandAction.kt @@ -8,6 +8,8 @@ import com.intellij.modcommand.* import com.intellij.openapi.progress.ProcessCanceledException import com.intellij.psi.PsiElement import com.intellij.util.concurrency.annotations.RequiresBackgroundThread +import org.jetbrains.kotlin.analysis.api.permissions.KaAllowAnalysisOnEdt +import org.jetbrains.kotlin.analysis.api.permissions.allowAnalysisOnEdt import org.jetbrains.kotlin.idea.codeinsight.api.applicable.ContextProvider import org.jetbrains.kotlin.idea.codeinsight.api.applicable.getElementContext import org.jetbrains.kotlin.psi.KtElement @@ -89,9 +91,12 @@ sealed class KotlinPsiUpdateModCommandAction( ) : KotlinPsiUpdateModCommandAction(null, elementClass), ContextProvider { + @OptIn(KaAllowAnalysisOnEdt::class) final override fun getElementContext( actionContext: ActionContext, element: E, - ): C? = getElementContext(element) + ): C? = allowAnalysisOnEdt { // TODO: remove this workaround when IJPL-193738 is fixed. + getElementContext(element) + } } } \ No newline at end of file