codegraphcontext 0.2.12__py3-none-any.whl → 0.3.0__py3-none-any.whl

codegraphcontext/cli/cli_helpers.py

@@ -6,6 +6,15 @@ from pathlib import Path
 import time
 from rich.console import Console
 from rich.table import Table
+from rich.progress import (
+    Progress,
+    SpinnerColumn,
+    TextColumn,
+    BarColumn,
+    TaskProgressColumn,
+    TimeRemainingColumn,
+    MofNCompleteColumn,
+)
 
 from ..core import get_database_manager
 from ..core.jobs import JobManager
@@ -67,6 +76,64 @@ def _initialize_services():
     return db_manager, graph_builder, code_finder
 
 
+async def _run_index_with_progress(graph_builder: GraphBuilder, path_obj: Path, is_dependency: bool = False):
+    """Internal helper to run indexing with a Live progress bar."""
+    job_id = graph_builder.job_manager.create_job(str(path_obj), is_dependency=is_dependency)
+    
+    # Create the progress bar
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        BarColumn(),
+        TaskProgressColumn(),
+        MofNCompleteColumn(),
+        TimeRemainingColumn(),
+        TextColumn("[dim]{task.fields[filename]}"),
+        console=console,
+        transient=True,
+    ) as progress:
+        
+        task_id = progress.add_task(
+            "Indexing...", 
+            total=None,  # Will be updated once file discovery is done
+            filename=""
+        )
+
+        indexing_task = asyncio.create_task(
+            graph_builder.build_graph_from_path_async(path_obj, is_dependency=is_dependency, job_id=job_id)
+        )
+
+        from ..core.jobs import JobStatus
+        
+        # Poll for updates
+        while not indexing_task.done():
+            job = graph_builder.job_manager.get_job(job_id)
+            if job:
+                if job.total_files > 0:
+                    progress.update(task_id, total=job.total_files, completed=job.processed_files)
+                
+                # Update the current filename in the UI
+                current_file = job.current_file or ""
+                if len(current_file) > 40:
+                    current_file = "..." + current_file[-37:]
+                progress.update(task_id, filename=current_file)
+
+                if job.status in [JobStatus.COMPLETED, JobStatus.FAILED, JobStatus.CANCELLED]:
+                    break
+            
+            await asyncio.sleep(0.1)
+
+        # Wait for actual completion and handle final state
+        try:
+            await indexing_task
+            job = graph_builder.job_manager.get_job(job_id)
+            if job and job.status == JobStatus.FAILED:
+                error_msg = job.errors[0] if job.errors else "Unknown error"
+                raise RuntimeError(error_msg)
+        except Exception as e:
+            raise e
+
+
 def index_helper(path: str):
     """Synchronously indexes a repository."""
     time_start = time.time()
@@ -107,13 +174,9 @@ def index_helper(path: str):
             console.print(f"[yellow]Warning: Could not check file count: {e}. Proceeding with indexing...[/yellow]")
 
     console.print(f"Starting indexing for: {path_obj}")
-    console.print("[yellow]This may take a few minutes for large repositories...[/yellow]")
-
-    async def do_index():
-        await graph_builder.build_graph_from_path_async(path_obj, is_dependency=False)
 
     try:
-        asyncio.run(do_index())
+        asyncio.run(_run_index_with_progress(graph_builder, path_obj, is_dependency=False))
         time_end = time.time()
         elapsed = time_end - time_start
         console.print(f"[green]Successfully finished indexing: {path} in {elapsed:.2f} seconds[/green]")
@@ -159,13 +222,9 @@ def add_package_helper(package_name: str, language: str):
         return
 
     console.print(f"Starting indexing for package '{package_name}' at: {package_path}")
-    console.print("[yellow]This may take a few minutes...[/yellow]")
-
-    async def do_index():
-        await graph_builder.build_graph_from_path_async(package_path, is_dependency=True)
 
     try:
-        asyncio.run(do_index())
+        asyncio.run(_run_index_with_progress(graph_builder, package_path, is_dependency=True))
         console.print(f"[green]Successfully finished indexing package: {package_name}[/green]")
     except Exception as e:
         console.print(f"[bold red]An error occurred during package indexing:[/bold red] {e}")
@@ -587,13 +646,9 @@ def reindex_helper(path: str):
             return
     
     console.print(f"[cyan]Re-indexing: {path_obj}[/cyan]")
-    console.print("[yellow]This may take a few minutes for large repositories...[/yellow]")
-
-    async def do_index():
-        await graph_builder.build_graph_from_path_async(path_obj, is_dependency=False)
-
+    
     try:
-        asyncio.run(do_index())
+        asyncio.run(_run_index_with_progress(graph_builder, path_obj, is_dependency=False))
         time_end = time.time()
         elapsed = time_end - time_start
         console.print(f"[green]Successfully re-indexed: {path} in {elapsed:.2f} seconds[/green]")

codegraphcontext/core/__init__.py

@@ -125,7 +125,14 @@ def get_database_manager() -> Union['DatabaseManager', 'FalkorDBManager', 'Falko
         else:
             raise ValueError(f"Unknown database type: '{db_type}'. Use 'kuzudb', 'falkordb', 'falkordb-remote', or 'neo4j'.")
 
-    # 4. Implicit Default -> FalkorDB Lite (Unix Zero Config)
+    # 4. Auto-detect: Remote FalkorDB (if FALKORDB_HOST is set)
+    # This takes priority over zero-config local backends because it's an explicit signal
+    if _is_falkordb_remote_configured():
+        from .database_falkordb_remote import FalkorDBRemoteManager
+        info_logger("Using remote FalkorDB (auto-detected via FALKORDB_HOST)")
+        return FalkorDBRemoteManager()
+
+    # 5. Implicit Default -> FalkorDB Lite (Unix Zero Config)
     if _is_falkordb_available():
         from .database_falkordb import FalkorDBManager, FalkorDBUnavailableError
         try:
@@ -139,18 +146,12 @@ def get_database_manager() -> Union['DatabaseManager', 'FalkorDBManager', 'Falko
             )
             # fall through to KùzuDB below
 
-    # 5. Implicit Default -> KùzuDB (Best Zero Config)
+    # 6. Implicit Default -> KùzuDB (Best Zero Config)
     if _is_kuzudb_available():
         from .database_kuzu import KuzuDBManager
         info_logger("Using KùzuDB (default)")
         return KuzuDBManager()
 
-    # 6. Auto-detect: Remote FalkorDB (if FALKORDB_HOST is set)
-    if _is_falkordb_remote_configured():
-        from .database_falkordb_remote import FalkorDBRemoteManager
-        info_logger("Using remote FalkorDB (auto-detected via FALKORDB_HOST)")
-        return FalkorDBRemoteManager()
-
     # 7. Fallback if configured
     if _is_neo4j_configured():
         from .database import DatabaseManager

{codegraphcontext-0.2.12.dist-info → codegraphcontext-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.2.12
+Version: 0.3.0
 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
@@ -147,7 +147,7 @@ A powerful **MCP server** and **CLI toolkit** that indexes local code into a gra
 ---
 
 ## Project Details
-- **Version:** 0.2.12
+- **Version:** 0.3.0
 - **Authors:** Shashank Shekhar Singh <shashankshekharsingh1205@gmail.com>
 - **License:** MIT License (See [LICENSE](LICENSE) for details)
 - **Website:** [CodeGraphContext](http://codegraphcontext.vercel.app/)

{codegraphcontext-0.2.12.dist-info → codegraphcontext-0.3.0.dist-info}/RECORD

@@ -4,14 +4,14 @@ codegraphcontext/prompts.py,sha256=E5P55paM0oHfBcNVfxkxpXRGZnya1kr_mKDg9i49FwM,6
 codegraphcontext/server.py,sha256=gcb6V4x0Oh8haBOCm2WBjTCO9FDukrSalAMMApL-oc8,12806
 codegraphcontext/tool_definitions.py,sha256=_0ahezQSURX_ewWydT2sFcvC6OFGdMoPA7KwgWNVcks,11482
 codegraphcontext/cli/__init__.py,sha256=v6CMDVKM5d_sXn3S5nZNf0phXn0IdrnhLazUoen9k9w,38
-codegraphcontext/cli/cli_helpers.py,sha256=e3WLdCdP63cbSji1RJZC9pwZt0mDgzM2ibeZqsdieJM,33996
+codegraphcontext/cli/cli_helpers.py,sha256=2T8eP1RmM_p2Z1XfZDv44wx3Hf01jF_b3KGoWVBG0lk,35945
 codegraphcontext/cli/config_manager.py,sha256=MK7GMGZ4hd8-ZOShXBd_KDtNqHQ3ngdbef5KM_naPrQ,15899
 codegraphcontext/cli/main.py,sha256=PSJ5BxQhbdNdCWWyYvW9-6COCiw852iVi-zhP43nYPA,86328
 codegraphcontext/cli/registry_commands.py,sha256=30rJm4SeS0n1jax4JVuhuv4zYLzyMmlHcCSnsDDhgc8,19964
 codegraphcontext/cli/setup_macos.py,sha256=Xjlv_9jk9qv8Gh7stpH1pvlalzC0Fg176y7jc5G1zh0,3575
 codegraphcontext/cli/setup_wizard.py,sha256=yVXKqvAtUM0UC4tUreiS6C5utJbBO5Et2MiPUPWit9s,41393
 codegraphcontext/cli/visualizer.py,sha256=rZOLPx44wlaxVXGLR6uOaiL2GaQvxEnpyDQ4tjRDVC8,45750
-codegraphcontext/core/__init__.py,sha256=tDXBV2Tr2Rt6uYYHZH2PsOTsvePTirzijbxVhkIxj6g,7188
+codegraphcontext/core/__init__.py,sha256=r-dC2b8hn5e4ZBVfWnCcBYMTQVcayxG5-eA5Do6uSM4,7278
 codegraphcontext/core/bundle_registry.py,sha256=kLvTLEl99QwfyAw8a7-4hqconLCtS64G9ECPVdfVBsM,7476
 codegraphcontext/core/cgc_bundle.py,sha256=YHL_uamATXYVweh5qbg9TQjT1lKlt69gXyF-xKKk67c,31353
 codegraphcontext/core/database.py,sha256=2UL8AyE6vX5mVHg1zcD74TYEREmddhKzN7Bjy2THQwM,11435
@@ -71,9 +71,9 @@ codegraphcontext/tools/query_tool_languages/typescript_toolkit.py,sha256=3S4hpmO
 codegraphcontext/utils/debug_log.py,sha256=Qg7jwyeg7x2h3Ur_2S34bdMCkHdlk_ngHfPwa97A9vE,2836
 codegraphcontext/utils/tree_sitter_manager.py,sha256=bIuKYN1aj1Zi6BnksGjZSLZzgBwuFRselZzyUpbToAU,9180
 codegraphcontext/utils/visualize_graph.py,sha256=Ntq8l8SvAOvsOp19QKByjEwU7-5rPa_XfcGLOrUehq4,5003
-codegraphcontext-0.2.12.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-codegraphcontext-0.2.12.dist-info/METADATA,sha256=EgsTEDqJ5yLjKXbLI4GQ2gTgI3xOEP2s3Qi5TGbeIVg,21095
-codegraphcontext-0.2.12.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codegraphcontext-0.2.12.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
-codegraphcontext-0.2.12.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
-codegraphcontext-0.2.12.dist-info/RECORD,,
+codegraphcontext-0.3.0.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+codegraphcontext-0.3.0.dist-info/METADATA,sha256=FH-FTUU-gexxDfZePS6rqliF6540hUmv3ARwXJ_qO-I,21093
+codegraphcontext-0.3.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+codegraphcontext-0.3.0.dist-info/entry_points.txt,sha256=LCxWCWMshdvYGoHBPuQZ8C-e4CiNSHCLXofrNSGHkoE,103
+codegraphcontext-0.3.0.dist-info/top_level.txt,sha256=CBgc6LAPZIO5FS0nSYYkylDifHsZTIqw3Gf5UwDxeGI,17
+codegraphcontext-0.3.0.dist-info/RECORD,,