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

codegraphcontext/tools/tree_sitter_parser.py

@@ -1,9 +1,10 @@
 """Tree-sitter parser dispatch by language name."""
 
 from pathlib import Path
-from typing import Dict, Optional
+from typing import TYPE_CHECKING, Dict
 
-from tree_sitter import Language, Parser
+if TYPE_CHECKING:
+    from tree_sitter import Language
 
 from ..utils.tree_sitter_manager import get_tree_sitter_manager
 
@@ -15,8 +16,8 @@ class TreeSitterParser:
         self.language_name = language_name
         self.ts_manager = get_tree_sitter_manager()
 
-        self.language: Language = self.ts_manager.get_language_safe(language_name)
-        self.parser = Parser(self.language)
+        self.language: "Language" = self.ts_manager.get_language_safe(language_name)
+        self.parser = self.ts_manager.create_parser(language_name)
 
         self.language_specific_parser = None
         if self.language_name == "python":

codegraphcontext/utils/tree_sitter_manager.py

@@ -11,11 +11,55 @@ Key design principles:
 4. Support optional tree-sitter dependency
 """
 
-from typing import Dict, Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Dict, Optional
 import threading
+import sys
+
+if TYPE_CHECKING:
+    from tree_sitter import Language, Parser
+
+Language = Any
+Parser = Any
+_tree_sitter_import_error: Optional[ImportError] = None
+_Language = None
+_Parser = None
+_get_language = None
+
+
+def _missing_tree_sitter_error(import_error: ImportError) -> ImportError:
+    """Return an actionable error for optional tree-sitter dependencies."""
+    if sys.version_info >= (3, 13):
+        return ImportError(
+            "Tree-sitter parsing is not available on Python 3.13 because "
+            "tree-sitter-language-pack does not publish cp313 wheels. "
+            "Install CodeGraphContext with Python 3.12 or 3.14 to use indexing/parsing."
+        )
+    return ImportError(
+        "tree-sitter and tree-sitter-language-pack are required for code parsing. "
+        "Install them with: pip install codegraphcontext[parsing]"
+    )
+
 
-from tree_sitter import Language, Parser
-from tree_sitter_language_pack import get_language
+def _load_tree_sitter_dependencies():
+    """Load optional tree-sitter dependencies only when parsing is used."""
+    global _tree_sitter_import_error, _Language, _Parser, _get_language
+
+    if _Language is not None and _Parser is not None and _get_language is not None:
+        return _Language, _Parser, _get_language
+
+    try:
+        from tree_sitter import Language as ImportedLanguage, Parser as ImportedParser
+        from tree_sitter_language_pack import get_language as imported_get_language
+    except ImportError as e:
+        _tree_sitter_import_error = e
+        raise _missing_tree_sitter_error(e) from e
+
+    _Language = ImportedLanguage
+    _Parser = ImportedParser
+    _get_language = imported_get_language
+    return _Language, _Parser, _get_language
 
 
 # Language name aliases for compatibility
@@ -122,6 +166,7 @@ class TreeSitterManager:
         """
         # Normalize the language name
         canonical_name = self._normalize_language_name(lang)
+        _, _, load_language = _load_tree_sitter_dependencies()
         
         # Check cache first (fast path, no lock needed for reads)
         if canonical_name in self._language_cache:
@@ -136,7 +181,7 @@ class TreeSitterManager:
             try:
                 # Map canonical name to language-pack name where they differ
                 pack_name = LANGUAGE_PACK_NAMES.get(canonical_name, canonical_name)
-                language = get_language(pack_name)
+                language = load_language(pack_name)
                 
                 self._language_cache[canonical_name] = language
                 return language
@@ -167,9 +212,10 @@ class TreeSitterManager:
             ValueError: If language is not supported
             Exception: If parser creation fails
         """
+        _, parser_cls, _ = _load_tree_sitter_dependencies()
         language = self.get_language_safe(lang)
         # In tree-sitter 0.25+, Parser takes language in constructor
-        parser = Parser(language)
+        parser = parser_cls(language)
         return parser
     
     def is_language_available(self, lang: str) -> bool:
@@ -256,7 +302,10 @@ def execute_query(language: Language, query_string: str, node):
         >>> for node, name in captures:
         ...     print(f'{name}: {node.type}')
     """
-    from tree_sitter import Query, QueryCursor
+    try:
+        from tree_sitter import Query, QueryCursor
+    except ImportError as e:
+        raise _missing_tree_sitter_error(e) from e
     
     try:
         # Create query and cursor
@@ -282,4 +331,3 @@ def execute_query(language: Language, query_string: str, node):
             f"Failed to execute query: {e}\n"
             f"Query string: {query_string[:100]}..."
         )
-

codegraphcontext/viz/server.py

@@ -81,13 +81,16 @@ async def get_graph(repo_path: Optional[str] = None, cypher_query: Optional[str]
             elif repo_path:
                 repo_path = str(Path(repo_path).resolve())
                 print(f"DEBUG: Fetching subgraph for: {repo_path}", flush=True)
+                # Get all nodes within the repository scope
                 query = """
                 MATCH (r:Repository {path: $repo_path})
                 OPTIONAL MATCH (r)-[:CONTAINS*0..]->(n)
-                WITH DISTINCT n
-                WHERE n IS NOT NULL
-                OPTIONAL MATCH (n)-[rel]->(m)
-                RETURN n, rel, m
+                WITH DISTINCT r, COLLECT(DISTINCT n) as repo_nodes
+                UNWIND repo_nodes as node
+                OPTIONAL MATCH (node)-[rel]->(target)
+                WITH r, node, rel, target, repo_nodes
+                WHERE target IN repo_nodes OR target = r
+                RETURN node as n, rel, target as m
                 """
                 result = session.run(query, repo_path=repo_path)
             else:

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

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.4.2
+Version: 0.4.4
 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
         
-        Copyright (c) 2025
+        Copyright (c) 2025-2026
         
         Permission is hereby granted, free of charge, to any person obtaining a copy
         of this software and associated documentation files (the "Software"), to deal
@@ -39,13 +39,13 @@ License-File: LICENSE
 Requires-Dist: neo4j>=5.15.0
 Requires-Dist: watchdog>=3.0.0
 Requires-Dist: stdlibs>=2023.11.18
-Requires-Dist: typer[all]>=0.9.0
+Requires-Dist: typer>=0.9.0
 Requires-Dist: rich>=13.7.0
 Requires-Dist: inquirerpy>=0.3.4
 Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: tree-sitter>=0.21.0
-Requires-Dist: tree-sitter-language-pack>=0.6.0
-Requires-Dist: tree-sitter-c-sharp>=0.21.0
+Requires-Dist: tree-sitter>=0.21.0; python_version != "3.13"
+Requires-Dist: tree-sitter-language-pack>=0.6.0; python_version != "3.13"
+Requires-Dist: tree-sitter-c-sharp>=0.21.0; python_version != "3.13"
 Requires-Dist: pyyaml
 Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
@@ -57,9 +57,9 @@ Requires-Dist: kuzu>=0.4.0; sys_platform == "win32" or (sys_platform != "win32"
 Requires-Dist: fastapi>=0.100.0
 Requires-Dist: uvicorn>=0.22.0
 Provides-Extra: parsing
-Requires-Dist: tree-sitter>=0.21.0; extra == "parsing"
-Requires-Dist: tree-sitter-language-pack>=0.6.0; extra == "parsing"
-Requires-Dist: tree-sitter-c-sharp>=0.21.0; extra == "parsing"
+Requires-Dist: tree-sitter>=0.21.0; python_version != "3.13" and extra == "parsing"
+Requires-Dist: tree-sitter-language-pack>=0.6.0; python_version != "3.13" and extra == "parsing"
+Requires-Dist: tree-sitter-c-sharp>=0.21.0; python_version != "3.13" and extra == "parsing"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
@@ -165,7 +165,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.4.2
+- **Version:** 0.4.4
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -251,12 +251,12 @@ _If you’re using CodeGraphContext in your project, feel free to open a PR and
 - `neo4j>=5.15.0`
 - `watchdog>=3.0.0`
 - `stdlibs>=2023.11.18`
-- `typer[all]>=0.9.0`
+- `typer>=0.9.0`
 - `rich>=13.7.0`
 - `inquirerpy>=0.3.7`
 - `python-dotenv>=1.0.0`
-- `tree-sitter>=0.21.0`
-- `tree-sitter-language-pack>=0.6.0`
+- `tree-sitter>=0.21.0` (not installed on Python 3.13)
+- `tree-sitter-language-pack>=0.6.0` (not installed on Python 3.13)
 - `pyyaml`
 - `pytest`
 - `nbformat`

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

@@ -30,7 +30,7 @@ codegraphcontext/tools/package_resolver.py,sha256=KtbdMReTezszjdsqYniL-Xb-QUsrAJ
 codegraphcontext/tools/scip_indexer.py,sha256=MdxesJ4GQBOwrE2q3SZjImkzh6BSPQeA1tYsR5Zerok,20021
 codegraphcontext/tools/scip_pb2.py,sha256=dwOMNKlu6VyLq5h8kTPZRDqxrwfVL8yw7I7ziokFG48,98229
 codegraphcontext/tools/system.py,sha256=goQOs1A9gvd7SvDPVZB6Jd0DRJKQmjEkn-uiTE40VNM,6046
-codegraphcontext/tools/tree_sitter_parser.py,sha256=3LW7ewMoU7X2hVKdqvSTBO2PvWv22ITKZWGJbQwqrtI,4456
+codegraphcontext/tools/tree_sitter_parser.py,sha256=x8gDBCQze65r97DwlIovflg02wrY5SKm4nCVxxIja2E,4500
 codegraphcontext/tools/handlers/analysis_handlers.py,sha256=ZhwPNIm6aLBcaDORiMeLx0Mq7QWj4hS64Gozh5KTerY,4792
 codegraphcontext/tools/handlers/indexing_handlers.py,sha256=RfmJf57oAnVYca8lOBQz8vcjxkSQn7NbixSgGDHm59g,5273
 codegraphcontext/tools/handlers/management_handlers.py,sha256=_sWVum15qdoMBSx6si7-CRJCr3fmBW4p7npxEoAStcU,13806
@@ -87,9 +87,9 @@ codegraphcontext/tools/query_tool_languages/typescript_toolkit.py,sha256=3S4hpmO
 codegraphcontext/utils/debug_log.py,sha256=Qg7jwyeg7x2h3Ur_2S34bdMCkHdlk_ngHfPwa97A9vE,2836
 codegraphcontext/utils/path_ignore.py,sha256=WPIcDmyAi6eL9692rPBQrUZunn5gHRb57nzNRHSIq28,2124
 codegraphcontext/utils/repo_path.py,sha256=2ydwYV4aK04eGPKqjhKMPZFZv7sNJitjig4hcl8cssw,1079
-codegraphcontext/utils/tree_sitter_manager.py,sha256=n8z45gxiOH_W1cbwe9Z7mvbB5gjGjUGRvV78JjftG_M,8993
+codegraphcontext/utils/tree_sitter_manager.py,sha256=OMegIfLi95l3zC-SkGoxYdFd2FidKfDA-hPRhoLONp8,10803
 codegraphcontext/utils/visualize_graph.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codegraphcontext/viz/server.py,sha256=LUK3aHPpapZATG9c9ZeTsx3624bzy0w4rocms--VbWc,13155
+codegraphcontext/viz/server.py,sha256=UThvOdqlRT3aA5_lhS32FLQlAZu-U9mhPWCHnGmFSI0,13393
 codegraphcontext/viz/dist/favicon.ico,sha256=AjhUxD_Eslt5XuSVHIAZ494Fk__rb5GLXR8qm0elfP4,1870
 codegraphcontext/viz/dist/index.html,sha256=g97sbeOQEOHA75sDelnPBEuzLfcaKGdH9DAQ7hfE6PU,1879
 codegraphcontext/viz/dist/placeholder.svg,sha256=ZLrfeqvaC5YwuHAg_7YJXLhYzLz2azVcKqCLEGOVTTs,3253
@@ -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.2.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-codegraphcontext-0.4.2.dist-info/METADATA,sha256=2lvZnJhWuYSikZZsvR7UoLO3v0oPZO29J-917UPht4M,22108
-codegraphcontext-0.4.2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codegraphcontext-0.4.2.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.4.2.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.4.2.dist-info/RECORD,,
+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.2.dist-info → codegraphcontext-0.4.4.dist-info}/licenses/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2025
+Copyright (c) 2025-2026
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal