codegraphcontext 0.6.12__tar.gz → 0.6.13__tar.gz

{codegraphcontext-0.6.12/src/codegraphcontext.egg-info → codegraphcontext-0.6.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.6.12
+Version: 0.6.13
 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

{codegraphcontext-0.6.12 → codegraphcontext-0.6.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraphcontext"
-version = "0.6.12"
+version = "0.6.13"
 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.6.12 → codegraphcontext-0.6.13}/src/codegraphcontext/core/database_embedded_kuzu.py

@@ -32,16 +32,59 @@ DEFAULT_EMBEDDED_BUFFER_POOL_BYTES = 4 * 1024**3
 MIN_DEFAULT_BUFFER_POOL_BYTES = 256 * 1024**2
 
 
+def _cgroup_available_bytes():
+    """Memory headroom under the container's cgroup limit, or None when
+    unconfined. /proc/meminfo shows the HOST inside a container, so a pod
+    capped at 1 GiB on a 64 GiB host would otherwise size a 4 GiB pool and
+    be OOM-killed natively at first touch."""
+    # cgroup v2 (unified hierarchy)
+    try:
+        with open("/sys/fs/cgroup/memory.max") as f:
+            raw = f.read().strip()
+        if raw != "max":
+            limit = int(raw)
+            usage = 0
+            try:
+                with open("/sys/fs/cgroup/memory.current") as f:
+                    usage = int(f.read().strip())
+            except (OSError, ValueError):
+                pass
+            return max(limit - usage, 0)
+    except (OSError, ValueError):
+        pass
+    # cgroup v1
+    try:
+        with open("/sys/fs/cgroup/memory/memory.limit_in_bytes") as f:
+            limit = int(f.read().strip())
+        if limit < 1 << 60:  # v1 reports ~8 EiB when unlimited
+            usage = 0
+            try:
+                with open("/sys/fs/cgroup/memory/memory.usage_in_bytes") as f:
+                    usage = int(f.read().strip())
+            except (OSError, ValueError):
+                pass
+            return max(limit - usage, 0)
+    except (OSError, ValueError):
+        pass
+    return None
+
+
 def _available_memory_bytes():
-    """MemAvailable from /proc/meminfo, or None where unavailable (macOS…)."""
+    """The tighter of host MemAvailable (/proc/meminfo) and the cgroup
+    memory headroom, or None where neither is readable (macOS…)."""
+    host = None
     try:
         with open("/proc/meminfo") as f:
             for line in f:
                 if line.startswith("MemAvailable:"):
-                    return int(line.split()[1]) * 1024
+                    host = int(line.split()[1]) * 1024
+                    break
     except OSError:
         pass
-    return None
+    cgroup = _cgroup_available_bytes()
+    if host is not None and cgroup is not None:
+        return min(host, cgroup)
+    return host if host is not None else cgroup
 
 
 def resolve_embedded_buffer_pool_size() -> int:

{codegraphcontext-0.6.12 → codegraphcontext-0.6.13}/src/codegraphcontext/server.py

@@ -803,6 +803,11 @@ class MCPServer:
     async def _run_loop(self, loop):
         request_count = 0
         while True:
+            # Reset per iteration: on a parse error `request` would otherwise
+            # still hold the PREVIOUS request, and the error response would
+            # reuse its id — a second response for an already-answered id,
+            # which desyncs clients that match responses by id.
+            request = None
             try:
                 if request_count and request_count % 50 == 0:
                     self.job_manager.cleanup_old_jobs(max_age_hours=24)
@@ -880,15 +885,19 @@ class MCPServer:
 
             except Exception as e:
                 error_logger(f"Error processing request: {e}\n{traceback.format_exc()}")
-                request_id = "unknown"
-                if 'request' in locals() and isinstance(request, dict):
-                    request_id = request.get('id', "unknown")
+                # JSON-RPC: parse errors respond with id null and -32700;
+                # errors while handling a parsed request keep that request's id.
+                request_id = request.get('id') if isinstance(request, dict) else None
+                is_parse_error = not isinstance(request, dict)
 
                 error_response = {
                     "jsonrpc": "2.0", "id": request_id,
                     "error": {
-                        "code": -32603,
-                        "message": f"Internal error: {str(e)}",
+                        "code": -32700 if is_parse_error else -32603,
+                        "message": (
+                            f"Parse error: {str(e)}" if is_parse_error
+                            else f"Internal error: {str(e)}"
+                        ),
                     },
                 }
                 print(json.dumps(error_response), flush=True)

{codegraphcontext-0.6.12 → codegraphcontext-0.6.13/src/codegraphcontext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraphcontext
-Version: 0.6.12
+Version: 0.6.13
 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