fix(restart): add explicit restart mode

2026-06-30 15:21:14 +08:00
parent d979597361
commit 4726ca0478
6 changed files with 105 additions and 4 deletions
@@ -218,6 +218,7 @@ class AgentLoop:
        preset_snapshot_loader: preset_helpers.PresetSnapshotLoader | None = None,
        runtime_events: RuntimeEventBus | None = None,
        runtime_model_publisher: Callable[[str, str | None], None] | None = None,
+        restart_mode: str = "auto",
    ):
        from nanobot.config.schema import ToolsConfig

@@ -227,6 +228,7 @@ class AgentLoop:
        self.runtime_events = runtime_events or RuntimeEventBus()
        self.runtime_event_publisher = RuntimeEventPublisher(self.runtime_events)
        self.channels_config = channels_config
+        self.restart_mode = restart_mode
        self.provider = provider
        self._provider_snapshot_loader = provider_snapshot_loader
        self._preset_snapshot_loader = preset_snapshot_loader
@@ -396,6 +398,7 @@ class AgentLoop:
            tools_config=config.tools,
            model_presets=preset_helpers.configured_model_presets(config),
            model_preset=defaults.model_preset,
+            restart_mode=config.gateway.restart_mode,
            provider_snapshot_loader=provider_snapshot_loader,
            preset_snapshot_loader=preset_snapshot_loader,
            **extra,