diff options
| author | uttamkanodia14 <[email protected]> | 2025-07-12 02:40:25 +0530 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-07-11 21:10:25 +0000 |
| commit | 5b5f496436a060124e57009d1f7f37bd4d27f0f3 (patch) | |
| tree | 7d213f79753071c20cb90c36ebdc6edbce4ffa21 /packages/core/src/telemetry/loggers.test.ts | |
| parent | 764809753ad85ecc209aa9b126efbb3390c03274 (diff) | |
Adds Flash Fallback logging and clearcut logging (#3843)
Diffstat (limited to 'packages/core/src/telemetry/loggers.test.ts')
| -rw-r--r-- | packages/core/src/telemetry/loggers.test.ts | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/packages/core/src/telemetry/loggers.test.ts b/packages/core/src/telemetry/loggers.test.ts index 13617946..7a24bcca 100644 --- a/packages/core/src/telemetry/loggers.test.ts +++ b/packages/core/src/telemetry/loggers.test.ts @@ -23,6 +23,7 @@ import { EVENT_CLI_CONFIG, EVENT_TOOL_CALL, EVENT_USER_PROMPT, + EVENT_FLASH_FALLBACK, } from './constants.js'; import { logApiRequest, @@ -30,6 +31,7 @@ import { logCliConfiguration, logUserPrompt, logToolCall, + logFlashFallback, } from './loggers.js'; import { ApiRequestEvent, @@ -38,6 +40,7 @@ import { ToolCallDecision, ToolCallEvent, UserPromptEvent, + FlashFallbackEvent, } from './types.js'; import * as metrics from './metrics.js'; import * as sdk from './sdk.js'; @@ -350,6 +353,29 @@ describe('loggers', () => { }); }); + describe('logFlashFallback', () => { + const mockConfig = { + getSessionId: () => 'test-session-id', + getUsageStatisticsEnabled: () => true, + } as unknown as Config; + + it('should log flash fallback event', () => { + const event = new FlashFallbackEvent(AuthType.USE_VERTEX_AI); + + logFlashFallback(mockConfig, event); + + expect(mockLogger.emit).toHaveBeenCalledWith({ + body: 'Switching to flash as Fallback.', + attributes: { + 'session.id': 'test-session-id', + 'event.name': EVENT_FLASH_FALLBACK, + 'event.timestamp': '2025-01-01T00:00:00.000Z', + auth_type: 'vertex-ai', + }, + }); + }); + }); + describe('logToolCall', () => { const cfg1 = { getSessionId: () => 'test-session-id', |
