Resolve sources for py_proto_library deps

PiperOrigin-RevId: 712090894

Author: SergejSalnikov

aspect/intellij_info_impl.bzl

@@ -348,6 +348,15 @@ def collect_py_info(target, ctx, semantics, ide_info, ide_info_file, output_grou
         py_launcher = None
 
     sources = sources_from_target(ctx)
+
+    # collect sources for py_proto_library rule
+    if hasattr(target[OutputGroupInfo], "python_proto_generated_sources"):
+        generated_sources = [
+            artifact_location(f)
+            for f in target[OutputGroupInfo].python_proto_generated_sources.to_list()
+        ]
+        sources = sources + generated_sources
+
     to_build = target[PyInfo].transitive_sources
     args = getattr(ctx.rule.attr, "args", [])
     data_deps = getattr(ctx.rule.attr, "data", [])