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

add: remove loop detect in yolo

Browse Source
This commit is contained in:
koalazf.99
2025-09-16 22:50:30 +08:00
parent a9a84014e4
commit a0d77f5a44
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
packages/core/src/core/client.ts
View File

@@ -564,17 +564,25 @@ export class GeminiClient {
const turn = new Turn(this.getChat(), prompt_id); const turn = new Turn(this.getChat(), prompt_id);
const loopDetected = await this.loopDetector.turnStarted(signal); // Disable loop detection entirely in YOLO mode
if (loopDetected) { const loopDetectionDisabled =
yield { type: GeminiEventType.LoopDetected }; this.config.getApprovalMode() === ApprovalMode.YOLO;
return turn;
if (!loopDetectionDisabled) {
const loopDetected = await this.loopDetector.turnStarted(signal);
if (loopDetected) {
yield { type: GeminiEventType.LoopDetected };
return turn;
}
} }
const resultStream = turn.run(request, signal); const resultStream = turn.run(request, signal);
for await (const event of resultStream) { for await (const event of resultStream) {
if (this.loopDetector.addAndCheck(event)) { if (!loopDetectionDisabled) {
yield { type: GeminiEventType.LoopDetected }; if (this.loopDetector.addAndCheck(event)) {
return turn; yield { type: GeminiEventType.LoopDetected };
return turn;
}
} }
yield event; yield event;
if (event.type === GeminiEventType.Error) { if (event.type === GeminiEventType.Error) {