abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-20 08:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log prompt id when a loop is detected (#4765)

Browse Source 
Co-authored-by: N. Taylor Mullen <ntaylormullen@google.com>
This commit is contained in:
Sandy Tao
2025-07-23 22:37:28 -07:00
committed by GitHub
parent 6380bfe35c
commit 0ef9c0b792
5 changed files with 23 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
packages/core/src/services/loopDetectionService.ts
View File

@@ -50,6 +50,7 @@ const SENTENCE_ENDING_PUNCTUATION_REGEX = /[.!?]+(?=\s|$)/;
*/
export class LoopDetectionService {
private readonly config: Config;
private promptId = '';
// Tool call tracking
private lastToolCallKey: string | null = null;
@@ -129,7 +130,10 @@ export class LoopDetectionService {
if (this.toolCallRepetitionCount >= TOOL_CALL_LOOP_THRESHOLD) {
logLoopDetected(
this.config,
new LoopDetectedEvent(LoopType.CONSECUTIVE_IDENTICAL_TOOL_CALLS),
new LoopDetectedEvent(
LoopType.CONSECUTIVE_IDENTICAL_TOOL_CALLS,
this.promptId,
),
);
return true;
}
@@ -170,7 +174,10 @@ export class LoopDetectionService {
if (this.sentenceRepetitionCount >= CONTENT_LOOP_THRESHOLD) {
logLoopDetected(
this.config,
new LoopDetectedEvent(LoopType.CHANTING_IDENTICAL_SENTENCES),
new LoopDetectedEvent(
LoopType.CHANTING_IDENTICAL_SENTENCES,
this.promptId,
),
);
return true;
}
@@ -234,7 +241,7 @@ Please analyze the conversation history to determine the possibility that the co
}
logLoopDetected(
this.config,
new LoopDetectedEvent(LoopType.LLM_DETECTED_LOOP),
new LoopDetectedEvent(LoopType.LLM_DETECTED_LOOP, this.promptId),
);
return true;
} else {
@@ -251,7 +258,8 @@ Please analyze the conversation history to determine the possibility that the co
/**
* Resets all loop detection state.
*/
reset(): void {
reset(promptId: string): void {
this.promptId = promptId;
this.resetToolCallCount();
this.resetSentenceCount();
this.resetLlmCheckTracking();