diff --git a/src/langbot/pkg/box/service.py b/src/langbot/pkg/box/service.py index c8554cc52..deac13110 100644 --- a/src/langbot/pkg/box/service.py +++ b/src/langbot/pkg/box/service.py @@ -278,7 +278,14 @@ class BoxService: skill_mgr = getattr(self.ap, 'skill_mgr', None) reload_skills = getattr(skill_mgr, 'reload_skills', None) if callable(reload_skills) and not self._cloud_managed: - await reload_skills() + binding = await self.ap.workspace_service.get_execution_binding() + await reload_skills( + ExecutionContext( + instance_uuid=binding.instance_uuid, + workspace_uuid=binding.workspace_uuid, + placement_generation=binding.placement_generation, + ) + ) self.ap.logger.info('Box runtime reconnected, sandbox features restored.') return except Exception as exc: diff --git a/tests/unit_tests/box/test_box_service.py b/tests/unit_tests/box/test_box_service.py index ce6e7bc5f..942c87432 100644 --- a/tests/unit_tests/box/test_box_service.py +++ b/tests/unit_tests/box/test_box_service.py @@ -565,7 +565,7 @@ async def test_box_service_reconnect_restores_workspace_and_runs_cleanup( connector.reconnect.assert_awaited_once() service._ensure_default_workspace.assert_called_once() service._purge_attachment_dirs.assert_awaited_once() - app.skill_mgr.reload_skills.assert_awaited_once() + app.skill_mgr.reload_skills.assert_awaited_once_with(_CONTEXT) assert service.available is True