mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-20 16:57:46 +00:00
Fix resetting loop counts on every other event (#4348)
This commit is contained in:
@@ -97,6 +97,27 @@ describe('LoopDetectionService', () => {
|
|||||||
expect(service.addAndCheck(event3)).toBe(false);
|
expect(service.addAndCheck(event3)).toBe(false);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('should not reset tool call counter for other event types', () => {
|
||||||
|
const toolCallEvent = createToolCallRequestEvent('testTool', {
|
||||||
|
param: 'value',
|
||||||
|
});
|
||||||
|
const otherEvent = {
|
||||||
|
type: 'thought',
|
||||||
|
} as unknown as ServerGeminiStreamEvent;
|
||||||
|
|
||||||
|
// Send events just below the threshold
|
||||||
|
for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD - 1; i++) {
|
||||||
|
expect(service.addAndCheck(toolCallEvent)).toBe(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send a different event type
|
||||||
|
expect(service.addAndCheck(otherEvent)).toBe(false);
|
||||||
|
|
||||||
|
// Send the tool call event again, which should now trigger the loop
|
||||||
|
expect(service.addAndCheck(toolCallEvent)).toBe(true);
|
||||||
|
expect(loggers.logLoopDetected).toHaveBeenCalledTimes(1);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
describe('Content Loop Detection', () => {
|
describe('Content Loop Detection', () => {
|
||||||
|
|||||||
@@ -54,7 +54,6 @@ export class LoopDetectionService {
|
|||||||
case GeminiEventType.Content:
|
case GeminiEventType.Content:
|
||||||
return this.checkContentLoop(event.value);
|
return this.checkContentLoop(event.value);
|
||||||
default:
|
default:
|
||||||
this.reset();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user