@vybestack/llxprt-code-core 0.1.12 → 0.1.13-hotfix1

package/dist/src/providers/types/IProviderConfig.d.ts

@@ -0,0 +1,102 @@
+/**
+ * Copyright 2025 Vybestack LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/**
+ * Provider configuration interface that provides a subset of settings
+ * required by provider implementations.
+ */
+export interface IProviderConfig {
+    /**
+     * API key for authenticating with the provider.
+     */
+    apiKey?: string;
+    /**
+     * Base URL for the provider's API endpoint.
+     * If not provided, the provider's default endpoint will be used.
+     */
+    baseUrl?: string;
+    /**
+     * Default model to use for this provider.
+     */
+    defaultModel?: string;
+    /**
+     * Maximum number of tokens to generate in a single request.
+     */
+    maxTokens?: number;
+    /**
+     * Temperature setting for response generation (0.0 - 2.0).
+     */
+    temperature?: number;
+    /**
+     * Whether to enable streaming responses.
+     */
+    streaming?: boolean;
+    /**
+     * Request timeout in milliseconds.
+     */
+    timeout?: number;
+    /**
+     * Custom headers to include in API requests.
+     */
+    customHeaders?: Record<string, string>;
+    /**
+     * Whether to enable text-based tool call parsing for models that
+     * don't natively support function calling.
+     */
+    enableTextToolCallParsing?: boolean;
+    /**
+     * List of model IDs that require text-based tool call parsing.
+     */
+    textToolCallModels?: string[];
+    /**
+     * Override tool format for specific providers.
+     * Maps provider name to tool format.
+     */
+    providerToolFormatOverrides?: Record<string, string>;
+    /**
+     * Whether to enable OpenAI Responses API for compatible models.
+     */
+    openaiResponsesEnabled?: boolean;
+    /**
+     * Organization ID for providers that support organization-level access.
+     */
+    organizationId?: string;
+    /**
+     * Project ID for providers that support project-level access.
+     */
+    projectId?: string;
+    /**
+     * Retry configuration for failed requests.
+     */
+    retryConfig?: {
+        /**
+         * Maximum number of retry attempts.
+         */
+        maxRetries?: number;
+        /**
+         * Base delay in milliseconds between retries.
+         */
+        retryDelay?: number;
+        /**
+         * Whether to use exponential backoff.
+         */
+        exponentialBackoff?: boolean;
+    };
+    /**
+     * Provider-specific configuration options.
+     * This allows for flexibility when different providers need unique settings.
+     */
+    providerSpecific?: Record<string, unknown>;
+}

package/dist/src/providers/types/IProviderConfig.js

@@ -0,0 +1,17 @@
+/**
+ * Copyright 2025 Vybestack LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+export {};
+//# sourceMappingURL=IProviderConfig.js.map

package/dist/src/providers/types/IProviderConfig.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IProviderConfig.js","sourceRoot":"","sources":["../../../../src/providers/types/IProviderConfig.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG"}

package/dist/src/providers/types.d.ts

@@ -3,26 +3,10 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import type { Part } from '@google/genai';
-/**
- * Represents a message in a provider's format
- */
-export interface ProviderMessage {
-    role: 'system' | 'user' | 'assistant' | 'tool';
-    content: string;
-    tool_calls?: ProviderToolCall[];
-    tool_call_id?: string;
-    name?: string;
-    usage?: {
-        prompt_tokens: number;
-        completion_tokens: number;
-        total_tokens: number;
-    };
-    parts?: Part[];
-}
-/**
- * Represents a tool call from a provider
- */
+export type { IMessage as ProviderMessage } from './IMessage.js';
+export type { ITool as ProviderTool } from './ITool.js';
+export type { IProvider as Provider } from './IProvider.js';
+export type { IProviderManager as ProviderManager } from './IProviderManager.js';
 export interface ProviderToolCall {
     id: string;
     type: 'function';
@@ -31,52 +15,3 @@ export interface ProviderToolCall {
         arguments: string;
     };
 }
-/**
- * Represents a tool definition for providers
- */
-export interface ProviderTool {
-    type: 'function';
-    function: {
-        name: string;
-        description?: string;
-        parameters?: Record<string, unknown>;
-    };
-}
-/**
- * Provider interface for external AI providers
- */
-export interface Provider {
-    name: string;
-    /**
-     * Generate a chat completion
-     * @param messages The conversation history
-     * @param tools Available tools for the model to use
-     * @returns An async iterator of response messages
-     */
-    generateChatCompletion(messages: ProviderMessage[], tools?: ProviderTool[]): AsyncIterableIterator<ProviderMessage>;
-    /**
-     * Get the current model being used
-     */
-    getCurrentModel(): string;
-    /**
-     * Set the model to use
-     */
-    setModel(modelId: string): void;
-}
-/**
- * Manager for handling multiple providers
- */
-export interface ProviderManager {
-    /**
-     * Check if a provider is currently active
-     */
-    hasActiveProvider(): boolean;
-    /**
-     * Get the currently active provider
-     */
-    getActiveProvider(): Provider | null;
-    /**
-     * Get the name of the active provider
-     */
-    getActiveProviderName(): string;
-}

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

@@ -112,6 +112,7 @@ export declare function createIdeContextStore(): {
     setActiveFileContext: (newActiveFile: ActiveFile) => void;
     getActiveFileContext: () => ActiveFile | undefined;
     subscribeToActiveFile: (subscriber: ActiveFileSubscriber) => () => void;
+    clearActiveFileContext: () => void;
 };
 /**
  * The default, shared instance of the IDE context store for the application.
@@ -120,5 +121,6 @@ export declare const ideContext: {
     setActiveFileContext: (newActiveFile: ActiveFile) => void;
     getActiveFileContext: () => ActiveFile | undefined;
     subscribeToActiveFile: (subscriber: ActiveFileSubscriber) => () => void;
+    clearActiveFileContext: () => void;
 };
 export {};

package/dist/src/services/ideContext.js

@@ -55,6 +55,13 @@ export function createIdeContextStore() {
         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`.
@@ -81,6 +88,7 @@ export function createIdeContextStore() {
         setActiveFileContext,
         getActiveFileContext,
         subscribeToActiveFile,
+        clearActiveFileContext,
     };
 }
 /**

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

@@ -1 +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;;;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;KACtB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,qBAAqB,EAAE,CAAC"}
+{"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.js

@@ -97,5 +97,15 @@ describe('ideContext - Active File', () => {
         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

@@ -1 +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"}
+{"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

@@ -10,12 +10,15 @@ import { Config } from '../config/config.js';
  * 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 config;
+    private turnsInCurrentPrompt;
+    private llmCheckInterval;
+    private lastCheckTurn;
     constructor(config: Config);
     private getToolCallKey;
     /**
@@ -24,12 +27,25 @@ export declare class LoopDetectionService {
      * @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

@@ -7,14 +7,40 @@ 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;
@@ -22,7 +48,10 @@ export class LoopDetectionService {
     lastRepeatedSentence = '';
     sentenceRepetitionCount = 0;
     partialContent = '';
-    config;
+    // LLM loop track tracking
+    turnsInCurrentPrompt = 0;
+    llmCheckInterval = DEFAULT_LLM_CHECK_INTERVAL;
+    lastCheckTurn = 0;
     constructor(config) {
         this.config = config;
     }
@@ -46,10 +75,28 @@ export class LoopDetectionService {
             case GeminiEventType.Content:
                 return this.checkContentLoop(event.value);
             default:
-                this.reset();
                 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) {
@@ -97,12 +144,75 @@ export class LoopDetectionService {
         }
         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;
@@ -113,5 +223,10 @@ export class LoopDetectionService {
         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

@@ -1 +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"}
+{"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.js

@@ -3,10 +3,11 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { describe, it, expect, beforeEach, vi } from 'vitest';
-import { LoopDetectionService } from './loopDetectionService.js';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { GeminiEventType, } from '../core/turn.js';
 import * as loggers from '../telemetry/loggers.js';
+import { LoopType } from '../telemetry/types.js';
+import { LoopDetectionService } from './loopDetectionService.js';
 vi.mock('../telemetry/loggers.js', () => ({
     logLoopDetected: vi.fn(),
 }));
@@ -76,6 +77,23 @@ describe('LoopDetectionService', () => {
                 expect(service.addAndCheck(event3)).toBe(false);
             }
         });
+        it('should not reset tool call counter for other event types', () => {
+            const toolCallEvent = createToolCallRequestEvent('testTool', {
+                param: 'value',
+            });
+            const otherEvent = {
+                type: 'thought',
+            };
+            // Send events just below the threshold
+            for (let i = 0; i < TOOL_CALL_LOOP_THRESHOLD - 1; i++) {
+                expect(service.addAndCheck(toolCallEvent)).toBe(false);
+            }
+            // Send a different event type
+            expect(service.addAndCheck(otherEvent)).toBe(false);
+            // Send the tool call event again, which should now trigger the loop
+            expect(service.addAndCheck(toolCallEvent)).toBe(true);
+            expect(loggers.logLoopDetected).toHaveBeenCalledTimes(1);
+        });
     });
     describe('Content Loop Detection', () => {
         it(`should not detect a loop for fewer than CONTENT_LOOP_THRESHOLD identical content strings`, () => {
@@ -229,4 +247,93 @@ describe('LoopDetectionService', () => {
         });
     });
 });
+describe('LoopDetectionService LLM Checks', () => {
+    let service;
+    let mockConfig;
+    let mockGeminiClient;
+    let abortController;
+    beforeEach(() => {
+        mockGeminiClient = {
+            getHistory: vi.fn().mockReturnValue([]),
+            generateJson: vi.fn(),
+        };
+        mockConfig = {
+            getGeminiClient: () => mockGeminiClient,
+            getDebugMode: () => false,
+            getTelemetryEnabled: () => true,
+        };
+        service = new LoopDetectionService(mockConfig);
+        abortController = new AbortController();
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    const advanceTurns = async (count) => {
+        for (let i = 0; i < count; i++) {
+            await service.turnStarted(abortController.signal);
+        }
+    };
+    it('should not trigger LLM check before LLM_CHECK_AFTER_TURNS', async () => {
+        await advanceTurns(29);
+        expect(mockGeminiClient.generateJson).not.toHaveBeenCalled();
+    });
+    it('should trigger LLM check on the 30th turn', async () => {
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockResolvedValue({ confidence: 0.1 });
+        await advanceTurns(30);
+        expect(mockGeminiClient.generateJson).toHaveBeenCalledTimes(1);
+    });
+    it('should detect a cognitive loop when confidence is high', async () => {
+        // First check at turn 30
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockResolvedValue({ confidence: 0.85, reasoning: 'Repetitive actions' });
+        await advanceTurns(30);
+        expect(mockGeminiClient.generateJson).toHaveBeenCalledTimes(1);
+        // The confidence of 0.85 will result in a low interval.
+        // The interval will be: 5 + (15 - 5) * (1 - 0.85) = 5 + 10 * 0.15 = 6.5 -> rounded to 7
+        await advanceTurns(6); // advance to turn 36
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockResolvedValue({ confidence: 0.95, reasoning: 'Repetitive actions' });
+        const finalResult = await service.turnStarted(abortController.signal); // This is turn 37
+        expect(finalResult).toBe(true);
+        expect(loggers.logLoopDetected).toHaveBeenCalledWith(mockConfig, expect.objectContaining({
+            'event.name': 'loop_detected',
+            loop_type: LoopType.LLM_DETECTED_LOOP,
+        }));
+    });
+    it('should not detect a loop when confidence is low', async () => {
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockResolvedValue({ confidence: 0.5, reasoning: 'Looks okay' });
+        await advanceTurns(30);
+        const result = await service.turnStarted(abortController.signal);
+        expect(result).toBe(false);
+        expect(loggers.logLoopDetected).not.toHaveBeenCalled();
+    });
+    it('should adjust the check interval based on confidence', async () => {
+        // Confidence is 0.0, so interval should be MAX_LLM_CHECK_INTERVAL (15)
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockResolvedValue({ confidence: 0.0 });
+        await advanceTurns(30); // First check at turn 30
+        expect(mockGeminiClient.generateJson).toHaveBeenCalledTimes(1);
+        await advanceTurns(14); // Advance to turn 44
+        expect(mockGeminiClient.generateJson).toHaveBeenCalledTimes(1);
+        await service.turnStarted(abortController.signal); // Turn 45
+        expect(mockGeminiClient.generateJson).toHaveBeenCalledTimes(2);
+    });
+    it('should handle errors from generateJson gracefully', async () => {
+        mockGeminiClient.generateJson = vi
+            .fn()
+            .mockRejectedValue(new Error('API error'));
+        await advanceTurns(30);
+        const result = await service.turnStarted(abortController.signal);
+        expect(result).toBe(false);
+        expect(loggers.logLoopDetected).not.toHaveBeenCalled();
+    });
+});
 //# sourceMappingURL=loopDetectionService.test.js.map