starred/tty7
1
0
Fork 0
mirror of https://github.com/l0ng-ai/tty7.git synced 2026-11-06 00:15:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): keep Pi and Oh My Pi panes working through retries and approvals

Browse Source 
agent_end closes every attempt, so a run Pi retries after a provider
error read as done while it was still retrying. The bridge now takes
Pi's agent_settled as the end of a run when the build sends it, and
Oh My Pi's auto_retry_start puts the pane back to working.

Oh My Pi asks before running a tool through tool_approval_requested and
tool_approval_resolved, not Pi's UI-prompt pair, so its pane said
working while the approval dialog waited. Those now report waiting and
restore the turn.
This commit is contained in:
l0ng-ai
2026-10-01 15:28:01 +08:00
parent d62c289377
commit 8db82ab48d
1 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/tty7-core/src/core/agent_hooks.rs
+33 -1
View File
@@ -2214,8 +2214,17 @@ export default function (pi: ExtensionAPI) {{
// What the pane showed before a UI prompt put it on "waiting", so closing
// the prompt can put it back.
let turn = "session-start";
// Whether this build says when a run is over for good. agent_end also ends
// each attempt an automatic retry, a compaction or a queued message picks
// up again, and reading that as "done" calls the pane finished while it
// still works. Pi's agent_settled comes once the run is really over; a Pi
// whose on() hands back an unsubscribe is new enough to always send it,
// and an older one that sends it shows so the first time it does.
let settles = false;
pi.on("agent_start", (_event, ctx) => emit((turn = "prompt-submit"), ctx));
pi.on("agent_end", (_event, ctx) => emit((turn = "stop"), ctx));
pi.on("agent_end", (_event, ctx) => {{
if (!settles) emit((turn = "stop"), ctx);
}});
pi.on("session_shutdown", (_event, ctx) => emit("session-end", ctx));
// Last, and guarded: the three above already worked, so a Pi build that
// rejects this event name must not take them — or the whole extension —
@@ -2240,6 +2249,25 @@ export default function (pi: ExtensionAPI) {{
try {{
pi.on("ui_prompt_end", (_event, ctx) => emit(turn, ctx));
}} catch {{}}
try {{
const off: unknown = pi.on("agent_settled" as any, (_event: unknown, ctx: SessionCtx) => {{
settles = true;
emit((turn = "stop"), ctx);
}});
if (typeof off === "function") settles = true;
}} catch {{}}
// Oh My Pi asks before running a tool through events of its own rather
// than Pi's UI-prompt pair, and announces its retries — which follow an
// agent_end — where Pi does not.
try {{
pi.on("tool_approval_requested" as any, (_event: unknown, ctx: SessionCtx) =>
emit("permission-request", ctx),
);
pi.on("tool_approval_resolved" as any, (_event: unknown, ctx: SessionCtx) => emit(turn, ctx));
pi.on("auto_retry_start" as any, (_event: unknown, ctx: SessionCtx) =>
emit((turn = "prompt-submit"), ctx),
);
}} catch {{}}
}}
"#
))
@@ -3826,6 +3854,10 @@ mod tests {
"session_shutdown",
"ui_prompt_start",
"ui_prompt_end",
"agent_settled",
"tool_approval_requested",
"tool_approval_resolved",
"auto_retry_start",
] {
assert!(
bridge.contains(&format!(r#"pi.on("{event}""#)),