Read CPU/RAM from cgroup limits instead of host

app.py

@@ -39,7 +39,79 @@ def _log(message):
     print(message.encode("ascii", "backslashreplace").decode("ascii"), flush=True)
 
 
-def _get_ram_bytes():
+def _read_cgroup_file(path):
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return f.read().strip()
+    except (FileNotFoundError, OSError):
+        return None
+
+
+def _parse_cpu_count():
+    """Read actual CPU allocation from cgroup limits."""
+    # cgroup v2: /sys/fs/cgroup/cpu.max -> "quota period" or "max period"
+    cpu_max = _read_cgroup_file("/sys/fs/cgroup/cpu.max")
+    if cpu_max:
+        parts = cpu_max.split()
+        if len(parts) == 2 and parts[0] != "max":
+            try:
+                quota, period = int(parts[0]), int(parts[1])
+                if period > 0:
+                    return quota // period
+            except ValueError:
+                pass
+
+    # cgroup v1: /sys/fs/cgroup/cpu/cpu.cfs_quota_us and cpu.cfs_period_us
+    quota = _read_cgroup_file("/sys/fs/cgroup/cpu/cpu.cfs_quota_us")
+    period = _read_cgroup_file("/sys/fs/cgroup/cpu/cpu.cfs_period_us")
+    if quota and period:
+        try:
+            q, p = int(quota), int(period)
+            if q > 0 and p > 0:
+                return q // p
+        except ValueError:
+            pass
+
+    # cpuset: /sys/fs/cgroup/cpuset.cpus -> "0-3" or "0,1,2,3"
+    cpuset = _read_cgroup_file("/sys/fs/cgroup/cpuset.cpus")
+    if cpuset:
+        count = 0
+        for part in cpuset.split(","):
+            if "-" in part:
+                start, end = part.split("-", 1)
+                count += int(end) - int(start) + 1
+            else:
+                count += 1
+        if count > 0:
+            return count
+
+    return os.cpu_count()
+
+
+def _parse_ram_bytes():
+    """Read actual memory limit from cgroup limits."""
+    # cgroup v2: /sys/fs/cgroup/memory.max
+    mem_max = _read_cgroup_file("/sys/fs/cgroup/memory.max")
+    if mem_max and mem_max != "max":
+        try:
+            val = int(mem_max)
+            # cgroup v2 reports in bytes
+            if val < 2**62:  # sanity check: not unlimited
+                return val
+        except ValueError:
+            pass
+
+    # cgroup v1: /sys/fs/cgroup/memory/memory.limit_in_bytes
+    mem_limit = _read_cgroup_file("/sys/fs/cgroup/memory/memory.limit_in_bytes")
+    if mem_limit:
+        try:
+            val = int(mem_limit)
+            if val < 2**62:  # sanity check: not unlimited (reported as huge number)
+                return val
+        except ValueError:
+            pass
+
+    # Fallback to /proc/meminfo
     try:
         with open("/proc/meminfo", "r", encoding="utf-8") as meminfo:
             for line in meminfo:
@@ -91,20 +163,20 @@ API_SPECS = {
 
 def _get_runtime_specs():
     storage = shutil.disk_usage(os.getcwd())
-    host_ram = _get_ram_bytes()
-    host_cores = os.cpu_count()
-
-    # ZeroGPU shares a large host machine.  /proc/meminfo and
-    # os.cpu_count() report the *full host*, not the Space's slice.
-    # The actual allocation depends on ZeroGPU scheduling.
     hardware = os.environ.get("SPACE_HARDWARE", "zero-a10g")
     is_zerogpu = "zero" in hardware.lower()
 
+    # Read actual container limits from cgroup, falling back to host values
+    allocated_cores = _parse_cpu_count()
+    allocated_ram = _parse_ram_bytes()
+    host_cores = os.cpu_count()
+
     return {
         "platform": platform.platform(),
         "python_version": platform.python_version(),
+        "cpu_cores": allocated_cores,
         "host_cpu_cores": host_cores,
-        "host_ram_bytes": host_ram,
+        "ram_bytes": allocated_ram,
         "storage_total_bytes": storage.total,
         "storage_used_bytes": storage.used,
         "storage_free_bytes": storage.free,
@@ -114,11 +186,6 @@ def _get_runtime_specs():
             "vram_bytes": 48 * 1024**3 if is_zerogpu else None,
             "shared_zero_gpu": is_zerogpu,
         } if is_zerogpu else None,
-        "note": (
-            "Host machine resources shown; Space gets a shared slice "
-            "via ZeroGPU dynamic allocation."
-            if is_zerogpu else None
-        ),
     }