@google/gemini-cli-core 0.1.12 → 0.1.13

package/dist/src/services/ideContext.js

@@ -0,0 +1,98 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+/**
+ * The reserved server name for the IDE's MCP server.
+ */
+export const IDE_SERVER_NAME = '_ide_server';
+/**
+ * Zod schema for validating a cursor position.
+ */
+export const CursorSchema = z.object({
+    line: z.number(),
+    character: z.number(),
+});
+/**
+ * Zod schema for validating an active file context from the IDE.
+ */
+export const ActiveFileSchema = z.object({
+    filePath: z.string(),
+    cursor: CursorSchema.optional(),
+});
+/**
+ * Zod schema for validating the 'ide/activeFileChanged' notification from the IDE.
+ */
+export const ActiveFileNotificationSchema = z.object({
+    method: z.literal('ide/activeFileChanged'),
+    params: ActiveFileSchema,
+});
+/**
+ * Creates a new store for managing the IDE's active file context.
+ * This factory function encapsulates the state and logic, allowing for the creation
+ * of isolated instances, which is particularly useful for testing.
+ *
+ * @returns An object with methods to interact with the active file context.
+ */
+export function createIdeContextStore() {
+    let activeFileContext = undefined;
+    const subscribers = new Set();
+    /**
+     * Notifies all registered subscribers about the current active file context.
+     */
+    function notifySubscribers() {
+        for (const subscriber of subscribers) {
+            subscriber(activeFileContext);
+        }
+    }
+    /**
+     * Sets the active file context and notifies all registered subscribers of the change.
+     * @param newActiveFile The new active file context from the IDE.
+     */
+    function setActiveFileContext(newActiveFile) {
+        activeFileContext = newActiveFile;
+        notifySubscribers();
+    }
+    /**
+     * Clears the active file context and notifies all registered subscribers of the change.
+     */
+    function clearActiveFileContext() {
+        activeFileContext = undefined;
+        notifySubscribers();
+    }
+    /**
+     * Retrieves the current active file context.
+     * @returns The `ActiveFile` object if a file is active, otherwise `undefined`.
+     */
+    function getActiveFileContext() {
+        return activeFileContext;
+    }
+    /**
+     * Subscribes to changes in the active file context.
+     *
+     * When the active file context changes, the provided `subscriber` function will be called.
+     * Note: The subscriber is not called with the current value upon subscription.
+     *
+     * @param subscriber The function to be called when the active file context changes.
+     * @returns A function that, when called, will unsubscribe the provided subscriber.
+     */
+    function subscribeToActiveFile(subscriber) {
+        subscribers.add(subscriber);
+        return () => {
+            subscribers.delete(subscriber);
+        };
+    }
+    return {
+        setActiveFileContext,
+        getActiveFileContext,
+        subscribeToActiveFile,
+        clearActiveFileContext,
+    };
+}
+/**
+ * The default, shared instance of the IDE context store for the application.
+ */
+export const ideContext = createIdeContextStore();
+//# sourceMappingURL=ideContext.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ideContext.js","sourceRoot":"","sources":["../../../src/services/ideContext.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,aAAa,CAAC;AAE7C;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAGH;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,MAAM,EAAE,YAAY,CAAC,QAAQ,EAAE;CAChC,CAAC,CAAC;AAGH;;GAEG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC;IACnD,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,uBAAuB,CAAC;IAC1C,MAAM,EAAE,gBAAgB;CACzB,CAAC,CAAC;AAIH;;;;;;GAMG;AACH,MAAM,UAAU,qBAAqB;IACnC,IAAI,iBAAiB,GAA2B,SAAS,CAAC;IAC1D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEpD;;OAEG;IACH,SAAS,iBAAiB;QACxB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,UAAU,CAAC,iBAAiB,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,SAAS,oBAAoB,CAAC,aAAyB;QACrD,iBAAiB,GAAG,aAAa,CAAC;QAClC,iBAAiB,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,SAAS,sBAAsB;QAC7B,iBAAiB,GAAG,SAAS,CAAC;QAC9B,iBAAiB,EAAE,CAAC;IACtB,CAAC;IAED;;;OAGG;IACH,SAAS,oBAAoB;QAC3B,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED;;;;;;;;OAQG;IACH,SAAS,qBAAqB,CAAC,UAAgC;QAC7D,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5B,OAAO,GAAG,EAAE;YACV,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO;QACL,oBAAoB;QACpB,oBAAoB;QACpB,qBAAqB;QACrB,sBAAsB;KACvB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,qBAAqB,EAAE,CAAC"}

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

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

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

@@ -0,0 +1,111 @@
+/**
+ * @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);
+    });
+    it('should clear the active file context', () => {
+        const testFile = {
+            filePath: '/path/to/test/file.ts',
+            cursor: { line: 5, character: 10 },
+        };
+        ideContext.setActiveFileContext(testFile);
+        expect(ideContext.getActiveFileContext()).toEqual(testFile);
+        ideContext.clearActiveFileContext();
+        expect(ideContext.getActiveFileContext()).toBeUndefined();
+    });
+});
+//# 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;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,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,CAAC,UAAU,CAAC,oBAAoB,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE5D,UAAU,CAAC,sBAAsB,EAAE,CAAC;QAEpC,MAAM,CAAC,UAAU,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,51 @@
+/**
+ * @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 readonly config;
+    private lastToolCallKey;
+    private toolCallRepetitionCount;
+    private lastRepeatedSentence;
+    private sentenceRepetitionCount;
+    private partialContent;
+    private turnsInCurrentPrompt;
+    private llmCheckInterval;
+    private lastCheckTurn;
+    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;
+    /**
+     * Signals the start of a new turn in the conversation.
+     *
+     * This method increments the turn counter and, if specific conditions are met,
+     * triggers an LLM-based check to detect potential conversation loops. The check
+     * is performed periodically based on the `llmCheckInterval`.
+     *
+     * @param signal - An AbortSignal to allow for cancellation of the asynchronous LLM check.
+     * @returns A promise that resolves to `true` if a loop is detected, and `false` otherwise.
+     */
+    turnStarted(signal: AbortSignal): Promise<boolean>;
+    private checkToolCallLoop;
+    private checkContentLoop;
+    private checkForLoopWithLLM;
+    /**
+     * Resets all loop detection state.
+     */
+    reset(): void;
+    private resetToolCallCount;
+    private resetSentenceCount;
+    private resetLlmCheckTracking;
+}

package/dist/src/services/loopDetectionService.js

@@ -0,0 +1,232 @@
+/**
+ * @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';
+import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/config.js';
+import { Type } from '@google/genai';
+const TOOL_CALL_LOOP_THRESHOLD = 5;
+const CONTENT_LOOP_THRESHOLD = 10;
+/**
+ * The number of recent conversation turns to include in the history when asking the LLM to check for a loop.
+ */
+const LLM_LOOP_CHECK_HISTORY_COUNT = 20;
+/**
+ * The number of turns that must pass in a single prompt before the LLM-based loop check is activated.
+ */
+const LLM_CHECK_AFTER_TURNS = 30;
+/**
+ * The default interval, in number of turns, at which the LLM-based loop check is performed.
+ * This value is adjusted dynamically based on the LLM's confidence.
+ */
+const DEFAULT_LLM_CHECK_INTERVAL = 3;
+/**
+ * The minimum interval for LLM-based loop checks.
+ * This is used when the confidence of a loop is high, to check more frequently.
+ */
+const MIN_LLM_CHECK_INTERVAL = 5;
+/**
+ * The maximum interval for LLM-based loop checks.
+ * This is used when the confidence of a loop is low, to check less frequently.
+ */
+const MAX_LLM_CHECK_INTERVAL = 15;
+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 {
+    config;
+    // Tool call tracking
+    lastToolCallKey = null;
+    toolCallRepetitionCount = 0;
+    // Content streaming tracking
+    lastRepeatedSentence = '';
+    sentenceRepetitionCount = 0;
+    partialContent = '';
+    // LLM loop track tracking
+    turnsInCurrentPrompt = 0;
+    llmCheckInterval = DEFAULT_LLM_CHECK_INTERVAL;
+    lastCheckTurn = 0;
+    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:
+                return false;
+        }
+    }
+    /**
+     * Signals the start of a new turn in the conversation.
+     *
+     * This method increments the turn counter and, if specific conditions are met,
+     * triggers an LLM-based check to detect potential conversation loops. The check
+     * is performed periodically based on the `llmCheckInterval`.
+     *
+     * @param signal - An AbortSignal to allow for cancellation of the asynchronous LLM check.
+     * @returns A promise that resolves to `true` if a loop is detected, and `false` otherwise.
+     */
+    async turnStarted(signal) {
+        this.turnsInCurrentPrompt++;
+        if (this.turnsInCurrentPrompt >= LLM_CHECK_AFTER_TURNS &&
+            this.turnsInCurrentPrompt - this.lastCheckTurn >= this.llmCheckInterval) {
+            this.lastCheckTurn = this.turnsInCurrentPrompt;
+            return await this.checkForLoopWithLLM(signal);
+        }
+        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;
+    }
+    async checkForLoopWithLLM(signal) {
+        const recentHistory = this.config
+            .getGeminiClient()
+            .getHistory()
+            .slice(-LLM_LOOP_CHECK_HISTORY_COUNT);
+        const prompt = `You are a sophisticated AI diagnostic agent specializing in identifying when a conversational AI is stuck in an unproductive state. Your task is to analyze the provided conversation history and determine if the assistant has ceased to make meaningful progress.
+
+An unproductive state is characterized by one or more of the following patterns over the last 5 or more assistant turns:
+
+Repetitive Actions: The assistant repeats the same tool calls or conversational responses a decent number of times. This includes simple loops (e.g., tool_A, tool_A, tool_A) and alternating patterns (e.g., tool_A, tool_B, tool_A, tool_B, ...).
+
+Cognitive Loop: The assistant seems unable to determine the next logical step. It might express confusion, repeatedly ask the same questions, or generate responses that don't logically follow from the previous turns, indicating it's stuck and not advancing the task.
+
+Crucially, differentiate between a true unproductive state and legitimate, incremental progress.
+For example, a series of 'tool_A' or 'tool_B' tool calls that make small, distinct changes to the same file (like adding docstrings to functions one by one) is considered forward progress and is NOT a loop. A loop would be repeatedly replacing the same text with the same content, or cycling between a small set of files with no net change.
+
+Please analyze the conversation history to determine the possibility that the conversation is stuck in a repetitive, non-productive state.`;
+        const contents = [
+            ...recentHistory,
+            { role: 'user', parts: [{ text: prompt }] },
+        ];
+        const schema = {
+            type: Type.OBJECT,
+            properties: {
+                reasoning: {
+                    type: Type.STRING,
+                    description: 'Your reasoning on if the conversation is looping without forward progress.',
+                },
+                confidence: {
+                    type: Type.NUMBER,
+                    description: 'A number between 0.0 and 1.0 representing your confidence that the conversation is in an unproductive state.',
+                },
+            },
+            required: ['reasoning', 'confidence'],
+        };
+        let result;
+        try {
+            result = await this.config
+                .getGeminiClient()
+                .generateJson(contents, schema, signal, DEFAULT_GEMINI_FLASH_MODEL);
+        }
+        catch (e) {
+            // Do nothing, treat it as a non-loop.
+            this.config.getDebugMode() ? console.error(e) : console.debug(e);
+            return false;
+        }
+        if (typeof result.confidence === 'number') {
+            if (result.confidence > 0.9) {
+                if (typeof result.reasoning === 'string' && result.reasoning) {
+                    console.warn(result.reasoning);
+                }
+                logLoopDetected(this.config, new LoopDetectedEvent(LoopType.LLM_DETECTED_LOOP));
+                return true;
+            }
+            else {
+                this.llmCheckInterval = Math.round(MIN_LLM_CHECK_INTERVAL +
+                    (MAX_LLM_CHECK_INTERVAL - MIN_LLM_CHECK_INTERVAL) *
+                        (1 - result.confidence));
+            }
+        }
+        return false;
+    }
+    /**
+     * Resets all loop detection state.
+     */
+    reset() {
+        this.resetToolCallCount();
+        this.resetSentenceCount();
+        this.resetLlmCheckTracking();
+    }
+    resetToolCallCount() {
+        this.lastToolCallKey = null;
+        this.toolCallRepetitionCount = 0;
+    }
+    resetSentenceCount() {
+        this.lastRepeatedSentence = '';
+        this.sentenceRepetitionCount = 0;
+        this.partialContent = '';
+    }
+    resetLlmCheckTracking() {
+        this.turnsInCurrentPrompt = 0;
+        this.llmCheckInterval = DEFAULT_LLM_CHECK_INTERVAL;
+        this.lastCheckTurn = 0;
+    }
+}
+//# 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;AACpE,OAAO,EAAU,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAe,IAAI,EAAE,MAAM,eAAe,CAAC;AAElD,MAAM,wBAAwB,GAAG,CAAC,CAAC;AACnC,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAElC;;GAEG;AACH,MAAM,4BAA4B,GAAG,EAAE,CAAC;AAExC;;GAEG;AACH,MAAM,qBAAqB,GAAG,EAAE,CAAC;AAEjC;;;GAGG;AACH,MAAM,0BAA0B,GAAG,CAAC,CAAC;AAErC;;;GAGG;AACH,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAEjC;;;GAGG;AACH,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAElC,MAAM,iCAAiC,GAAG,gBAAgB,CAAC;AAE3D;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IACd,MAAM,CAAS;IAEhC,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;IAEpC,0BAA0B;IAClB,oBAAoB,GAAG,CAAC,CAAC;IACzB,gBAAgB,GAAG,0BAA0B,CAAC;IAC9C,aAAa,GAAG,CAAC,CAAC;IAE1B,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,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,WAAW,CAAC,MAAmB;QACnC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAE5B,IACE,IAAI,CAAC,oBAAoB,IAAI,qBAAqB;YAClD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,gBAAgB,EACvE,CAAC;YACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC;YAC/C,OAAO,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,KAAK,CAAC;IACf,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;IAEO,KAAK,CAAC,mBAAmB,CAAC,MAAmB;QACnD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM;aAC9B,eAAe,EAAE;aACjB,UAAU,EAAE;aACZ,KAAK,CAAC,CAAC,4BAA4B,CAAC,CAAC;QAExC,MAAM,MAAM,GAAG;;;;;;;;;;;2IAWwH,CAAC;QACxI,MAAM,QAAQ,GAAG;YACf,GAAG,aAAa;YAChB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE;SAC5C,CAAC;QACF,MAAM,MAAM,GAAgB;YAC1B,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,UAAU,EAAE;gBACV,SAAS,EAAE;oBACT,IAAI,EAAE,IAAI,CAAC,MAAM;oBACjB,WAAW,EACT,4EAA4E;iBAC/E;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,IAAI,CAAC,MAAM;oBACjB,WAAW,EACT,8GAA8G;iBACjH;aACF;YACD,QAAQ,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC;SACtC,CAAC;QACF,IAAI,MAAM,CAAC;QACX,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM;iBACvB,eAAe,EAAE;iBACjB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,0BAA0B,CAAC,CAAC;QACxE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,sCAAsC;YACtC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACjE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC1C,IAAI,MAAM,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;gBAC5B,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBAC7D,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACjC,CAAC;gBACD,eAAe,CACb,IAAI,CAAC,MAAM,EACX,IAAI,iBAAiB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAClD,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAChC,sBAAsB;oBACpB,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;wBAC/C,CAAC,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,CAC5B,CAAC;YACJ,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;QAC1B,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,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;IAEO,qBAAqB;QAC3B,IAAI,CAAC,oBAAoB,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,gBAAgB,GAAG,0BAA0B,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;IACzB,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 {};