fix(agent): reset recovery chains across injections

Reset both the recovered segments and retry budget whenever injected input starts a new logical answer. Cover fatal tool-error boundaries and rename the prompt test module so pytest can collect the full suite.
2026-07-27 01:39:46 +08:00
parent 154cbc1974
commit 1d2ed6e4d2
3 changed files with 25 additions and 8 deletions
@@ -520,6 +520,8 @@ class AgentRunner:
                    )
                    if should_continue:
                        had_injections = True
+                        length_recovery_count = 0
+                        length_recovery_parts.clear()
                        continue
                    break
                await self._emit_checkpoint(
@@ -635,6 +637,7 @@ class AgentRunner:
                await hook.on_stream_end(context, resuming=should_continue)

            if should_continue:
+                length_recovery_count = 0
                length_recovery_parts.clear()
                await hook.after_iteration(context)
                continue
@@ -657,6 +660,7 @@ class AgentRunner:
                )
                if should_continue:
                    had_injections = True
+                    length_recovery_count = 0
                    length_recovery_parts.clear()
                    continue
                break
@@ -675,6 +679,7 @@ class AgentRunner:
                )
                if should_continue:
                    had_injections = True
+                    length_recovery_count = 0
                    length_recovery_parts.clear()
                    continue
                break