diff --git a/packages/core/src/tools/shell.ts b/packages/core/src/tools/shell.ts index e5e375b9ef4..7abe7067d69 100644 --- a/packages/core/src/tools/shell.ts +++ b/packages/core/src/tools/shell.ts @@ -32,7 +32,7 @@ import type { ShellOutputEvent, } from '../services/shellExecutionService.js'; import { ShellExecutionService } from '../services/shellExecutionService.js'; -import { formatMemoryUsage } from '../utils/formatters.js'; +import { formatBytes } from '../utils/formatters.js'; import type { AnsiOutput } from '../utils/terminalSerializer.js'; import { getCommandRoots, @@ -231,7 +231,7 @@ export class ShellToolInvocation extends BaseToolInvocation< break; case 'binary_progress': isBinaryStream = true; - cumulativeOutput = `[Receiving binary output... ${formatMemoryUsage( + cumulativeOutput = `[Receiving binary output... ${formatBytes( event.bytesReceived, )} received]`; if (Date.now() - lastUpdateTime > OUTPUT_UPDATE_INTERVAL_MS) { diff --git a/packages/core/src/utils/formatters.test.ts b/packages/core/src/utils/formatters.test.ts index 305b2d0adb1..cf6ba3a790d 100644 --- a/packages/core/src/utils/formatters.test.ts +++ b/packages/core/src/utils/formatters.test.ts @@ -6,7 +6,7 @@ import { describe, it, expect } from 'vitest'; -import { bytesToMB, formatMemoryUsage } from './formatters.js'; +import { bytesToMB, formatBytes } from './formatters.js'; describe('bytesToMB', () => { it('converts bytes to megabytes', () => { @@ -16,16 +16,16 @@ describe('bytesToMB', () => { }); }); -describe('formatMemoryUsage', () => { +describe('formatBytes', () => { it('formats values below one megabyte in KB', () => { - expect(formatMemoryUsage(512 * 1024)).toBe('512.0 KB'); + expect(formatBytes(512 * 1024)).toBe('512.0 KB'); }); it('formats values below one gigabyte in MB', () => { - expect(formatMemoryUsage(5 * 1024 * 1024)).toBe('5.0 MB'); + expect(formatBytes(5 * 1024 * 1024)).toBe('5.0 MB'); }); it('formats values of one gigabyte or larger in GB', () => { - expect(formatMemoryUsage(2 * 1024 * 1024 * 1024)).toBe('2.00 GB'); + expect(formatBytes(2 * 1024 * 1024 * 1024)).toBe('2.00 GB'); }); }); diff --git a/packages/core/src/utils/formatters.ts b/packages/core/src/utils/formatters.ts index 88946ae910b..28657a8c90f 100644 --- a/packages/core/src/utils/formatters.ts +++ b/packages/core/src/utils/formatters.ts @@ -6,7 +6,7 @@ export const bytesToMB = (bytes: number): number => bytes / (1024 * 1024); -export const formatMemoryUsage = (bytes: number): string => { +export const formatBytes = (bytes: number): string => { const gb = bytes / (1024 * 1024 * 1024); if (bytes < 1024 * 1024) { return `${(bytes / 1024).toFixed(1)} KB`;