@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/safety/context-builder.test.js

@@ -0,0 +1,49 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { ContextBuilder } from './context-builder.js';
+describe('ContextBuilder', () => {
+    let contextBuilder;
+    let mockConfig;
+    const mockHistory = [
+        { user: { text: 'hello' }, model: { text: 'hi' } },
+    ];
+    const mockCwd = '/home/user/project';
+    const mockWorkspaces = ['/home/user/project'];
+    beforeEach(() => {
+        vi.spyOn(process, 'cwd').mockReturnValue(mockCwd);
+        mockConfig = {
+            getWorkspaceContext: vi.fn().mockReturnValue({
+                getDirectories: vi.fn().mockReturnValue(mockWorkspaces),
+            }),
+            apiKey: 'secret-api-key',
+            somePublicConfig: 'public-value',
+            nested: {
+                secretToken: 'hidden',
+                public: 'visible',
+            },
+        };
+        contextBuilder = new ContextBuilder(mockConfig, mockHistory);
+    });
+    it('should build full context with all fields', () => {
+        const context = contextBuilder.buildFullContext();
+        expect(context.environment.cwd).toBe(mockCwd);
+        expect(context.environment.workspaces).toEqual(mockWorkspaces);
+        expect(context.history?.turns).toEqual(mockHistory);
+    });
+    it('should build minimal context with only required keys', () => {
+        const context = contextBuilder.buildMinimalContext(['environment']);
+        expect(context).toHaveProperty('environment');
+        expect(context).not.toHaveProperty('config');
+        expect(context).not.toHaveProperty('history');
+    });
+    it('should handle missing history', () => {
+        contextBuilder = new ContextBuilder(mockConfig);
+        const context = contextBuilder.buildFullContext();
+        expect(context.history?.turns).toEqual([]);
+    });
+});
+//# sourceMappingURL=context-builder.test.js.map

package/dist/src/safety/context-builder.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-builder.test.js","sourceRoot":"","sources":["../../../src/safety/context-builder.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAItD,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,cAA8B,CAAC;IACnC,IAAI,UAAkB,CAAC;IACvB,MAAM,WAAW,GAAuB;QACtC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;KACnD,CAAC;IACF,MAAM,OAAO,GAAG,oBAAoB,CAAC;IACrC,MAAM,cAAc,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAE9C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAClD,UAAU,GAAG;YACX,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC3C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,cAAc,CAAC;aACxD,CAAC;YACF,MAAM,EAAE,gBAAgB;YACxB,gBAAgB,EAAE,cAAc;YAChC,MAAM,EAAE;gBACN,WAAW,EAAE,QAAQ;gBACrB,MAAM,EAAE,SAAS;aAClB;SACmB,CAAC;QACvB,cAAc,GAAG,IAAI,cAAc,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,OAAO,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAC/D,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,OAAO,GAAG,cAAc,CAAC,mBAAmB,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,cAAc,GAAG,IAAI,cAAc,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/safety/protocol.d.ts

@@ -0,0 +1,88 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FunctionCall } from '@google/genai';
+/**
+ * Represents a single turn in the conversation between the user and the model.
+ * This provides semantic context for why a tool call might be happening.
+ */
+export interface ConversationTurn {
+    user: {
+        text: string;
+    };
+    model: {
+        text?: string;
+        toolCalls?: FunctionCall[];
+    };
+}
+/**
+ * The data structure passed from the CLI to a safety checker process via stdin.
+ */
+export interface SafetyCheckInput {
+    /**
+     * The semantic version of the protocol (e.g., "1.0.0"). This allows
+     * for introducing breaking changes in the future while maintaining
+     * support for older checkers.
+     */
+    protocolVersion: '1.0.0';
+    /**
+     * The specific tool call that is being validated.
+     */
+    toolCall: FunctionCall;
+    /**
+     * A container for all contextual information from the CLI's internal state.
+     * By grouping data into categories, we can easily add new context in the
+     * future without creating a flat, unmanageable object.
+     */
+    context: {
+        /**
+         * Information about the user's file system and execution environment.
+         */
+        environment: {
+            cwd: string;
+            workspaces: string[];
+        };
+        /**
+         * The recent history of the conversation. This can be used by checkers
+         * that need to understand the intent behind a tool call.
+         */
+        history?: {
+            turns: ConversationTurn[];
+        };
+    };
+    /**
+     * Configuration for the safety checker.
+     * This allows checkers to be parameterized (e.g. allowed paths).
+     */
+    config?: unknown;
+}
+/**
+ * The possible decisions a safety checker can make.
+ */
+export declare enum SafetyCheckDecision {
+    ALLOW = "allow",
+    DENY = "deny",
+    ASK_USER = "ask_user"
+}
+/**
+ * The data structure returned by a safety checker process via stdout.
+ */
+export type SafetyCheckResult = {
+    /**
+     * The decision made by the safety checker.
+     */
+    decision: SafetyCheckDecision.ALLOW;
+    /**
+     * If not allowed, a message explaining why the tool call was blocked.
+     * This will be shown to the user.
+     */
+    reason?: string;
+} | {
+    decision: SafetyCheckDecision.DENY;
+    reason: string;
+} | {
+    decision: SafetyCheckDecision.ASK_USER;
+    reason: string;
+};

package/dist/src/safety/protocol.js

@@ -0,0 +1,15 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * The possible decisions a safety checker can make.
+ */
+export var SafetyCheckDecision;
+(function (SafetyCheckDecision) {
+    SafetyCheckDecision["ALLOW"] = "allow";
+    SafetyCheckDecision["DENY"] = "deny";
+    SafetyCheckDecision["ASK_USER"] = "ask_user";
+})(SafetyCheckDecision || (SafetyCheckDecision = {}));
+//# sourceMappingURL=protocol.js.map

package/dist/src/safety/protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../../src/safety/protocol.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgEH;;GAEG;AACH,MAAM,CAAN,IAAY,mBAIX;AAJD,WAAY,mBAAmB;IAC7B,sCAAe,CAAA;IACf,oCAAa,CAAA;IACb,4CAAqB,CAAA;AACvB,CAAC,EAJW,mBAAmB,KAAnB,mBAAmB,QAI9B"}

package/dist/src/safety/registry.d.ts

@@ -0,0 +1,26 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type InProcessChecker } from './built-in.js';
+/**
+ * Registry for managing safety checker resolution.
+ */
+export declare class CheckerRegistry {
+    private readonly checkersPath;
+    private static readonly BUILT_IN_EXTERNAL_CHECKERS;
+    private static readonly BUILT_IN_IN_PROCESS_CHECKERS;
+    private static readonly VALID_NAME_PATTERN;
+    constructor(checkersPath: string);
+    /**
+     * Resolves an external checker name to an absolute executable path.
+     */
+    resolveExternal(name: string): string;
+    /**
+     * Resolves an in-process checker name to a checker instance.
+     */
+    resolveInProcess(name: string): InProcessChecker;
+    private static isValidCheckerName;
+    static getBuiltInCheckers(): string[];
+}

package/dist/src/safety/registry.js

@@ -0,0 +1,65 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as path from 'node:path';
+import * as fs from 'node:fs';
+import { AllowedPathChecker } from './built-in.js';
+import { InProcessCheckerType } from '../policy/types.js';
+/**
+ * Registry for managing safety checker resolution.
+ */
+export class CheckerRegistry {
+    checkersPath;
+    static BUILT_IN_EXTERNAL_CHECKERS = new Map([
+    // No external built-ins for now
+    ]);
+    static BUILT_IN_IN_PROCESS_CHECKERS = new Map([[InProcessCheckerType.ALLOWED_PATH, new AllowedPathChecker()]]);
+    // Regex to validate checker names (alphanumeric and hyphens only)
+    static VALID_NAME_PATTERN = /^[a-z0-9-]+$/;
+    constructor(checkersPath) {
+        this.checkersPath = checkersPath;
+    }
+    /**
+     * Resolves an external checker name to an absolute executable path.
+     */
+    resolveExternal(name) {
+        if (!CheckerRegistry.isValidCheckerName(name)) {
+            throw new Error(`Invalid checker name "${name}". Checker names must contain only lowercase letters, numbers, and hyphens.`);
+        }
+        const builtInPath = CheckerRegistry.BUILT_IN_EXTERNAL_CHECKERS.get(name);
+        if (builtInPath) {
+            const fullPath = path.join(this.checkersPath, builtInPath);
+            if (!fs.existsSync(fullPath)) {
+                throw new Error(`Built-in checker "${name}" not found at ${fullPath}`);
+            }
+            return fullPath;
+        }
+        // TODO: Phase 5 - Add support for custom external checkers
+        throw new Error(`Unknown external checker "${name}".`);
+    }
+    /**
+     * Resolves an in-process checker name to a checker instance.
+     */
+    resolveInProcess(name) {
+        if (!CheckerRegistry.isValidCheckerName(name)) {
+            throw new Error(`Invalid checker name "${name}".`);
+        }
+        const checker = CheckerRegistry.BUILT_IN_IN_PROCESS_CHECKERS.get(name);
+        if (checker) {
+            return checker;
+        }
+        throw new Error(`Unknown in-process checker "${name}". Available: ${Array.from(CheckerRegistry.BUILT_IN_IN_PROCESS_CHECKERS.keys()).join(', ')}`);
+    }
+    static isValidCheckerName(name) {
+        return this.VALID_NAME_PATTERN.test(name) && !name.includes('..');
+    }
+    static getBuiltInCheckers() {
+        return [
+            ...Array.from(this.BUILT_IN_EXTERNAL_CHECKERS.keys()),
+            ...Array.from(this.BUILT_IN_IN_PROCESS_CHECKERS.keys()),
+        ];
+    }
+}
+//# sourceMappingURL=registry.js.map

package/dist/src/safety/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/safety/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAyB,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAC1E,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE1D;;GAEG;AACH,MAAM,OAAO,eAAe;IAaG;IAZrB,MAAM,CAAU,0BAA0B,GAAG,IAAI,GAAG,CAAiB;IAC3E,gCAAgC;KACjC,CAAC,CAAC;IAEK,MAAM,CAAU,4BAA4B,GAAG,IAAI,GAAG,CAG5D,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,IAAI,kBAAkB,EAAE,CAAC,CAAC,CAAC,CAAC;IAEnE,kEAAkE;IAC1D,MAAM,CAAU,kBAAkB,GAAG,cAAc,CAAC;IAE5D,YAA6B,YAAoB;QAApB,iBAAY,GAAZ,YAAY,CAAQ;IAAG,CAAC;IAErD;;OAEG;IACH,eAAe,CAAC,IAAY;QAC1B,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CACb,yBAAyB,IAAI,6EAA6E,CAC3G,CAAC;QACJ,CAAC;QAED,MAAM,WAAW,GAAG,eAAe,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzE,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YAC3D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,kBAAkB,QAAQ,EAAE,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,2DAA2D;QAC3D,MAAM,IAAI,KAAK,CAAC,6BAA6B,IAAI,IAAI,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,IAAY;QAC3B,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,IAAI,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,OAAO,GAAG,eAAe,CAAC,4BAA4B,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACvE,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,MAAM,IAAI,KAAK,CACb,+BAA+B,IAAI,iBAAiB,KAAK,CAAC,IAAI,CAC5D,eAAe,CAAC,4BAA4B,CAAC,IAAI,EAAE,CACpD,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACf,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,kBAAkB,CAAC,IAAY;QAC5C,OAAO,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,CAAC,kBAAkB;QACvB,OAAO;YACL,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,CAAC;YACrD,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,CAAC;SACxD,CAAC;IACJ,CAAC"}

package/dist/src/safety/registry.test.d.ts

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

package/dist/src/safety/registry.test.js

@@ -0,0 +1,31 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { CheckerRegistry } from './registry.js';
+import { InProcessCheckerType } from '../policy/types.js';
+import { AllowedPathChecker } from './built-in.js';
+describe('CheckerRegistry', () => {
+    let registry;
+    const mockCheckersPath = '/mock/checkers/path';
+    beforeEach(() => {
+        registry = new CheckerRegistry(mockCheckersPath);
+    });
+    it('should resolve built-in in-process checkers', () => {
+        const checker = registry.resolveInProcess(InProcessCheckerType.ALLOWED_PATH);
+        expect(checker).toBeInstanceOf(AllowedPathChecker);
+    });
+    it('should throw for unknown in-process checkers', () => {
+        expect(() => registry.resolveInProcess('unknown-checker')).toThrow('Unknown in-process checker "unknown-checker"');
+    });
+    it('should validate checker names', () => {
+        expect(() => registry.resolveInProcess('invalid name!')).toThrow('Invalid checker name');
+        expect(() => registry.resolveInProcess('../escape')).toThrow('Invalid checker name');
+    });
+    it('should throw for unknown external checkers (for now)', () => {
+        expect(() => registry.resolveExternal('some-external')).toThrow('Unknown external checker "some-external"');
+    });
+});
+//# sourceMappingURL=registry.test.js.map

package/dist/src/safety/registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.test.js","sourceRoot":"","sources":["../../../src/safety/registry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,QAAyB,CAAC;IAC9B,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;IAE/C,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,eAAe,CAAC,gBAAgB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,OAAO,GAAG,QAAQ,CAAC,gBAAgB,CACvC,oBAAoB,CAAC,YAAY,CAClC,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAChE,8CAA8C,CAC/C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAC9D,sBAAsB,CACvB,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAC1D,sBAAsB,CACvB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAC7D,0CAA0C,CAC3C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,32 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content } from '@google/genai';
+import type { Config } from '../config/config.js';
+import type { GeminiChat } from '../core/geminiChat.js';
+import { type ChatCompressionInfo } from '../core/turn.js';
+/**
+ * Default threshold for compression token count as a fraction of the model's
+ * token limit. If the chat history exceeds this threshold, it will be compressed.
+ */
+export declare const DEFAULT_COMPRESSION_TOKEN_THRESHOLD = 0.5;
+/**
+ * The fraction of the latest chat history to keep. A value of 0.3
+ * means that only the last 30% of the chat history will be kept after compression.
+ */
+export declare const COMPRESSION_PRESERVE_THRESHOLD = 0.3;
+/**
+ * Returns the index of the oldest item to keep when compressing. May return
+ * contents.length which indicates that everything should be compressed.
+ *
+ * Exported for testing purposes.
+ */
+export declare function findCompressSplitPoint(contents: Content[], fraction: number): number;
+export declare class ChatCompressionService {
+    compress(chat: GeminiChat, promptId: string, force: boolean, model: string, config: Config, hasFailedCompressionAttempt: boolean): Promise<{
+        newHistory: Content[] | null;
+        info: ChatCompressionInfo;
+    }>;
+}

package/dist/src/services/chatCompressionService.js

@@ -0,0 +1,162 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CompressionStatus } from '../core/turn.js';
+import { tokenLimit } from '../core/tokenLimits.js';
+import { getCompressionPrompt } from '../core/prompts.js';
+import { getResponseText } from '../utils/partUtils.js';
+import { logChatCompression } from '../telemetry/loggers.js';
+import { makeChatCompressionEvent } from '../telemetry/types.js';
+import { getInitialChatHistory } from '../utils/environmentContext.js';
+/**
+ * Default threshold for compression token count as a fraction of the model's
+ * token limit. If the chat history exceeds this threshold, it will be compressed.
+ */
+export const DEFAULT_COMPRESSION_TOKEN_THRESHOLD = 0.5;
+/**
+ * The fraction of the latest chat history to keep. A value of 0.3
+ * means that only the last 30% of the chat history will be kept after compression.
+ */
+export const COMPRESSION_PRESERVE_THRESHOLD = 0.3;
+/**
+ * Returns the index of the oldest item to keep when compressing. May return
+ * contents.length which indicates that everything should be compressed.
+ *
+ * Exported for testing purposes.
+ */
+export function findCompressSplitPoint(contents, fraction) {
+    if (fraction <= 0 || fraction >= 1) {
+        throw new Error('Fraction must be between 0 and 1');
+    }
+    const charCounts = contents.map((content) => JSON.stringify(content).length);
+    const totalCharCount = charCounts.reduce((a, b) => a + b, 0);
+    const targetCharCount = totalCharCount * fraction;
+    let lastSplitPoint = 0; // 0 is always valid (compress nothing)
+    let cumulativeCharCount = 0;
+    for (let i = 0; i < contents.length; i++) {
+        const content = contents[i];
+        if (content.role === 'user' &&
+            !content.parts?.some((part) => !!part.functionResponse)) {
+            if (cumulativeCharCount >= targetCharCount) {
+                return i;
+            }
+            lastSplitPoint = i;
+        }
+        cumulativeCharCount += charCounts[i];
+    }
+    // We found no split points after targetCharCount.
+    // Check if it's safe to compress everything.
+    const lastContent = contents[contents.length - 1];
+    if (lastContent?.role === 'model' &&
+        !lastContent?.parts?.some((part) => part.functionCall)) {
+        return contents.length;
+    }
+    // Can't compress everything so just compress at last splitpoint.
+    return lastSplitPoint;
+}
+export class ChatCompressionService {
+    async compress(chat, promptId, force, model, config, hasFailedCompressionAttempt) {
+        const curatedHistory = chat.getHistory(true);
+        // Regardless of `force`, don't do anything if the history is empty.
+        if (curatedHistory.length === 0 ||
+            (hasFailedCompressionAttempt && !force)) {
+            return {
+                newHistory: null,
+                info: {
+                    originalTokenCount: 0,
+                    newTokenCount: 0,
+                    compressionStatus: CompressionStatus.NOOP,
+                },
+            };
+        }
+        const originalTokenCount = chat.getLastPromptTokenCount();
+        // Don't compress if not forced and we are under the limit.
+        if (!force) {
+            const threshold = (await config.getCompressionThreshold()) ??
+                DEFAULT_COMPRESSION_TOKEN_THRESHOLD;
+            if (originalTokenCount < threshold * tokenLimit(model)) {
+                return {
+                    newHistory: null,
+                    info: {
+                        originalTokenCount,
+                        newTokenCount: originalTokenCount,
+                        compressionStatus: CompressionStatus.NOOP,
+                    },
+                };
+            }
+        }
+        const splitPoint = findCompressSplitPoint(curatedHistory, 1 - COMPRESSION_PRESERVE_THRESHOLD);
+        const historyToCompress = curatedHistory.slice(0, splitPoint);
+        const historyToKeep = curatedHistory.slice(splitPoint);
+        if (historyToCompress.length === 0) {
+            return {
+                newHistory: null,
+                info: {
+                    originalTokenCount,
+                    newTokenCount: originalTokenCount,
+                    compressionStatus: CompressionStatus.NOOP,
+                },
+            };
+        }
+        const summaryResponse = await config.getContentGenerator().generateContent({
+            model,
+            contents: [
+                ...historyToCompress,
+                {
+                    role: 'user',
+                    parts: [
+                        {
+                            text: 'First, reason in your scratchpad. Then, generate the <state_snapshot>.',
+                        },
+                    ],
+                },
+            ],
+            config: {
+                systemInstruction: { text: getCompressionPrompt() },
+            },
+        }, promptId);
+        const summary = getResponseText(summaryResponse) ?? '';
+        const extraHistory = [
+            {
+                role: 'user',
+                parts: [{ text: summary }],
+            },
+            {
+                role: 'model',
+                parts: [{ text: 'Got it. Thanks for the additional context!' }],
+            },
+            ...historyToKeep,
+        ];
+        // Use a shared utility to construct the initial history for an accurate token count.
+        const fullNewHistory = await getInitialChatHistory(config, extraHistory);
+        // Estimate token count 1 token ≈ 4 characters
+        const newTokenCount = Math.floor(fullNewHistory.reduce((total, content) => total + JSON.stringify(content).length, 0) / 4);
+        logChatCompression(config, makeChatCompressionEvent({
+            tokens_before: originalTokenCount,
+            tokens_after: newTokenCount,
+        }));
+        if (newTokenCount > originalTokenCount) {
+            return {
+                newHistory: null,
+                info: {
+                    originalTokenCount,
+                    newTokenCount,
+                    compressionStatus: CompressionStatus.COMPRESSION_FAILED_INFLATED_TOKEN_COUNT,
+                },
+            };
+        }
+        else {
+            return {
+                newHistory: extraHistory,
+                info: {
+                    originalTokenCount,
+                    newTokenCount,
+                    compressionStatus: CompressionStatus.COMPRESSED,
+                },
+            };
+        }
+    }
+}
+//# sourceMappingURL=chatCompressionService.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatCompressionService.js","sourceRoot":"","sources":["../../../src/services/chatCompressionService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,EAA4B,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAEvE;;;GAGG;AACH,MAAM,CAAC,MAAM,mCAAmC,GAAG,GAAG,CAAC;AAEvD;;;GAGG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAElD;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAmB,EACnB,QAAgB;IAEhB,IAAI,QAAQ,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,MAAM,eAAe,GAAG,cAAc,GAAG,QAAQ,CAAC;IAElD,IAAI,cAAc,GAAG,CAAC,CAAC,CAAC,uCAAuC;IAC/D,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC5B,IACE,OAAO,CAAC,IAAI,KAAK,MAAM;YACvB,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,EACvD,CAAC;YACD,IAAI,mBAAmB,IAAI,eAAe,EAAE,CAAC;gBAC3C,OAAO,CAAC,CAAC;YACX,CAAC;YACD,cAAc,GAAG,CAAC,CAAC;QACrB,CAAC;QACD,mBAAmB,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,kDAAkD;IAClD,6CAA6C;IAC7C,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClD,IACE,WAAW,EAAE,IAAI,KAAK,OAAO;QAC7B,CAAC,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,EACtD,CAAC;QACD,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,iEAAiE;IACjE,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,OAAO,sBAAsB;IACjC,KAAK,CAAC,QAAQ,CACZ,IAAgB,EAChB,QAAgB,EAChB,KAAc,EACd,KAAa,EACb,MAAc,EACd,2BAAoC;QAEpC,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAE7C,oEAAoE;QACpE,IACE,cAAc,CAAC,MAAM,KAAK,CAAC;YAC3B,CAAC,2BAA2B,IAAI,CAAC,KAAK,CAAC,EACvC,CAAC;YACD,OAAO;gBACL,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE;oBACJ,kBAAkB,EAAE,CAAC;oBACrB,aAAa,EAAE,CAAC;oBAChB,iBAAiB,EAAE,iBAAiB,CAAC,IAAI;iBAC1C;aACF,CAAC;QACJ,CAAC;QAED,MAAM,kBAAkB,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAE1D,2DAA2D;QAC3D,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,SAAS,GACb,CAAC,MAAM,MAAM,CAAC,uBAAuB,EAAE,CAAC;gBACxC,mCAAmC,CAAC;YACtC,IAAI,kBAAkB,GAAG,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvD,OAAO;oBACL,UAAU,EAAE,IAAI;oBAChB,IAAI,EAAE;wBACJ,kBAAkB;wBAClB,aAAa,EAAE,kBAAkB;wBACjC,iBAAiB,EAAE,iBAAiB,CAAC,IAAI;qBAC1C;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,sBAAsB,CACvC,cAAc,EACd,CAAC,GAAG,8BAA8B,CACnC,CAAC;QAEF,MAAM,iBAAiB,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QAC9D,MAAM,aAAa,GAAG,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAEvD,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,OAAO;gBACL,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE;oBACJ,kBAAkB;oBAClB,aAAa,EAAE,kBAAkB;oBACjC,iBAAiB,EAAE,iBAAiB,CAAC,IAAI;iBAC1C;aACF,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,mBAAmB,EAAE,CAAC,eAAe,CACxE;YACE,KAAK;YACL,QAAQ,EAAE;gBACR,GAAG,iBAAiB;gBACpB;oBACE,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EAAE,wEAAwE;yBAC/E;qBACF;iBACF;aACF;YACD,MAAM,EAAE;gBACN,iBAAiB,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,EAAE;aACpD;SACF,EACD,QAAQ,CACT,CAAC;QACF,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAEvD,MAAM,YAAY,GAAc;YAC9B;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3B;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,4CAA4C,EAAE,CAAC;aAChE;YACD,GAAG,aAAa;SACjB,CAAC;QAEF,qFAAqF;QACrF,MAAM,cAAc,GAAG,MAAM,qBAAqB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAEzE,8CAA8C;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAC9B,cAAc,CAAC,MAAM,CACnB,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,EAC1D,CAAC,CACF,GAAG,CAAC,CACN,CAAC;QAEF,kBAAkB,CAChB,MAAM,EACN,wBAAwB,CAAC;YACvB,aAAa,EAAE,kBAAkB;YACjC,YAAY,EAAE,aAAa;SAC5B,CAAC,CACH,CAAC;QAEF,IAAI,aAAa,GAAG,kBAAkB,EAAE,CAAC;YACvC,OAAO;gBACL,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE;oBACJ,kBAAkB;oBAClB,aAAa;oBACb,iBAAiB,EACf,iBAAiB,CAAC,uCAAuC;iBAC5D;aACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO;gBACL,UAAU,EAAE,YAAY;gBACxB,IAAI,EAAE;oBACJ,kBAAkB;oBAClB,aAAa;oBACb,iBAAiB,EAAE,iBAAiB,CAAC,UAAU;iBAChD;aACF,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

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

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