diff --git a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt index 00f1bf7889..8339d80f2d 100644 --- a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt +++ b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt @@ -9,6 +9,9 @@ import com.intellij.psi.PsiFile import com.jetbrains.python.psi.PyClass import com.jetbrains.python.psi.PyFile import com.jetbrains.python.psi.PyFunction +import com.jetbrains.python.sdk.PythonSdkType +import org.jetbrains.kotlin.idea.util.projectStructure.module +import org.jetbrains.kotlin.idea.util.projectStructure.sdk import org.utbot.intellij.plugin.language.agnostic.LanguageAssistant object PythonLanguageAssistant : LanguageAssistant() { @@ -45,6 +48,9 @@ object PythonLanguageAssistant : LanguageAssistant() { val file = e.getData(CommonDataKeys.PSI_FILE) as? PyFile ?: return null val element = findPsiElement(file, editor) ?: return null + if (file.module?.sdk?.sdkType !is PythonSdkType) + return null + val containingFunction = getContainingElement(element) val containingClass = getContainingElement(element)