chore: fix type check for worker and cpu monitor

livekit · Nov 11, 2024 · 2e1f551 · 2e1f551
1 parent 28e8e7b
commit 2e1f551
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/livekit-agents/livekit/agents/utils/hw/cpu.py b/livekit-agents/livekit/agents/utils/hw/cpu.py
@@ -21,7 +21,7 @@ def cpu_percent(self, interval: float = 0.5) -> float:
 
 class DefaultCPUMonitor(CPUMonitor):
     def cpu_count(self) -> float:
-        return psutil.cpu_count()
+        return psutil.cpu_count() or 0
 
     def cpu_percent(self, interval: float = 0.5) -> float:
         return psutil.cpu_percent(interval) / 100.0

diff --git a/livekit-agents/livekit/agents/worker.py b/livekit-agents/livekit/agents/worker.py
@@ -569,6 +569,9 @@ async def _recv_task():
             await utils.aio.gracefully_cancel(*tasks)
 
     async def _reload_jobs(self, jobs: list[RunningJobInfo]) -> None:
+        if not self._opts.api_secret:
+            raise RuntimeError("api_secret is required to reload jobs")
+
         for aj in jobs:
             logger.log(
                 DEV_LEVEL,