codegraph-engine 2.1.7__py3-none-any.whl → 2.2.0__py3-none-any.whl

codegraph/mcp/server.py

@@ -173,6 +173,8 @@ def create_server(
     repository: Path,
     settings: Settings | None = None,
     profile: str = "full",
+    host: str = "127.0.0.1",
+    port: int = 8765,
 ) -> Any:
     """Build an MCP FastMCP server lazily so normal CLI use needs no MCP dependency."""
     try:
@@ -184,7 +186,12 @@ def create_server(
 
     indexer = Indexer(repository, settings)
     memory = MemoryStore(indexer.db_path)
-    app = FastMCP("CodeGraph MCP", instructions=render_agent_rules("antigravity"))
+    app = FastMCP(
+        "CodeGraph MCP",
+        instructions=render_agent_rules("antigravity"),
+        host=host,
+        port=port,
+    )
     max_read_bytes = indexer.settings.max_read_bytes
 
     # -----------------------------------------------------------------------
@@ -486,25 +493,79 @@ def create_server(
             return git_analyze_change_impact(con, indexer.repository, since=since, until=until)
 
     def get_repository_status() -> dict[str, object]:
-        """Get repository indexing status, freshness, symbol counts, and health."""
+        """Get repository indexing status, freshness, symbol counts, and health (automatically hot-reloads in-memory caches when `.codegraph.sqlite3` is created or updated by `codegraph init` / `codegraph index`)."""
+        from codegraph.resources.cache import get_graph_cache, get_parse_cache
+
+        get_graph_cache().clear()
+        get_parse_cache().clear()
+        if not indexer.db_path.exists():
+            return {
+                "status": "error",
+                "repository": str(indexer.repository),
+                "generation": 0,
+                "freshness": "NOT_INDEXED",
+                "freshness_detail": "No CodeGraph index exists for this repository.",
+                "files": 0,
+                "files_indexed": 0,
+                "symbols": 0,
+                "symbols_indexed": 0,
+                "graph_edges": 0,
+                "framework_routes": 0,
+                "reloaded": True,
+                "error": {
+                    "code": "INDEX_NOT_FOUND",
+                    "message": "No CodeGraph index exists for this repository.",
+                    "next_action": {
+                        "command": "codegraph init",
+                        "reason": "Initialize the repository before querying it.",
+                    },
+                },
+            }
         with indexer.session() as con:
             report = check_freshness(indexer.repository, con)
-            file_count = con.execute("SELECT count(*) FROM files").fetchone()[0]
-            symbol_count = con.execute("SELECT count(*) FROM symbols").fetchone()[0]
-            edge_count = con.execute("SELECT count(*) FROM graph_edges").fetchone()[0]
-            route_count = con.execute("SELECT count(*) FROM framework_routes").fetchone()[0]
+            file_count = int(con.execute("SELECT count(*) FROM files").fetchone()[0])
+            symbol_count = int(con.execute("SELECT count(*) FROM symbols").fetchone()[0])
+            edge_count = int(con.execute("SELECT count(*) FROM graph_edges").fetchone()[0])
+            route_count = int(con.execute("SELECT count(*) FROM framework_routes").fetchone()[0])
+            if file_count == 0:
+                return {
+                    "status": "error",
+                    "repository": str(indexer.repository),
+                    "generation": report.index_generation,
+                    "freshness": "NOT_INDEXED",
+                    "freshness_detail": "No files are indexed in this repository.",
+                    "files": 0,
+                    "files_indexed": 0,
+                    "symbols": 0,
+                    "symbols_indexed": 0,
+                    "graph_edges": 0,
+                    "framework_routes": 0,
+                    "reloaded": True,
+                    "error": {
+                        "code": "INDEX_NOT_FOUND",
+                        "message": "No CodeGraph index exists for this repository.",
+                        "next_action": {
+                            "command": "codegraph init",
+                            "reason": "Initialize the repository before querying it.",
+                        },
+                    },
+                }
             return {
+                "status": "ok",
                 "repository": str(indexer.repository),
                 "generation": report.index_generation,
                 "freshness": report.status.value,
                 "freshness_detail": report.detail,
+                "files": file_count,
                 "files_indexed": file_count,
+                "symbols": symbol_count,
                 "symbols_indexed": symbol_count,
                 "graph_edges": edge_count,
                 "framework_routes": route_count,
                 "modified_files": report.modified_files,
                 "deleted_files": report.deleted_files,
                 "parse_failed_files": report.parse_failed_files,
+                "reloaded": True,
             }
 
     def trace_call(
@@ -1024,12 +1085,41 @@ def create_server(
     else:
         selected_names = set(all_tools.keys())
 
+    _last_db_sig: list[tuple[int, int, int, int] | None] = [None]
+
+    def _compute_db_sig() -> tuple[int, int, int, int] | None:
+        db_file = indexer.db_path
+        if not db_file.exists():
+            return None
+        try:
+            st = db_file.stat()
+            wal_file = Path(f"{db_file}-wal")
+            wal_mtime = wal_file.stat().st_mtime_ns if wal_file.exists() else 0
+            return (st.st_ino, st.st_mtime_ns, st.st_size, wal_mtime)
+        except OSError:
+            return None
+
+    _last_db_sig[0] = _compute_db_sig()
+
+    def _ensure_hot_reloaded() -> bool:
+        """Detect if `.codegraph.sqlite3` was created, replaced, or updated on disk (e.g. via `codegraph init`)."""
+        cur_sig = _compute_db_sig()
+        if cur_sig != _last_db_sig[0]:
+            _last_db_sig[0] = cur_sig
+            from codegraph.resources.cache import get_graph_cache, get_parse_cache
+
+            get_graph_cache().clear()
+            get_parse_cache().clear()
+            return True
+        return False
+
     for tool_name in sorted(selected_names):
         target_fn: Any = all_tools[tool_name]
 
         def _make_guarded(fn: Any) -> Any:
             @functools.wraps(fn)
             def _guarded(*args: Any, **kwargs: Any) -> Any:
+                _ensure_hot_reloaded()
                 with indexer.governor.task_scope(TaskPriority.INTERACTIVE_HIGH):
                     return fn(*args, **kwargs)
 
@@ -1151,4 +1241,5 @@ def create_server(
             "2. Run `get_context` with intent='REVIEW'."
         )
 
+    app._codegraph_indexer = indexer  # type: ignore[attr-defined]
     return app

codegraph/mcp_diagnostics.py

@@ -111,14 +111,17 @@ class MCPDoctorReport:
             "discovered_tools": list(self.discovered_tools),
         }
 
-    def format_human(self) -> str:
+    def format_human(self, *, unicode_override: bool | None = None) -> str:
+        from codegraph.cli_output import get_cli_symbols
+
+        sym = get_cli_symbols(unicode_override=unicode_override)
         lines = [
             "CodeGraph MCP Doctor",
             "--------------------",
             "",
         ]
         for chk in self.checks:
-            mark = "✓" if chk.passed else "✗"
+            mark = sym.ok if chk.passed else sym.error
             lines.append(f"{mark} {chk.name}")
         return "\n".join(lines)
 

codegraph/monorepo.py

@@ -29,7 +29,7 @@ from codegraph.indexing.classifier import FileCategory, classify_file
 from codegraph.security import is_sensitive, safe_path
 
 try:
-    import yaml  # type: ignore[import-untyped]
+    import yaml
     _HAS_YAML = True
 except ImportError:
     _HAS_YAML = False