diff options
Diffstat (limited to 'packages/core/src/telemetry')
| -rw-r--r-- | packages/core/src/telemetry/index.ts | 9 | ||||
| -rw-r--r-- | packages/core/src/telemetry/loggers.test.ts | 6 | ||||
| -rw-r--r-- | packages/core/src/telemetry/loggers.ts | 4 |
3 files changed, 14 insertions, 5 deletions
diff --git a/packages/core/src/telemetry/index.ts b/packages/core/src/telemetry/index.ts index e8248bf9..32e98144 100644 --- a/packages/core/src/telemetry/index.ts +++ b/packages/core/src/telemetry/index.ts @@ -4,6 +4,15 @@ * SPDX-License-Identifier: Apache-2.0 */ +export enum TelemetryTarget { + GCP = 'gcp', + LOCAL = 'local', +} + +const DEFAULT_TELEMETRY_TARGET = TelemetryTarget.LOCAL; +const DEFAULT_OTLP_ENDPOINT = 'http://localhost:4317'; + +export { DEFAULT_TELEMETRY_TARGET, DEFAULT_OTLP_ENDPOINT }; export { initializeTelemetry, shutdownTelemetry, diff --git a/packages/core/src/telemetry/loggers.test.ts b/packages/core/src/telemetry/loggers.test.ts index 2f909c22..6ec73853 100644 --- a/packages/core/src/telemetry/loggers.test.ts +++ b/packages/core/src/telemetry/loggers.test.ts @@ -60,7 +60,7 @@ describe('loggers', () => { vertexai: true, codeAssist: false, }), - getTelemetryLogUserPromptsEnabled: () => true, + getTelemetryLogPromptsEnabled: () => true, getFileFilteringRespectGitIgnore: () => true, getDebugMode: () => true, getMcpServers: () => ({ @@ -99,7 +99,7 @@ describe('loggers', () => { describe('logUserPrompt', () => { const mockConfig = { getSessionId: () => 'test-session-id', - getTelemetryLogUserPromptsEnabled: () => true, + getTelemetryLogPromptsEnabled: () => true, } as unknown as Config; it('should log a user prompt', () => { @@ -125,7 +125,7 @@ describe('loggers', () => { it('should not log prompt if disabled', () => { const mockConfig = { getSessionId: () => 'test-session-id', - getTelemetryLogUserPromptsEnabled: () => false, + getTelemetryLogPromptsEnabled: () => false, } as unknown as Config; const event = { prompt: 'test-prompt', diff --git a/packages/core/src/telemetry/loggers.ts b/packages/core/src/telemetry/loggers.ts index e788119c..01e83908 100644 --- a/packages/core/src/telemetry/loggers.ts +++ b/packages/core/src/telemetry/loggers.ts @@ -37,7 +37,7 @@ import { } from '@google/genai'; const shouldLogUserPrompts = (config: Config): boolean => - config.getTelemetryLogUserPromptsEnabled() ?? false; + config.getTelemetryLogPromptsEnabled() ?? false; function getCommonAttributes(config: Config): LogAttributes { return { @@ -86,7 +86,7 @@ export function logCliConfiguration(config: Config): void { api_key_enabled: !!generatorConfig.apiKey, vertex_ai_enabled: !!generatorConfig.vertexai, code_assist_enabled: !!generatorConfig.codeAssist, - log_user_prompts_enabled: config.getTelemetryLogUserPromptsEnabled(), + log_user_prompts_enabled: config.getTelemetryLogPromptsEnabled(), file_filtering_respect_git_ignore: config.getFileFilteringRespectGitIgnore(), debug_mode: config.getDebugMode(), |
