Initial public release

hydra/config/profiles.py

@@ -0,0 +1,49 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class ProfileLimits:
+    name: str
+    core_mem_gb: int
+    redis_mem_gb: int
+    db_mem_gb: int
+    cpus: int
+    ai_enabled: bool
+    db_backend: str  # "sqlite" or "timescaledb"
+
+
+PROFILES: dict[str, ProfileLimits] = {
+    "lite": ProfileLimits(
+        name="lite",
+        core_mem_gb=2,
+        redis_mem_gb=1,
+        db_mem_gb=0,
+        cpus=2,
+        ai_enabled=False,
+        db_backend="sqlite",
+    ),
+    "pro": ProfileLimits(
+        name="pro",
+        core_mem_gb=4,
+        redis_mem_gb=2,
+        db_mem_gb=4,
+        cpus=4,
+        ai_enabled=True,
+        db_backend="timescaledb",
+    ),
+    "expert": ProfileLimits(
+        name="expert",
+        core_mem_gb=8,
+        redis_mem_gb=4,
+        db_mem_gb=8,
+        cpus=8,
+        ai_enabled=True,
+        db_backend="timescaledb",
+    ),
+}
+
+
+def get_profile(name: str) -> ProfileLimits:
+    if name not in PROFILES:
+        raise ValueError(f"Unknown profile '{name}'. Choose: {list(PROFILES)}")
+    return PROFILES[name]