diff --git a/agent/agent.py b/agent/agent.py index 59cdfed..7c3a3a4 100644 --- a/agent/agent.py +++ b/agent/agent.py @@ -554,7 +554,10 @@ async def handle_job(ctx: JobContext) -> None: logger.info("user state -> %s", ev.new_state) # ── Background task events → room data channel ───────────────────────── - from agent.task_registry import registry as task_registry + try: + from agent.task_registry import registry as task_registry + except ImportError: + from task_registry import registry as task_registry async def _publish_task_event(task_id: str, event: dict) -> None: payload = json.dumps({"type": "task_event", "task_id": task_id, **event})