codegraphcontext 0.2.3__tar.gz → 0.2.5__tar.gz

{codegraphcontext-0.2.3/src/codegraphcontext.egg-info → codegraphcontext-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.2.3
+Version: 0.2.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
@@ -51,7 +51,6 @@ Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: falkordblite>=0.1.0; sys_platform != "win32" and python_version >= "3.12"
-Requires-Dist: requests
 Provides-Extra: parsing
 Requires-Dist: tree-sitter>=0.21.0; extra == "parsing"
 Requires-Dist: tree-sitter-language-pack>=0.6.0; extra == "parsing"
@@ -147,7 +146,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.2.3
+- **Version:** 0.2.5
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/README.md

@@ -84,7 +84,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.2.3
+- **Version:** 0.2.5
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.2.3"
+version = "0.2.5"
 description = "An MCP server that indexes local code into a graph database to provide context to AI assistants."
 authors = [{ name = "Shashank Shekhar Singh", email = "shashankshekharsingh1205@gmail.com" }]
 readme = "README.md"
@@ -29,8 +29,7 @@ dependencies = [
     "nbformat",
     "nbconvert>=7.16.6",
     "pathspec>=0.12.1",
-    "falkordblite>=0.1.0; sys_platform != 'win32' and python_version >= '3.12'",
-    "requests"
+    "falkordblite>=0.1.0; sys_platform != 'win32' and python_version >= '3.12'"
 ]
 
 [project.optional-dependencies]

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/src/codegraphcontext/cli/setup_wizard.py

@@ -140,7 +140,7 @@ def _configure_ide(mcp_config):
     questions = [
         {
             "type": "confirm",
-            "message": "Automatically configure your IDE/CLI (VS Code, Cursor, Windsurf, Claude, Gemini, Cline, RooCode, ChatGPT Codex, Amazon Q Developer, Aider, Kiro)?",
+            "message": "Automatically configure your IDE/CLI (VS Code, Cursor, Windsurf, Claude, Gemini, Cline, RooCode, ChatGPT Codex, Amazon Q Developer, Aider, Kiro, Antigravity)?",
             "name": "configure_ide",
             "default": True,
         }
@@ -154,7 +154,7 @@ def _configure_ide(mcp_config):
         {
             "type": "list",
             "message": "Choose your IDE/CLI to configure:",
-            "choices": ["VS Code", "Cursor", "Windsurf", "Claude code", "Gemini CLI", "ChatGPT Codex", "Cline", "RooCode", "Amazon Q Developer", "JetBrainsAI", "Aider", "Kiro", "None of the above"],
+            "choices": ["VS Code", "Cursor", "Windsurf", "Claude code", "Gemini CLI", "ChatGPT Codex", "Cline", "RooCode", "Amazon Q Developer", "JetBrainsAI", "Aider", "Kiro", "Antigravity", "None of the above"],
             "name": "ide_choice",
         }
     ]
@@ -166,7 +166,7 @@ def _configure_ide(mcp_config):
         return
 
 
-    if ide_choice in ["VS Code", "Cursor/CLI", "Claude code", "Gemini CLI", "ChatGPT Codex", "Cline", "Windsurf", "RooCode", "Amazon Q Developer , JetBrainsAI", "Aider", "Kiro"]:
+    if ide_choice in ["VS Code", "Cursor", "Claude code", "Gemini CLI", "ChatGPT Codex", "Cline", "Windsurf", "RooCode", "Amazon Q Developer", "JetBrainsAI", "Aider", "Kiro", "Antigravity"]:
         console.print(f"\n[bold cyan]Configuring for {ide_choice}...[/bold cyan]")
 
         if ide_choice == "Amazon Q Developer":
@@ -229,6 +229,11 @@ def _configure_ide(mcp_config):
                 Path.home() / ".kiro" / "settings" / "mcp.json",                                   # macOS / Linux / Windows (user-level global)
                 Path.home() / ".config" / "kiro" / "settings" / "mcp.json",                         # Linux (XDG config)
                 Path.home() / "AppData" / "Roaming" / "Kiro" / "settings" / "mcp.json",             # Windows
+            ],
+            "Antigravity": [
+                Path.home() / ".antigravity" / "mcp_settings.json",                                # macOS / Linux / Windows (user-level global)
+                Path.home() / ".config" / "antigravity" / "mcp_settings.json",                     # Linux (XDG config)
+                Path.home() / "AppData" / "Roaming" / "Antigravity" / "mcp_settings.json",         # Windows
             ]
         }
 

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/src/codegraphcontext/core/database_falkordb.py

@@ -15,6 +15,21 @@ from typing import Optional, Tuple
 
 from codegraphcontext.utils.debug_log import debug_log, info_logger, error_logger, warning_logger
 
+# ---------------------------------------------------------------------------
+# Compatibility patch: redis-py >= 5.x added OpenTelemetry error telemetry that
+# accesses conn.port inside its error handler. UnixDomainSocketConnection never
+# had a 'port' attribute, so any exception raised during a Unix-socket connection
+# (e.g. the sentinel-detection INFO call inside FalkorDB.__init__) would produce
+# a secondary AttributeError masking the real problem.
+# Patching the class at import time costs nothing and fixes all call-sites.
+# ---------------------------------------------------------------------------
+try:
+    from redis.connection import UnixDomainSocketConnection as _UDSC
+    if not hasattr(_UDSC, 'port'):
+        _UDSC.port = 0  # type: ignore[attr-defined]
+except Exception:
+    pass  # redis not installed or class structure changed — safe to ignore
+
 class FalkorDBManager:
     """
     Manages the FalkorDB Lite database connection as a singleton.

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/src/codegraphcontext/tools/code_finder.py

@@ -14,9 +14,22 @@ class CodeFinder:
     def __init__(self, db_manager: DatabaseManager):
         self.db_manager = db_manager
         self.driver = self.db_manager.get_driver()
+        self._is_falkordb = getattr(db_manager, 'get_backend_type', lambda: 'neo4j')() != 'neo4j'
 
     def format_query(self, find_by: Literal["Class", "Function"], fuzzy_search:bool) -> str:
         """Format the search query based on the search type and fuzzy search settings."""
+        if self._is_falkordb:
+            # FalkorDB does not support CALL db.idx.fulltext.queryNodes.
+            # Fall back to a pure Cypher CONTAINS/toLower match on node name.
+            name_filter = "toLower(node.name) CONTAINS toLower($search_term)"
+            return f"""
+                MATCH (node:{find_by})
+                WHERE {name_filter}
+                RETURN node.name as name, node.path as path, node.line_number as line_number,
+                    node.source as source, node.docstring as docstring, node.is_dependency as is_dependency
+                ORDER BY node.is_dependency ASC, node.name
+                LIMIT 20
+            """
         return f"""
             CALL db.index.fulltext.queryNodes("code_search_index", $search_term) YIELD node, score
                 WITH node, score
@@ -40,8 +53,10 @@ class CodeFinder:
                 """, name=search_term)
                 return result.data()
             
-            # Fuzzy search using fulltext index
-            formatted_search_term = f"name:{search_term}"
+            # Fuzzy search using fulltext index (Neo4j) or CONTAINS fallback (FalkorDB)
+            # On FalkorDB, format_query uses CONTAINS so we pass the raw term; on Neo4j
+            # we need the Lucene field-selector prefix.
+            formatted_search_term = search_term if self._is_falkordb else f"name:{search_term}"
             result = session.run(self.format_query("Function", fuzzy_search), search_term=formatted_search_term)
             return result.data()
 
@@ -58,8 +73,10 @@ class CodeFinder:
                 """, name=search_term)
                 return result.data()
 
-            # Fuzzy search using fulltext index
-            formatted_search_term = f"name:{search_term}"
+            # Fuzzy search using fulltext index (Neo4j) or CONTAINS fallback (FalkorDB)
+            # On FalkorDB, format_query uses CONTAINS so we pass the raw term; on Neo4j
+            # we need the Lucene field-selector prefix.
+            formatted_search_term = search_term if self._is_falkordb else f"name:{search_term}"
             result = session.run(self.format_query("Class", fuzzy_search), search_term=formatted_search_term)
             return result.data()
 
@@ -79,6 +96,8 @@ class CodeFinder:
 
     def find_by_content(self, search_term: str) -> List[Dict]:
         """Find code by content matching in source or docstrings using the full-text index."""
+        if self._is_falkordb:
+            return self._find_by_content_falkordb(search_term)
         with self.driver.session() as session:
             result = session.run("""
                 CALL db.index.fulltext.queryNodes("code_search_index", $search_term) YIELD node, score
@@ -86,10 +105,10 @@ class CodeFinder:
                 WHERE node:Function OR node:Class OR node:Variable
                 MATCH (node)<-[:CONTAINS]-(f:File)
                 RETURN
-                    CASE 
+                    CASE
                         WHEN node:Function THEN 'function'
                         WHEN node:Class THEN 'class'
-                        ELSE 'variable' 
+                        ELSE 'variable'
                     END as type,
                     node.name as name, f.path as path,
                     node.line_number as line_number, node.source as source,
@@ -98,6 +117,32 @@ class CodeFinder:
                 LIMIT 20
             """, search_term=search_term)
             return result.data()
+
+    def _find_by_content_falkordb(self, search_term: str) -> List[Dict]:
+        """FalkorDB-compatible content search using pure Cypher CONTAINS matching.
+        FalkorDB does not support CALL db.idx.fulltext.queryNodes, so we fall back
+        to substring matching on name, source, and docstring fields."""
+        all_results = []
+        with self.driver.session() as session:
+            for label, type_name in [('Function', 'function'), ('Class', 'class')]:
+                try:
+                    result = session.run(f"""
+                        MATCH (node:{label})
+                        WHERE toLower(node.name) CONTAINS toLower($search_term)
+                            OR (node.source IS NOT NULL AND toLower(node.source) CONTAINS toLower($search_term))
+                            OR (node.docstring IS NOT NULL AND toLower(node.docstring) CONTAINS toLower($search_term))
+                        RETURN
+                            '{type_name}' as type,
+                            node.name as name, node.path as path,
+                            node.line_number as line_number, node.source as source,
+                            node.docstring as docstring, node.is_dependency as is_dependency
+                        ORDER BY node.is_dependency ASC, node.name
+                        LIMIT 20
+                    """, search_term=search_term)
+                    all_results.extend(result.data())
+                except Exception:
+                    logger.debug(f"FalkorDB content query failed for label {label}", exc_info=True)
+        return all_results[:20]
     
     def find_by_module_name(self, search_term: str) -> List[Dict]:
         """Find modules by name matching"""
@@ -130,6 +175,12 @@ class CodeFinder:
 
     def find_related_code(self, user_query: str, fuzzy_search: bool, edit_distance: int) -> Dict[str, Any]:
         """Find code related to a query using multiple search strategies"""
+        # FalkorDB does not support Lucene-style fuzzy edit-distance syntax (e.g. term~2).
+        # On FalkorDB, always use the plain query so that the CONTAINS-based fallbacks work.
+        if fuzzy_search and self._is_falkordb:
+            logger.debug("FalkorDB backend: ignoring fuzzy edit-distance normalisation; using plain CONTAINS search.")
+            fuzzy_search = False
+
         if fuzzy_search:
             user_query_normalized = " ".join(map(lambda x: f"{x}~{edit_distance}", user_query.split(" ")))
         else:

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/src/codegraphcontext/tools/graph_builder.py

@@ -105,6 +105,7 @@ class GraphBuilder:
             '.cpp': TreeSitterParser('cpp'),
             '.h': TreeSitterParser('cpp'),
             '.hpp': TreeSitterParser('cpp'),
+            '.hh': TreeSitterParser('cpp'),
             '.rs': TreeSitterParser('rust'),
             '.c': TreeSitterParser('c'),
             # '.h': TreeSitterParser('c'), # Need to write an algo for distinguishing C vs C++ headers
@@ -149,11 +150,20 @@ class GraphBuilder:
                 session.run("CREATE INDEX function_lang IF NOT EXISTS FOR (f:Function) ON (f.lang)")
                 session.run("CREATE INDEX class_lang IF NOT EXISTS FOR (c:Class) ON (c.lang)")
                 session.run("CREATE INDEX annotation_lang IF NOT EXISTS FOR (a:Annotation) ON (a.lang)")
-                session.run("""
-                    CREATE FULLTEXT INDEX code_search_index IF NOT EXISTS 
-                    FOR (n:Function|Class|Variable) 
-                    ON EACH [n.name, coalesce(n.source, ''), coalesce(n.docstring, '')]
-                """ )
+                is_falkordb = getattr(self.db_manager, 'get_backend_type', lambda: 'neo4j')() != 'neo4j'
+                if is_falkordb:
+                    # FalkorDB uses db.idx.fulltext.createNodeIndex per label
+                    for label in ['Function', 'Class']:
+                        try:
+                            session.run(f"CALL db.idx.fulltext.createNodeIndex('{label}', 'name', 'source', 'docstring')")
+                        except Exception:
+                            pass  # Index may already exist
+                else:
+                    session.run("""
+                        CREATE FULLTEXT INDEX code_search_index IF NOT EXISTS
+                        FOR (n:Function|Class|Variable)
+                        ON EACH [n.name, n.source, n.docstring]
+                    """)
                 
                 info_logger("Database schema verified/created successfully")
             except Exception as e:
@@ -209,6 +219,9 @@ class GraphBuilder:
         if '.hpp' in files_by_lang:
             from .languages import cpp as cpp_lang_module
             imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hpp'], self.parsers['.hpp']))
+        if '.hh' in files_by_lang:
+            from .languages import cpp as cpp_lang_module
+            imports_map.update(cpp_lang_module.pre_scan_cpp(files_by_lang['.hh'], self.parsers['.hh']))
         if '.rs' in files_by_lang:
             from .languages import rust as rust_lang_module
             imports_map.update(rust_lang_module.pre_scan_rust(files_by_lang['.rs'], self.parsers['.rs']))

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.2.3
+Version: 0.2.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
@@ -51,7 +51,6 @@ Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: falkordblite>=0.1.0; sys_platform != "win32" and python_version >= "3.12"
-Requires-Dist: requests
 Provides-Extra: parsing
 Requires-Dist: tree-sitter>=0.21.0; extra == "parsing"
 Requires-Dist: tree-sitter-language-pack>=0.6.0; extra == "parsing"
@@ -147,7 +146,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.2.3
+- **Version:** 0.2.5
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.2.3 → codegraphcontext-0.2.5}/src/codegraphcontext.egg-info/requires.txt

@@ -12,7 +12,6 @@ pytest
 nbformat
 nbconvert>=7.16.6
 pathspec>=0.12.1
-requests
 
 [:sys_platform != "win32" and python_version >= "3.12"]
 falkordblite>=0.1.0