Fix workflow progress pin lifecycle

1 files changed, 10 insertions, 3 deletions

diff --git a/init.lua b/init.lua
index 2e4cab1..a104324 100644
--- a/init.lua
+++ b/init.lua
@@ -153,6 +153,7 @@ local function activate()
 
     -- Escape reaches the children through the turn, not through a tool: the
     -- primary is parked inside a tool call when they are running.
+    progress.begin_replay()
     -- Startup session replay fires tool lifecycle events before the first
     -- live turn. After that boundary, progress.claim must stay presentation-
     -- only and never rescan the child catalog for ordinary calls.
@@ -193,7 +194,9 @@ local function activate()
         },
         handler = function(input, context)
             progress.bind(context)
-            return run.handle(input, profiles)
+            local result = run.handle(input, profiles)
+            progress.settle(context)
+            return result
         end,
     }
 
@@ -241,7 +244,9 @@ local function activate()
         },
         handler = function(input, context)
             progress.bind(context)
-            return luatool.handle(input, profiles)
+            local result = luatool.handle(input, profiles)
+            progress.settle(context)
+            return result
         end,
     }
 
@@ -278,7 +283,9 @@ local function activate()
         },
         handler = function(input, context)
             progress.bind(context)
-            return toml_workflows.handle(input, profiles)
+            local result = toml_workflows.handle(input, profiles)
+            progress.settle(context)
+            return result
         end,
     }