Merge pull request #2237 from dgageot/board/reset-loop-count-on-error-in-tui-fa5544d2

Reset tool loop detector after degenerate loop error

Author: dgageot

pkg/runtime/loop.go

@@ -352,6 +352,7 @@ func (r *LocalRuntime) RunStream(ctx context.Context, sess *session.Session) <-c
 					loopDetector.consecutive, toolName)
 				events <- Error(errMsg)
 				r.executeNotificationHooks(ctx, a, sess.ID, "error", errMsg)
+				loopDetector.reset()
 				return
 			}
 

pkg/runtime/tool_loop_detector.go

@@ -24,6 +24,12 @@ func newToolLoopDetector(threshold int) *toolLoopDetector {
 	return &toolLoopDetector{threshold: threshold}
 }
 
+// reset clears the detector state so it can be reused after recovery.
+func (d *toolLoopDetector) reset() {
+	d.lastSignature = ""
+	d.consecutive = 0
+}
+
 // record updates the detector with the latest tool call batch and returns
 // true if the consecutive-duplicate threshold has been reached.
 func (d *toolLoopDetector) record(calls []tools.ToolCall) bool {