@vybestack/llxprt-code-core 0.1.12

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

@@ -0,0 +1,347 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { describe, it, expect, vi } from 'vitest';
+import { CoreToolScheduler, convertToFunctionResponse, } from './coreToolScheduler.js';
+import { BaseTool, ToolConfirmationOutcome, } from '../index.js';
+class MockTool extends BaseTool {
+    shouldConfirm = false;
+    executeFn = vi.fn();
+    constructor(name = 'mockTool') {
+        super(name, name, 'A mock tool', {});
+    }
+    async shouldConfirmExecute(_params, _abortSignal) {
+        if (this.shouldConfirm) {
+            return {
+                type: 'exec',
+                title: 'Confirm Mock Tool',
+                command: 'do_thing',
+                rootCommand: 'do_thing',
+                onConfirm: async () => { },
+            };
+        }
+        return false;
+    }
+    async execute(params, _abortSignal) {
+        this.executeFn(params);
+        return { llmContent: 'Tool executed', returnDisplay: 'Tool executed' };
+    }
+}
+class MockModifiableTool extends MockTool {
+    constructor(name = 'mockModifiableTool') {
+        super(name);
+        this.shouldConfirm = true;
+    }
+    getModifyContext(_abortSignal) {
+        return {
+            getFilePath: () => 'test.txt',
+            getCurrentContent: async () => 'old content',
+            getProposedContent: async () => 'new content',
+            createUpdatedParams: (_oldContent, modifiedProposedContent, _originalParams) => ({ newContent: modifiedProposedContent }),
+        };
+    }
+    async shouldConfirmExecute(_params, _abortSignal) {
+        if (this.shouldConfirm) {
+            return {
+                type: 'edit',
+                title: 'Confirm Mock Tool',
+                fileName: 'test.txt',
+                fileDiff: 'diff',
+                onConfirm: async () => { },
+            };
+        }
+        return false;
+    }
+}
+describe('CoreToolScheduler', () => {
+    it('should cancel a tool call if the signal is aborted before confirmation', async () => {
+        const mockTool = new MockTool();
+        mockTool.shouldConfirm = true;
+        const toolRegistry = {
+            getTool: () => mockTool,
+            getFunctionDeclarations: () => [],
+            tools: new Map(),
+            discovery: {},
+            registerTool: () => { },
+            getToolByName: () => mockTool,
+            getToolByDisplayName: () => mockTool,
+            getTools: () => [],
+            discoverTools: async () => { },
+            getAllTools: () => [],
+            getToolsByServer: () => [],
+        };
+        const onAllToolCallsComplete = vi.fn();
+        const onToolCallsUpdate = vi.fn();
+        const mockConfig = {
+            getSessionId: () => 'test-session-id',
+            getUsageStatisticsEnabled: () => true,
+            getDebugMode: () => false,
+        };
+        const scheduler = new CoreToolScheduler({
+            config: mockConfig,
+            toolRegistry: Promise.resolve(toolRegistry),
+            onAllToolCallsComplete,
+            onToolCallsUpdate,
+            getPreferredEditor: () => 'vscode',
+        });
+        const abortController = new AbortController();
+        const request = {
+            callId: '1',
+            name: 'mockTool',
+            args: {},
+            isClientInitiated: false,
+            prompt_id: 'prompt-id-1',
+        };
+        abortController.abort();
+        await scheduler.schedule([request], abortController.signal);
+        const _waitingCall = onToolCallsUpdate.mock
+            .calls[1][0][0];
+        const confirmationDetails = await mockTool.shouldConfirmExecute({}, abortController.signal);
+        if (confirmationDetails) {
+            await scheduler.handleConfirmationResponse('1', confirmationDetails.onConfirm, ToolConfirmationOutcome.ProceedOnce, abortController.signal);
+        }
+        expect(onAllToolCallsComplete).toHaveBeenCalled();
+        const completedCalls = onAllToolCallsComplete.mock
+            .calls[0][0];
+        expect(completedCalls[0].status).toBe('cancelled');
+    });
+});
+describe('CoreToolScheduler with payload', () => {
+    it('should update args and diff and execute tool when payload is provided', async () => {
+        const mockTool = new MockModifiableTool();
+        const toolRegistry = {
+            getTool: () => mockTool,
+            getFunctionDeclarations: () => [],
+            tools: new Map(),
+            discovery: {},
+            registerTool: () => { },
+            getToolByName: () => mockTool,
+            getToolByDisplayName: () => mockTool,
+            getTools: () => [],
+            discoverTools: async () => { },
+            getAllTools: () => [],
+            getToolsByServer: () => [],
+        };
+        const onAllToolCallsComplete = vi.fn();
+        const onToolCallsUpdate = vi.fn();
+        const mockConfig = {
+            getSessionId: () => 'test-session-id',
+            getUsageStatisticsEnabled: () => true,
+            getDebugMode: () => false,
+        };
+        const scheduler = new CoreToolScheduler({
+            config: mockConfig,
+            toolRegistry: Promise.resolve(toolRegistry),
+            onAllToolCallsComplete,
+            onToolCallsUpdate,
+            getPreferredEditor: () => 'vscode',
+        });
+        const abortController = new AbortController();
+        const request = {
+            callId: '1',
+            name: 'mockModifiableTool',
+            args: {},
+            isClientInitiated: false,
+            prompt_id: 'prompt-id-2',
+        };
+        await scheduler.schedule([request], abortController.signal);
+        const confirmationDetails = await mockTool.shouldConfirmExecute({}, abortController.signal);
+        if (confirmationDetails) {
+            const payload = { newContent: 'final version' };
+            await scheduler.handleConfirmationResponse('1', confirmationDetails.onConfirm, ToolConfirmationOutcome.ProceedOnce, abortController.signal, payload);
+        }
+        expect(onAllToolCallsComplete).toHaveBeenCalled();
+        const completedCalls = onAllToolCallsComplete.mock
+            .calls[0][0];
+        expect(completedCalls[0].status).toBe('success');
+        expect(mockTool.executeFn).toHaveBeenCalledWith({
+            newContent: 'final version',
+        });
+    });
+});
+describe('convertToFunctionResponse', () => {
+    const toolName = 'testTool';
+    const callId = 'call1';
+    it('should handle simple string llmContent', () => {
+        const llmContent = 'Simple text output';
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Simple text output' },
+            },
+        });
+    });
+    it('should handle llmContent as a single Part with text', () => {
+        const llmContent = { text: 'Text from Part object' };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Text from Part object' },
+            },
+        });
+    });
+    it('should handle llmContent as a PartListUnion array with a single text Part', () => {
+        const llmContent = [{ text: 'Text from array' }];
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Text from array' },
+            },
+        });
+    });
+    it('should handle llmContent with inlineData', () => {
+        const llmContent = {
+            inlineData: { mimeType: 'image/png', data: 'base64...' },
+        };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: {
+                    output: 'Binary content of type image/png was processed.',
+                    binaryContent: llmContent,
+                },
+            },
+        });
+    });
+    it('should handle llmContent with fileData', () => {
+        const llmContent = {
+            fileData: { mimeType: 'application/pdf', fileUri: 'gs://...' },
+        };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: {
+                    output: 'Binary content of type application/pdf was processed.',
+                    binaryContent: llmContent,
+                },
+            },
+        });
+    });
+    it('should handle llmContent as an array of multiple Parts (text and inlineData)', () => {
+        const llmContent = [
+            { text: 'Some textual description' },
+            { inlineData: { mimeType: 'image/jpeg', data: 'base64data...' } },
+            { text: 'Another text part' },
+        ];
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual([
+            {
+                functionResponse: {
+                    name: toolName,
+                    id: callId,
+                    response: { output: 'Tool execution succeeded.' },
+                },
+            },
+            ...llmContent,
+        ]);
+    });
+    it('should handle llmContent as an array with a single inlineData Part', () => {
+        const llmContent = [
+            { inlineData: { mimeType: 'image/gif', data: 'gifdata...' } },
+        ];
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: {
+                    output: 'Binary content of type image/gif was processed.',
+                    binaryContent: llmContent[0],
+                },
+            },
+        });
+    });
+    it('should handle llmContent as a generic Part (not text, inlineData, or fileData)', () => {
+        const llmContent = { functionCall: { name: 'test', args: {} } };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Tool execution succeeded.' },
+            },
+        });
+    });
+    it('should handle empty string llmContent', () => {
+        const llmContent = '';
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: '' },
+            },
+        });
+    });
+    it('should handle llmContent as an empty array', () => {
+        const llmContent = [];
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual([
+            {
+                functionResponse: {
+                    name: toolName,
+                    id: callId,
+                    response: { output: 'Tool execution succeeded.' },
+                },
+            },
+        ]);
+    });
+    it('should handle llmContent as a Part with undefined inlineData/fileData/text', () => {
+        const llmContent = {}; // An empty part object
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Tool execution succeeded.' },
+            },
+        });
+    });
+    it('should ensure correct id when llmContent contains functionResponse without id', () => {
+        const llmContent = {
+            functionResponse: {
+                name: 'originalTool',
+                response: { output: 'Tool completed successfully' },
+            },
+        };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                name: toolName,
+                id: callId,
+                response: { output: 'Tool completed successfully' },
+            },
+        });
+    });
+    it('should override id when llmContent contains functionResponse with different id', () => {
+        const llmContent = {
+            functionResponse: {
+                id: 'wrong_id',
+                name: 'originalTool',
+                response: { output: 'Tool completed successfully' },
+            },
+        };
+        const result = convertToFunctionResponse(toolName, callId, llmContent);
+        expect(result).toEqual({
+            functionResponse: {
+                id: callId, // Should use the provided callId, not 'wrong_id'
+                name: toolName, // Should use the provided toolName
+                response: { output: 'Tool completed successfully' },
+            },
+        });
+    });
+});
+//# sourceMappingURL=coreToolScheduler.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"coreToolScheduler.test.js","sourceRoot":"","sources":["../../../src/core/coreToolScheduler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,uDAAuD;AACvD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EACL,iBAAiB,EAGjB,yBAAyB,GAC1B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,QAAQ,EAER,uBAAuB,GAIxB,MAAM,aAAa,CAAC;AAKrB,MAAM,QAAS,SAAQ,QAA6C;IAClE,aAAa,GAAG,KAAK,CAAC;IACtB,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAEpB,YAAY,IAAI,GAAG,UAAU;QAC3B,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAAgC,EAChC,YAAyB;QAEzB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,mBAAmB;gBAC1B,OAAO,EAAE,UAAU;gBACnB,WAAW,EAAE,UAAU;gBACvB,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;aAC1B,CAAC;QACJ,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAA+B,EAC/B,YAAyB;QAEzB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvB,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC;IACzE,CAAC;CACF;AAED,MAAM,kBACJ,SAAQ,QAAQ;IAGhB,YAAY,IAAI,GAAG,oBAAoB;QACrC,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,gBAAgB,CACd,YAAyB;QAEzB,OAAO;YACL,WAAW,EAAE,GAAG,EAAE,CAAC,UAAU;YAC7B,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;YAC5C,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;YAC7C,mBAAmB,EAAE,CACnB,WAAmB,EACnB,uBAA+B,EAC/B,eAAwC,EACxC,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC;SAC/C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAAgC,EAChC,YAAyB;QAEzB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,mBAAmB;gBAC1B,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,MAAM;gBAChB,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;aAC1B,CAAC;QACJ,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;QAChC,QAAQ,CAAC,aAAa,GAAG,IAAI,CAAC;QAC9B,MAAM,YAAY,GAAG;YACnB,OAAO,EAAE,GAAG,EAAE,CAAC,QAAQ;YACvB,uBAAuB,EAAE,GAAG,EAAE,CAAC,EAAE;YACjC,KAAK,EAAE,IAAI,GAAG,EAAE;YAChB,SAAS,EAAE,EAAS;YACpB,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;YACtB,aAAa,EAAE,GAAG,EAAE,CAAC,QAAQ;YAC7B,oBAAoB,EAAE,GAAG,EAAE,CAAC,QAAQ;YACpC,QAAQ,EAAE,GAAG,EAAE,CAAC,EAAE;YAClB,aAAa,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;YAC7B,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE;YACrB,gBAAgB,EAAE,GAAG,EAAE,CAAC,EAAE;SAC3B,CAAC;QAEF,MAAM,sBAAsB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAElC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;YACrC,yBAAyB,EAAE,GAAG,EAAE,CAAC,IAAI;YACrC,YAAY,EAAE,GAAG,EAAE,CAAC,KAAK;SACL,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC;YACtC,MAAM,EAAE,UAAU;YAClB,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,YAAmB,CAAC;YAClD,sBAAsB;YACtB,iBAAiB;YACjB,kBAAkB,EAAE,GAAG,EAAE,CAAC,QAAQ;SACnC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,EAAE;YACR,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,aAAa;SACzB,CAAC;QAEF,eAAe,CAAC,KAAK,EAAE,CAAC;QACxB,MAAM,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QAE5D,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI;aACxC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAuB,CAAC;QACxC,MAAM,mBAAmB,GAAG,MAAM,QAAQ,CAAC,oBAAoB,CAC7D,EAAE,EACF,eAAe,CAAC,MAAM,CACvB,CAAC;QACF,IAAI,mBAAmB,EAAE,CAAC;YACxB,MAAM,SAAS,CAAC,0BAA0B,CACxC,GAAG,EACH,mBAAmB,CAAC,SAAS,EAC7B,uBAAuB,CAAC,WAAW,EACnC,eAAe,CAAC,MAAM,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,sBAAsB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,cAAc,GAAG,sBAAsB,CAAC,IAAI;aAC/C,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAe,CAAC;QAC7B,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAG;YACnB,OAAO,EAAE,GAAG,EAAE,CAAC,QAAQ;YACvB,uBAAuB,EAAE,GAAG,EAAE,CAAC,EAAE;YACjC,KAAK,EAAE,IAAI,GAAG,EAAE;YAChB,SAAS,EAAE,EAAS;YACpB,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;YACtB,aAAa,EAAE,GAAG,EAAE,CAAC,QAAQ;YAC7B,oBAAoB,EAAE,GAAG,EAAE,CAAC,QAAQ;YACpC,QAAQ,EAAE,GAAG,EAAE,CAAC,EAAE;YAClB,aAAa,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;YAC7B,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE;YACrB,gBAAgB,EAAE,GAAG,EAAE,CAAC,EAAE;SAC3B,CAAC;QAEF,MAAM,sBAAsB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAElC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;YACrC,yBAAyB,EAAE,GAAG,EAAE,CAAC,IAAI;YACrC,YAAY,EAAE,GAAG,EAAE,CAAC,KAAK;SACL,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC;YACtC,MAAM,EAAE,UAAU;YAClB,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,YAAmB,CAAC;YAClD,sBAAsB;YACtB,iBAAiB;YACjB,kBAAkB,EAAE,GAAG,EAAE,CAAC,QAAQ;SACnC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,oBAAoB;YAC1B,IAAI,EAAE,EAAE;YACR,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,aAAa;SACzB,CAAC;QAEF,MAAM,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QAE5D,MAAM,mBAAmB,GAAG,MAAM,QAAQ,CAAC,oBAAoB,CAC7D,EAAE,EACF,eAAe,CAAC,MAAM,CACvB,CAAC;QAEF,IAAI,mBAAmB,EAAE,CAAC;YACxB,MAAM,OAAO,GAA4B,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC;YACzE,MAAM,SAAS,CAAC,0BAA0B,CACxC,GAAG,EACH,mBAAmB,CAAC,SAAS,EAC7B,uBAAuB,CAAC,WAAW,EACnC,eAAe,CAAC,MAAM,EACtB,OAAO,CACR,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,sBAAsB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,cAAc,GAAG,sBAAsB,CAAC,IAAI;aAC/C,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAe,CAAC;QAC7B,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC;YAC9C,UAAU,EAAE,eAAe;SAC5B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,MAAM,QAAQ,GAAG,UAAU,CAAC;IAC5B,MAAM,MAAM,GAAG,OAAO,CAAC;IAEvB,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,UAAU,GAAG,oBAAoB,CAAC;QACxC,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE;aAC3C;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,UAAU,GAAS,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC;QAC3D,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,uBAAuB,EAAE;aAC9C;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;QACnF,MAAM,UAAU,GAAkB,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE;aACxC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAS;YACvB,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE;SACzD,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE;oBACR,MAAM,EAAE,iDAAiD;oBACzD,aAAa,EAAE,UAAU;iBAC1B;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,UAAU,GAAS;YACvB,QAAQ,EAAE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,OAAO,EAAE,UAAU,EAAE;SAC/D,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE;oBACR,MAAM,EAAE,uDAAuD;oBAC/D,aAAa,EAAE,UAAU;iBAC1B;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,MAAM,UAAU,GAAkB;YAChC,EAAE,IAAI,EAAE,0BAA0B,EAAE;YACpC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE;YACjE,EAAE,IAAI,EAAE,mBAAmB,EAAE;SAC9B,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,gBAAgB,EAAE;oBAChB,IAAI,EAAE,QAAQ;oBACd,EAAE,EAAE,MAAM;oBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,2BAA2B,EAAE;iBAClD;aACF;YACD,GAAG,UAAU;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,MAAM,UAAU,GAAkB;YAChC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;SAC9D,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE;oBACR,MAAM,EAAE,iDAAiD;oBACzD,aAAa,EAAE,UAAU,CAAC,CAAC,CAAC;iBAC7B;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,UAAU,GAAS,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;QACtE,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,2BAA2B,EAAE;aAClD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,UAAU,GAAG,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;aACzB;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,UAAU,GAAkB,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB;gBACE,gBAAgB,EAAE;oBAChB,IAAI,EAAE,QAAQ;oBACd,EAAE,EAAE,MAAM;oBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,2BAA2B,EAAE;iBAClD;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;QACpF,MAAM,UAAU,GAAS,EAAE,CAAC,CAAC,uBAAuB;QACpD,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,2BAA2B,EAAE;aAClD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;QACvF,MAAM,UAAU,GAAS;YACvB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE;aACpD;SACF,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE;aACpD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,UAAU,GAAS;YACvB,gBAAgB,EAAE;gBAChB,EAAE,EAAE,UAAU;gBACd,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE;aACpD;SACF,CAAC;QACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,gBAAgB,EAAE;gBAChB,EAAE,EAAE,MAAM,EAAE,iDAAiD;gBAC7D,IAAI,EAAE,QAAQ,EAAE,mCAAmC;gBACnD,QAAQ,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE;aACpD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,116 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GenerateContentResponse, Content, GenerateContentConfig, SendMessageParameters, GenerateContentResponseUsageMetadata } from '@google/genai';
+import { ContentGenerator } from './contentGenerator.js';
+import { Config } from '../config/config.js';
+/**
+ * Chat session that enables sending messages to the model with previous
+ * conversation context.
+ *
+ * @remarks
+ * The session maintains all the turns between user and model.
+ */
+export declare class GeminiChat {
+    private readonly config;
+    private readonly contentGenerator;
+    private readonly generationConfig;
+    private history;
+    private sendPromise;
+    constructor(config: Config, contentGenerator: ContentGenerator, generationConfig?: GenerateContentConfig, history?: Content[]);
+    private _getRequestTextFromContents;
+    private _logApiRequest;
+    private _logApiResponse;
+    private _logApiError;
+    /**
+     * Handles fallback to Flash model when persistent 429 errors occur for OAuth users.
+     * Uses a fallback handler if provided by the config, otherwise returns null.
+     */
+    private handleFlashFallback;
+    /**
+     * Sends a message to the model and returns the response.
+     *
+     * @remarks
+     * This method will wait for the previous message to be processed before
+     * sending the next message.
+     *
+     * @see {@link Chat#sendMessageStream} for streaming method.
+     * @param params - parameters for sending messages within a chat session.
+     * @returns The model's response.
+     *
+     * @example
+     * ```ts
+     * const chat = ai.chats.create({model: 'gemini-2.0-flash'});
+     * const response = await chat.sendMessage({
+     *   message: 'Why is the sky blue?'
+     * });
+     * console.log(response.text);
+     * ```
+     */
+    sendMessage(params: SendMessageParameters, prompt_id: string): Promise<GenerateContentResponse>;
+    /**
+     * Sends a message to the model and returns the response in chunks.
+     *
+     * @remarks
+     * This method will wait for the previous message to be processed before
+     * sending the next message.
+     *
+     * @see {@link Chat#sendMessage} for non-streaming method.
+     * @param params - parameters for sending the message.
+     * @return The model's response.
+     *
+     * @example
+     * ```ts
+     * const chat = ai.chats.create({model: 'gemini-2.0-flash'});
+     * const response = await chat.sendMessageStream({
+     *   message: 'Why is the sky blue?'
+     * });
+     * for await (const chunk of response) {
+     *   console.log(chunk.text);
+     * }
+     * ```
+     */
+    sendMessageStream(params: SendMessageParameters, prompt_id: string): Promise<AsyncGenerator<GenerateContentResponse>>;
+    /**
+     * Returns the chat history.
+     *
+     * @remarks
+     * The history is a list of contents alternating between user and model.
+     *
+     * There are two types of history:
+     * - The `curated history` contains only the valid turns between user and
+     * model, which will be included in the subsequent requests sent to the model.
+     * - The `comprehensive history` contains all turns, including invalid or
+     *   empty model outputs, providing a complete record of the history.
+     *
+     * The history is updated after receiving the response from the model,
+     * for streaming response, it means receiving the last chunk of the response.
+     *
+     * The `comprehensive history` is returned by default. To get the `curated
+     * history`, set the `curated` parameter to `true`.
+     *
+     * @param curated - whether to return the curated history or the comprehensive
+     *     history.
+     * @return History contents alternating between user and model for the entire
+     *     chat session.
+     */
+    getHistory(curated?: boolean): Content[];
+    /**
+     * Clears the chat history.
+     */
+    clearHistory(): void;
+    /**
+     * Adds a new entry to the chat history.
+     *
+     * @param content - The content to add to the history.
+     */
+    addHistory(content: Content): void;
+    setHistory(history: Content[]): void;
+    getFinalUsageMetadata(chunks: GenerateContentResponse[]): GenerateContentResponseUsageMetadata | undefined;
+    private processStreamResponse;
+    private recordHistory;
+    private isTextContent;
+    private isThoughtContent;
+}