summaryrefslogtreecommitdiff
path: root/packages/core/src/telemetry/loggers.test.ts
diff options
context:
space:
mode:
authoruttamkanodia14 <[email protected]>2025-07-12 02:40:25 +0530
committerGitHub <[email protected]>2025-07-11 21:10:25 +0000
commit5b5f496436a060124e57009d1f7f37bd4d27f0f3 (patch)
tree7d213f79753071c20cb90c36ebdc6edbce4ffa21 /packages/core/src/telemetry/loggers.test.ts
parent764809753ad85ecc209aa9b126efbb3390c03274 (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.ts26
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',