@vybestack/llxprt-code-core 0.1.12

package/dist/src/services/ideContext.test.js

@@ -0,0 +1,101 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { createIdeContextStore } from './ideContext.js';
+describe('ideContext - Active File', () => {
+    let ideContext;
+    beforeEach(() => {
+        // Create a fresh, isolated instance for each test
+        ideContext = createIdeContextStore();
+    });
+    it('should return undefined initially for active file context', () => {
+        expect(ideContext.getActiveFileContext()).toBeUndefined();
+    });
+    it('should set and retrieve the active file context', () => {
+        const testFile = {
+            filePath: '/path/to/test/file.ts',
+            cursor: { line: 5, character: 10 },
+        };
+        ideContext.setActiveFileContext(testFile);
+        const activeFile = ideContext.getActiveFileContext();
+        expect(activeFile).toEqual(testFile);
+    });
+    it('should update the active file context when called multiple times', () => {
+        const firstFile = {
+            filePath: '/path/to/first.js',
+            cursor: { line: 1, character: 1 },
+        };
+        ideContext.setActiveFileContext(firstFile);
+        const secondFile = {
+            filePath: '/path/to/second.py',
+            cursor: { line: 20, character: 30 },
+        };
+        ideContext.setActiveFileContext(secondFile);
+        const activeFile = ideContext.getActiveFileContext();
+        expect(activeFile).toEqual(secondFile);
+    });
+    it('should handle empty string for file path', () => {
+        const testFile = {
+            filePath: '',
+            cursor: { line: 0, character: 0 },
+        };
+        ideContext.setActiveFileContext(testFile);
+        expect(ideContext.getActiveFileContext()).toEqual(testFile);
+    });
+    it('should notify subscribers when active file context changes', () => {
+        const subscriber1 = vi.fn();
+        const subscriber2 = vi.fn();
+        ideContext.subscribeToActiveFile(subscriber1);
+        ideContext.subscribeToActiveFile(subscriber2);
+        const testFile = {
+            filePath: '/path/to/subscribed.ts',
+            cursor: { line: 15, character: 25 },
+        };
+        ideContext.setActiveFileContext(testFile);
+        expect(subscriber1).toHaveBeenCalledTimes(1);
+        expect(subscriber1).toHaveBeenCalledWith(testFile);
+        expect(subscriber2).toHaveBeenCalledTimes(1);
+        expect(subscriber2).toHaveBeenCalledWith(testFile);
+        // Test with another update
+        const newFile = {
+            filePath: '/path/to/new.js',
+            cursor: { line: 1, character: 1 },
+        };
+        ideContext.setActiveFileContext(newFile);
+        expect(subscriber1).toHaveBeenCalledTimes(2);
+        expect(subscriber1).toHaveBeenCalledWith(newFile);
+        expect(subscriber2).toHaveBeenCalledTimes(2);
+        expect(subscriber2).toHaveBeenCalledWith(newFile);
+    });
+    it('should stop notifying a subscriber after unsubscribe', () => {
+        const subscriber1 = vi.fn();
+        const subscriber2 = vi.fn();
+        const unsubscribe1 = ideContext.subscribeToActiveFile(subscriber1);
+        ideContext.subscribeToActiveFile(subscriber2);
+        ideContext.setActiveFileContext({
+            filePath: '/path/to/file1.txt',
+            cursor: { line: 1, character: 1 },
+        });
+        expect(subscriber1).toHaveBeenCalledTimes(1);
+        expect(subscriber2).toHaveBeenCalledTimes(1);
+        unsubscribe1();
+        ideContext.setActiveFileContext({
+            filePath: '/path/to/file2.txt',
+            cursor: { line: 2, character: 2 },
+        });
+        expect(subscriber1).toHaveBeenCalledTimes(1); // Should not be called again
+        expect(subscriber2).toHaveBeenCalledTimes(2);
+    });
+    it('should allow the cursor to be optional', () => {
+        const testFile = {
+            filePath: '/path/to/test/file.ts',
+        };
+        ideContext.setActiveFileContext(testFile);
+        const activeFile = ideContext.getActiveFileContext();
+        expect(activeFile).toEqual(testFile);
+    });
+});
+//# sourceMappingURL=ideContext.test.js.map

package/dist/src/services/ideContext.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ideContext.test.js","sourceRoot":"","sources":["../../../src/services/ideContext.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAExD,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,UAAoD,CAAC;IAEzD,UAAU,CAAC,GAAG,EAAE;QACd,kDAAkD;QAClD,UAAU,GAAG,qBAAqB,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,UAAU,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,uBAAuB;YACjC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE;SACnC,CAAC;QAEF,UAAU,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,UAAU,GAAG,UAAU,CAAC,oBAAoB,EAAE,CAAC;QACrD,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,SAAS,GAAG;YAChB,QAAQ,EAAE,mBAAmB;YAC7B,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SAClC,CAAC;QACF,UAAU,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAE3C,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,oBAAoB;YAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;SACpC,CAAC;QACF,UAAU,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAE5C,MAAM,UAAU,GAAG,UAAU,CAAC,oBAAoB,EAAE,CAAC;QACrD,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SAClC,CAAC;QACF,UAAU,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,oBAAoB,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE5B,UAAU,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;QAC9C,UAAU,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;QAE9C,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,wBAAwB;YAClC,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;SACpC,CAAC;QACF,UAAU,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAEnD,2BAA2B;QAC3B,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE,iBAAiB;YAC3B,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SAClC,CAAC;QACF,UAAU,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE5B,MAAM,YAAY,GAAG,UAAU,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;QACnE,UAAU,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;QAE9C,UAAU,CAAC,oBAAoB,CAAC;YAC9B,QAAQ,EAAE,oBAAoB;YAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SAClC,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE7C,YAAY,EAAE,CAAC;QAEf,UAAU,CAAC,oBAAoB,CAAC;YAC9B,QAAQ,EAAE,oBAAoB;YAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SAClC,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B;QAC3E,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,uBAAuB;SAClC,CAAC;QAEF,UAAU,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,UAAU,GAAG,UAAU,CAAC,oBAAoB,EAAE,CAAC;QACrD,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/services/loopDetectionService.d.ts

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ServerGeminiStreamEvent } from '../core/turn.js';
+import { Config } from '../config/config.js';
+/**
+ * Service for detecting and preventing infinite loops in AI responses.
+ * Monitors tool call repetitions and content sentence repetitions.
+ */
+export declare class LoopDetectionService {
+    private lastToolCallKey;
+    private toolCallRepetitionCount;
+    private lastRepeatedSentence;
+    private sentenceRepetitionCount;
+    private partialContent;
+    private config;
+    constructor(config: Config);
+    private getToolCallKey;
+    /**
+     * Processes a stream event and checks for loop conditions.
+     * @param event - The stream event to process
+     * @returns true if a loop is detected, false otherwise
+     */
+    addAndCheck(event: ServerGeminiStreamEvent): boolean;
+    private checkToolCallLoop;
+    private checkContentLoop;
+    /**
+     * Resets all loop detection state.
+     */
+    reset(): void;
+    private resetToolCallCount;
+    private resetSentenceCount;
+}

package/dist/src/services/loopDetectionService.js

@@ -0,0 +1,117 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { createHash } from 'crypto';
+import { GeminiEventType } from '../core/turn.js';
+import { logLoopDetected } from '../telemetry/loggers.js';
+import { LoopDetectedEvent, LoopType } from '../telemetry/types.js';
+const TOOL_CALL_LOOP_THRESHOLD = 5;
+const CONTENT_LOOP_THRESHOLD = 10;
+const SENTENCE_ENDING_PUNCTUATION_REGEX = /[.!?]+(?=\s|$)/;
+/**
+ * Service for detecting and preventing infinite loops in AI responses.
+ * Monitors tool call repetitions and content sentence repetitions.
+ */
+export class LoopDetectionService {
+    // Tool call tracking
+    lastToolCallKey = null;
+    toolCallRepetitionCount = 0;
+    // Content streaming tracking
+    lastRepeatedSentence = '';
+    sentenceRepetitionCount = 0;
+    partialContent = '';
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    getToolCallKey(toolCall) {
+        const argsString = JSON.stringify(toolCall.args);
+        const keyString = `${toolCall.name}:${argsString}`;
+        return createHash('sha256').update(keyString).digest('hex');
+    }
+    /**
+     * Processes a stream event and checks for loop conditions.
+     * @param event - The stream event to process
+     * @returns true if a loop is detected, false otherwise
+     */
+    addAndCheck(event) {
+        switch (event.type) {
+            case GeminiEventType.ToolCallRequest:
+                // content chanting only happens in one single stream, reset if there
+                // is a tool call in between
+                this.resetSentenceCount();
+                return this.checkToolCallLoop(event.value);
+            case GeminiEventType.Content:
+                return this.checkContentLoop(event.value);
+            default:
+                this.reset();
+                return false;
+        }
+    }
+    checkToolCallLoop(toolCall) {
+        const key = this.getToolCallKey(toolCall);
+        if (this.lastToolCallKey === key) {
+            this.toolCallRepetitionCount++;
+        }
+        else {
+            this.lastToolCallKey = key;
+            this.toolCallRepetitionCount = 1;
+        }
+        if (this.toolCallRepetitionCount >= TOOL_CALL_LOOP_THRESHOLD) {
+            logLoopDetected(this.config, new LoopDetectedEvent(LoopType.CONSECUTIVE_IDENTICAL_TOOL_CALLS));
+            return true;
+        }
+        return false;
+    }
+    checkContentLoop(content) {
+        this.partialContent += content;
+        if (!SENTENCE_ENDING_PUNCTUATION_REGEX.test(this.partialContent)) {
+            return false;
+        }
+        const completeSentences = this.partialContent.match(/[^.!?]+[.!?]+(?=\s|$)/g) || [];
+        if (completeSentences.length === 0) {
+            return false;
+        }
+        const lastSentence = completeSentences[completeSentences.length - 1];
+        const lastCompleteIndex = this.partialContent.lastIndexOf(lastSentence);
+        const endOfLastSentence = lastCompleteIndex + lastSentence.length;
+        this.partialContent = this.partialContent.slice(endOfLastSentence);
+        for (const sentence of completeSentences) {
+            const trimmedSentence = sentence.trim();
+            if (trimmedSentence === '') {
+                continue;
+            }
+            if (this.lastRepeatedSentence === trimmedSentence) {
+                this.sentenceRepetitionCount++;
+            }
+            else {
+                this.lastRepeatedSentence = trimmedSentence;
+                this.sentenceRepetitionCount = 1;
+            }
+            if (this.sentenceRepetitionCount >= CONTENT_LOOP_THRESHOLD) {
+                logLoopDetected(this.config, new LoopDetectedEvent(LoopType.CHANTING_IDENTICAL_SENTENCES));
+                return true;
+            }
+        }
+        return false;
+    }
+    /**
+     * Resets all loop detection state.
+     */
+    reset() {
+        this.resetToolCallCount();
+        this.resetSentenceCount();
+    }
+    resetToolCallCount() {
+        this.lastToolCallKey = null;
+        this.toolCallRepetitionCount = 0;
+    }
+    resetSentenceCount() {
+        this.lastRepeatedSentence = '';
+        this.sentenceRepetitionCount = 0;
+        this.partialContent = '';
+    }
+}
+//# sourceMappingURL=loopDetectionService.js.map

package/dist/src/services/loopDetectionService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loopDetectionService.js","sourceRoot":"","sources":["../../../src/services/loopDetectionService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,eAAe,EAA2B,MAAM,iBAAiB,CAAC;AAC3E,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAGpE,MAAM,wBAAwB,GAAG,CAAC,CAAC;AACnC,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAClC,MAAM,iCAAiC,GAAG,gBAAgB,CAAC;AAE3D;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IAC/B,qBAAqB;IACb,eAAe,GAAkB,IAAI,CAAC;IACtC,uBAAuB,GAAW,CAAC,CAAC;IAE5C,6BAA6B;IACrB,oBAAoB,GAAW,EAAE,CAAC;IAClC,uBAAuB,GAAW,CAAC,CAAC;IACpC,cAAc,GAAW,EAAE,CAAC;IAC5B,MAAM,CAAS;IAEvB,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEO,cAAc,CAAC,QAAwC;QAC7D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,SAAS,GAAG,GAAG,QAAQ,CAAC,IAAI,IAAI,UAAU,EAAE,CAAC;QACnD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC;IAED;;;;OAIG;IACH,WAAW,CAAC,KAA8B;QACxC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,eAAe,CAAC,eAAe;gBAClC,qEAAqE;gBACrE,4BAA4B;gBAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC1B,OAAO,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC7C,KAAK,eAAe,CAAC,OAAO;gBAC1B,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC5C;gBACE,IAAI,CAAC,KAAK,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,QAAwC;QAChE,MAAM,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,IAAI,CAAC,eAAe,KAAK,GAAG,EAAE,CAAC;YACjC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;YAC3B,IAAI,CAAC,uBAAuB,GAAG,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,CAAC,uBAAuB,IAAI,wBAAwB,EAAE,CAAC;YAC7D,eAAe,CACb,IAAI,CAAC,MAAM,EACX,IAAI,iBAAiB,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CACjE,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,gBAAgB,CAAC,OAAe;QACtC,IAAI,CAAC,cAAc,IAAI,OAAO,CAAC;QAE/B,IAAI,CAAC,iCAAiC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;YACjE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,iBAAiB,GACrB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC;QAC5D,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrE,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACxE,MAAM,iBAAiB,GAAG,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC;QAClE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAEnE,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;YACzC,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,IAAI,eAAe,KAAK,EAAE,EAAE,CAAC;gBAC3B,SAAS;YACX,CAAC;YAED,IAAI,IAAI,CAAC,oBAAoB,KAAK,eAAe,EAAE,CAAC;gBAClD,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,oBAAoB,GAAG,eAAe,CAAC;gBAC5C,IAAI,CAAC,uBAAuB,GAAG,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,IAAI,CAAC,uBAAuB,IAAI,sBAAsB,EAAE,CAAC;gBAC3D,eAAe,CACb,IAAI,CAAC,MAAM,EACX,IAAI,iBAAiB,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAC7D,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAEO,kBAAkB;QACxB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC5B,IAAI,CAAC,uBAAuB,GAAG,CAAC,CAAC;IACnC,CAAC;IAEO,kBAAkB;QACxB,IAAI,CAAC,oBAAoB,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,uBAAuB,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;IAC3B,CAAC;CACF"}

package/dist/src/services/loopDetectionService.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/services/loopDetectionService.test.js

@@ -0,0 +1,232 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { LoopDetectionService } from './loopDetectionService.js';
+import { GeminiEventType, } from '../core/turn.js';
+import * as loggers from '../telemetry/loggers.js';
+vi.mock('../telemetry/loggers.js', () => ({
+    logLoopDetected: vi.fn(),
+}));
+const TOOL_CALL_LOOP_THRESHOLD = 5;
+const CONTENT_LOOP_THRESHOLD = 10;
+describe('LoopDetectionService', () => {
+    let service;
+    let mockConfig;
+    beforeEach(() => {
+        mockConfig = {
+            getTelemetryEnabled: () => true,
+        };
+        service = new LoopDetectionService(mockConfig);
+        vi.clearAllMocks();
+    });
+    const createToolCallRequestEvent = (name, args) => ({
+        type: GeminiEventType.ToolCallRequest,
+        value: {
+            name,
+            args,
+            callId: 'test-id',
+            isClientInitiated: false,
+            prompt_id: 'test-prompt-id',
+        },
+    });
+    const createContentEvent = (content) => ({
+        type: GeminiEventType.Content,
+        value: content,
+    });
+    describe('Tool Call Loop Detection', () => {
+        it(`should not detect a loop for fewer than TOOL_CALL_LOOP_THRESHOLD identical calls`, () => {
+            const event = createToolCallRequestEvent('testTool', { param: 'value' });
+            for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(event)).toBe(false);
+            }
+            expect(loggers.logLoopDetected).not.toHaveBeenCalled();
+        });
+        it(`should detect a loop on the TOOL_CALL_LOOP_THRESHOLD-th identical call`, () => {
+            const event = createToolCallRequestEvent('testTool', { param: 'value' });
+            for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD - 1; i++) {
+                service.addAndCheck(event);
+            }
+            expect(service.addAndCheck(event)).toBe(true);
+            expect(loggers.logLoopDetected).toHaveBeenCalledTimes(1);
+        });
+        it('should detect a loop on subsequent identical calls', () => {
+            const event = createToolCallRequestEvent('testTool', { param: 'value' });
+            for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD; i++) {
+                service.addAndCheck(event);
+            }
+            expect(service.addAndCheck(event)).toBe(true);
+            expect(loggers.logLoopDetected).toHaveBeenCalledTimes(2);
+        });
+        it('should not detect a loop for different tool calls', () => {
+            const event1 = createToolCallRequestEvent('testTool', {
+                param: 'value1',
+            });
+            const event2 = createToolCallRequestEvent('testTool', {
+                param: 'value2',
+            });
+            const event3 = createToolCallRequestEvent('anotherTool', {
+                param: 'value1',
+            });
+            for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD - 2; i++) {
+                expect(service.addAndCheck(event1)).toBe(false);
+                expect(service.addAndCheck(event2)).toBe(false);
+                expect(service.addAndCheck(event3)).toBe(false);
+            }
+        });
+    });
+    describe('Content Loop Detection', () => {
+        it(`should not detect a loop for fewer than CONTENT_LOOP_THRESHOLD identical content strings`, () => {
+            const event = createContentEvent('This is a test sentence.');
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(event)).toBe(false);
+            }
+            expect(loggers.logLoopDetected).not.toHaveBeenCalled();
+        });
+        it(`should detect a loop on the CONTENT_LOOP_THRESHOLD-th identical content string`, () => {
+            const event = createContentEvent('This is a test sentence.');
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                service.addAndCheck(event);
+            }
+            expect(service.addAndCheck(event)).toBe(true);
+            expect(loggers.logLoopDetected).toHaveBeenCalledTimes(1);
+        });
+        it('should not detect a loop for different content strings', () => {
+            const event1 = createContentEvent('Sentence A');
+            const event2 = createContentEvent('Sentence B');
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 2; i++) {
+                expect(service.addAndCheck(event1)).toBe(false);
+                expect(service.addAndCheck(event2)).toBe(false);
+            }
+            expect(loggers.logLoopDetected).not.toHaveBeenCalled();
+        });
+    });
+    describe('Sentence Extraction and Punctuation', () => {
+        it('should not check for loops when content has no sentence-ending punctuation', () => {
+            const eventNoPunct = createContentEvent('This has no punctuation');
+            expect(service.addAndCheck(eventNoPunct)).toBe(false);
+            const eventWithPunct = createContentEvent('This has punctuation!');
+            expect(service.addAndCheck(eventWithPunct)).toBe(false);
+        });
+        it('should not treat function calls or method calls as sentence endings', () => {
+            // These should not trigger sentence detection, so repeating them many times should never cause a loop
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD + 2; i++) {
+                expect(service.addAndCheck(createContentEvent('console.log()'))).toBe(false);
+            }
+            service.reset();
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD + 2; i++) {
+                expect(service.addAndCheck(createContentEvent('obj.method()'))).toBe(false);
+            }
+            service.reset();
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD + 2; i++) {
+                expect(service.addAndCheck(createContentEvent('arr.filter().map()'))).toBe(false);
+            }
+            service.reset();
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD + 2; i++) {
+                expect(service.addAndCheck(createContentEvent('if (condition) { return true; }'))).toBe(false);
+            }
+        });
+        it('should correctly identify actual sentence endings and trigger loop detection', () => {
+            // These should trigger sentence detection, so repeating them should eventually cause a loop
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(createContentEvent('This is a sentence.'))).toBe(false);
+            }
+            expect(service.addAndCheck(createContentEvent('This is a sentence.'))).toBe(true);
+            service.reset();
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(createContentEvent('Is this a question? '))).toBe(false);
+            }
+            expect(service.addAndCheck(createContentEvent('Is this a question? '))).toBe(true);
+            service.reset();
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(createContentEvent('What excitement!\n'))).toBe(false);
+            }
+            expect(service.addAndCheck(createContentEvent('What excitement!\n'))).toBe(true);
+        });
+        it('should handle content with mixed punctuation', () => {
+            service.addAndCheck(createContentEvent('Question?'));
+            service.addAndCheck(createContentEvent('Exclamation!'));
+            service.addAndCheck(createContentEvent('Period.'));
+            // Repeat one of them multiple times
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                service.addAndCheck(createContentEvent('Period.'));
+            }
+            expect(service.addAndCheck(createContentEvent('Period.'))).toBe(true);
+        });
+        it('should handle empty sentences after trimming', () => {
+            service.addAndCheck(createContentEvent('   .'));
+            expect(service.addAndCheck(createContentEvent('Normal sentence.'))).toBe(false);
+        });
+        it('should require at least two sentences for loop detection', () => {
+            const event = createContentEvent('Only one sentence.');
+            expect(service.addAndCheck(event)).toBe(false);
+            // Even repeating the same single sentence shouldn't trigger detection
+            for (let i = 0; i < 5; i++) {
+                expect(service.addAndCheck(event)).toBe(false);
+            }
+        });
+    });
+    describe('Performance Optimizations', () => {
+        it('should cache sentence extraction and only re-extract when content grows significantly', () => {
+            // Add initial content
+            service.addAndCheck(createContentEvent('First sentence.'));
+            service.addAndCheck(createContentEvent('Second sentence.'));
+            // Add small amounts of content (shouldn't trigger re-extraction)
+            for (let i = 0; i < 10; i++) {
+                service.addAndCheck(createContentEvent('X'));
+            }
+            service.addAndCheck(createContentEvent('.'));
+            // Should still work correctly
+            expect(service.addAndCheck(createContentEvent('Test.'))).toBe(false);
+        });
+        it('should re-extract sentences when content grows by more than 100 characters', () => {
+            service.addAndCheck(createContentEvent('Initial sentence.'));
+            // Add enough content to trigger re-extraction
+            const longContent = 'X'.repeat(101);
+            service.addAndCheck(createContentEvent(longContent + '.'));
+            // Should work correctly after re-extraction
+            expect(service.addAndCheck(createContentEvent('Test.'))).toBe(false);
+        });
+        it('should use indexOf for efficient counting instead of regex', () => {
+            const repeatedSentence = 'This is a repeated sentence.';
+            // Build up content with the sentence repeated
+            for (let i = 0; i < CONTENT_LOOP_THRESHOLD - 1; i++) {
+                service.addAndCheck(createContentEvent(repeatedSentence));
+            }
+            // The threshold should be reached
+            expect(service.addAndCheck(createContentEvent(repeatedSentence))).toBe(true);
+        });
+    });
+    describe('Edge Cases', () => {
+        it('should handle empty content', () => {
+            const event = createContentEvent('');
+            expect(service.addAndCheck(event)).toBe(false);
+        });
+    });
+    describe('Reset Functionality', () => {
+        it('tool call should reset content count', () => {
+            const contentEvent = createContentEvent('Some content.');
+            const toolEvent = createToolCallRequestEvent('testTool', {
+                param: 'value',
+            });
+            for (let i = 0; i < 9; i++) {
+                service.addAndCheck(contentEvent);
+            }
+            service.addAndCheck(toolEvent);
+            // Should start fresh
+            expect(service.addAndCheck(createContentEvent('Fresh content.'))).toBe(false);
+        });
+    });
+    describe('General Behavior', () => {
+        it('should return false for unhandled event types', () => {
+            const otherEvent = {
+                type: 'unhandled_event',
+            };
+            expect(service.addAndCheck(otherEvent)).toBe(false);
+            expect(service.addAndCheck(otherEvent)).toBe(false);
+        });
+    });
+});
+//# sourceMappingURL=loopDetectionService.test.js.map

package/dist/src/services/loopDetectionService.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loopDetectionService.test.js","sourceRoot":"","sources":["../../../src/services/loopDetectionService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EACL,eAAe,GAGhB,MAAM,iBAAiB,CAAC;AAGzB,OAAO,KAAK,OAAO,MAAM,yBAAyB,CAAC;AAEnD,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AAEJ,MAAM,wBAAwB,GAAG,CAAC,CAAC;AACnC,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAElC,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,OAA6B,CAAC;IAClC,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI;SACX,CAAC;QACvB,OAAO,GAAG,IAAI,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAC/C,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,MAAM,0BAA0B,GAAG,CACjC,IAAY,EACZ,IAA6B,EACK,EAAE,CAAC,CAAC;QACtC,IAAI,EAAE,eAAe,CAAC,eAAe;QACrC,KAAK,EAAE;YACL,IAAI;YACJ,IAAI;YACJ,MAAM,EAAE,SAAS;YACjB,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,gBAAgB;SAC5B;KACF,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAAG,CAAC,OAAe,EAA4B,EAAE,CAAC,CAAC;QACzE,IAAI,EAAE,eAAe,CAAC,OAAO;QAC7B,KAAK,EAAE,OAAO;KACf,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,KAAK,GAAG,0BAA0B,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,wBAAwB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,KAAK,GAAG,0BAA0B,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,wBAAwB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtD,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,KAAK,GAAG,0BAA0B,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,wBAAwB,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClD,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,MAAM,GAAG,0BAA0B,CAAC,UAAU,EAAE;gBACpD,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,0BAA0B,CAAC,UAAU,EAAE;gBACpD,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,0BAA0B,CAAC,aAAa,EAAE;gBACvD,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YAEH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,wBAAwB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,0FAA0F,EAAE,GAAG,EAAE;YAClG,MAAM,KAAK,GAAG,kBAAkB,CAAC,0BAA0B,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,KAAK,GAAG,kBAAkB,CAAC,0BAA0B,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;YAChD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClD,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qCAAqC,EAAE,GAAG,EAAE;QACnD,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,YAAY,GAAG,kBAAkB,CAAC,yBAAyB,CAAC,CAAC;YACnE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAEtD,MAAM,cAAc,GAAG,kBAAkB,CAAC,uBAAuB,CAAC,CAAC;YACnE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,sGAAsG;YACtG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CACnE,KAAK,CACN,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAClE,KAAK,CACN,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC,CAC9D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YAED,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CACJ,OAAO,CAAC,WAAW,CACjB,kBAAkB,CAAC,iCAAiC,CAAC,CACtD,CACF,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,4FAA4F;YAC5F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,CAAC,CAC/D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YACD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,CAAC,CAC/D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,sBAAsB,CAAC,CAAC,CAChE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YACD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,sBAAsB,CAAC,CAAC,CAChE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC,CAC9D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;YACD,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC,CAC9D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC;YACrD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,CAAC;YACxD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC;YAEnD,oCAAoC;YACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC;YACrD,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,IAAI,CACtE,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,KAAK,GAAG,kBAAkB,CAAC,oBAAoB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE/C,sEAAsE;YACtE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,uFAAuF,EAAE,GAAG,EAAE;YAC/F,sBAAsB;YACtB,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAE5D,iEAAiE;YACjE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/C,CAAC;YACD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC;YAE7C,8BAA8B;YAC9B,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAE7D,8CAA8C;YAC9C,MAAM,WAAW,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACpC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,WAAW,GAAG,GAAG,CAAC,CAAC,CAAC;YAE3D,4CAA4C;YAC5C,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,gBAAgB,GAAG,8BAA8B,CAAC;YAExD,8CAA8C;YAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAC5D,CAAC;YAED,kCAAkC;YAClC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,IAAI,CACpE,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,KAAK,GAAG,kBAAkB,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,YAAY,GAAG,kBAAkB,CAAC,eAAe,CAAC,CAAC;YACzD,MAAM,SAAS,GAAG,0BAA0B,CAAC,UAAU,EAAE;gBACvD,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;YACpC,CAAC;YAED,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YAE/B,qBAAqB;YACrB,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,IAAI,CACpE,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,UAAU,GAAG;gBACjB,IAAI,EAAE,iBAAiB;aACc,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Buffer } from 'buffer';
+import { StartSessionEvent, EndSessionEvent, UserPromptEvent, ToolCallEvent, ApiRequestEvent, ApiResponseEvent, ApiErrorEvent, FlashFallbackEvent, LoopDetectedEvent } from '../types.js';
+import { Config } from '../../config/config.js';
+export interface LogResponse {
+    nextRequestWaitMs?: number;
+}
+export declare class ClearcutLogger {
+    private static instance;
+    private config?;
+    private readonly events;
+    private last_flush_time;
+    private flush_interval_ms;
+    private constructor();
+    static getInstance(_config?: Config): ClearcutLogger | undefined;
+    enqueueLogEvent(event: any): void;
+    createLogEvent(name: string, data: object[]): object;
+    flushIfNeeded(): void;
+    flushToClearcut(): Promise<LogResponse>;
+    decodeLogResponse(buf: Buffer): LogResponse | undefined;
+    logStartSessionEvent(event: StartSessionEvent): void;
+    logNewPromptEvent(event: UserPromptEvent): void;
+    logToolCallEvent(event: ToolCallEvent): void;
+    logApiRequestEvent(event: ApiRequestEvent): void;
+    logApiResponseEvent(event: ApiResponseEvent): void;
+    logApiErrorEvent(event: ApiErrorEvent): void;
+    logFlashFallbackEvent(event: FlashFallbackEvent): void;
+    logLoopDetectedEvent(event: LoopDetectedEvent): void;
+    logEndSessionEvent(event: EndSessionEvent): void;
+    shutdown(): void;
+}