ctxgraph-code 0.2.0__tar.gz → 0.2.1__tar.gz

{ctxgraph_code-0.2.0 → ctxgraph_code-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ctxgraph-code
-Version: 0.2.0
+Version: 0.2.1
 Summary: Code knowledge graph for Claude Code. Build a relationship graph of your Python codebase and query it during coding sessions.
 Author: ctxgraph-code contributors
 License: MIT

{ctxgraph_code-0.2.0 → ctxgraph_code-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ctxgraph-code"
-version = "0.2.0"
+version = "0.2.1"
 description = "Code knowledge graph for Claude Code. Build a relationship graph of your Python codebase and query it during coding sessions."
 readme = "README.md"
 license = {text = "MIT"}

{ctxgraph_code-0.2.0 → ctxgraph_code-0.2.1}/src/ctxgraph_code/cli.py

@@ -261,6 +261,76 @@ def _build_single_graph(path, exts, user_patterns, db_path, label):
     return stats
 
 
+def _build_dir_worker(path, exts, user_patterns, db_path, label):
+    """Silent build worker for parallel execution. No console output."""
+    stats = build_graph(
+        path,
+        db_path=db_path,
+        exclude_patterns=user_patterns,
+        extensions=exts,
+    )
+    return label, stats
+
+
+def _build_dirs_parallel(path, exts, user_patterns, top_dirs, graphs_dir, jobs):
+    """Build per-directory graphs in parallel using a thread pool."""
+    from concurrent.futures import ThreadPoolExecutor, as_completed
+
+    n_workers = max(1, jobs) if jobs > 0 else os.cpu_count() or 1
+    console.print(f"[dim]Building {len(top_dirs)} graphs with {n_workers} workers...[/dim]")
+
+    futures = []
+    with ThreadPoolExecutor(max_workers=n_workers) as pool:
+        for d in top_dirs:
+            db_path = graphs_dir / f"{d.name}.db"
+            fut = pool.submit(_build_dir_worker, path, exts, user_patterns, db_path, d.name)
+            futures.append(fut)
+
+        results = []
+        for f in as_completed(futures):
+            try:
+                label, stats = f.result()
+                results.append((label, stats))
+            except Exception as e:
+                results.append(("error", str(e)))
+
+    results.sort(key=lambda x: x[0] if isinstance(x[0], str) else "")
+
+    summary = Table(title="Build Summary")
+    summary.add_column("Directory", style="cyan")
+    summary.add_column("Files", style="green")
+    summary.add_column("Nodes", style="blue")
+    summary.add_column("Edges", style="yellow")
+    summary.add_column("Time", style="magenta")
+
+    total_files = 0
+    total_nodes = 0
+    total_edges = 0
+    total_time = 0.0
+
+    for label, stats in results:
+        if isinstance(stats, str):
+            summary.add_row(f"{label}/", "[red]ERROR[/red]", stats, "", "")
+        else:
+            files = stats.get("files_analyzed", 0)
+            nodes = stats.get("total_nodes", 0)
+            edges = stats.get("total_edges", 0)
+            t = stats.get("elapsed_seconds", 0)
+            summary.add_row(f"{label}/", str(files), str(nodes), str(edges), f"{t}s")
+            total_files += files
+            total_nodes += nodes
+            total_edges += edges
+            total_time = max(total_time, t)
+
+    console.print(summary)
+    console.print(
+        f"\n[green]Built {len(top_dirs)} graphs: "
+        f"{total_files} files, {total_nodes} nodes, {total_edges} edges "
+        f"in {total_time}s[/green]"
+    )
+    return results
+
+
 @app.command()
 def build(
     repo_path: Optional[str] = typer.Argument(
@@ -275,10 +345,13 @@ def build(
     all_graph: bool = typer.Option(
         False, "--all", "-a", help="Build a single combined graph instead of per-directory"
     ),
+    jobs: int = typer.Option(
+        0, "--jobs", "-j", help="Number of parallel workers (0 = auto, default: CPU count)"
+    ),
 ):
     """Build the knowledge graph from source files.
 
-    Default: builds a separate graph per top-level directory.
+    Default: builds a separate graph per top-level directory in parallel.
     Use --all to build one combined graph instead.
     """
     path = Path(repo_path).resolve() if repo_path else Path.cwd()
@@ -299,7 +372,6 @@ def build(
         db_path = ctx_dir / "graph.db"
         _build_single_graph(path, exts, user_patterns, db_path, "combined")
     else:
-        # Per-directory: find top-level dirs (skip excluded ones)
         from ctxgraph_code.exclude.patterns import should_exclude
         top_dirs = sorted([
             d for d in path.iterdir()
@@ -314,28 +386,7 @@ def build(
         else:
             graphs_dir = ctx_dir / "graphs"
             graphs_dir.mkdir(parents=True, exist_ok=True)
-
-            total_nodes = 0
-            total_edges = 0
-            for d in top_dirs:
-                db_path = graphs_dir / f"{d.name}.db"
-                s = _build_single_graph(path, exts, user_patterns, db_path, d.name)
-                total_nodes += s.get("total_nodes", 0)
-                total_edges += s.get("total_edges", 0)
-
-            summary = Table(title="Build Summary")
-            summary.add_column("Directory", style="cyan")
-            for d in top_dirs:
-                db = graphs_dir / f"{d.name}.db"
-                if db.exists():
-                    from ctxgraph_code.graph.storage import Storage
-                    st = Storage(db)
-                    st.connect()
-                    nd = st.stats()["nodes"]
-                    ed = st.stats()["edges"]
-                    summary.add_row(f"{d.name}/: {nd} nodes, {ed} edges")
-            console.print(summary)
-            console.print(f"\n[green]Built {len(top_dirs)} graphs in {graphs_dir}[/green]")
+            _build_dirs_parallel(path, exts, user_patterns, top_dirs, graphs_dir, jobs)
 
 
 @app.command()
@@ -490,6 +541,9 @@ def setup(
         False, "--project-slash",
         help="Install slash command in project .claude/ instead of globally",
     ),
+    jobs: int = typer.Option(
+        0, "--jobs", "-j", help="Number of parallel workers (0 = auto, default: CPU count)"
+    ),
 ):
     """Initialize config, build the graph, and install the Claude Code slash command.
 
@@ -529,7 +583,6 @@ def setup(
     init_project(path, extensions=exts, exclude_patterns=excl)
     console.print("[green][OK] Initialized .ctxgraph/[/green]")
 
-    # Build per-directory graphs
     ctx_dir = path / ".ctxgraph"
     graphs_dir = ctx_dir / "graphs"
     graphs_dir.mkdir(parents=True, exist_ok=True)
@@ -539,10 +592,7 @@ def setup(
     ])
 
     if top_dirs:
-        for d in top_dirs:
-            db_path = graphs_dir / f"{d.name}.db"
-            _build_single_graph(path, exts, excl, db_path, d.name)
-        console.print(f"[green][OK] Built graphs for {len(top_dirs)} directories[/green]")
+        _build_dirs_parallel(path, exts, excl, top_dirs, graphs_dir, jobs)
     else:
         db_path = ctx_dir / "graph.db"
         _build_single_graph(path, exts, excl, db_path, "combined")
@@ -684,7 +734,7 @@ def version():
     try:
         ver = _v("ctxgraph-code")
     except Exception:
-        ver = "0.2.0"
+        ver = "0.2.1"
     console.print(f"ctxgraph-code version [bold]{ver}[/bold]")
 
 

{ctxgraph_code-0.2.0 → ctxgraph_code-0.2.1}/src/ctxgraph_code.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ctxgraph-code
-Version: 0.2.0
+Version: 0.2.1
 Summary: Code knowledge graph for Claude Code. Build a relationship graph of your Python codebase and query it during coding sessions.
 Author: ctxgraph-code contributors
 License: MIT