diff --git a/x-pack/plugin/esql/build.gradle b/x-pack/plugin/esql/build.gradle
index c8d704cd2b8bf..df4a2a64db0d1 100644
--- a/x-pack/plugin/esql/build.gradle
+++ b/x-pack/plugin/esql/build.gradle
@@ -1,3 +1,7 @@
+plugins {
+  id 'idea'
+}
+
 import org.elasticsearch.gradle.internal.info.BuildParams
 import org.elasticsearch.gradle.internal.precommit.CheckForbiddenApisTask;
 import org.elasticsearch.gradle.internal.util.SourceDirectoryCommandLineArgumentProvider;
@@ -51,14 +55,19 @@ dependencies {
   internalClusterTestImplementation project(":modules:mapper-extras")
 }
 
+def generatedPath = "src/main/generated"
 def projectDirectory = project.layout.projectDirectory
-def generatedSourceDir = projectDirectory.dir("src/main/generated")
+def generatedSourceDir = projectDirectory.dir(generatedPath)
 tasks.named("compileJava").configure {
   options.compilerArgumentProviders.add(new SourceDirectoryCommandLineArgumentProvider(generatedSourceDir))
   // IntelliJ sticks generated files here and we can't stop it....
   exclude { it.file.toString().contains("src/main/generated-src/generated") }
 }
 
+idea.module {
+  sourceDirs += file(generatedPath)
+}
+
 interface Injected {
   @Inject FileSystemOperations getFs()
 }