codegraphcontext 0.4.4__py3-none-any.whl → 0.4.5__py3-none-any.whl

codegraphcontext/cli/cli_helpers.py

@@ -721,19 +721,31 @@ def watch_helper(path: str, context: Optional[str] = None):
         # Add the directory to watch
         if is_indexed:
             console.print("[green]✓[/green] Already indexed (no initial scan needed)")
-            watcher.watch_directory(str(path_obj), perform_initial_scan=False)
+            watcher.watch_directory(
+                str(path_obj),
+                perform_initial_scan=False,
+                cgcignore_path=ctx.cgcignore_path,
+            )
         else:
             console.print("[yellow]⚠[/yellow]  Not indexed yet. Performing initial scan...")
             
             # Index the repository first (like MCP does)
             async def do_index():
-                await graph_builder.build_graph_from_path_async(path_obj, is_dependency=False)
+                await graph_builder.build_graph_from_path_async(
+                    path_obj,
+                    is_dependency=False,
+                    cgcignore_path=ctx.cgcignore_path,
+                )
             
             asyncio.run(do_index())
             console.print("[green]✓[/green] Initial scan complete")
             
             # Now start watching (without another scan)
-            watcher.watch_directory(str(path_obj), perform_initial_scan=False)
+            watcher.watch_directory(
+                str(path_obj),
+                perform_initial_scan=False,
+                cgcignore_path=ctx.cgcignore_path,
+            )
         
         console.print("[bold green]👀 Monitoring for file changes...[/bold green] (Press Ctrl+C to stop)")
         console.print("[dim]💡 Tip: Open a new terminal window to continue working[/dim]\n")

codegraphcontext/cli/main.py

@@ -854,11 +854,11 @@ def doctor():
         elif default_db == "kuzudb":
             from importlib.util import find_spec
 
-            if find_spec("kuzu") is not None:
+            if find_spec("real_ladybug") is not None:
                 console.print(f"   [green]✓[/green] KuzuDB is installed")
             else:
                 console.print(f"   [red]✗[/red] KuzuDB is not installed")
-                console.print(f"       Run: pip install kuzu")
+                console.print(f"       Run: pip install real_ladybug")
                 all_checks_passed = False
         else:
             # FalkorDB

codegraphcontext/core/__init__.py

@@ -22,7 +22,7 @@ import importlib.util
 def _is_kuzudb_available() -> bool:
     """Check if KùzuDB is installed."""
     try:
-        return importlib.util.find_spec("kuzu") is not None
+        return importlib.util.find_spec("real_ladybug") is not None
     except ImportError:
         return False
 
@@ -70,7 +70,7 @@ def get_database_manager(db_path: Optional[str] = None) -> Union['DatabaseManage
         db_type = db_type.lower()
         if db_type == 'kuzudb':
             if not _is_kuzudb_available():
-                raise ValueError("Database set to 'kuzudb' but Kùzu is not installed.\nRun 'pip install kuzu'")
+                raise ValueError("Database set to 'kuzudb' but Kùzu is not installed.\nRun 'pip install real_ladybug'")
             from .database_kuzu import KuzuDBManager
             info_logger(f"Using KùzuDB (explicit) at {db_path or 'default path'}")
             return KuzuDBManager(db_path=db_path)
@@ -147,7 +147,7 @@ def get_database_manager(db_path: Optional[str] = None) -> Union['DatabaseManage
         return DatabaseManager()
 
     error_msg = "No database backend available.\n"
-    error_msg += "Recommended: Install KùzuDB for zero-config ('pip install kuzu')\n"
+    error_msg += "Recommended: Install KùzuDB for zero-config ('pip install real_ladybug')\n"
 
     if platform.system() != "Windows":
         error_msg += "Alternative: Install FalkorDB Lite ('pip install falkordblite')\n"

codegraphcontext/core/database_kuzu.py

@@ -66,7 +66,7 @@ class KuzuDBManager:
         if self._conn is None:
             with self._lock:
                 if self._conn is None:
-                    import kuzu
+                    import real_ladybug as kuzu
                     max_retries = 5
                     for attempt in range(max_retries):
                         try:
@@ -77,7 +77,7 @@ class KuzuDBManager:
                             info_logger("KùzuDB connection established and schema verified")
                             break
                         except ImportError:
-                            error_logger("KùzuDB is not installed. Run 'pip install kuzu'")
+                            error_logger("KùzuDB is not installed. Run 'pip install real_ladybug'")
                             raise ValueError("KùzuDB missing.")
                         except Exception as e:
                             if "lock" in str(e).lower() and attempt < max_retries - 1:
@@ -156,6 +156,26 @@ class KuzuDBManager:
                     warning_logger(f"Kuzu Schema Rel Error ({table_name}): {e}")
                     debug_log(f"Kuzu Schema Rel Error ({table_name}): {e}")
 
+        self._run_schema_migrations()
+
+    def _run_schema_migrations(self):
+        """Add columns introduced after older local Kùzu databases were created."""
+        migrations = [
+            ("Module", "full_import_name", "STRING"),
+            ("IMPORTS", "full_import_name", "STRING"),
+            ("IMPORTS", "imported_name", "STRING"),
+        ]
+
+        for table_name, column_name, column_type in migrations:
+            try:
+                self._conn.execute(f"ALTER TABLE `{table_name}` ADD {column_name} {column_type}")
+            except Exception as e:
+                err = str(e).lower()
+                if "already exists" in err or "duplicate" in err:
+                    continue
+                warning_logger(f"Kuzu Schema Migration Error ({table_name}.{column_name}): {e}")
+                debug_log(f"Kuzu Schema Migration Error ({table_name}.{column_name}): {e}")
+
     def close_driver(self):
         """Closes the connection."""
         if self._conn is not None:
@@ -188,10 +208,10 @@ class KuzuDBManager:
     @staticmethod
     def test_connection(db_path: str = None) -> Tuple[bool, Optional[str]]:
         try:
-            import kuzu
+            import real_ladybug as kuzu
             return True, None
         except ImportError:
-            return False, "KùzuDB is not installed. Run 'pip install kuzu'"
+            return False, "KùzuDB is not installed. Run 'pip install real_ladybug'"
 
 class KuzuDriverWrapper:
     def __init__(self, conn):

codegraphcontext/core/watcher.py

@@ -10,9 +10,13 @@ from watchdog.observers import Observer
 from watchdog.events import FileSystemEventHandler
 
 if typing.TYPE_CHECKING:
+    from pathspec import PathSpec
     from codegraphcontext.tools.graph_builder import GraphBuilder
     from codegraphcontext.core.jobs import JobManager
 
+from codegraphcontext.core.cgcignore import build_ignore_spec
+from codegraphcontext.tools.indexing.constants import DEFAULT_IGNORE_PATTERNS
+from codegraphcontext.cli.config_manager import get_config_value
 from codegraphcontext.utils.debug_log import debug_log, info_logger, error_logger, warning_logger
 
 class RepositoryEventHandler(FileSystemEventHandler):
@@ -23,7 +27,15 @@ class RepositoryEventHandler(FileSystemEventHandler):
     to build a baseline and then uses this cached state to perform efficient
     updates when files are changed, created, or deleted.
     """
-    def __init__(self, graph_builder: "GraphBuilder", repo_path: Path, debounce_interval=2.0, perform_initial_scan: bool = True):
+    def __init__(
+        self,
+        graph_builder: "GraphBuilder",
+        repo_path: Path,
+        debounce_interval=2.0,
+        perform_initial_scan: bool = True,
+        cgcignore_path: str = None,
+        ignore_spec: "PathSpec" = None,
+    ):
         """
         Initializes the event handler.
 
@@ -32,12 +44,17 @@ class RepositoryEventHandler(FileSystemEventHandler):
             repo_path: The absolute path to the repository directory to watch.
             debounce_interval: The time in seconds to wait for more changes before processing an event.
             perform_initial_scan: Whether to perform an initial scan of the repository.
+            cgcignore_path: Optional explicit .cgcignore path from the active context.
+            ignore_spec: Optional precompiled ignore spec, useful for tests.
         """
         super().__init__()
         self.graph_builder = graph_builder
-        self.repo_path = repo_path
+        self.repo_path = repo_path.resolve()
         self.debounce_interval = debounce_interval
         self.timers = {} # A dictionary to manage debounce timers for file paths.
+        self.ignore_root = self.repo_path
+        self.ignore_spec = ignore_spec
+        self._load_ignore_spec(cgcignore_path)
         
         # Caches for the repository's state.
         self.all_file_data = []
@@ -47,11 +64,65 @@ class RepositoryEventHandler(FileSystemEventHandler):
         if perform_initial_scan:
             self._initial_scan()
 
+    def _load_ignore_spec(self, cgcignore_path: str = None) -> None:
+        """Load .cgcignore rules using the same defaults as repository indexing."""
+        if self.ignore_spec is not None:
+            return
+        try:
+            self.ignore_spec, resolved_cgcignore = build_ignore_spec(
+                ignore_root=self.ignore_root,
+                default_patterns=DEFAULT_IGNORE_PATTERNS,
+                explicit_path=cgcignore_path,
+            )
+            if resolved_cgcignore:
+                debug_log(
+                    f"Watcher using .cgcignore at {resolved_cgcignore} "
+                    f"(filtering relative to {self.ignore_root})"
+                )
+        except OSError as e:
+            self.ignore_spec = None
+            warning_logger(f"Could not load/create watcher .cgcignore: {e}")
+
+    def _should_ignore(self, path: str | Path) -> bool:
+        """Return True when a path is excluded by .cgcignore or IGNORE_DIRS."""
+        path_obj = Path(path).resolve()
+        ignore_root = getattr(self, "ignore_root", getattr(self, "repo_path", None))
+
+        ignore_dirs_str = get_config_value("IGNORE_DIRS") or ""
+        if ignore_dirs_str and ignore_root:
+            ignore_dirs = {d.strip().lower() for d in ignore_dirs_str.split(",") if d.strip()}
+            try:
+                parts = {p.lower() for p in path_obj.relative_to(ignore_root).parent.parts}
+                if parts.intersection(ignore_dirs):
+                    return True
+            except ValueError:
+                pass
+
+        ignore_spec = getattr(self, "ignore_spec", None)
+        if not ignore_spec or not ignore_root:
+            return False
+
+        try:
+            rel_path = path_obj.relative_to(ignore_root).as_posix()
+        except ValueError:
+            return False
+        return ignore_spec.match_file(rel_path)
+
+    def _is_supported_code_file(self, path: str | Path) -> bool:
+        path_obj = Path(path)
+        return path_obj.is_file() and path_obj.suffix in self.graph_builder.parsers and not self._should_ignore(path_obj)
+
+    def _iter_supported_files(self) -> list[Path]:
+        supported_extensions = self.graph_builder.parsers.keys()
+        return [
+            f for f in self.repo_path.rglob("*")
+            if f.is_file() and f.suffix in supported_extensions and not self._should_ignore(f)
+        ]
+
     def _initial_scan(self):
         """Scans the entire repository, parses all files, and builds the initial graph."""
         info_logger(f"Performing initial scan for watcher: {self.repo_path}")
-        supported_extensions = self.graph_builder.parsers.keys()
-        all_files = [f for f in self.repo_path.rglob("*") if f.is_file() and f.suffix in supported_extensions]
+        all_files = self._iter_supported_files()
         
         # 1. Pre-scan all files to get a global map of where every symbol is defined.
         self.imports_map = self.graph_builder.pre_scan_imports(all_files)
@@ -126,6 +197,10 @@ class RepositoryEventHandler(FileSystemEventHandler):
         """
         info_logger(f"File change detected (incremental update): {event_path_str}")
         changed_path = Path(event_path_str)
+        if self._should_ignore(changed_path):
+            debug_log(f"Ignored watcher update based on .cgcignore: {changed_path}")
+            return
+
         changed_path_str = str(changed_path.resolve())
         supported_extensions = self.graph_builder.parsers.keys()
 
@@ -160,7 +235,7 @@ class RepositoryEventHandler(FileSystemEventHandler):
         subset_file_data = []
         for path_str in affected_paths:
             p = Path(path_str)
-            if p.exists() and p.suffix in supported_extensions:
+            if p.exists() and p.suffix in supported_extensions and not self._should_ignore(p):
                 parsed = self.graph_builder.parse_file(self.repo_path, p)
                 if "error" not in parsed:
                     subset_file_data.append(parsed)
@@ -177,22 +252,22 @@ class RepositoryEventHandler(FileSystemEventHandler):
 
     # The following methods are called by the watchdog observer when a file event occurs.
     def on_created(self, event):
-        if not event.is_directory and Path(event.src_path).suffix in self.graph_builder.parsers:
+        if not event.is_directory and self._is_supported_code_file(event.src_path):
             self._debounce(event.src_path, lambda: self._handle_modification(event.src_path))
 
     def on_modified(self, event):
-        if not event.is_directory and Path(event.src_path).suffix in self.graph_builder.parsers:
+        if not event.is_directory and self._is_supported_code_file(event.src_path):
             self._debounce(event.src_path, lambda: self._handle_modification(event.src_path))
 
     def on_deleted(self, event):
-        if not event.is_directory and Path(event.src_path).suffix in self.graph_builder.parsers:
+        if not event.is_directory and Path(event.src_path).suffix in self.graph_builder.parsers and not self._should_ignore(event.src_path):
             self._debounce(event.src_path, lambda: self._handle_modification(event.src_path))
 
     def on_moved(self, event):
         if not event.is_directory:
-            if Path(event.src_path).suffix in self.graph_builder.parsers:
+            if Path(event.src_path).suffix in self.graph_builder.parsers and not self._should_ignore(event.src_path):
                 self._debounce(event.src_path, lambda: self._handle_modification(event.src_path))
-            if Path(event.dest_path).suffix in self.graph_builder.parsers:
+            if Path(event.dest_path).suffix in self.graph_builder.parsers and not self._should_ignore(event.dest_path):
                 self._debounce(event.dest_path, lambda: self._handle_modification(event.dest_path))
 
 
@@ -207,7 +282,7 @@ class CodeWatcher:
         self.watched_paths = set() # Keep track of paths already being watched.
         self.watches = {} # Store watch objects to allow unscheduling
 
-    def watch_directory(self, path: str, perform_initial_scan: bool = True):
+    def watch_directory(self, path: str, perform_initial_scan: bool = True, cgcignore_path: str = None):
         """Schedules a directory to be watched for changes."""
         path_obj = Path(path).resolve()
         path_str = str(path_obj)
@@ -217,7 +292,12 @@ class CodeWatcher:
             return {"message": f"Path already being watched: {path_str}"}
         
         # Create a new, dedicated event handler for this specific repository path.
-        event_handler = RepositoryEventHandler(self.graph_builder, path_obj, perform_initial_scan=perform_initial_scan)
+        event_handler = RepositoryEventHandler(
+            self.graph_builder,
+            path_obj,
+            perform_initial_scan=perform_initial_scan,
+            cgcignore_path=cgcignore_path,
+        )
         
         watch = self.observer.schedule(event_handler, path_str, recursive=True)
         self.watches[path_str] = watch

codegraphcontext/tools/code_finder.py

@@ -864,6 +864,42 @@ class CodeFinder:
         """Find all dependencies and dependents of a module"""
         with self.driver.session() as session:
             repo_filter = "AND file.path STARTS WITH $repo_path" if repo_path else ""
+            backend = getattr(self.db_manager, "get_backend_type", lambda: "")()
+
+            # KuzuDB is stricter about OPTIONAL MATCH variable scoping, and nested
+            # repository ownership is already represented in the file path.
+            if backend == "kuzudb":
+                importers_result = session.run(f"""
+                    MATCH (file:File)-[imp:IMPORTS]->(module:Module)
+                    WHERE (module.name = $module_name OR module.full_import_name CONTAINS $module_name) {repo_filter}
+                    RETURN DISTINCT
+                        file.path as importer_file_path,
+                        imp.line_number as import_line_number,
+                        file.is_dependency as file_is_dependency,
+                        '' as repository_name
+                    ORDER BY file_is_dependency ASC, importer_file_path
+                    LIMIT 50
+                """, module_name=module_name, repo_path=repo_path)
+
+                imports_result = session.run(f"""
+                    MATCH (file:File)-[:IMPORTS]->(target_module:Module)
+                    WHERE (target_module.name = $module_name OR target_module.full_import_name CONTAINS $module_name) {repo_filter}
+                    WITH file, target_module
+                    MATCH (file)-[imp:IMPORTS]->(other_module:Module)
+                    WHERE other_module.name <> target_module.name
+                    RETURN DISTINCT
+                        other_module.name as imported_module,
+                        imp.alias as import_alias
+                    ORDER BY imported_module
+                    LIMIT 50
+                """, module_name=module_name, repo_path=repo_path)
+
+                return {
+                    "module_name": module_name,
+                    "importers": importers_result.data(),
+                    "imports": imports_result.data()
+                }
+
             # Find files that import this module (who imports this module)
             importers_result = session.run(f"""
                 MATCH (file:File)-[imp:IMPORTS]->(module:Module {{name: $module_name}})

codegraphcontext/tools/indexing/persistence/writer.py

@@ -278,19 +278,28 @@ class GraphWriter:
             js_imports = []
             other_imports = []
             for imp in file_data.get("imports", []):
-                if lang == "javascript":
+                if lang in {"javascript", "typescript", "tsx"}:
                     module_name = imp.get("source")
                     if module_name:
                         js_imports.append(
                             {
                                 "module_name": module_name,
                                 "imported_name": imp.get("name", "*"),
-                                "alias": imp.get("alias"),
-                                "line_number": imp.get("line_number"),
+                                "alias": imp.get("alias") or "",
+                                "line_number": imp.get("line_number") or 0,
                             }
                         )
                 else:
-                    other_imports.append(imp)
+                    module_name = imp.get("name") or imp.get("source") or imp.get("full_import_name")
+                    if module_name:
+                        other_imports.append(
+                            {
+                                "name": module_name,
+                                "alias": imp.get("alias") or "",
+                                "full_import_name": imp.get("full_import_name") or module_name,
+                                "line_number": imp.get("line_number") or 0,
+                            }
+                        )
 
             if js_imports:
                 session.run(

codegraphcontext/tools/indexing/resolution/calls.py

@@ -20,6 +20,7 @@ def resolve_function_call(
     if called_name in __builtins__:
         return None
 
+    resolved_called_name = called_name
     resolved_path = None
     full_call = call.get("full_name", called_name)
     base_obj = full_call.split(".")[0] if "." in full_call else None
@@ -43,6 +44,14 @@ def resolve_function_call(
 
     if not resolved_path:
         possible_paths = imports_map.get(lookup_name, [])
+        if not possible_paths and lookup_name in local_imports:
+            imported_name = local_imports[lookup_name]
+            alias_paths = imports_map.get(imported_name, [])
+            if alias_paths:
+                possible_paths = alias_paths
+                lookup_name = imported_name
+                if called_name == base_obj or called_name == call["name"]:
+                    resolved_called_name = imported_name
         if len(possible_paths) == 1:
             resolved_path = possible_paths[0]
         elif len(possible_paths) > 1:
@@ -74,8 +83,8 @@ def resolve_function_call(
         if called_name in local_names:
             resolved_path = caller_file_path
             is_unresolved_external = False
-        elif called_name in imports_map and imports_map[called_name]:
-            candidates = imports_map[called_name]
+        elif resolved_called_name in imports_map and imports_map[resolved_called_name]:
+            candidates = imports_map[resolved_called_name]
             for path in candidates:
                 for imp_name in local_imports.values():
                     if imp_name.replace(".", "/") in path:
@@ -100,7 +109,7 @@ def resolve_function_call(
             "caller_name": caller_name,
             "caller_file_path": caller_file_path,
             "caller_line_number": caller_line_number,
-            "called_name": called_name,
+            "called_name": resolved_called_name,
             "called_file_path": resolved_path,
             "line_number": call["line_number"],
             "args": call.get("args", []),
@@ -109,7 +118,7 @@ def resolve_function_call(
     return {
         "type": "file",
         "caller_file_path": caller_file_path,
-        "called_name": called_name,
+        "called_name": resolved_called_name,
         "called_file_path": resolved_path,
         "line_number": call["line_number"],
         "args": call.get("args", []),

codegraphcontext/tools/languages/dart.py

@@ -51,6 +51,23 @@ DART_QUERIES = {
     """,
 }
 
+SIGNATURE_TYPES = (
+    "function_signature",
+    "method_signature",
+    "getter_signature",
+    "setter_signature",
+    "constructor_signature",
+    "factory_constructor_signature",
+    "operator_signature",
+)
+CONTAINER_TYPES = (
+    "class_definition",
+    "mixin_declaration",
+    "extension_declaration",
+)
+_CALL_PUNCTUATION = {".", "?.", "..", "?..", ";", ",", "(", ")", "[", "]"}
+
+
 class DartTreeSitterParser:
     """A Dart-specific parser using tree-sitter, encapsulating language-specific logic."""
 
@@ -65,15 +82,101 @@ class DartTreeSitterParser:
         if not node: return ""
         return node.text.decode('utf-8')
 
-    def _get_parent_context(self, node, types=('function_signature', 'class_definition', 'mixin_declaration', 'extension_declaration')):
+    def _get_parent_context(self, node, types=SIGNATURE_TYPES + CONTAINER_TYPES):
         curr = node.parent
         while curr:
             if curr.type in types:
                 name_node = curr.child_by_field_name('name')
+                if name_node is None:
+                    for child in curr.children:
+                        if child.type == "identifier":
+                            name_node = child
+                            break
                 return self._get_node_text(name_node) if name_node else None, curr.type, curr.start_point[0] + 1
+
+            # Dart places function_body next to its signature under class_body/program,
+            # so calls inside a body cannot find the signature through parent links.
+            if curr.type == "function_body":
+                parent = curr.parent
+                if parent is not None:
+                    siblings = list(parent.children)
+                    try:
+                        idx = siblings.index(curr)
+                    except ValueError:
+                        idx = -1
+
+                    for i in range(idx - 1, -1, -1):
+                        sibling = siblings[i]
+                        if sibling.type not in types:
+                            continue
+
+                        target = sibling
+                        for child in sibling.children:
+                            if child.type in SIGNATURE_TYPES:
+                                target = child
+                                break
+
+                        name_node = target.child_by_field_name("name")
+                        if name_node is None:
+                            for child in target.children:
+                                if child.type == "identifier":
+                                    name_node = child
+                                    break
+                        if name_node is not None:
+                            return self._get_node_text(name_node), target.type, target.start_point[0] + 1
             curr = curr.parent
         return None, None, None
 
+    def _last_identifier_in(self, node):
+        """Return the deepest-last identifier under node."""
+        last = None
+        for child in node.children:
+            if child.type == "identifier":
+                last = child
+            else:
+                deeper = self._last_identifier_in(child)
+                if deeper is not None:
+                    last = deeper
+        return last
+
+    def _name_node_for_call_selector(self, call_node):
+        """Find the receiver/member name paired with a captured argument selector."""
+        parent = call_node.parent
+        if parent is None:
+            return None
+
+        siblings = list(parent.children)
+        try:
+            idx = siblings.index(call_node)
+        except ValueError:
+            return None
+
+        for i in range(idx - 1, -1, -1):
+            sibling = siblings[i]
+            if sibling.type in _CALL_PUNCTUATION:
+                continue
+            if sibling.type == "identifier":
+                return sibling
+            name_node = self._last_identifier_in(sibling)
+            if name_node is not None:
+                return name_node
+            break
+        return None
+
+    def _has_call_selector_sibling(self, name_node):
+        """Return True if an identifier is a receiver beside a call selector."""
+        parent = name_node.parent
+        if parent is None:
+            return False
+        for sibling in parent.children:
+            if sibling is name_node:
+                continue
+            if sibling.type == "selector":
+                for child in sibling.children:
+                    if child.type == "argument_part":
+                        return True
+        return False
+
     def _calculate_complexity(self, node):
         complexity_nodes = {
             "if_statement", "for_statement", "while_statement", "do_statement",
@@ -293,23 +396,14 @@ class DartTreeSitterParser:
         
         for node, capture_name in execute_query(self.language, query_str, root_node):
             if capture_name in ("name", "call"):
-                # Ensure we are at the right node level
-                target_node = node
                 if capture_name == "call":
-                    name_node = None
-                    for child in node.children:
-                        if child.type == 'identifier':
-                            name_node = child
-                            break
-                        if child.type == 'selector':
-                            for sub in child.children:
-                                if sub.type == 'identifier':
-                                    name_node = sub
-                                    break
-                    if name_node:
-                        target_node = name_node
-                    else:
+                    target_node = self._name_node_for_call_selector(node)
+                    if target_node is None:
+                        continue
+                else:
+                    if self._has_call_selector_sibling(node):
                         continue
+                    target_node = node
 
                 # Deduplicate by start byte
                 node_id = target_node.start_byte

codegraphcontext/tools/languages/python.py

@@ -145,6 +145,7 @@ class PythonTreeSitterParser:
             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 @@ class PythonTreeSitterParser:
                 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

codegraphcontext/tools/scip_indexer.py

@@ -226,9 +226,11 @@ class ScipIndexParser:
         """
         try:
             from . import scip_pb2  # type: ignore
-        except ImportError:
+        except Exception as e:
             error_logger(
-                "scip_pb2.py not found in tools directory."
+                "Failed to import codegraphcontext.tools.scip_pb2. "
+                "Ensure protobuf>=3.20,<3.21 is installed in the CodeGraphContext environment. "
+                f"Original error: {e}"
             )
             return {}
 

{codegraphcontext-0.4.4.dist-info → codegraphcontext-0.4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.4.4
+Version: 0.4.5
 Summary: An MCP server that indexes local code into a graph database to provide context to AI assistants.
 Author-email: Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 License: MIT License
@@ -52,8 +52,9 @@ Requires-Dist: nbconvert>=7.16.6
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: falkordb>=0.1.0
 Requires-Dist: requests>=2.28.0
+Requires-Dist: protobuf<3.21,>=3.20
 Requires-Dist: falkordblite>=0.1.0; sys_platform != "win32" and python_version >= "3.12"
-Requires-Dist: kuzu>=0.4.0; sys_platform == "win32" or (sys_platform != "win32" and python_version >= "3.10" and python_version < "3.12")
+Requires-Dist: real_ladybug; sys_platform == "win32" or (sys_platform != "win32" and python_version >= "3.10")
 Requires-Dist: fastapi>=0.100.0
 Requires-Dist: uvicorn>=0.22.0
 Provides-Extra: parsing
@@ -165,7 +166,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.4.4
+- **Version:** 0.4.5
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -228,7 +229,7 @@ CodeGraphContext supports multiple graph database backends to suit your environm
 | **Setup** | Zero-config / Embedded | Zero-config / In-process | Docker / External |
 | **Platform** | **All (Windows Native, macOS, Linux)** | Unix-only (Linux/macOS/WSL) | All Platforms |
 | **Use Case** | Desktop, IDE, Local development | Specialized Unix development | Enterprise, Massive graphs |
-| **Requirement**| `pip install kuzu` | `pip install falkordblite` | Neo4j Server / Docker |
+| **Requirement**| `pip install real_ladybug` | `pip install falkordblite` | Neo4j Server / Docker |
 | **Speed** | ⚡ Extremely Fast | ⚡ Fast | 🚀 Scalable |
 | **Persistence**| Yes (to disk) | Yes (to disk) | Yes (to disk) |
 
@@ -339,7 +340,7 @@ Use CodeGraphContext as an **MCP server** for AI assistants:
 
 2.  **Database Setup (Automatic)**
     
-    - **KùzuDB (default on Windows):** Runs natively on Windows, macOS, and Linux. On Windows it is the usual embedded choice; `pip install kuzu` if needed.
+    - **KùzuDB (default on Windows):** Runs natively on Windows, macOS, and Linux. On Windows it is the usual embedded choice; `pip install real_ladybug` if needed.
     - **FalkorDB Lite (typical default on Unix):** When Python 3.12+ and `falkordblite` are available on Unix/macOS/WSL, the embedded backend prefers FalkorDB Lite; otherwise KùzuDB is used.
     - **Neo4j (Alternative):** To use Neo4j instead, or if you prefer a server-based approach, run: `cgc neo4j setup`
 

{codegraphcontext-0.4.4.dist-info → codegraphcontext-0.4.5.dist-info}/RECORD

@@ -4,30 +4,30 @@ codegraphcontext/prompts.py,sha256=E5P55paM0oHfBcNVfxkxpXRGZnya1kr_mKDg9i49FwM,6
 codegraphcontext/server.py,sha256=AMe3BVvon0cNE1uyPB-E2a0XedBTWqSiIZv19KEPZh4,19120
 codegraphcontext/tool_definitions.py,sha256=kDRI9WrfkV9TienmYvetGlimNdb1AdiOWuuYid_ZtAk,13177
 codegraphcontext/cli/__init__.py,sha256=v6CMDVKM5d_sXn3S5nZNf0phXn0IdrnhLazUoen9k9w,38
-codegraphcontext/cli/cli_helpers.py,sha256=toXzoiiig7UmOZ7-Yr7euKTb-71FEyxYt4qhCj3PjqQ,32683
+codegraphcontext/cli/cli_helpers.py,sha256=-_7N4rCZuunyDiUsD1G1ik9MS40JWLnP5zItmYxYtuk,32993
 codegraphcontext/cli/config_manager.py,sha256=s13JltwsmHF0jWK65XFD7gM3OIhDNPsBWpm2qNPZrkU,38522
-codegraphcontext/cli/main.py,sha256=g6ITjieGo54w3LKJcaxigefDlZOT3jLmCWOw2QudALQ,94571
+codegraphcontext/cli/main.py,sha256=_dXQI_PQkqTD_P-iaDfRc4Iz9kuLWkPYg_ZxX3c6YHM,94587
 codegraphcontext/cli/registry_commands.py,sha256=mAJDYw1XUbzhj_SMM9PgO4MzECQj0OR3LU54x-OpPeU,15947
 codegraphcontext/cli/setup_macos.py,sha256=Xjlv_9jk9qv8Gh7stpH1pvlalzC0Fg176y7jc5G1zh0,3575
 codegraphcontext/cli/setup_wizard.py,sha256=jrh8504YwKAp-7b-gOW4fwRTD_yxlBYaVPfRk0TYlvQ,42573
 codegraphcontext/cli/visualizer.py,sha256=YgqXA-o1l3qRABat_s-wT0gTM_rmHmOhpUtxS7Rhcfk,1861
-codegraphcontext/core/__init__.py,sha256=t0VsF48zSsw0yAGsg3__7RH_R4IraF0iqnGLOtA_9G8,7245
+codegraphcontext/core/__init__.py,sha256=fOkVEnsA7ikCn3xAhGGSjAokzVn4WgKWJMwBYWDZdEY,7269
 codegraphcontext/core/bundle_registry.py,sha256=WR0apx71czZLFrTP7a8k3l0vXYsDGyZKn-qzRTzuu7o,7866
 codegraphcontext/core/cgc_bundle.py,sha256=lqhOQQegm6fCmdndcag7eoS0DV5KNrF8Sm-2VNZ5HR4,35070
 codegraphcontext/core/cgcignore.py,sha256=uoydGpSR8LVdWYS40C3o0YFkV0s_v9IouxPLF_GsTxY,4172
 codegraphcontext/core/database.py,sha256=2UL8AyE6vX5mVHg1zcD74TYEREmddhKzN7Bjy2THQwM,11435
 codegraphcontext/core/database_falkordb.py,sha256=7jBLK_Y4u4ozeVV59vZr9E90Z3XZ_Cq6p4DUyq3WNnc,19741
 codegraphcontext/core/database_falkordb_remote.py,sha256=aMiuJTB_-2rBUpbAmX8ElZiZrYZbOXK-On3ncgMCyhY,7118
-codegraphcontext/core/database_kuzu.py,sha256=eznuzOZrd_QnXiZKF_nWLmJ30juOwtqhgrgiyTv5kv4,31551
+codegraphcontext/core/database_kuzu.py,sha256=g_ceVfg0GZUU8UNzvqwI8yL8ETJ9VmHBCUmHbpgzaJQ,32492
 codegraphcontext/core/falkor_worker.py,sha256=bMcTOac6e8IzSPjCd-YulGVh-S7QsNtXs2c8hx6M6gk,4958
 codegraphcontext/core/jobs.py,sha256=d6v_IERdEcDlIsz3CW3p0QMp3N-6dgQICs3FZRPgPgU,4809
-codegraphcontext/core/watcher.py,sha256=np5dBVUnNVZ2Wslvk-q9QWZotj4VirqhJMv8WqTUQLQ,13213
+codegraphcontext/core/watcher.py,sha256=XcBDALfUx3ypMmVPmMcjq4q9khLLVQkNmO-IiNAAbj8,16508
 codegraphcontext/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codegraphcontext/tools/advanced_language_query_tool.py,sha256=APtC-KSkXXSrVLwiw3kgi6Q-xaqXl6skLWHPyuNh358,3670
-codegraphcontext/tools/code_finder.py,sha256=TrdNmN6VHUguGdX4izY2MsM82vlx4E2B9SpZsVpyz30,60463
+codegraphcontext/tools/code_finder.py,sha256=hS3_rVxmIplDJyazMNrwQVFTvDvFXg36UekDq3CDYaA,62345
 codegraphcontext/tools/graph_builder.py,sha256=jiHHI1XIvmDJ8-BlWPe_WBNEVtXBUCfwhHTkUka-sAg,13219
 codegraphcontext/tools/package_resolver.py,sha256=KtbdMReTezszjdsqYniL-Xb-QUsrAJWtf1NSiyIPkLI,18704
-codegraphcontext/tools/scip_indexer.py,sha256=MdxesJ4GQBOwrE2q3SZjImkzh6BSPQeA1tYsR5Zerok,20021
+codegraphcontext/tools/scip_indexer.py,sha256=bVMkSfvemeXciS6XiXrC5ZgFz33YiW80x6OASfU9zwQ,20169
 codegraphcontext/tools/scip_pb2.py,sha256=dwOMNKlu6VyLq5h8kTPZRDqxrwfVL8yw7I7ziokFG48,98229
 codegraphcontext/tools/system.py,sha256=goQOs1A9gvd7SvDPVZB6Jd0DRJKQmjEkn-uiTE40VNM,6046
 codegraphcontext/tools/tree_sitter_parser.py,sha256=x8gDBCQze65r97DwlIovflg02wrY5SKm4nCVxxIja2E,4500
@@ -46,14 +46,14 @@ codegraphcontext/tools/indexing/schema.py,sha256=Lyzc_06fJx7eMb7a1OiSvEJGzkyXuUX
 codegraphcontext/tools/indexing/schema_contract.py,sha256=0705pzAw3h9tO9EIfRbdXl5_WudY2l6drE6zX843120,1054
 codegraphcontext/tools/indexing/scip_pipeline.py,sha256=Y9pvJM4HfbrDQC1nFj7Dgil42nVOwjoNarfIvk7O-vA,5362
 codegraphcontext/tools/indexing/persistence/__init__.py,sha256=-fSSq1MMR2TB0iuzT80cxGAoKs55mzdYkduwyIzXCh0,59
-codegraphcontext/tools/indexing/persistence/writer.py,sha256=ByoUZRBnxZWQXm7D6vZpGtXJs7SUCoB1Uv5EVXHiNd8,32990
+codegraphcontext/tools/indexing/persistence/writer.py,sha256=41bY9inZKEEtchEP8tYaxK0ciH-oGlKm6k1XDCgb79Q,33538
 codegraphcontext/tools/indexing/resolution/__init__.py,sha256=gsnYNNawrafAyBCKhG7SOF2Um5GblZ-rhgz6S5J2v1s,249
-codegraphcontext/tools/indexing/resolution/calls.py,sha256=BROGHHtRdqzLdsNNi56ZSzbrTGnaOoWZdyCQbQ2r-sM,7743
+codegraphcontext/tools/indexing/resolution/calls.py,sha256=0nxb7Em2A6LwPl3DjA7cXADvAC0NRSliD-nT2ssbCgI,8256
 codegraphcontext/tools/indexing/resolution/inheritance.py,sha256=NZsYDwJpWoOdRSMNl8bPrBgP07iaMD5z1wgm2ELel0I,3292
 codegraphcontext/tools/languages/c.py,sha256=mrXQWEvMtzLVElATYNZ7VThNk1ZNZdw2UV8i6r2y2y4,22291
 codegraphcontext/tools/languages/cpp.py,sha256=aXBF_MX43PpRTr5U7UBcn5mZvP6fjP4Q8g21JuCho38,25761
 codegraphcontext/tools/languages/csharp.py,sha256=0ZIpV531i8yhz4vktaKkzbHDg-TCxgSZPC69SBkmfYU,22594
-codegraphcontext/tools/languages/dart.py,sha256=Qql2NhCBpAgh2EmWYJWbkqDAt6grvfrSLyKBs7HdRNU,15442
+codegraphcontext/tools/languages/dart.py,sha256=qt6NkN1IZZWQCcT6lpUqDdpBScXI498dJNjS1ZQPf3k,18705
 codegraphcontext/tools/languages/elixir.py,sha256=mFsMvi_5ZlCsrB5QQ7Kklj94tLo3D6aFU7ukuhGmGSI,18815
 codegraphcontext/tools/languages/go.py,sha256=cBfeM3Qt9W73KKb39VPhS6LlcWLdVJpvrAnnUeSiH4E,20456
 codegraphcontext/tools/languages/haskell.py,sha256=LboJms-iUZVBY32Q5GvWlyHCGXT4ov05JVO-zbdOFlQ,17285
@@ -62,7 +62,7 @@ codegraphcontext/tools/languages/javascript.py,sha256=XB-icOtzIERAtackWWmaayGAWR
 codegraphcontext/tools/languages/kotlin.py,sha256=6HvrClCgCNiGJnTjDhyHtuAy2PyGZwJN2Mx1zOA6T2c,28051
 codegraphcontext/tools/languages/perl.py,sha256=EHFpcGO2NLPPxyx4s0kmtqQ5_ODFLx7QHFDIvwlRb84,10004
 codegraphcontext/tools/languages/php.py,sha256=mqCoahKTrKzc0lsE-1K3S_UBT-pyIMi8_CrCJ2kHCUg,23350
-codegraphcontext/tools/languages/python.py,sha256=qB1rMjHKMtx4MHxg0zAOkFU34Zvete4JsdwW33b7_xk,25731
+codegraphcontext/tools/languages/python.py,sha256=abyN-POHLoA2PLkk8NikDgAi3368_II4Lk0-wNjky4Y,27189
 codegraphcontext/tools/languages/ruby.py,sha256=sd803ZpfQbsWuqXlUvX4UShM2NkkgUf9xOkl160X8oM,21993
 codegraphcontext/tools/languages/rust.py,sha256=TD7H4SsSsPWJnIUoThkA8gxIKnQhqL6_cVGnAaOBeEE,11991
 codegraphcontext/tools/languages/scala.py,sha256=-seQcdX0UJfDPiyhgf4kg6kDXJsSNcfVH472aSv1dH8,23003
@@ -124,9 +124,9 @@ codegraphcontext/viz/dist/wasm/tree-sitter-typescript.wasm,sha256=hRVATc7tOOHthq
 codegraphcontext/viz/dist/wasm/tree-sitter.wasm,sha256=CCeVuI_hXktkBD-DW01xYPv5XoAYVRIqzJUJI5te-RY,196763
 codegraphcontext/viz/dist/wasm/web-tree-sitter.js,sha256=DIaCNqRylrT_PBVw8g4ImeSnhP9uXNe_ycOlUiVGPko,153666
 codegraphcontext/viz/dist/wasm/web-tree-sitter.wasm,sha256=CCeVuI_hXktkBD-DW01xYPv5XoAYVRIqzJUJI5te-RY,196763
-codegraphcontext-0.4.4.dist-info/licenses/LICENSE,sha256=rh8M-bJpQYJnw2vtRVgt0t7piMZXh5QzaKeNEI0vqqA,1061
-codegraphcontext-0.4.4.dist-info/METADATA,sha256=FDddbhcyfIBGfUlyBl9UHaQgdtotbA4sed_Jo_M59rM,22330
-codegraphcontext-0.4.4.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codegraphcontext-0.4.4.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.4.4.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.4.4.dist-info/RECORD,,
+codegraphcontext-0.4.5.dist-info/licenses/LICENSE,sha256=rh8M-bJpQYJnw2vtRVgt0t7piMZXh5QzaKeNEI0vqqA,1061
+codegraphcontext-0.4.5.dist-info/METADATA,sha256=IJH89toSZECngKn505IurSIunzimNBsqsT2Dy_n0MWU,22355
+codegraphcontext-0.4.5.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+codegraphcontext-0.4.5.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
+codegraphcontext-0.4.5.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
+codegraphcontext-0.4.5.dist-info/RECORD,,