vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-04-29 12:11:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 32

Introduce loop detection service that breaks simple loop (#3919)

Browse source 
Co-authored-by: Scott Densmore <scottdensmore@mac.com>
Co-authored-by: N. Taylor Mullen <ntaylormullen@google.com>
This commit is contained in:
Sandy Tao 2025-07-14 20:25:16 -07:00 • committed by GitHub
parent 7ffe8038ef
commit 734da8b9d2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 456 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
packages/cli/src/ui/hooks/useGeminiStream.ts
View file

@ -141,6 +141,8 @@ export const useGeminiStream = (
[toolCalls],
);
const loopDetectedRef = useRef(false);
const onExec = useCallback(async (done: Promise<void>) => {
setIsResponding(true);
await done;
@ -450,6 +452,16 @@ export const useGeminiStream = (
[addItem, config],
);
const handleLoopDetectedEvent = useCallback(() => {
addItem(
{
type: 'info',
text: `A potential loop was detected. This can happen due to repetitive tool calls or other model behavior. The request has been halted.`,
},
Date.now(),
);
}, [addItem]);
const processGeminiStreamEvents = useCallback(
async (
stream: AsyncIterable<GeminiEvent>,
@ -489,6 +501,11 @@ export const useGeminiStream = (
case ServerGeminiEventType.MaxSessionTurns:
handleMaxSessionTurnsEvent();
break;
case ServerGeminiEventType.LoopDetected:
// handle later because we want to move pending history to history
// before we add loop detected message to history
loopDetectedRef.current = true;
break;
default: {
// enforces exhaustive switch-case
const unreachable: never = event;
@ -579,6 +596,10 @@ export const useGeminiStream = (
addItem(pendingHistoryItemRef.current, userMessageTimestamp);
setPendingHistoryItem(null);
}
if (loopDetectedRef.current) {
loopDetectedRef.current = false;
handleLoopDetectedEvent();
}
} catch (error: unknown) {
if (error instanceof UnauthorizedError) {
onAuthError();
@ -616,6 +637,7 @@ export const useGeminiStream = (
config,
startNewPrompt,
getPromptCount,
handleLoopDetectedEvent,
],
);