codegraphcontext 0.1.17__tar.gz → 0.1.18__tar.gz

{codegraphcontext-0.1.17/src/codegraphcontext.egg-info → codegraphcontext-0.1.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.17
+Version: 0.1.18
 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
@@ -33,7 +33,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
-Requires-Python: >=3.9
+Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: neo4j>=5.15.0
@@ -51,6 +51,7 @@ Requires-Dist: pytest
 Requires-Dist: nbformat
 Requires-Dist: nbconvert>=7.16.6
 Requires-Dist: pathspec>=0.12.1
+Requires-Dist: falkordblite>=0.1.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.4.0; extra == "dev"
 Requires-Dist: black>=23.11.0; extra == "dev"
@@ -125,6 +126,11 @@ If you’re using CodeGraphContext in your project, feel free to open a PR and a
 - `python-dotenv>=1.0.0`
 - `tree-sitter==0.20.4`
 - `tree-sitter-languages==1.10.2`
+- `pyyaml`
+- `pytest`
+- `nbformat`
+- `nbconvert>=7.16.6`
+- `pathspec>=0.12.1`
 
 ## Getting Started
 

{codegraphcontext-0.1.17 → codegraphcontext-0.1.18}/README.md

@@ -66,6 +66,11 @@ If you’re using CodeGraphContext in your project, feel free to open a PR and a
 - `python-dotenv>=1.0.0`
 - `tree-sitter==0.20.4`
 - `tree-sitter-languages==1.10.2`
+- `pyyaml`
+- `pytest`
+- `nbformat`
+- `nbconvert>=7.16.6`
+- `pathspec>=0.12.1`
 
 ## Getting Started
 

{codegraphcontext-0.1.17 → codegraphcontext-0.1.18}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.17"
+version = "0.1.18"
 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"
 license = { file = "LICENSE" }
-requires-python = ">=3.9"
+requires-python = ">=3.12"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: MIT License",
@@ -29,7 +29,8 @@ dependencies = [
     "pytest",
     "nbformat",
     "nbconvert>=7.16.6",
-    "pathspec>=0.12.1"
+    "pathspec>=0.12.1",
+    "falkordblite>=0.1.0"
 ]
 
 [project.urls]

{codegraphcontext-0.1.17 → codegraphcontext-0.1.18}/src/codegraphcontext/cli/cli_helpers.py

@@ -7,7 +7,7 @@ import time
 from rich.console import Console
 from rich.table import Table
 
-from ..core.database import DatabaseManager
+from ..core import get_database_manager
 from ..core.jobs import JobManager
 from ..tools.code_finder import CodeFinder
 from ..tools.graph_builder import GraphBuilder
@@ -19,7 +19,12 @@ console = Console()
 def _initialize_services():
     """Initializes and returns core service managers."""
     console.print("[dim]Initializing services and database connection...[/dim]")
-    db_manager = DatabaseManager()
+    try:
+        db_manager = get_database_manager()
+    except ValueError as e:
+        console.print(f"[bold red]Database Configuration Error:[/bold red] {e}")
+        return None, None, None
+
     try:
         db_manager.get_driver()
     except ValueError as e:
@@ -188,14 +193,134 @@ def cypher_helper(query: str):
         db_manager.close_driver()
 
 
+import webbrowser
+
 def visualize_helper(query: str):
-    """Generates a URL to visualize a Cypher query."""
+    """Generates a visualization."""
+    services = _initialize_services()
+    if not all(services):
+        return
+
+    db_manager, _, _ = services
+    
+    # Check if FalkorDB
+    if "FalkorDB" in db_manager.__class__.__name__:
+        _visualize_falkordb(db_manager)
+    else:
+        try:
+            encoded_query = urllib.parse.quote(query)
+            visualization_url = f"http://localhost:7474/browser/?cmd=edit&arg={encoded_query}"
+            console.print("[green]Graph visualization URL:[/green]")
+            console.print(visualization_url)
+            console.print("Open the URL in your browser to see the graph.")
+        except Exception as e:
+            console.print(f"[bold red]An error occurred while generating URL:[/bold red] {e}")
+        finally:
+            db_manager.close_driver()
+
+def _visualize_falkordb(db_manager):
+    console.print("[dim]Generating FalkorDB visualization (showing up to 500 relationships)...[/dim]")
     try:
-        encoded_query = urllib.parse.quote(query)
-        visualization_url = f"http://localhost:7474/browser/?cmd=edit&arg={encoded_query}"
-        console.print("[green]Graph visualization URL:[/green]")
-        console.print(visualization_url)
-        console.print("Open the URL in your browser to see the graph.")
+        data_nodes = []
+        data_edges = []
+        
+        with db_manager.get_driver().session() as session:
+            # Fetch nodes and edges
+            q = "MATCH (n)-[r]->(m) RETURN n, r, m LIMIT 500"
+            result = session.run(q)
+            
+            seen_nodes = set()
+            
+            for record in result:
+                # record values are Node/Relationship objects from falkordb client
+                n = record['n']
+                r = record['r']
+                m = record['m']
+                
+                # Process Node helper
+                def process_node(node):
+                    nid = getattr(node, 'id', -1)
+                    labels = getattr(node, 'labels', [])
+                    lbl = list(labels)[0] if labels else "Node"
+                    props = getattr(node, 'properties', {})
+                    name = props.get('name', str(nid))
+                    
+                    if nid not in seen_nodes:
+                        seen_nodes.add(nid)
+                        color = "#97c2fc" # Default blue
+                        if "Repository" in labels: color = "#ffb3ba" # Red
+                        elif "File" in labels: color = "#baffc9" # Green
+                        elif "Class" in labels: color = "#bae1ff" # Light Blue
+                        elif "Function" in labels: color = "#ffffba" # Yellow
+                        elif "Package" in labels: color = "#ffdfba" # Orange
+                        
+                        data_nodes.append({
+                            "id": nid, 
+                            "label": name, 
+                            "group": lbl, 
+                            "title": str(props),
+                            "color": color
+                        })
+                    return nid
+
+                nid = process_node(n)
+                mid = process_node(m)
+                
+                # Check Edge
+                e_type = getattr(r, 'relation', '') or getattr(r, 'type', 'REL')
+                data_edges.append({
+                    "from": nid,
+                    "to": mid,
+                    "label": e_type,
+                    "arrows": "to"
+                })
+        
+        filename = "codegraph_viz.html"
+        html_content = f"""
+<!DOCTYPE html>
+<html>
+<head>
+  <title>CodeGraphContext Visualization</title>
+  <script type="text/javascript" src="https://unpkg.com/vis-network/standalone/umd/vis-network.min.js"></script>
+  <style type="text/css">
+    #mynetwork {{
+      width: 100%;
+      height: 100vh;
+      border: 1px solid lightgray;
+    }}
+  </style>
+</head>
+<body>
+  <div id="mynetwork"></div>
+  <script type="text/javascript">
+    var nodes = new vis.DataSet({json.dumps(data_nodes)});
+    var edges = new vis.DataSet({json.dumps(data_edges)});
+    var container = document.getElementById('mynetwork');
+    var data = {{ nodes: nodes, edges: edges }};
+    var options = {{
+        nodes: {{ shape: 'dot', size: 16 }},
+        physics: {{ stabilization: false }},
+        layout: {{ improvedLayout: false }}
+    }};
+    var network = new vis.Network(container, data, options);
+  </script>
+</body>
+</html>
+"""
+        
+        out_path = Path(filename).resolve()
+        with open(out_path, "w") as f:
+            f.write(html_content)
+            
+        console.print(f"[green]Visualization generated at:[/green] {out_path}")
+        console.print("Opening in default browser...")
+        webbrowser.open(f"file://{out_path}")
+
     except Exception as e:
-        console.print(f"[bold red]An error occurred while generating URL:[/bold red] {e}")
+        console.print(f"[bold red]Visualization failed:[/bold red] {e}")
+        import traceback
+        traceback.print_exc()
+    finally:
+        db_manager.close_driver()
+
 

{codegraphcontext-0.1.17 → codegraphcontext-0.1.18}/src/codegraphcontext/cli/main.py

@@ -18,7 +18,7 @@ import logging
 import json
 import os
 from pathlib import Path
-from dotenv import load_dotenv, find_dotenv
+from dotenv import load_dotenv, find_dotenv, set_key
 from importlib.metadata import version as pkg_version, PackageNotFoundError
 
 from codegraphcontext.server import MCPServer
@@ -63,10 +63,17 @@ def get_version() -> str:
 @app.command()
 def setup():
     """
-    Runs the interactive setup wizard to configure the server and database connection.
-    This helps users set up a local Docker-based Neo4j instance or connect to a remote one.
+    Configure Neo4j Database Connection.
+    
+    Note: CodeGraphContext works out of the box with FalkorDB Lite (Default).
+    This setup is ONLY required if you want to use an external Neo4j database.
     """
-    run_setup_wizard()
+    console.print("\n[bold cyan]CodeGraphContext Setup[/bold cyan]")
+    console.print("CodeGraphContext works out of the box with FalkorDB. This setup is only required if you want to use Neo4j.\n")
+    if typer.confirm("Do you want to configure Neo4j now?", default=True):
+         run_setup_wizard()
+    else:
+         console.print("Setup cancelled. You can continue using FalkorDB Lite.")
 
 def _load_credentials():
     """
@@ -112,6 +119,33 @@ def _load_credentials():
         console.print(f"[bold red]Error loading .env file:[/bold red] {e}")
 
 
+# Create a subcommand group for "default"
+default_app = typer.Typer(help="Manage default configurations")
+app.add_typer(default_app, name="default")
+
+@default_app.command("database")
+def set_default_database(db_type: str = typer.Argument(..., help="Backend type: 'falkordb' or 'neo4j'")):
+    """
+    Set the default database backend.
+    
+    This preference is stored in ~/.codegraphcontext/.env
+    """
+    db_type = db_type.lower()
+    if db_type not in ['falkordb', 'neo4j']:
+        console.print(f"[bold red]Invalid database type: {db_type}.[/bold red] Must be 'falkordb' or 'neo4j'")
+        raise typer.Exit(code=1)
+    
+    env_path = Path.home() / ".codegraphcontext" / ".env"
+    env_path.parent.mkdir(parents=True, exist_ok=True)
+    if not env_path.exists():
+        env_path.touch()
+        
+    set_key(env_path, "DEFAULT_DATABASE", db_type)
+    console.print(f"[green]✔ Default database set to {db_type}[/green]")
+
+
+
+
 @app.command()
 def start():
     """
@@ -168,6 +202,7 @@ def visualize(query: Optional[str] = typer.Argument(None, help="The Cypher query
     """
     if query is None:
         query = "MATCH p=()-->() RETURN p"
+    _load_credentials()
     visualize_helper(query)
 
 @app.command(name="list-repos")
@@ -239,6 +274,7 @@ def version_cmd():
 @app.callback(invoke_without_command=True)
 def main(
     ctx: typer.Context,
+    database: Optional[str] = typer.Option(None, "--database", "-d", help="Override default database backend (falkordb or neo4j)"),
     version_: bool = typer.Option(
         None,
         "--version",
@@ -251,6 +287,9 @@ def main(
     Main entry point for the cgc CLI application.
     If no subcommand is provided, it displays a welcome message with instructions.
     """
+    if database:
+        os.environ["CGC_RUNTIME_DB_TYPE"] = database
+
     if version_:
         console.print(f"CodeGraphContext [bold cyan]{get_version()}[/bold cyan]")
         raise typer.Exit()
@@ -262,3 +301,6 @@ def main(
         console.print("👉 Run [cyan]cgc help[/cyan] to see all available commands.\n")
         console.print("👉 Run [cyan]cgc --version[/cyan] to check the version.\n")
         console.print("👉 Running [green]codegraphcontext [white]works the same as using [green]cgc")
+
+if __name__ == "__main__":
+    app()

codegraphcontext-0.1.18/src/codegraphcontext/core/__init__.py

@@ -0,0 +1,92 @@
+# src/codegraphcontext/core/__init__.py
+"""
+Core database management module.
+
+Supports both Neo4j and FalkorDB Lite backends.
+Use DATABASE_TYPE environment variable to switch:
+- DATABASE_TYPE=falkordb - Uses embedded FalkorDB Lite (recommended for lite-version)
+- DATABASE_TYPE=neo4j - Uses Neo4j server
+- If not set, auto-detects based on what's available
+"""
+import os
+from typing import Union
+
+import importlib.util
+
+def _is_falkordb_available() -> bool:
+    """Check if FalkorDB Lite is installed (without importing native modules)."""
+    try:
+        # Check for redislite/falkordb-client spec without loading it
+        return importlib.util.find_spec("redislite") is not None
+    except ImportError:
+        return False
+
+def _is_neo4j_configured() -> bool:
+    """Check if Neo4j is configured with credentials."""
+    return all([
+        os.getenv('NEO4J_URI'),
+        os.getenv('NEO4J_USERNAME'),
+        os.getenv('NEO4J_PASSWORD')
+    ])
+
+def get_database_manager() -> Union['DatabaseManager', 'FalkorDBManager']:
+    """
+    Factory function to get the appropriate database manager based on configuration.
+    
+    Selection logic:
+    1. Runtime Override: 'CGC_RUNTIME_DB_TYPE' (set via --database flag)
+    2. Configured Default: 'DEFAULT_DATABASE' (set via 'cgc default database')
+    3. Legacy Env Var: 'DATABASE_TYPE'
+    4. Implicit Default: FalkorDB (if available)
+    5. Fallback: Neo4j (if configured)
+    """
+    from codegraphcontext.utils.debug_log import info_logger
+    
+    # 1. Runtime Override (CLI flag) or Config/Env
+    db_type = os.getenv('CGC_RUNTIME_DB_TYPE')
+    if not db_type:
+        db_type = os.getenv('DEFAULT_DATABASE')
+    if not db_type:
+        db_type = os.getenv('DATABASE_TYPE')
+
+    if db_type:
+        db_type = db_type.lower()
+        if db_type == 'falkordb':
+            if not _is_falkordb_available():
+                 raise ValueError("Database set to 'falkordb' but FalkorDB Lite is not installed.\nRun 'pip install falkordblite'")
+            from .database_falkordb import FalkorDBManager
+            info_logger("Using FalkorDB Lite (explicit)")
+            return FalkorDBManager()
+            
+        elif db_type == 'neo4j':
+            if not _is_neo4j_configured():
+                 raise ValueError("Database set to 'neo4j' but it is not configured.\nRun 'cgc setup' to configure Neo4j.")
+            from .database import DatabaseManager
+            info_logger("Using Neo4j Server (explicit)")
+            return DatabaseManager()
+        else:
+            raise ValueError(f"Unknown database type: '{db_type}'. Use 'falkordb' or 'neo4j'.")
+
+    # 4. Implicit Default -> FalkorDB (Zero Config)
+    if _is_falkordb_available():
+        from .database_falkordb import FalkorDBManager
+        info_logger("Using FalkorDB Lite (default)")
+        return FalkorDBManager()
+        
+    # 5. Fallback if FalkorDB missing but Neo4j is ready
+    if _is_neo4j_configured():
+        from .database import DatabaseManager
+        info_logger("Using Neo4j Server (auto-detected)")
+        return DatabaseManager()
+
+    raise ValueError(
+        "No database backend available.\n"
+        "Recommended: Install FalkorDB Lite ('pip install falkordblite')\n"
+        "Alternative: Run 'cgc setup' to configure Neo4j."
+    )
+
+# For backward compatibility, export DatabaseManager
+from .database import DatabaseManager
+from .database_falkordb import FalkorDBManager
+
+__all__ = ['DatabaseManager', 'FalkorDBManager', 'get_database_manager']