@hibanacloud/core 0.3.15

package/dist/src/utils/openaiLogger.test.js

@@ -0,0 +1,304 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import * as path from 'node:path';
+import * as os from 'os';
+import { promises as fs } from 'node:fs';
+import { OpenAILogger } from './openaiLogger.js';
+describe('OpenAILogger', () => {
+    let originalCwd;
+    let testTempDir;
+    const createdDirs = [];
+    beforeEach(() => {
+        originalCwd = process.cwd();
+        testTempDir = path.join(os.tmpdir(), `openai-logger-test-${Date.now()}`);
+        createdDirs.length = 0; // Clear array
+    });
+    afterEach(async () => {
+        // Clean up all created directories
+        const cleanupPromises = [
+            testTempDir,
+            ...createdDirs,
+            path.resolve(process.cwd(), 'relative-logs'),
+            path.resolve(process.cwd(), 'custom-logs'),
+            path.resolve(process.cwd(), 'test-relative-logs'),
+            path.join(os.homedir(), 'custom-logs'),
+            path.join(os.homedir(), 'test-openai-logs'),
+        ].map(async (dir) => {
+            try {
+                await fs.rm(dir, { recursive: true, force: true });
+            }
+            catch {
+                // Ignore cleanup errors
+            }
+        });
+        await Promise.all(cleanupPromises);
+        process.chdir(originalCwd);
+    });
+    describe('constructor', () => {
+        it('should use default directory when no custom directory is provided', () => {
+            const logger = new OpenAILogger();
+            // We can't directly access private logDir, but we can verify behavior
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should accept absolute path as custom directory', () => {
+            const customDir = '/absolute/path/to/logs';
+            const logger = new OpenAILogger(customDir);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should resolve relative path to absolute path', async () => {
+            const relativeDir = 'custom-logs';
+            const logger = new OpenAILogger(relativeDir);
+            const expectedDir = path.resolve(process.cwd(), relativeDir);
+            createdDirs.push(expectedDir);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should expand ~ to home directory', () => {
+            const customDir = '~/custom-logs';
+            const logger = new OpenAILogger(customDir);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should expand ~/ to home directory', () => {
+            const customDir = '~/custom-logs';
+            const logger = new OpenAILogger(customDir);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should handle just ~ as home directory', () => {
+            const customDir = '~';
+            const logger = new OpenAILogger(customDir);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+    });
+    describe('initialize', () => {
+        it('should create directory if it does not exist', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const dirExists = await fs
+                .access(testTempDir)
+                .then(() => true)
+                .catch(() => false);
+            expect(dirExists).toBe(true);
+        });
+        it('should create nested directories recursively', async () => {
+            const nestedDir = path.join(testTempDir, 'nested', 'deep', 'path');
+            const logger = new OpenAILogger(nestedDir);
+            await logger.initialize();
+            const dirExists = await fs
+                .access(nestedDir)
+                .then(() => true)
+                .catch(() => false);
+            expect(dirExists).toBe(true);
+        });
+        it('should not throw if directory already exists', async () => {
+            await fs.mkdir(testTempDir, { recursive: true });
+            const logger = new OpenAILogger(testTempDir);
+            await expect(logger.initialize()).resolves.not.toThrow();
+        });
+    });
+    describe('logInteraction', () => {
+        it('should create log file with correct format', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            expect(logPath).toContain(testTempDir);
+            expect(logPath).toMatch(/openai-\d{4}-\d{2}-\d{2}T\d{2}-\d{2}-\d{2}\.\d{3}Z-[a-f0-9]{8}\.json/);
+            const fileExists = await fs
+                .access(logPath)
+                .then(() => true)
+                .catch(() => false);
+            expect(fileExists).toBe(true);
+        });
+        it('should write correct log data structure', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            const logContent = JSON.parse(await fs.readFile(logPath, 'utf-8'));
+            expect(logContent).toHaveProperty('timestamp');
+            expect(logContent).toHaveProperty('request', request);
+            expect(logContent).toHaveProperty('response', response);
+            expect(logContent).toHaveProperty('error', null);
+            expect(logContent).toHaveProperty('system');
+            expect(logContent.system).toHaveProperty('hostname');
+            expect(logContent.system).toHaveProperty('platform');
+            expect(logContent.system).toHaveProperty('release');
+            expect(logContent.system).toHaveProperty('nodeVersion');
+        });
+        it('should log error when provided', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const error = new Error('Test error');
+            const logPath = await logger.logInteraction(request, undefined, error);
+            const logContent = JSON.parse(await fs.readFile(logPath, 'utf-8'));
+            expect(logContent).toHaveProperty('error');
+            expect(logContent.error).toHaveProperty('message', 'Test error');
+            expect(logContent.error).toHaveProperty('stack');
+            expect(logContent.response).toBeNull();
+        });
+        it('should use custom directory when provided', async () => {
+            const customDir = path.join(testTempDir, 'custom-logs');
+            const logger = new OpenAILogger(customDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            expect(logPath).toContain(customDir);
+            expect(logPath.startsWith(customDir)).toBe(true);
+        });
+        it('should resolve relative path correctly', async () => {
+            const relativeDir = 'relative-logs';
+            const logger = new OpenAILogger(relativeDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            const expectedDir = path.resolve(process.cwd(), relativeDir);
+            createdDirs.push(expectedDir);
+            expect(logPath).toContain(expectedDir);
+        });
+        it('should expand ~ correctly', async () => {
+            const customDir = '~/test-openai-logs';
+            const logger = new OpenAILogger(customDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            const expectedDir = path.join(os.homedir(), 'test-openai-logs');
+            createdDirs.push(expectedDir);
+            expect(logPath).toContain(expectedDir);
+        });
+    });
+    describe('getLogFiles', () => {
+        it('should return empty array when directory does not exist', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            const files = await logger.getLogFiles();
+            expect(files).toEqual([]);
+        });
+        it('should return log files after initialization', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            await logger.logInteraction(request, response);
+            const files = await logger.getLogFiles();
+            expect(files.length).toBeGreaterThan(0);
+            expect(files[0]).toMatch(/openai-.*\.json$/);
+        });
+        it('should return only log files matching pattern', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            // Create a log file
+            await logger.logInteraction({ test: 'request' }, { test: 'response' });
+            // Create a non-log file
+            await fs.writeFile(path.join(testTempDir, 'other-file.txt'), 'content');
+            const files = await logger.getLogFiles();
+            expect(files.length).toBe(1);
+            expect(files[0]).toMatch(/openai-.*\.json$/);
+        });
+        it('should respect limit parameter', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            // Create multiple log files
+            for (let i = 0; i < 5; i++) {
+                await logger.logInteraction({ test: `request-${i}` }, { test: `response-${i}` });
+                // Small delay to ensure different timestamps
+                await new Promise((resolve) => setTimeout(resolve, 10));
+            }
+            const allFiles = await logger.getLogFiles();
+            expect(allFiles.length).toBe(5);
+            const limitedFiles = await logger.getLogFiles(3);
+            expect(limitedFiles.length).toBe(3);
+        });
+        it('should return files sorted by most recent first', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const files = [];
+            for (let i = 0; i < 3; i++) {
+                const logPath = await logger.logInteraction({ test: `request-${i}` }, { test: `response-${i}` });
+                files.push(logPath);
+                await new Promise((resolve) => setTimeout(resolve, 10));
+            }
+            const retrievedFiles = await logger.getLogFiles();
+            expect(retrievedFiles[0]).toBe(files[2]); // Most recent first
+            expect(retrievedFiles[1]).toBe(files[1]);
+            expect(retrievedFiles[2]).toBe(files[0]);
+        });
+    });
+    describe('readLogFile', () => {
+        it('should read and parse log file correctly', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            await logger.initialize();
+            const request = {
+                model: 'gpt-4',
+                messages: [{ role: 'user', content: 'test' }],
+            };
+            const response = { id: 'test-id', choices: [] };
+            const logPath = await logger.logInteraction(request, response);
+            const logData = await logger.readLogFile(logPath);
+            expect(logData).toHaveProperty('timestamp');
+            expect(logData).toHaveProperty('request', request);
+            expect(logData).toHaveProperty('response', response);
+        });
+        it('should throw error when file does not exist', async () => {
+            const logger = new OpenAILogger(testTempDir);
+            const nonExistentPath = path.join(testTempDir, 'non-existent.json');
+            await expect(logger.readLogFile(nonExistentPath)).rejects.toThrow();
+        });
+    });
+    describe('path resolution', () => {
+        it('should normalize absolute paths', () => {
+            const absolutePath = '/tmp/test/logs';
+            const logger = new OpenAILogger(absolutePath);
+            expect(logger).toBeInstanceOf(OpenAILogger);
+        });
+        it('should resolve relative paths based on current working directory', async () => {
+            const relativePath = 'test-relative-logs';
+            const logger = new OpenAILogger(relativePath);
+            await logger.initialize();
+            const request = { test: 'request' };
+            const response = { test: 'response' };
+            const logPath = await logger.logInteraction(request, response);
+            const expectedBaseDir = path.resolve(process.cwd(), relativePath);
+            createdDirs.push(expectedBaseDir);
+            expect(logPath).toContain(expectedBaseDir);
+        });
+        it('should handle paths with special characters', async () => {
+            const specialPath = path.join(testTempDir, 'logs-with-special-chars');
+            const logger = new OpenAILogger(specialPath);
+            await logger.initialize();
+            const request = { test: 'request' };
+            const response = { test: 'response' };
+            const logPath = await logger.logInteraction(request, response);
+            expect(logPath).toContain(specialPath);
+        });
+    });
+});
+//# sourceMappingURL=openaiLogger.test.js.map

package/dist/src/utils/openaiLogger.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openaiLogger.test.js","sourceRoot":"","sources":["../../../src/utils/openaiLogger.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,WAAmB,CAAC;IACxB,IAAI,WAAmB,CAAC;IACxB,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;QAC5B,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACzE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,cAAc;IACxC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,mCAAmC;QACnC,MAAM,eAAe,GAAG;YACtB,WAAW;YACX,GAAG,WAAW;YACd,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,eAAe,CAAC;YAC5C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,oBAAoB,CAAC;YACjD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC;YACtC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,kBAAkB,CAAC;SAC5C,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YAClB,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACnC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAClC,sEAAsE;YACtE,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,SAAS,GAAG,wBAAwB,CAAC;YAC3C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,WAAW,GAAG,aAAa,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;YAC7D,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,SAAS,GAAG,eAAe,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,SAAS,GAAG,eAAe,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,SAAS,GAAG,MAAM,EAAE;iBACvB,MAAM,CAAC,WAAW,CAAC;iBACnB,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;iBAChB,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;YACtB,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,SAAS,GAAG,MAAM,EAAE;iBACvB,MAAM,CAAC,SAAS,CAAC;iBACjB,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;iBAChB,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;YACtB,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE/D,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CACrB,sEAAsE,CACvE,CAAC;YAEF,MAAM,UAAU,GAAG,MAAM,EAAE;iBACxB,MAAM,CAAC,OAAO,CAAC;iBACf,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;iBAChB,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;YACtB,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEnE,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACxD,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAEtC,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEnE,MAAM,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YACjE,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE/D,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACrC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,WAAW,GAAG,eAAe,CAAC;YACpC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;YAC7D,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAE9B,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,SAAS,GAAG,oBAAoB,CAAC;YACvC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAChE,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAE9B,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,oBAAoB;YACpB,MAAM,MAAM,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YAEvE,wBAAwB;YACxB,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,EAAE,SAAS,CAAC,CAAC;YAExE,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,4BAA4B;YAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,MAAM,CAAC,cAAc,CACzB,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,EACxB,EAAE,IAAI,EAAE,YAAY,CAAC,EAAE,EAAE,CAC1B,CAAC;gBACF,6CAA6C;gBAC7C,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEhC,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CACzC,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,EACxB,EAAE,IAAI,EAAE,YAAY,CAAC,EAAE,EAAE,CAC1B,CAAC;gBACF,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,oBAAoB;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAEhD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAElD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;YAEpE,MAAM,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,YAAY,GAAG,gBAAgB,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,YAAY,GAAG,oBAAoB,CAAC;YAC1C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAEtC,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,YAAY,CAAC,CAAC;YAClE,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAElC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,yBAAyB,CAAC,CAAC;YACtE,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,OAAO,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAEtC,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/partUtils.d.ts

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { GenerateContentResponse, PartListUnion, PartUnion } from '@google/genai';
+/**
+ * Converts a PartListUnion into a string.
+ * If verbose is true, includes summary representations of non-text parts.
+ */
+export declare function partToString(value: PartListUnion, options?: {
+    verbose?: boolean;
+}): string;
+export declare function getResponseText(response: GenerateContentResponse): string | null;
+/**
+ * Asynchronously maps over a PartListUnion, applying a transformation function
+ * to the text content of each text-based part.
+ *
+ * @param parts The PartListUnion to process.
+ * @param transform A function that takes a string of text and returns a Promise
+ *   resolving to an array of new PartUnions.
+ * @returns A Promise that resolves to a new array of PartUnions with the
+ *   transformations applied.
+ */
+export declare function flatMapTextParts(parts: PartListUnion, transform: (text: string) => Promise<PartUnion[]>): Promise<PartUnion[]>;
+/**
+ * Appends a string of text to the last text part of a prompt, or adds a new
+ * text part if the last part is not a text part.
+ *
+ * @param prompt The prompt to modify.
+ * @param textToAppend The text to append to the prompt.
+ * @param separator The separator to add between existing text and the new text.
+ * @returns The modified prompt.
+ */
+export declare function appendToLastTextPart(prompt: PartUnion[], textToAppend: string, separator?: string): PartUnion[];

package/dist/src/utils/partUtils.js

@@ -0,0 +1,133 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Converts a PartListUnion into a string.
+ * If verbose is true, includes summary representations of non-text parts.
+ */
+export function partToString(value, options) {
+    if (!value) {
+        return '';
+    }
+    if (typeof value === 'string') {
+        return value;
+    }
+    if (Array.isArray(value)) {
+        return value.map((part) => partToString(part, options)).join('');
+    }
+    // Cast to Part, assuming it might contain project-specific fields
+    const part = value;
+    if (options?.verbose) {
+        if (part.videoMetadata !== undefined) {
+            return `[Video Metadata]`;
+        }
+        if (part.thought !== undefined) {
+            return `[Thought: ${part.thought}]`;
+        }
+        if (part.codeExecutionResult !== undefined) {
+            return `[Code Execution Result]`;
+        }
+        if (part.executableCode !== undefined) {
+            return `[Executable Code]`;
+        }
+        // Standard Part fields
+        if (part.fileData !== undefined) {
+            return `[File Data]`;
+        }
+        if (part.functionCall !== undefined) {
+            return `[Function Call: ${part.functionCall.name}]`;
+        }
+        if (part.functionResponse !== undefined) {
+            return `[Function Response: ${part.functionResponse.name}]`;
+        }
+        if (part.inlineData !== undefined) {
+            return `<${part.inlineData.mimeType}>`;
+        }
+    }
+    return part.text ?? '';
+}
+export function getResponseText(response) {
+    if (response.candidates && response.candidates.length > 0) {
+        const candidate = response.candidates[0];
+        if (candidate.content &&
+            candidate.content.parts &&
+            candidate.content.parts.length > 0) {
+            return candidate.content.parts
+                .filter((part) => part.text)
+                .map((part) => part.text)
+                .join('');
+        }
+    }
+    return null;
+}
+/**
+ * Asynchronously maps over a PartListUnion, applying a transformation function
+ * to the text content of each text-based part.
+ *
+ * @param parts The PartListUnion to process.
+ * @param transform A function that takes a string of text and returns a Promise
+ *   resolving to an array of new PartUnions.
+ * @returns A Promise that resolves to a new array of PartUnions with the
+ *   transformations applied.
+ */
+export async function flatMapTextParts(parts, transform) {
+    const result = [];
+    const partArray = Array.isArray(parts)
+        ? parts
+        : typeof parts === 'string'
+            ? [{ text: parts }]
+            : [parts];
+    for (const part of partArray) {
+        let textToProcess;
+        if (typeof part === 'string') {
+            textToProcess = part;
+        }
+        else if ('text' in part) {
+            textToProcess = part.text;
+        }
+        if (textToProcess !== undefined) {
+            const transformedParts = await transform(textToProcess);
+            result.push(...transformedParts);
+        }
+        else {
+            // Pass through non-text parts unmodified.
+            result.push(part);
+        }
+    }
+    return result;
+}
+/**
+ * Appends a string of text to the last text part of a prompt, or adds a new
+ * text part if the last part is not a text part.
+ *
+ * @param prompt The prompt to modify.
+ * @param textToAppend The text to append to the prompt.
+ * @param separator The separator to add between existing text and the new text.
+ * @returns The modified prompt.
+ */
+export function appendToLastTextPart(prompt, textToAppend, separator = '\n\n') {
+    if (!textToAppend) {
+        return prompt;
+    }
+    if (prompt.length === 0) {
+        return [{ text: textToAppend }];
+    }
+    const newPrompt = [...prompt];
+    const lastPart = newPrompt.at(-1);
+    if (typeof lastPart === 'string') {
+        newPrompt[newPrompt.length - 1] = `${lastPart}${separator}${textToAppend}`;
+    }
+    else if (lastPart && 'text' in lastPart) {
+        newPrompt[newPrompt.length - 1] = {
+            ...lastPart,
+            text: `${lastPart.text}${separator}${textToAppend}`,
+        };
+    }
+    else {
+        newPrompt.push({ text: `${separator}${textToAppend}` });
+    }
+    return newPrompt;
+}
+//# sourceMappingURL=partUtils.js.map

package/dist/src/utils/partUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"partUtils.js","sourceRoot":"","sources":["../../../src/utils/partUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH;;;GAGG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAoB,EACpB,OAA+B;IAE/B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,kEAAkE;IAClE,MAAM,IAAI,GAAG,KAKZ,CAAC;IAEF,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO,kBAAkB,CAAC;QAC5B,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,aAAa,IAAI,CAAC,OAAO,GAAG,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,mBAAmB,KAAK,SAAS,EAAE,CAAC;YAC3C,OAAO,yBAAyB,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YACtC,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QAED,uBAAuB;QACvB,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACpC,OAAO,mBAAmB,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC;QACtD,CAAC;QACD,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACxC,OAAO,uBAAuB,IAAI,CAAC,gBAAgB,CAAC,IAAI,GAAG,CAAC;QAC9D,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,GAAG,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,QAAiC;IAEjC,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAEzC,IACE,SAAS,CAAC,OAAO;YACjB,SAAS,CAAC,OAAO,CAAC,KAAK;YACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAClC,CAAC;YACD,OAAO,SAAS,CAAC,OAAO,CAAC,KAAK;iBAC3B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC3B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxB,IAAI,CAAC,EAAE,CAAC,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAoB,EACpB,SAAiD;IAEjD,MAAM,MAAM,GAAgB,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACpC,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ;YACzB,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YACnB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAEd,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC7B,IAAI,aAAiC,CAAC;QACtC,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,aAAa,GAAG,IAAI,CAAC;QACvB,CAAC;aAAM,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YAC1B,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC;QAC5B,CAAC;QAED,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YAChC,MAAM,gBAAgB,GAAG,MAAM,SAAS,CAAC,aAAa,CAAC,CAAC;YACxD,MAAM,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,0CAA0C;YAC1C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAmB,EACnB,YAAoB,EACpB,SAAS,GAAG,MAAM;IAElB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;IAC9B,MAAM,QAAQ,GAAG,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAElC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,GAAG,QAAQ,GAAG,SAAS,GAAG,YAAY,EAAE,CAAC;IAC7E,CAAC;SAAM,IAAI,QAAQ,IAAI,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC1C,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG;YAChC,GAAG,QAAQ;YACX,IAAI,EAAE,GAAG,QAAQ,CAAC,IAAI,GAAG,SAAS,GAAG,YAAY,EAAE;SACpD,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,SAAS,GAAG,YAAY,EAAE,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/src/utils/partUtils.test.d.ts

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