codegraph-engine 2.1.1__py3-none-any.whl → 2.1.2__py3-none-any.whl

codegraph/__init__.py

@@ -1,6 +1,6 @@
 """CodeGraph MCP: local, evidence-backed codebase intelligence."""
 
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 from codegraph.errors import (
     CodeGraphError,
     ErrorCode,

codegraph/resources/__init__.py

@@ -16,7 +16,8 @@ from .cache import (
 )
 from .coalescer import RequestCoalescer, get_global_coalescer
 from .debouncer import FileChangeDebouncer
-from .governor import ResourceGovernor, get_global_governor, get_process_memory_mb
+from .governor import ResourceGovernor, get_global_governor
+from .monitor import get_process_memory_mb
 from .policy import (
     ActivityMode,
     PressureLevel,

codegraph/resources/governor.py

@@ -5,13 +5,12 @@ throttling background indexing, managing process memory, and cooperatively yield
 """
 from __future__ import annotations
 
-import resource
-import sys
 import threading
 import time
 from collections.abc import Iterator
 from contextlib import contextmanager
 
+from .monitor import get_process_memory_mb
 from .policy import (
     ActivityMode,
     PressureLevel,
@@ -22,18 +21,6 @@ from .policy import (
 )
 
 
-def get_process_memory_mb() -> float:
-    """Return estimated resident process memory in megabytes portably."""
-    try:
-        usage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
-        if sys.platform == "darwin":
-            return round(usage / (1024 * 1024), 2)
-        else:
-            return round(usage / 1024, 2)
-    except Exception:
-        return 0.0
-
-
 class ResourceGovernor:
     """Thread-safe central resource governor and priority scheduler."""
 
@@ -97,17 +84,17 @@ class ResourceGovernor:
         with self._lock:
             active_inter = self._active_interactive
 
-        if limit_mb and mem_mb >= limit_mb * 0.90:
+        if mem_mb is not None and limit_mb and mem_mb >= limit_mb * 0.90:
             return PressureLevel.CRITICAL
         if active_inter >= self.policy.max_interactive_workers * 2:
             return PressureLevel.CRITICAL
 
-        if limit_mb and mem_mb >= limit_mb * 0.75:
+        if mem_mb is not None and limit_mb and mem_mb >= limit_mb * 0.75:
             return PressureLevel.HIGH
         if active_inter > self.policy.max_interactive_workers:
             return PressureLevel.HIGH
 
-        if limit_mb and mem_mb >= limit_mb * 0.60:
+        if mem_mb is not None and limit_mb and mem_mb >= limit_mb * 0.60:
             return PressureLevel.ELEVATED
         if active_inter > 0:
             return PressureLevel.ELEVATED

codegraph/resources/monitor.py

@@ -0,0 +1,102 @@
+"""Cross-platform process resource monitoring for CodeGraph.
+
+Provides portable resident memory (RSS) measurements across Windows, macOS,
+and Linux without requiring Unix-only APIs at import or runtime.
+"""
+from __future__ import annotations
+
+import sys
+from typing import Any
+
+
+def _get_unix_memory_mb() -> float | None:
+    """Read process RSS using standard library resource module on Unix."""
+    try:
+        import resource  # Unix-only standard library module
+
+        usage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
+        if sys.platform == "darwin":
+            # macOS ru_maxrss is returned in bytes
+            return round(usage / (1024 * 1024), 2)
+        else:
+            # Linux and BSD ru_maxrss is returned in kilobytes
+            return round(usage / 1024, 2)
+    except Exception:
+        return None
+
+
+def _get_psutil_memory_mb() -> float | None:
+    """Read process RSS using psutil if installed."""
+    try:
+        import psutil  # type: ignore[import-untyped]
+
+        rss_bytes = float(psutil.Process().memory_info().rss)
+        return round(rss_bytes / (1024 * 1024), 2)
+    except Exception:
+        return None
+
+
+def _get_windows_memory_mb_ctypes() -> float | None:
+    """Read process RSS on Windows using standard library ctypes (zero dependencies)."""
+    try:
+        import ctypes
+        from ctypes import wintypes
+
+        class PROCESS_MEMORY_COUNTERS(ctypes.Structure):
+            _fields_ = [
+                ("cb", wintypes.DWORD),
+                ("PageFaultCount", wintypes.DWORD),
+                ("PeakWorkingSetSize", ctypes.c_size_t),
+                ("WorkingSetSize", ctypes.c_size_t),
+                ("QuotaPeakPagedPoolUsage", ctypes.c_size_t),
+                ("QuotaPagedPoolUsage", ctypes.c_size_t),
+                ("QuotaPeakNonPagedPoolUsage", ctypes.c_size_t),
+                ("QuotaNonPagedPoolUsage", ctypes.c_size_t),
+                ("PagefileUsage", ctypes.c_size_t),
+                ("PeakPagefileUsage", ctypes.c_size_t),
+            ]
+
+        windll: Any = getattr(ctypes, "windll", None)
+        if windll is None:
+            return None
+
+        handle = windll.kernel32.GetCurrentProcess()
+        counters = PROCESS_MEMORY_COUNTERS()
+        counters.cb = ctypes.sizeof(PROCESS_MEMORY_COUNTERS)
+
+        # On Windows 7+, K32GetProcessMemoryInfo is in kernel32; otherwise in psapi
+        get_info = getattr(windll.kernel32, "K32GetProcessMemoryInfo", None)
+        if get_info is None:
+            psapi = getattr(windll, "psapi", None)
+            if psapi is not None:
+                get_info = getattr(psapi, "GetProcessMemoryInfo", None)
+
+        if get_info is not None and bool(get_info(handle, ctypes.byref(counters), counters.cb)):
+            # WorkingSetSize is the resident set size in bytes
+            return round(float(counters.WorkingSetSize) / (1024 * 1024), 2)
+    except Exception:
+        return None
+    return None
+
+
+def get_process_memory_mb() -> float | None:
+    """Return estimated resident process memory in megabytes cross-platform.
+
+    Returns:
+        float: Estimated memory in MB if measurement is supported and succeeds.
+        None: If measurement is unsupported or fails (degrades gracefully to UNKNOWN).
+    """
+    if sys.platform == "win32":
+        # 1. On Windows: try psutil first, then ctypes Windows API
+        mem = _get_psutil_memory_mb()
+        if mem is not None:
+            return mem
+        return _get_windows_memory_mb_ctypes()
+
+    # 2. On Unix-like platforms (macOS, Linux): try standard resource module
+    mem = _get_unix_memory_mb()
+    if mem is not None:
+        return mem
+
+    # 3. Fallback to psutil on any platform if resource module failed or is unavailable
+    return _get_psutil_memory_mb()

codegraph/resources/policy.py

@@ -113,7 +113,7 @@ class ResourceState:
     queued_tasks: int = 0
     recent_cpu_time_ms: float = 0.0
     recent_wall_time_ms: float = 0.0
-    estimated_memory_mb: float = 0.0
+    estimated_memory_mb: float | None = None
     pressure_level: str = PressureLevel.NORMAL.value
     activity_mode: str = ActivityMode.IDLE.value
     indexing_active: bool = False

{codegraph_engine-2.1.1.dist-info → codegraph_engine-2.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraph-engine
-Version: 2.1.1
+Version: 2.1.2
 Summary: Evidence-backed local codebase intelligence for MCP clients
 Author: CodeGraph contributors
 License-Expression: MIT
@@ -11,6 +11,8 @@ Requires-Dist: pydantic>=2.7
 Requires-Dist: typer>=0.12
 Provides-Extra: mcp
 Requires-Dist: mcp<2,>=1.0; extra == "mcp"
+Provides-Extra: system
+Requires-Dist: psutil>=5.9; extra == "system"
 Provides-Extra: dev
 Requires-Dist: pytest>=8; extra == "dev"
 Requires-Dist: pytest-cov>=5; extra == "dev"

{codegraph_engine-2.1.1.dist-info → codegraph_engine-2.1.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-codegraph/__init__.py,sha256=_fvuBpqIOyFtG6zV5ouDbAzcHh6kaZOODpnBo5zKE8A,845
+codegraph/__init__.py,sha256=79b7HvT1to9TpPMt04Kt_qVbBIwZwVr06vQh9ArQxHU,845
 codegraph/agent.py,sha256=We0TWEXSMBBz6t2NQEGPB6gv-2P1IwyCmxoFGPN_CBA,911
 codegraph/architecture.py,sha256=m6RZgXDMoDkSx8KO0xuQHtz5rT0h5B7Jjh1yVE2H2LI,12069
 codegraph/audit.py,sha256=j6fFL6b4vKfnAdUYU7goqQBv4ysrxy-ecBJvT8rYoSI,3301
@@ -43,20 +43,21 @@ codegraph/mcp/__init__.py,sha256=BmjrnksGkbG7TPqwbyQvgYj9uei8pFSFpfkRpaGVdJU,63
 codegraph/mcp/server.py,sha256=CnBdY5Qa9T5LkAFpA3TwwpTkGetkauY7JmiGO2U9ocg,31395
 codegraph/memory/__init__.py,sha256=yfQgSZ9Tii6PPZiLW0ImwC2k-Bg6m7mnUm_K1laBems,58
 codegraph/memory/store.py,sha256=K3iD9VWF5if4OmYCTb5KxtZ3zj45kfEKmz8yivWzP-M,1490
-codegraph/resources/__init__.py,sha256=14FNWGRA0ZXr_m-smhFUxY8dvhgH2f1EqmyXaV6TxHA,1276
+codegraph/resources/__init__.py,sha256=KPpsePEf6cNVvpMI4tnIbYVSw6Q77kaOP6wcmEko7tE,1296
 codegraph/resources/cache.py,sha256=mdhlrXjhbBGChiF_rR1SnMN27oVxjpx9f4zu5jV-hbI,4054
 codegraph/resources/coalescer.py,sha256=bKbTmX8q-R0FjNJjZQrh3Us2ZcuYsIKTrH0WimWcxQs,2510
 codegraph/resources/debouncer.py,sha256=LjDv7uGxixlH-N7vVNkphvlhvtGh6t4jGKFAGg_nX7A,3084
-codegraph/resources/governor.py,sha256=mk9pgxlflvrvm87aZQqexqUt5H6aiJlOlY-AjwiaWWg,8971
-codegraph/resources/policy.py,sha256=pJScYxmhJiSk8c9FlfORTpxf68ZpM2UwNgd4U_LhUWQ,4026
+codegraph/resources/governor.py,sha256=JFZrjwXb19YtPUGoVg2I9yiKBg8Dbw3q1H8Q2J9RW68,8681
+codegraph/resources/monitor.py,sha256=TZMpxvtNS4nwG1w511LAkpef4XThQjNCYZP8qu__Iy0,3753
+codegraph/resources/policy.py,sha256=Xv0Bd1ZzvCcXhAObrbABxTAxdKxFoUUgCrVNSLY5u5w,4034
 codegraph/search/__init__.py,sha256=LFdgDMGE_cGWFFBk--uoRappyC_LnKvfKedogv-p5kQ,504
 codegraph/search/hybrid.py,sha256=pqdRZSwe7wEEETxmUjV5Ns7_oxus60jggnOPNRfFpg0,9617
 codegraph/search/semantic.py,sha256=QZaJVEoOZIZrp4O6SmUAzMiqMPfopgmPSpqzL3T7GVg,836
 codegraph/security/__init__.py,sha256=QsucVn1YOuC1IFrd9-EHeZd9kdHv8XUQgRZ9bTz_6_4,142
 codegraph/security/paths.py,sha256=BtiVvDgMzfL7_Dj3COt6B2Aui4A7hH13QwPUgl5-7No,1241
-codegraph_engine-2.1.1.dist-info/licenses/LICENSE,sha256=I6wm9APPQhaH4Cj9H9z8I2jahgYh7ik-9x7wFrmsGOo,1079
-codegraph_engine-2.1.1.dist-info/METADATA,sha256=A5yelC6QUOe9FKbvQfbQnMHra-hAPoxfQZY19e8B4bI,13240
-codegraph_engine-2.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-codegraph_engine-2.1.1.dist-info/entry_points.txt,sha256=U2xQ3yhqOaaDyFtV9p7N0FDref1VEk56RE_Bnvrnycc,48
-codegraph_engine-2.1.1.dist-info/top_level.txt,sha256=RqBj9sPbifZTb9aeHtnbxTgKJvfHIQdmYE1Brv8Wdkg,10
-codegraph_engine-2.1.1.dist-info/RECORD,,
+codegraph_engine-2.1.2.dist-info/licenses/LICENSE,sha256=I6wm9APPQhaH4Cj9H9z8I2jahgYh7ik-9x7wFrmsGOo,1079
+codegraph_engine-2.1.2.dist-info/METADATA,sha256=1TV7fwB6BD6GdRxajmumnqA-z5iuDRmd9Uslo8b6za4,13309
+codegraph_engine-2.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+codegraph_engine-2.1.2.dist-info/entry_points.txt,sha256=U2xQ3yhqOaaDyFtV9p7N0FDref1VEk56RE_Bnvrnycc,48
+codegraph_engine-2.1.2.dist-info/top_level.txt,sha256=RqBj9sPbifZTb9aeHtnbxTgKJvfHIQdmYE1Brv8Wdkg,10
+codegraph_engine-2.1.2.dist-info/RECORD,,