@hibanacloud/core 0.3.15

package/dist/src/core/baseLlmClient.test.d.ts

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

package/dist/src/core/baseLlmClient.test.js

@@ -0,0 +1,323 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { BaseLlmClient } from './baseLlmClient.js';
+import { AuthType } from './contentGenerator.js';
+import { reportError } from '../utils/errorReporting.js';
+import { retryWithBackoff } from '../utils/retry.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { getFunctionCalls } from '../utils/generateContentResponseUtilities.js';
+vi.mock('../utils/errorReporting.js');
+vi.mock('../utils/errors.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getErrorMessage: vi.fn((e) => (e instanceof Error ? e.message : String(e))),
+    };
+});
+vi.mock('../utils/generateContentResponseUtilities.js', () => ({
+    getFunctionCalls: vi.fn(),
+}));
+vi.mock('../utils/retry.js', () => ({
+    retryWithBackoff: vi.fn(async (fn) => await fn()),
+}));
+const mockGenerateContent = vi.fn();
+const mockEmbedContent = vi.fn();
+const mockContentGenerator = {
+    generateContent: mockGenerateContent,
+    embedContent: mockEmbedContent,
+};
+const mockConfig = {
+    getSessionId: vi.fn().mockReturnValue('test-session-id'),
+    getContentGeneratorConfig: vi
+        .fn()
+        .mockReturnValue({ authType: AuthType.USE_GEMINI }),
+    getEmbeddingModel: vi.fn().mockReturnValue('test-embedding-model'),
+};
+// Helper to create a mock GenerateContentResponse with function call
+const createMockResponseWithFunctionCall = (args) => ({
+    candidates: [
+        {
+            content: {
+                role: 'model',
+                parts: [
+                    {
+                        functionCall: {
+                            name: 'respond_in_schema',
+                            args,
+                        },
+                    },
+                ],
+            },
+            index: 0,
+        },
+    ],
+});
+// Helper to create a mock response without function call (for error cases)
+const createMockResponseWithoutFunctionCall = () => ({
+    candidates: [
+        {
+            content: {
+                role: 'model',
+                parts: [{ text: 'some text' }],
+            },
+            index: 0,
+        },
+    ],
+});
+describe('BaseLlmClient', () => {
+    let client;
+    let abortController;
+    let defaultOptions;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        // Reset the mocked implementation for getErrorMessage for accurate error message assertions
+        vi.mocked(getErrorMessage).mockImplementation((e) => e instanceof Error ? e.message : String(e));
+        client = new BaseLlmClient(mockContentGenerator, mockConfig);
+        abortController = new AbortController();
+        defaultOptions = {
+            contents: [{ role: 'user', parts: [{ text: 'Give me a color.' }] }],
+            schema: { type: 'object', properties: { color: { type: 'string' } } },
+            model: 'test-model',
+            abortSignal: abortController.signal,
+            promptId: 'test-prompt-id',
+        };
+    });
+    afterEach(() => {
+        abortController.abort();
+    });
+    describe('generateJson - Success Scenarios', () => {
+        it('should call generateContent with correct parameters using function declarations', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({
+                color: 'blue',
+            });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'blue' } },
+            ]);
+            const result = await client.generateJson(defaultOptions);
+            expect(result).toEqual({ color: 'blue' });
+            // Ensure the retry mechanism was engaged
+            expect(retryWithBackoff).toHaveBeenCalledTimes(1);
+            expect(retryWithBackoff).toHaveBeenCalledWith(expect.any(Function), expect.objectContaining({
+                maxAttempts: 5,
+            }));
+            // Validate the parameters passed to the underlying generator
+            expect(mockGenerateContent).toHaveBeenCalledTimes(1);
+            expect(mockGenerateContent).toHaveBeenCalledWith({
+                model: 'test-model',
+                contents: defaultOptions.contents,
+                config: {
+                    abortSignal: defaultOptions.abortSignal,
+                    temperature: 0,
+                    topP: 1,
+                    tools: [
+                        {
+                            functionDeclarations: [
+                                {
+                                    name: 'respond_in_schema',
+                                    description: 'Provide the response in provided schema',
+                                    parameters: defaultOptions.schema,
+                                },
+                            ],
+                        },
+                    ],
+                    // Crucial: systemInstruction should NOT be in the config object if not provided
+                },
+            }, 'test-prompt-id');
+        });
+        it('should respect configuration overrides', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({ color: 'red' });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'red' } },
+            ]);
+            const options = {
+                ...defaultOptions,
+                config: { temperature: 0.8, topK: 10 },
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.objectContaining({
+                config: expect.objectContaining({
+                    temperature: 0.8,
+                    topP: 1, // Default should remain if not overridden
+                    topK: 10,
+                    tools: expect.any(Array),
+                }),
+            }), expect.any(String));
+        });
+        it('should include system instructions when provided', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({
+                color: 'green',
+            });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'green' } },
+            ]);
+            const systemInstruction = 'You are a helpful assistant.';
+            const options = {
+                ...defaultOptions,
+                systemInstruction,
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.objectContaining({
+                config: expect.objectContaining({
+                    systemInstruction,
+                }),
+            }), expect.any(String));
+        });
+        it('should use the provided promptId', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({
+                color: 'yellow',
+            });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'yellow' } },
+            ]);
+            const customPromptId = 'custom-id-123';
+            const options = {
+                ...defaultOptions,
+                promptId: customPromptId,
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.any(Object), customPromptId);
+        });
+        it('should pass maxAttempts to retryWithBackoff when provided', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({
+                color: 'cyan',
+            });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'cyan' } },
+            ]);
+            const customMaxAttempts = 3;
+            const options = {
+                ...defaultOptions,
+                maxAttempts: customMaxAttempts,
+            };
+            await client.generateJson(options);
+            expect(retryWithBackoff).toHaveBeenCalledTimes(1);
+            expect(retryWithBackoff).toHaveBeenCalledWith(expect.any(Function), expect.objectContaining({
+                maxAttempts: customMaxAttempts,
+            }));
+        });
+        it('should call retryWithBackoff with default maxAttempts when not provided', async () => {
+            const mockResponse = createMockResponseWithFunctionCall({
+                color: 'indigo',
+            });
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue([
+                { name: 'respond_in_schema', args: { color: 'indigo' } },
+            ]);
+            // No maxAttempts in defaultOptions
+            await client.generateJson(defaultOptions);
+            expect(retryWithBackoff).toHaveBeenCalledWith(expect.any(Function), expect.objectContaining({
+                maxAttempts: 5,
+            }));
+        });
+        it('should return empty object when no function calls are returned', async () => {
+            const mockResponse = createMockResponseWithoutFunctionCall();
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            vi.mocked(getFunctionCalls).mockReturnValue(undefined);
+            const result = await client.generateJson(defaultOptions);
+            expect(result).toEqual({});
+        });
+    });
+    describe('generateJson - Error Handling', () => {
+        it('should throw and report generic API errors', async () => {
+            const apiError = new Error('Service Unavailable (503)');
+            // Simulate the generator failing
+            mockGenerateContent.mockRejectedValue(apiError);
+            await expect(client.generateJson(defaultOptions)).rejects.toThrow('Failed to generate JSON content: Service Unavailable (503)');
+            // Verify generic error reporting
+            expect(reportError).toHaveBeenCalledTimes(1);
+            expect(reportError).toHaveBeenCalledWith(apiError, 'Error generating JSON content via API.', defaultOptions.contents, 'generateJson-api');
+        });
+        it('should throw immediately without reporting if aborted', async () => {
+            const abortError = new DOMException('Aborted', 'AbortError');
+            // Simulate abortion happening during the API call
+            mockGenerateContent.mockImplementation(() => {
+                abortController.abort(); // Ensure the signal is aborted when the service checks
+                throw abortError;
+            });
+            const options = {
+                ...defaultOptions,
+                abortSignal: abortController.signal,
+            };
+            await expect(client.generateJson(options)).rejects.toThrow(abortError);
+            // Crucially, it should not report a cancellation as an application error
+            expect(reportError).not.toHaveBeenCalled();
+        });
+        it('should not throw for empty response message check', async () => {
+            const emptyResponseError = new Error('API returned an empty response for generateJson.');
+            mockGenerateContent.mockRejectedValue(emptyResponseError);
+            await expect(client.generateJson(defaultOptions)).rejects.toThrow('API returned an empty response for generateJson.');
+            // Should not double-report this specific error
+            expect(reportError).not.toHaveBeenCalled();
+        });
+    });
+    describe('generateEmbedding', () => {
+        const texts = ['hello world', 'goodbye world'];
+        const testEmbeddingModel = 'test-embedding-model';
+        it('should call embedContent with correct parameters and return embeddings', async () => {
+            const mockEmbeddings = [
+                [0.1, 0.2, 0.3],
+                [0.4, 0.5, 0.6],
+            ];
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [
+                    { values: mockEmbeddings[0] },
+                    { values: mockEmbeddings[1] },
+                ],
+            });
+            const result = await client.generateEmbedding(texts);
+            expect(mockEmbedContent).toHaveBeenCalledTimes(1);
+            expect(mockEmbedContent).toHaveBeenCalledWith({
+                model: testEmbeddingModel,
+                contents: texts,
+            });
+            expect(result).toEqual(mockEmbeddings);
+        });
+        it('should return an empty array if an empty array is passed', async () => {
+            const result = await client.generateEmbedding([]);
+            expect(result).toEqual([]);
+            expect(mockEmbedContent).not.toHaveBeenCalled();
+        });
+        it('should throw an error if API response has no embeddings array', async () => {
+            mockEmbedContent.mockResolvedValue({});
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('No embeddings found in API response.');
+        });
+        it('should throw an error if API response has an empty embeddings array', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [],
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('No embeddings found in API response.');
+        });
+        it('should throw an error if API returns a mismatched number of embeddings', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [1, 2, 3] }], // Only one for two texts
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned a mismatched number of embeddings. Expected 2, got 1.');
+        });
+        it('should throw an error if any embedding has nullish values', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [1, 2, 3] }, { values: undefined }], // Second one is bad
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned an empty embedding for input text at index 1: "goodbye world"');
+        });
+        it('should throw an error if any embedding has an empty values array', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [] }, { values: [1, 2, 3] }], // First one is bad
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned an empty embedding for input text at index 0: "hello world"');
+        });
+        it('should propagate errors from the API call', async () => {
+            mockEmbedContent.mockRejectedValue(new Error('API Failure'));
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API Failure');
+        });
+    });
+});
+//# sourceMappingURL=baseLlmClient.test.js.map

package/dist/src/core/baseLlmClient.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"baseLlmClient.test.js","sourceRoot":"","sources":["../../../src/core/baseLlmClient.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAGhB,OAAO,EAAE,aAAa,EAA4B,MAAM,oBAAoB,CAAC;AAG7E,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8CAA8C,CAAC;AAEhF,EAAE,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AACtC,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACrD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAuC,CAAC;IAC3E,OAAO;QACL,GAAG,MAAM;QACT,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;KAC5E,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7D,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACpC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAEjC,MAAM,oBAAoB,GAAG;IAC3B,eAAe,EAAE,mBAAmB;IACpC,YAAY,EAAE,gBAAgB;CACQ,CAAC;AAEzC,MAAM,UAAU,GAAG;IACjB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;IACxD,yBAAyB,EAAE,EAAE;SAC1B,EAAE,EAAE;SACJ,eAAe,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC;IACrD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,sBAAsB,CAAC;CACtC,CAAC;AAE/B,qEAAqE;AACrE,MAAM,kCAAkC,GAAG,CACzC,IAA6B,EACJ,EAAE,CAC3B,CAAC;IACC,UAAU,EAAE;QACV;YACE,OAAO,EAAE;gBACP,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,YAAY,EAAE;4BACZ,IAAI,EAAE,mBAAmB;4BACzB,IAAI;yBACL;qBACF;iBACF;aACF;YACD,KAAK,EAAE,CAAC;SACT;KACF;CACF,CAA4B,CAAC;AAEhC,2EAA2E;AAC3E,MAAM,qCAAqC,GAAG,GAA4B,EAAE,CAC1E,CAAC;IACC,UAAU,EAAE;QACV;YACE,OAAO,EAAE;gBACP,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;aAC/B;YACD,KAAK,EAAE,CAAC;SACT;KACF;CACF,CAA4B,CAAC;AAEhC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,MAAqB,CAAC;IAC1B,IAAI,eAAgC,CAAC;IACrC,IAAI,cAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,4FAA4F;QAC5F,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAClD,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAC3C,CAAC;QACF,MAAM,GAAG,IAAI,aAAa,CAAC,oBAAoB,EAAE,UAAU,CAAC,CAAC;QAC7D,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QACxC,cAAc,GAAG;YACf,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,EAAE,CAAC;YACnE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;YACrE,KAAK,EAAE,YAAY;YACnB,WAAW,EAAE,eAAe,CAAC,MAAM;YACnC,QAAQ,EAAE,gBAAgB;SAC3B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,YAAY,GAAG,kCAAkC,CAAC;gBACtD,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YACH,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;aACvD,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAE1C,yCAAyC;YACzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,CAAC;aACf,CAAC,CACH,CAAC;YAEF,6DAA6D;YAC7D,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACrD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C;gBACE,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,MAAM,EAAE;oBACN,WAAW,EAAE,cAAc,CAAC,WAAW;oBACvC,WAAW,EAAE,CAAC;oBACd,IAAI,EAAE,CAAC;oBACP,KAAK,EAAE;wBACL;4BACE,oBAAoB,EAAE;gCACpB;oCACE,IAAI,EAAE,mBAAmB;oCACzB,WAAW,EAAE,yCAAyC;oCACtD,UAAU,EAAE,cAAc,CAAC,MAAM;iCAClC;6BACF;yBACF;qBACF;oBACD,gFAAgF;iBACjF;aACF,EACD,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,YAAY,GAAG,kCAAkC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;aACtD,CAAC,CAAC;YAEH,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE;aACvC,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC9B,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,CAAC,EAAE,0CAA0C;oBACnD,IAAI,EAAE,EAAE;oBACR,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;iBACzB,CAAC;aACH,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,YAAY,GAAG,kCAAkC,CAAC;gBACtD,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;YACH,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;aACxD,CAAC,CAAC;YACH,MAAM,iBAAiB,GAAG,8BAA8B,CAAC;YAEzD,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,iBAAiB;aAClB,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC9B,iBAAiB;iBAClB,CAAC;aACH,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,YAAY,GAAG,kCAAkC,CAAC;gBACtD,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YACH,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;aACzD,CAAC,CAAC;YACH,MAAM,cAAc,GAAG,eAAe,CAAC;YAEvC,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,QAAQ,EAAE,cAAc;aACzB,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,YAAY,GAAG,kCAAkC,CAAC;gBACtD,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YACH,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;aACvD,CAAC,CAAC;YACH,MAAM,iBAAiB,GAAG,CAAC,CAAC;YAE5B,MAAM,OAAO,GAAwB;gBACnC,GAAG,cAAc;gBACjB,WAAW,EAAE,iBAAiB;aAC/B,CAAC;YAEF,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAEnC,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,iBAAiB;aAC/B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,MAAM,YAAY,GAAG,kCAAkC,CAAC;gBACtD,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YACH,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC1C,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;aACzD,CAAC,CAAC;YAEH,mCAAmC;YACnC,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAE1C,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,CAAC;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,YAAY,GAAG,qCAAqC,EAAE,CAAC;YAC7D,mBAAmB,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACpD,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACxD,iCAAiC;YACjC,mBAAmB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,4DAA4D,CAC7D,CAAC;YAEF,iCAAiC;YACjC,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,QAAQ,EACR,wCAAwC,EACxC,cAAc,CAAC,QAAQ,EACvB,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAE7D,kDAAkD;YAClD,mBAAmB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1C,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC,uDAAuD;gBAChF,MAAM,UAAU,CAAC;YACnB,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG;gBACd,GAAG,cAAc;gBACjB,WAAW,EAAE,eAAe,CAAC,MAAM;aACpC,CAAC;YAEF,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAEvE,yEAAyE;YACzE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,kBAAkB,GAAG,IAAI,KAAK,CAClC,kDAAkD,CACnD,CAAC;YACF,mBAAmB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAE1D,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,kDAAkD,CACnD,CAAC;YAEF,+CAA+C;YAC/C,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,MAAM,KAAK,GAAG,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC/C,MAAM,kBAAkB,GAAG,sBAAsB,CAAC;QAElD,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,cAAc,GAAG;gBACrB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBACf,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;aAChB,CAAC;YACF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE;oBACV,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAC7B,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAErD,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,kBAAkB;gBACzB,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,sCAAsC,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,sCAAsC,CACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,yBAAyB;aAC/D,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,oEAAoE,CACrE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,oBAAoB;aACjF,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,4EAA4E,CAC7E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,gBAAgB,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,mBAAmB;aACzE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,0EAA0E,CAC3E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,aAAa,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/client.d.ts

@@ -0,0 +1,56 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content, GenerateContentConfig, GenerateContentResponse, PartListUnion } from '@google/genai';
+import { type Config } from '../config/config.js';
+import { GeminiChat } from './geminiChat.js';
+import { Turn, type ChatCompressionInfo, type ServerGeminiStreamEvent } from './turn.js';
+import { type ChatRecordingService } from '../services/chatRecordingService.js';
+import { LoopDetectionService } from '../services/loopDetectionService.js';
+export declare function isThinkingSupported(model: string): boolean;
+export declare function isThinkingDefault(model: string): boolean;
+export declare class GeminiClient {
+    private readonly config;
+    private chat?;
+    private readonly generateContentConfig;
+    private sessionTurnCount;
+    private readonly loopDetector;
+    private lastPromptId;
+    private lastSentIdeContext;
+    private forceFullIdeContext;
+    /**
+     * At any point in this conversation, was compression triggered without
+     * being forced and did it fail?
+     */
+    private hasFailedCompressionAttempt;
+    constructor(config: Config);
+    initialize(): Promise<void>;
+    private getContentGeneratorOrFail;
+    addHistory(content: Content): Promise<void>;
+    getChat(): GeminiChat;
+    isInitialized(): boolean;
+    getHistory(): Content[];
+    stripThoughtsFromHistory(): void;
+    setHistory(history: Content[]): void;
+    setTools(): Promise<void>;
+    resetChat(): Promise<void>;
+    /**
+     * Recreate the chat with a new system instruction while preserving history and tools.
+     * Used when the system prompt changes mid-session.
+     */
+    resetChatWithSystemInstruction(sysInstr: string): Promise<void>;
+    getChatRecordingService(): ChatRecordingService | undefined;
+    getLoopDetectionService(): LoopDetectionService;
+    addDirectoryContext(): Promise<void>;
+    startChat(extraHistory?: Content[]): Promise<GeminiChat>;
+    private getIdeContextParts;
+    sendMessageStream(request: PartListUnion, signal: AbortSignal, prompt_id: string, turns?: number): AsyncGenerator<ServerGeminiStreamEvent, Turn>;
+    generateContent(contents: Content[], generationConfig: GenerateContentConfig, abortSignal: AbortSignal, model: string): Promise<GenerateContentResponse>;
+    tryCompressChat(prompt_id: string, force?: boolean): Promise<ChatCompressionInfo>;
+}
+export declare const TEST_ONLY: {
+    COMPRESSION_PRESERVE_THRESHOLD: number;
+    COMPRESSION_TOKEN_THRESHOLD: number;
+};