codegraphcontext 0.1.41__tar.gz → 0.2.1__tar.gz

{codegraphcontext-0.1.41/src/codegraphcontext.egg-info → codegraphcontext-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.41
+Version: 0.2.1
 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
@@ -113,6 +113,7 @@ Dynamic: license-file
   </a>
 </p>
 
+
 A powerful **MCP server** and **CLI toolkit** that indexes local code into a graph database to provide context to AI assistants and developers. Use it as a standalone CLI for comprehensive code analysis or connect it to your favorite AI IDE via MCP for AI-powered code understanding.
 
 ---
@@ -128,6 +129,12 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 
 ## ✨ Experience CGC
 
+
+### 👨🏻‍💻 Installation and CLI
+> Install in seconds with pip and unlock a powerful CLI for code graph analysis.
+![Install and unlock the CLI instantly](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/install&cli.gif)
+
+
 ### 🛠️ Indexing in Seconds
 > The CLI intelligently parses your tree-sitter nodes to build the graph.
 ![Indexing using an MCP client](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/Indexing.gif)
@@ -139,7 +146,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.1.41
+- **Version:** 0.2.1
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -418,8 +425,6 @@ Add the following server configuration to your client's settings file (e.g., VS
         "NEO4J_URI": "YOUR_NEO4J_URI",
         "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
         "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
-      },
-        "disabled": false
       },
       "disabled": false,
       "alwaysAllow": []

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/README.md

@@ -51,6 +51,7 @@
   </a>
 </p>
 
+
 A powerful **MCP server** and **CLI toolkit** that indexes local code into a graph database to provide context to AI assistants and developers. Use it as a standalone CLI for comprehensive code analysis or connect it to your favorite AI IDE via MCP for AI-powered code understanding.
 
 ---
@@ -66,6 +67,12 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 
 ## ✨ Experience CGC
 
+
+### 👨🏻‍💻 Installation and CLI
+> Install in seconds with pip and unlock a powerful CLI for code graph analysis.
+![Install and unlock the CLI instantly](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/install&cli.gif)
+
+
 ### 🛠️ Indexing in Seconds
 > The CLI intelligently parses your tree-sitter nodes to build the graph.
 ![Indexing using an MCP client](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/Indexing.gif)
@@ -77,7 +84,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.1.41
+- **Version:** 0.2.1
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -356,8 +363,6 @@ Add the following server configuration to your client's settings file (e.g., VS
         "NEO4J_URI": "YOUR_NEO4J_URI",
         "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
         "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
-      },
-        "disabled": false
       },
       "disabled": false,
       "alwaysAllow": []

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.1.41"
+version = "0.2.1"
 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"

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/cli/config_manager.py

@@ -38,7 +38,7 @@ DEFAULT_CONFIG = {
     "PARALLEL_WORKERS": "4",
     "CACHE_ENABLED": "true",
     "IGNORE_DIRS": "node_modules,venv,.venv,env,.env,dist,build,target,out,.git,.idea,.vscode,__pycache__",
-    "INDEX_SOURCE": "false",
+    "INDEX_SOURCE": "true",
 }
 
 # Configuration key descriptions
@@ -61,7 +61,7 @@ CONFIG_DESCRIPTIONS = {
     "PARALLEL_WORKERS": "Number of parallel indexing workers",
     "CACHE_ENABLED": "Enable caching for faster re-indexing",
     "IGNORE_DIRS": "Comma-separated list of directory names to ignore during indexing",
-    "INDEX_SOURCE": "Store full source code in graph database (recommended false)",
+    "INDEX_SOURCE": "Store full source code in graph database (for faster indexing use false, for better performance use true)",
 }
 
 # Valid values for each config key

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/cli/main.py

@@ -1332,7 +1332,7 @@ def find_by_content_search(
             results = code_finder.find_by_content(query)
         except Exception as e:
             error_msg = str(e).lower()
-            if 'fulltext' in error_msg or 'db.index.fulltext' in error_msg:
+            if ('fulltext' in error_msg or 'db.index.fulltext' in error_msg) and "Falkor" in db_manager.__class__.__name__:
                 console.print("\n[bold red]❌ Full-text search is not supported on FalkorDB[/bold red]\n")
                 console.print("[yellow]💡 You have two options:[/yellow]\n")
                 console.print("  1. [cyan]Switch to Neo4j:[/cyan]")

codegraphcontext-0.2.1/src/codegraphcontext/core/bundle_registry.py

@@ -0,0 +1,171 @@
+import requests
+from pathlib import Path
+from typing import Optional, List, Dict, Any, Tuple
+import logging
+
+logger = logging.getLogger(__name__)
+
+GITHUB_ORG = "CodeGraphContext"
+GITHUB_REPO = "CodeGraphContext"
+REGISTRY_API_URL = f"https://api.github.com/repos/{GITHUB_ORG}/{GITHUB_REPO}/releases"
+MANIFEST_URL = f"https://github.com/{GITHUB_ORG}/{GITHUB_REPO}/releases/download/on-demand-bundles/manifest.json"
+
+class BundleRegistry:
+    """
+    Core logic for interacting with the CodeGraphContext bundle registry.
+    Handles fetching metadata, searching, and downloading bundles without CLI dependencies.
+    """
+
+    @staticmethod
+    def fetch_available_bundles() -> List[Dict[str, Any]]:
+        """
+        Fetch all available bundles from GitHub Releases and the on-demand manifest.
+        Returns a list of bundle dictionaries with metadata.
+        Preserves all versions - no deduplication.
+        """
+        all_bundles = []
+        
+        # 1. Fetch on-demand bundles from manifest
+        try:
+            response = requests.get(MANIFEST_URL, timeout=10)
+            if response.status_code == 200:
+                manifest = response.json()
+                if manifest.get('bundles'):
+                    for bundle in manifest['bundles']:
+                        bundle['source'] = 'on-demand'
+                        # Ensure bundle has a full_name field (with version info)
+                        if 'bundle_name' in bundle:
+                            # Extract full name without .cgc extension
+                            bundle['full_name'] = bundle['bundle_name'].replace('.cgc', '')
+                        all_bundles.append(bundle)
+        except Exception as e:
+            logger.warning(f"Could not fetch on-demand bundles from manifest: {e}")
+        
+        # 2. Fetch weekly pre-indexed bundles
+        try:
+            response = requests.get(REGISTRY_API_URL, timeout=10)
+            if response.status_code == 200:
+                releases = response.json()
+                
+                # Find weekly releases (bundles-YYYYMMDD pattern)
+                weekly_releases = [r for r in releases if r['tag_name'].startswith('bundles-') and r['tag_name'] != 'bundles-latest']
+                
+                if weekly_releases:
+                    # Get the most recent weekly release
+                    latest_weekly = weekly_releases[0]
+                    
+                    for asset in latest_weekly.get('assets', []):
+                        if asset['name'].endswith('.cgc'):
+                            # Full bundle name without extension
+                            full_name = asset['name'].replace('.cgc', '')
+                            
+                            # Parse bundle name
+                            name_parts = full_name.split('-')
+                            bundle = {
+                                'name': name_parts[0],  # Base package name
+                                'full_name': full_name,  # Complete name with version
+                                'repo': f"{name_parts[0]}/{name_parts[0]}",  # Simplified
+                                'bundle_name': asset['name'],
+                                'version': name_parts[1] if len(name_parts) > 1 else 'latest',
+                                'commit': name_parts[2] if len(name_parts) > 2 else 'unknown',
+                                'size_bytes': asset.get('size', 0),
+                                'size': f"{asset['size'] / 1024 / 1024:.1f}MB",
+                                'download_url': asset['browser_download_url'],
+                                'generated_at': asset['updated_at'],
+                                'source': 'weekly'
+                            }
+                            all_bundles.append(bundle)
+        except Exception as e:
+            logger.warning(f"Could not fetch weekly bundles from GitHub API: {e}")
+        
+        # Normalize all bundles to have required fields
+        for bundle in all_bundles:
+            # Ensure 'name' field exists (base package name)
+            if 'name' not in bundle:
+                repo = bundle.get('repo', '')
+                if '/' in repo:
+                    bundle['name'] = repo.split('/')[-1]
+                else:
+                    # Extract from full_name or bundle_name
+                    full_name = bundle.get('full_name', bundle.get('bundle_name', 'unknown'))
+                    bundle['name'] = full_name.split('-')[0]
+            
+            # Ensure 'full_name' exists
+            if 'full_name' not in bundle:
+                bundle['full_name'] = bundle.get('bundle_name', bundle.get('name', 'unknown')).replace('.cgc', '')
+        
+        return all_bundles
+
+    @staticmethod
+    def find_bundle_download_info(name: str) -> Tuple[Optional[str], Optional[Dict[str, Any]], str]:
+        """
+        Find a download URL and metadata for a bundle by name.
+        
+        Strategies:
+        1. Exact match on full_name (e.g., 'flask-main-abc123')
+        2. Match on base name (e.g., 'flask') - returns most recent version
+        
+        Returns:
+            (download_url, bundle_metadata, error_message)
+        """
+        bundles = BundleRegistry.fetch_available_bundles()
+        
+        if not bundles:
+            return None, None, "Could not fetch bundle registry."
+        
+        name_lower = name.lower()
+        
+        # Strategy 1: Exact match on full_name
+        for b in bundles:
+            if b.get('full_name', '').lower() == name_lower:
+                url = b.get('download_url')
+                if url:
+                    return url, b, ""
+                return None, b, f"No download URL found for bundle '{name}'"
+        
+        # Strategy 2: Match base package name (most recent)
+        matching_bundles = []
+        for b in bundles:
+            if b.get('name', '').lower() == name_lower:
+                matching_bundles.append(b)
+        
+        if matching_bundles:
+            # Sort by timestamp (newest first)
+            matching_bundles.sort(key=lambda x: x.get('generated_at', ''), reverse=True)
+            bundle = matching_bundles[0]
+            url = bundle.get('download_url')
+            if url:
+                return url, bundle, ""
+            return None, bundle, f"No download URL found for bundle '{name}'"
+            
+        return None, None, f"Bundle '{name}' not found in registry."
+
+    @staticmethod
+    def download_file(url: str, output_path: Path, progress_callback=None) -> bool:
+        """
+        Download a file from a URL to a local path.
+        
+        Args:
+            url: The URL to download from
+            output_path: Local path to save the file
+            progress_callback: Optional callable(chunk_size) to report progress
+            
+        Returns:
+            True if successful, raises exception otherwise
+        """
+        try:
+            response = requests.get(url, stream=True, timeout=30)
+            response.raise_for_status()
+            
+            with open(output_path, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    if chunk:
+                        f.write(chunk)
+                        if progress_callback:
+                            progress_callback(len(chunk))
+            return True
+        except Exception as e:
+            # Clean up partial file
+            if output_path.exists():
+                output_path.unlink()
+            raise e

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/core/cgc_bundle.py

@@ -179,10 +179,21 @@ class CGCBundle:
                 info_logger(f"Loading bundle: {metadata.get('repo', 'unknown')}")
                 info_logger(f"Bundle version: {metadata.get('cgc_version', 'unknown')}")
                 
-                # Step 4: Clear existing data if requested
+                # Step 4: Handle existing data
+                repo_name = metadata.get('repo', 'unknown')
+                repo_path = metadata.get('repo_path')
+                
                 if clear_existing:
-                    info_logger("Clearing existing graph data...")
+                    # User explicitly wants to clear - remove everything
+                    info_logger("Clearing all existing graph data...")
                     self._clear_graph()
+                else:
+                    # Check if this repository already exists (only when NOT clearing)
+                    existing_repo = self._check_existing_repository(repo_name, repo_path)
+                    
+                    if existing_repo:
+                        return False, f"Repository '{repo_name}' already exists in the database. Use clear_existing=True to replace it."
+                
                 
                 # Step 5: Create schema
                 info_logger("Creating schema...")
@@ -584,6 +595,63 @@ cgc import <bundle-file>.cgc
         
         return True, "Valid bundle"
     
+    def _check_existing_repository(self, repo_name: str, repo_path: Optional[str]) -> bool:
+        """Check if a repository already exists in the database."""
+        with self.db_manager.get_driver().session() as session:
+            # Try to find by name first
+            result = session.run(
+                "MATCH (r:Repository {name: $name}) RETURN r LIMIT 1",
+                name=repo_name
+            )
+            if result.single():
+                return True
+            
+            # If repo_path is provided, also check by path
+            if repo_path:
+                result = session.run(
+                    "MATCH (r:Repository {path: $path}) RETURN r LIMIT 1",
+                    path=repo_path
+                )
+                if result.single():
+                    return True
+        
+        return False
+    
+    def _delete_repository(self, repo_identifier: str):
+        """Delete a specific repository and all its related nodes from the graph."""
+        with self.db_manager.get_driver().session() as session:
+            # First, try to find the repository by name or path
+            result = session.run("""
+                MATCH (r:Repository)
+                WHERE r.name = $identifier OR r.path = $identifier
+                RETURN r.path as path
+                LIMIT 1
+            """, identifier=repo_identifier)
+            
+            record = result.single()
+            if not record:
+                warning_logger(f"Repository '{repo_identifier}' not found for deletion")
+                return
+            
+            repo_path = record['path']
+            
+            # Delete all nodes that belong to this repository
+            # Files, Functions, Classes, Modules all have paths that start with repo_path
+            session.run("""
+                MATCH (n)
+                WHERE n.path STARTS WITH $repo_path
+                DETACH DELETE n
+            """, repo_path=repo_path)
+            
+            # Delete the repository node itself
+            session.run("""
+                MATCH (r:Repository)
+                WHERE r.path = $repo_path
+                DELETE r
+            """, repo_path=repo_path)
+            
+            info_logger(f"Deleted repository: {repo_identifier}")
+    
     def _clear_graph(self):
         """Clear all nodes and relationships from the graph."""
         with self.db_manager.get_driver().session() as session:

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/core/database.py

@@ -136,7 +136,8 @@ class DatabaseManager:
             Tuple[bool, Optional[str]]: (is_valid, error_message)
         """
         # Validate URI format
-        uri_pattern = r'^(neo4j|neo4j\+s|neo4j\+ssc|bolt|bolt\+s|bolt\+ssc)://[^:]+:\d+$'
+        # Modified regex to make port optional "(:\\d+)?"
+        uri_pattern = r'^(neo4j|neo4j\+s|neo4j\+ssc|bolt|bolt\+s|bolt\+ssc)://[^:]+(:\d+)?$'
         if not re.match(uri_pattern, uri):
             return False, (
                 "Invalid Neo4j URI format.\n"
@@ -177,8 +178,12 @@ class DatabaseManager:
             try:
                 # Extract host and port from URI
                 host_port = uri.split('://')[1]
-                host = host_port.split(':')[0]
-                port = int(host_port.split(':')[1])
+                if ':' in host_port:
+                    host = host_port.split(':')[0]
+                    port = int(host_port.split(':')[1])
+                else:
+                    host = host_port
+                    port = 7687 # Default Neo4j port
                 
                 # Test socket connection
                 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/prompts.py

@@ -76,6 +76,7 @@ You are an expert AI pair programmer. Your primary goal is to help a developer u
     * `cyclomatic_complexity` (int)
     * `decorators` (list)
     * `lang` (string)
+    * `source` (string, the full source code of the function)
     * `is_dependency` (boolean)
 * **`Class`**
     * `name` (string)
@@ -85,6 +86,7 @@ You are an expert AI pair programmer. Your primary goal is to help a developer u
     * `bases` (list)
     * `decorators` (list)
     * `lang` (string)
+    * `source` (string, the full source code of the class)
     * `is_dependency` (boolean)
 
 ### Relationships

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/tool_definitions.py

@@ -59,7 +59,7 @@ TOOLS = {
     },
     "execute_cypher_query": {
         "name": "execute_cypher_query",
-        "description": "Fallback tool to run a direct, read-only Cypher query against the code graph. Use this for complex questions not covered by other tools. The graph contains nodes representing code structures and relationships between them. **Schema Overview:**\n- **Nodes:** `Repository`, `File`, `Module`, `Class`, `Function`.\n- **Properties:** Nodes have properties like `name`, `path`, `cyclomatic_complexity` (on Function nodes), and `code`.\n- **Relationships:** `CONTAINS` (e.g., File-[:CONTAINS]->Function), `CALLS` (Function-[:CALLS]->Function or File-[:CALLS]->Function), `IMPORTS` (File-[:IMPORTS]->Module), `INHERITS` (Class-[:INHERITS]->Class).",
+        "description": "Fallback tool to run a direct, read-only Cypher query against the code graph. Use this for complex questions not covered by other tools. The graph contains nodes representing code structures and relationships between them. **Schema Overview:**\n- **Nodes:** `Repository`, `File`, `Module`, `Class`, `Function`.\n- **Properties:** Nodes have properties like `name`, `path`, `cyclomatic_complexity` (on Function nodes), and `source`.\n- **Relationships:** `CONTAINS` (e.g., File-[:CONTAINS]->Function), `CALLS` (Function-[:CALLS]->Function or File-[:CALLS]->Function), `IMPORTS` (File-[:IMPORTS]->Module), `INHERITS` (Class-[:INHERITS]->Class).",
         "inputSchema": {
             "type": "object",
             "properties": { "cypher_query": {"type": "string", "description": "The read-only Cypher query to execute."} },

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/tools/code_finder.py

@@ -84,13 +84,14 @@ class CodeFinder:
                 CALL db.index.fulltext.queryNodes("code_search_index", $search_term) YIELD node, score
                 WITH node, score
                 WHERE node:Function OR node:Class OR node:Variable
+                MATCH (node)<-[:CONTAINS]-(f:File)
                 RETURN
                     CASE 
                         WHEN node:Function THEN 'function'
                         WHEN node:Class THEN 'class'
                         ELSE 'variable' 
                     END as type,
-                    node.name as name, node.path as path,
+                    node.name as name, f.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 score DESC

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/tools/graph_builder.py

@@ -277,7 +277,12 @@ class GraphBuilder:
             """, path=file_path_str, name=file_name, relative_path=relative_path, is_dependency=is_dependency)
 
             file_path_obj = Path(file_path_str)
-            repo_path_obj = Path(repo_result['path'])
+            if repo_result:
+                repo_path_obj = Path(repo_result['path'])
+            else:
+                # Fallback to the path we queried for
+                warning_logger(f"Repository node not found for {file_data['repo_path']} during indexing of {file_name}. Using original path.")
+                repo_path_obj = Path(file_data['repo_path']).resolve()
             
             relative_path_to_file = file_path_obj.relative_to(repo_path_obj)
             

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext/tools/handlers/management_handlers.py

@@ -116,7 +116,8 @@ def list_jobs(job_manager: JobManager) -> Dict[str, Any]:
 def load_bundle(code_finder: CodeFinder, **args) -> Dict[str, Any]:
     """Tool to load a .cgc bundle into the database."""
     from pathlib import Path
-    from ...cli.registry_commands import load_bundle_command
+    from ...core.bundle_registry import BundleRegistry
+    from ...core.cgc_bundle import CGCBundle
     
     bundle_name = args.get("bundle_name")
     clear_existing = args.get("clear_existing", False)
@@ -127,22 +128,60 @@ def load_bundle(code_finder: CodeFinder, **args) -> Dict[str, Any]:
     try:
         debug_log(f"Loading bundle: {bundle_name}")
         
-        # Use the existing load_bundle_command from CLI
-        # This handles both local files and auto-download from registry
-        success, message, stats = load_bundle_command(
-            bundle_name=bundle_name,
+        # Check if bundle exists locally
+        bundle_path = Path(bundle_name)
+        
+        # If it doesn't exist as-is, try with .cgc extension
+        if not bundle_path.exists() and not str(bundle_name).endswith('.cgc'):
+            bundle_path = Path(f"{bundle_name}.cgc")
+        
+        if not bundle_path.exists():
+            # Try to download from registry
+            debug_log(f"Bundle {bundle_name} not found locally, checking registry...")
+            download_url, bundle_meta, error = BundleRegistry.find_bundle_download_info(bundle_name)
+            
+            if not download_url:
+                return {"error": f"Bundle not found locally or in registry: {bundle_name}. {error}"}
+            
+            # Determine output filename from metadata
+            filename = bundle_meta.get('bundle_name', f"{bundle_name}.cgc")
+            # Save to current working directory
+            target_path = Path.cwd() / filename
+            
+            debug_log(f"Downloading bundle to {target_path}...")
+            try:
+                BundleRegistry.download_file(download_url, target_path)
+                bundle_path = target_path
+                debug_log(f"Successfully downloaded to {bundle_path}")
+            except Exception as e:
+                return {"error": f"Failed to download bundle: {str(e)}"}
+            
+            # Verify the downloaded file exists
+            if not bundle_path.exists():
+                return {"error": f"Download completed but file not found at {bundle_path}"}
+
+        # Load the bundle using CGCBundle core class
+        bundle = CGCBundle(code_finder.db_manager)
+        success, message = bundle.import_from_bundle(
+            bundle_path=bundle_path,
             clear_existing=clear_existing
         )
         
         if success:
+            stats = {}
+            # Parse simple stats from message if possible, or just return success
+            if "Nodes:" in message:
+                 # Best effort parsing, not critical
+                 pass
+                 
             return {
                 "success": True,
                 "message": message,
                 "stats": stats
             }
         else:
-            return {"error": message}
-    
+             return {"error": message}
+
     except Exception as e:
         debug_log(f"Error loading bundle: {str(e)}")
         return {"error": f"Failed to load bundle: {str(e)}"}
@@ -150,7 +189,7 @@ def load_bundle(code_finder: CodeFinder, **args) -> Dict[str, Any]:
 
 def search_registry_bundles(code_finder: CodeFinder, **args) -> Dict[str, Any]:
     """Tool to search for bundles in the registry."""
-    from ...cli.registry_commands import fetch_available_bundles
+    from ...core.bundle_registry import BundleRegistry
     
     query = args.get("query", "").lower()
     unique_only = args.get("unique_only", False)
@@ -158,8 +197,8 @@ def search_registry_bundles(code_finder: CodeFinder, **args) -> Dict[str, Any]:
     try:
         debug_log(f"Searching registry for: {query}")
         
-        # Fetch all bundles from registry
-        bundles = fetch_available_bundles()
+        # Fetch directly from core registry
+        bundles = BundleRegistry.fetch_available_bundles()
         
         if not bundles:
             return {

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.1.41
+Version: 0.2.1
 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
@@ -113,6 +113,7 @@ Dynamic: license-file
   </a>
 </p>
 
+
 A powerful **MCP server** and **CLI toolkit** that indexes local code into a graph database to provide context to AI assistants and developers. Use it as a standalone CLI for comprehensive code analysis or connect it to your favorite AI IDE via MCP for AI-powered code understanding.
 
 ---
@@ -128,6 +129,12 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 
 ## ✨ Experience CGC
 
+
+### 👨🏻‍💻 Installation and CLI
+> Install in seconds with pip and unlock a powerful CLI for code graph analysis.
+![Install and unlock the CLI instantly](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/install&cli.gif)
+
+
 ### 🛠️ Indexing in Seconds
 > The CLI intelligently parses your tree-sitter nodes to build the graph.
 ![Indexing using an MCP client](https://github.com/CodeGraphContext/CodeGraphContext/blob/main/images/Indexing.gif)
@@ -139,7 +146,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.1.41
+- **Version:** 0.2.1
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)
@@ -418,8 +425,6 @@ Add the following server configuration to your client's settings file (e.g., VS
         "NEO4J_URI": "YOUR_NEO4J_URI",
         "NEO4J_USERNAME": "YOUR_NEO4J_USERNAME",
         "NEO4J_PASSWORD": "YOUR_NEO4J_PASSWORD"
-      },
-        "disabled": false
       },
       "disabled": false,
       "alwaysAllow": []

{codegraphcontext-0.1.41 → codegraphcontext-0.2.1}/src/codegraphcontext.egg-info/SOURCES.txt

@@ -22,6 +22,7 @@ src/codegraphcontext/cli/setup_macos.py
 src/codegraphcontext/cli/setup_wizard.py
 src/codegraphcontext/cli/visualizer.py
 src/codegraphcontext/core/__init__.py
+src/codegraphcontext/core/bundle_registry.py
 src/codegraphcontext/core/cgc_bundle.py
 src/codegraphcontext/core/database.py
 src/codegraphcontext/core/database_falkordb.py