vedanthvdev
diff --git a/‎src/codegraphcontext/tools/languages/python.py‎
Lines changed: 37 additions & 1 deletion b/‎src/codegraphcontext/tools/languages/python.py‎
Lines changed: 37 additions & 1 deletion
diff --git a/‎tests/unit/parsers/test_python_parser.py‎
Lines changed: 21 additions & 1 deletion b/‎tests/unit/parsers/test_python_parser.py‎
Lines changed: 21 additions & 1 deletion
@@ -145,6 +145,7 @@ def parse(self, path: Path, is_dependency: bool = False, is_notebook: bool = Fal
             classes = self._find_classes(root_node)
             imports = self._find_imports(root_node)
             function_calls = self._find_calls(root_node)
+            self._attach_module_context(functions, function_calls, root_node, index_source)
             variables = self._find_variables(root_node)
 
             return {
@@ -165,6 +166,41 @@ def parse(self, path: Path, is_dependency: bool = False, is_notebook: bool = Fal
                 os.remove(temp_py_file)
                 info_logger(f"Removed temporary file: {temp_py_file}")
 
+    def _attach_module_context(self, functions, function_calls, root_node, index_source: bool = False):
+        """Represent module-level executable code as Python's <module> frame."""
+        module_level_calls = [
+            call for call in function_calls
+            if not call.get("context") or call["context"][0] is None
+        ]
+        if not module_level_calls:
+            return
+
+        has_module_frame = any(
+            func.get("name") == "<module>" and func.get("line_number") == 1
+            for func in functions
+        )
+        if not has_module_frame:
+            module_func = {
+                "name": "<module>",
+                "line_number": 1,
+                "end_line": root_node.end_point[0] + 1,
+                "args": [],
+                "cyclomatic_complexity": 1,
+                "context": None,
+                "context_type": "module",
+                "class_context": None,
+                "decorators": [],
+                "lang": self.language_name,
+                "is_dependency": False,
+            }
+            if index_source:
+                module_func["source"] = self._get_node_text(root_node)
+                module_func["docstring"] = self._get_docstring(root_node)
+            functions.append(module_func)
+
+        for call in module_level_calls:
+            call["context"] = ("<module>", "module", 1)
+
     def _find_lambda_assignments(self, root_node, index_source: bool = False):
         functions = []
         query_str = PY_QUERIES.get('lambda_assignments')
@@ -573,4 +609,4 @@ def pre_scan_python(files: list[Path], parser_wrapper) -> dict:
         finally:
             if temp_py_file and temp_py_file.exists():
                 os.remove(temp_py_file)
-    return imports_map
+    return imports_map
@@ -46,6 +46,27 @@ def test_parse_simple_function(self, parser, temp_test_dir):
         assert len(funcs) == 1
         assert funcs[0]["name"] == "hello"
 
+    def test_module_level_call_uses_module_context(self, parser, temp_test_dir):
+        """Top-level executable calls should be linked from a synthetic module frame."""
+        code = "from pkg.utils import helper\n\nresult = helper()\n"
+        f = temp_test_dir / "__main__.py"
+        f.write_text(code)
+
+        result = parser.parse(str(f))
+
+        module_func = next(
+            func for func in result["functions"]
+            if func["name"] == "<module>"
+        )
+        helper_call = next(
+            call for call in result["function_calls"]
+            if call["name"] == "helper"
+        )
+
+        assert module_func["line_number"] == 1
+        assert module_func["context_type"] == "module"
+        assert helper_call["context"] == ("<module>", "module", 1)
+
     def test_parse_class_with_method(self, parser, temp_test_dir):
         """Parse a class with a method."""
         code = """
@@ -67,4 +88,3 @@ def greet(self, name):
         # Depending on implementation, methods might be in 'functions' with parent info
         # or inside 'classes'.
         # Let's assume they are captured.
-