@office-ai/aioncli-core 0.30.0 → 0.30.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/docs/assets/theme-ansi-dark.png +0 -0
- package/dist/docs/assets/theme-atom-one-dark.png +0 -0
- package/dist/docs/assets/theme-ayu-dark.png +0 -0
- package/dist/docs/assets/theme-default-dark.png +0 -0
- package/dist/docs/assets/theme-dracula-dark.png +0 -0
- package/dist/docs/assets/theme-github-dark.png +0 -0
- package/dist/docs/assets/theme-holiday-dark.png +0 -0
- package/dist/docs/assets/theme-shades-of-purple-dark.png +0 -0
- package/dist/docs/assets/theme-solarized-dark.png +0 -0
- package/dist/docs/assets/theme-solarized-light.png +0 -0
- package/dist/docs/cli/notifications.md +58 -0
- package/dist/docs/redirects.json +20 -0
- package/dist/docs/reference/commands.md +563 -0
- package/dist/docs/reference/configuration.md +1804 -0
- package/dist/docs/reference/keyboard-shortcuts.md +168 -0
- package/dist/docs/reference/memport.md +246 -0
- package/dist/docs/reference/policy-engine.md +386 -0
- package/dist/docs/reference/tools.md +106 -0
- package/dist/docs/resources/faq.md +175 -0
- package/dist/docs/resources/quota-and-pricing.md +199 -0
- package/dist/docs/resources/tos-privacy.md +102 -0
- package/dist/docs/resources/troubleshooting.md +176 -0
- package/dist/docs/resources/uninstall.md +56 -0
- package/dist/src/agents/a2a-errors.d.ts +65 -0
- package/dist/src/agents/a2a-errors.js +164 -0
- package/dist/src/agents/a2a-errors.js.map +1 -0
- package/dist/src/agents/a2a-errors.test.d.ts +6 -0
- package/dist/src/agents/a2a-errors.test.js +183 -0
- package/dist/src/agents/a2a-errors.test.js.map +1 -0
- package/dist/src/agents/auth-provider/api-key-provider.d.ts +30 -0
- package/dist/src/agents/auth-provider/api-key-provider.js +66 -0
- package/dist/src/agents/auth-provider/api-key-provider.js.map +1 -0
- package/dist/src/agents/auth-provider/api-key-provider.test.d.ts +6 -0
- package/dist/src/agents/auth-provider/api-key-provider.test.js +130 -0
- package/dist/src/agents/auth-provider/api-key-provider.test.js.map +1 -0
- package/dist/src/agents/auth-provider/http-provider.d.ts +28 -0
- package/dist/src/agents/auth-provider/http-provider.js +73 -0
- package/dist/src/agents/auth-provider/http-provider.js.map +1 -0
- package/dist/src/agents/auth-provider/http-provider.test.d.ts +6 -0
- package/dist/src/agents/auth-provider/http-provider.test.js +112 -0
- package/dist/src/agents/auth-provider/http-provider.test.js.map +1 -0
- package/dist/src/agents/auth-provider/oauth2-provider.d.ts +65 -0
- package/dist/src/agents/auth-provider/oauth2-provider.js +233 -0
- package/dist/src/agents/auth-provider/oauth2-provider.js.map +1 -0
- package/dist/src/agents/auth-provider/oauth2-provider.test.d.ts +6 -0
- package/dist/src/agents/auth-provider/oauth2-provider.test.js +490 -0
- package/dist/src/agents/auth-provider/oauth2-provider.test.js.map +1 -0
- package/dist/src/agents/browser/analyzeScreenshot.d.ts +35 -0
- package/dist/src/agents/browser/analyzeScreenshot.js +183 -0
- package/dist/src/agents/browser/analyzeScreenshot.js.map +1 -0
- package/dist/src/agents/browser/analyzeScreenshot.test.d.ts +6 -0
- package/dist/src/agents/browser/analyzeScreenshot.test.js +161 -0
- package/dist/src/agents/browser/analyzeScreenshot.test.js.map +1 -0
- package/dist/src/agents/browser/automationOverlay.d.ts +26 -0
- package/dist/src/agents/browser/automationOverlay.js +100 -0
- package/dist/src/agents/browser/automationOverlay.js.map +1 -0
- package/dist/src/agents/browser/browserAgentDefinition.d.ts +50 -0
- package/dist/src/agents/browser/browserAgentDefinition.js +141 -0
- package/dist/src/agents/browser/browserAgentDefinition.js.map +1 -0
- package/dist/src/agents/browser/browserAgentFactory.d.ts +42 -0
- package/dist/src/agents/browser/browserAgentFactory.js +116 -0
- package/dist/src/agents/browser/browserAgentFactory.js.map +1 -0
- package/dist/src/agents/browser/browserAgentFactory.test.d.ts +6 -0
- package/dist/src/agents/browser/browserAgentFactory.test.js +240 -0
- package/dist/src/agents/browser/browserAgentFactory.test.js.map +1 -0
- package/dist/src/agents/browser/browserAgentInvocation.d.ts +34 -0
- package/dist/src/agents/browser/browserAgentInvocation.js +386 -0
- package/dist/src/agents/browser/browserAgentInvocation.js.map +1 -0
- package/dist/src/agents/browser/browserAgentInvocation.test.d.ts +6 -0
- package/dist/src/agents/browser/browserAgentInvocation.test.js +382 -0
- package/dist/src/agents/browser/browserAgentInvocation.test.js.map +1 -0
- package/dist/src/agents/browser/browserManager.d.ts +115 -0
- package/dist/src/agents/browser/browserManager.js +370 -0
- package/dist/src/agents/browser/browserManager.js.map +1 -0
- package/dist/src/agents/browser/browserManager.test.d.ts +6 -0
- package/dist/src/agents/browser/browserManager.test.js +382 -0
- package/dist/src/agents/browser/browserManager.test.js.map +1 -0
- package/dist/src/agents/browser/mcpToolWrapper.d.ts +45 -0
- package/dist/src/agents/browser/mcpToolWrapper.js +358 -0
- package/dist/src/agents/browser/mcpToolWrapper.js.map +1 -0
- package/dist/src/agents/browser/mcpToolWrapper.test.d.ts +6 -0
- package/dist/src/agents/browser/mcpToolWrapper.test.js +126 -0
- package/dist/src/agents/browser/mcpToolWrapper.test.js.map +1 -0
- package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.d.ts +6 -0
- package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.js +59 -0
- package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.js.map +1 -0
- package/dist/src/agents/browser/modelAvailability.d.ts +23 -0
- package/dist/src/agents/browser/modelAvailability.js +23 -0
- package/dist/src/agents/browser/modelAvailability.js.map +1 -0
- package/dist/src/billing/billing.d.ts +80 -0
- package/dist/src/billing/billing.js +128 -0
- package/dist/src/billing/billing.js.map +1 -0
- package/dist/src/billing/billing.test.d.ts +6 -0
- package/dist/src/billing/billing.test.js +182 -0
- package/dist/src/billing/billing.test.js.map +1 -0
- package/dist/src/billing/index.d.ts +6 -0
- package/dist/src/billing/index.js +7 -0
- package/dist/src/billing/index.js.map +1 -0
- package/dist/src/code_assist/types.d.ts +14 -14
- package/dist/src/config/agent-loop-context.d.ts +22 -0
- package/dist/src/config/agent-loop-context.js +7 -0
- package/dist/src/config/agent-loop-context.js.map +1 -0
- package/dist/src/config/trackerFeatureFlag.test.d.ts +6 -0
- package/dist/src/config/trackerFeatureFlag.test.js +43 -0
- package/dist/src/config/trackerFeatureFlag.test.js.map +1 -0
- package/dist/src/config/userHintService.d.ts +46 -0
- package/dist/src/config/userHintService.js +81 -0
- package/dist/src/config/userHintService.js.map +1 -0
- package/dist/src/config/userHintService.test.d.ts +6 -0
- package/dist/src/config/userHintService.test.js +62 -0
- package/dist/src/config/userHintService.test.js.map +1 -0
- package/dist/src/core/localLiteRtLmClient.d.ts +24 -0
- package/dist/src/core/localLiteRtLmClient.js +77 -0
- package/dist/src/core/localLiteRtLmClient.js.map +1 -0
- package/dist/src/core/localLiteRtLmClient.test.d.ts +6 -0
- package/dist/src/core/localLiteRtLmClient.test.js +87 -0
- package/dist/src/core/localLiteRtLmClient.test.js.map +1 -0
- package/dist/src/core/openaiContentGenerator.d.ts +1 -0
- package/dist/src/core/openaiContentGenerator.js +15 -14
- package/dist/src/core/openaiContentGenerator.js.map +1 -1
- package/dist/src/generated/git-commit.d.ts +2 -2
- package/dist/src/generated/git-commit.js +2 -2
- package/dist/src/hooks/runtimeHooks.test.d.ts +6 -0
- package/dist/src/hooks/runtimeHooks.test.js +100 -0
- package/dist/src/hooks/runtimeHooks.test.js.map +1 -0
- package/dist/src/ide/types.d.ts +10 -10
- package/dist/src/mcp/mcp-oauth-provider.d.ts +43 -0
- package/dist/src/mcp/mcp-oauth-provider.js +67 -0
- package/dist/src/mcp/mcp-oauth-provider.js.map +1 -0
- package/dist/src/mcp/mcp-oauth-provider.test.d.ts +6 -0
- package/dist/src/mcp/mcp-oauth-provider.test.js +63 -0
- package/dist/src/mcp/mcp-oauth-provider.test.js.map +1 -0
- package/dist/src/policy/integrity.d.ts +45 -0
- package/dist/src/policy/integrity.js +121 -0
- package/dist/src/policy/integrity.js.map +1 -0
- package/dist/src/policy/integrity.test.d.ts +6 -0
- package/dist/src/policy/integrity.test.js +132 -0
- package/dist/src/policy/integrity.test.js.map +1 -0
- package/dist/src/policy/policies/conseca.toml +6 -0
- package/dist/src/policy/workspace-policy.test.js +231 -0
- package/dist/src/policy/workspace-policy.test.js.map +1 -0
- package/dist/src/routing/strategies/approvalModeStrategy.d.ts +18 -0
- package/dist/src/routing/strategies/approvalModeStrategy.js +59 -0
- package/dist/src/routing/strategies/approvalModeStrategy.js.map +1 -0
- package/dist/src/routing/strategies/approvalModeStrategy.test.d.ts +6 -0
- package/dist/src/routing/strategies/approvalModeStrategy.test.js +140 -0
- package/dist/src/routing/strategies/approvalModeStrategy.test.js.map +1 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.d.ts +14 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.js +182 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.js.map +1 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.test.d.ts +6 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.test.js +218 -0
- package/dist/src/routing/strategies/gemmaClassifierStrategy.test.js.map +1 -0
- package/dist/src/safety/conseca/conseca.d.ts +31 -0
- package/dist/src/safety/conseca/conseca.js +105 -0
- package/dist/src/safety/conseca/conseca.js.map +1 -0
- package/dist/src/safety/conseca/conseca.test.js +226 -0
- package/dist/src/safety/conseca/conseca.test.js.map +1 -0
- package/dist/src/safety/conseca/integration.test.js +19 -0
- package/dist/src/safety/conseca/integration.test.js.map +1 -0
- package/dist/src/safety/conseca/policy-enforcer.d.ts +13 -0
- package/dist/src/safety/conseca/policy-enforcer.js +135 -0
- package/dist/src/safety/conseca/policy-enforcer.js.map +1 -0
- package/dist/src/safety/conseca/policy-enforcer.test.js +141 -0
- package/dist/src/safety/conseca/policy-enforcer.test.js.map +1 -0
- package/dist/src/safety/conseca/policy-generator.d.ts +15 -0
- package/dist/src/safety/conseca/policy-generator.js +144 -0
- package/dist/src/safety/conseca/policy-generator.js.map +1 -0
- package/dist/src/safety/conseca/policy-generator.test.d.ts +6 -0
- package/dist/src/safety/conseca/policy-generator.test.js +84 -0
- package/dist/src/safety/conseca/policy-generator.test.js.map +1 -0
- package/dist/src/safety/conseca/types.d.ts +15 -0
- package/dist/src/safety/conseca/types.js +7 -0
- package/dist/src/safety/conseca/types.js.map +1 -0
- package/dist/src/scheduler/scheduler_parallel.test.d.ts +6 -0
- package/dist/src/scheduler/scheduler_parallel.test.js +401 -0
- package/dist/src/scheduler/scheduler_parallel.test.js.map +1 -0
- package/dist/src/services/FolderTrustDiscoveryService.d.ts +32 -0
- package/dist/src/services/FolderTrustDiscoveryService.js +169 -0
- package/dist/src/services/FolderTrustDiscoveryService.js.map +1 -0
- package/dist/src/services/FolderTrustDiscoveryService.test.d.ts +6 -0
- package/dist/src/services/FolderTrustDiscoveryService.test.js +118 -0
- package/dist/src/services/FolderTrustDiscoveryService.test.js.map +1 -0
- package/dist/src/services/fileKeychain.d.ts +24 -0
- package/dist/src/services/fileKeychain.js +123 -0
- package/dist/src/services/fileKeychain.js.map +1 -0
- package/dist/src/services/keychainService.d.ts +51 -0
- package/dist/src/services/keychainService.js +133 -0
- package/dist/src/services/keychainService.js.map +1 -0
- package/dist/src/services/keychainService.test.d.ts +6 -0
- package/dist/src/services/keychainService.test.js +150 -0
- package/dist/src/services/keychainService.test.js.map +1 -0
- package/dist/src/services/keychainTypes.d.ts +41 -0
- package/dist/src/services/keychainTypes.js +18 -0
- package/dist/src/services/keychainTypes.js.map +1 -0
- package/dist/src/services/trackerService.d.ts +49 -0
- package/dist/src/services/trackerService.js +193 -0
- package/dist/src/services/trackerService.js.map +1 -0
- package/dist/src/services/trackerService.test.d.ts +6 -0
- package/dist/src/services/trackerService.test.js +117 -0
- package/dist/src/services/trackerService.test.js.map +1 -0
- package/dist/src/services/trackerTypes.d.ts +51 -0
- package/dist/src/services/trackerTypes.js +33 -0
- package/dist/src/services/trackerTypes.js.map +1 -0
- package/dist/src/telemetry/billingEvents.d.ts +75 -0
- package/dist/src/telemetry/billingEvents.js +181 -0
- package/dist/src/telemetry/billingEvents.js.map +1 -0
- package/dist/src/telemetry/billingEvents.test.d.ts +6 -0
- package/dist/src/telemetry/billingEvents.test.js +139 -0
- package/dist/src/telemetry/billingEvents.test.js.map +1 -0
- package/dist/src/telemetry/conseca-logger.d.ts +9 -0
- package/dist/src/telemetry/conseca-logger.js +91 -0
- package/dist/src/telemetry/conseca-logger.js.map +1 -0
- package/dist/src/telemetry/conseca-logger.test.d.ts +6 -0
- package/dist/src/telemetry/conseca-logger.test.js +90 -0
- package/dist/src/telemetry/conseca-logger.test.js.map +1 -0
- package/dist/src/telemetry/trace.test.d.ts +6 -0
- package/dist/src/telemetry/trace.test.js +116 -0
- package/dist/src/telemetry/trace.test.js.map +1 -0
- package/dist/src/tools/definitions/trackerTools.d.ts +12 -0
- package/dist/src/tools/definitions/trackerTools.js +146 -0
- package/dist/src/tools/definitions/trackerTools.js.map +1 -0
- package/dist/src/tools/diff-utils.d.ts +9 -0
- package/dist/src/tools/diff-utils.js +66 -0
- package/dist/src/tools/diff-utils.js.map +1 -0
- package/dist/src/tools/diff-utils.test.d.ts +6 -0
- package/dist/src/tools/diff-utils.test.js +53 -0
- package/dist/src/tools/diff-utils.test.js.map +1 -0
- package/dist/src/tools/grep-utils.d.ts +49 -0
- package/dist/src/tools/grep-utils.js +147 -0
- package/dist/src/tools/grep-utils.js.map +1 -0
- package/dist/src/tools/omissionPlaceholderDetector.d.ts +15 -0
- package/dist/src/tools/omissionPlaceholderDetector.js +90 -0
- package/dist/src/tools/omissionPlaceholderDetector.js.map +1 -0
- package/dist/src/tools/omissionPlaceholderDetector.test.d.ts +6 -0
- package/dist/src/tools/omissionPlaceholderDetector.test.js +49 -0
- package/dist/src/tools/omissionPlaceholderDetector.test.js.map +1 -0
- package/dist/src/tools/trackerTools.d.ts +122 -0
- package/dist/src/tools/trackerTools.js +365 -0
- package/dist/src/tools/trackerTools.js.map +1 -0
- package/dist/src/tools/trackerTools.test.d.ts +6 -0
- package/dist/src/tools/trackerTools.test.js +97 -0
- package/dist/src/tools/trackerTools.test.js.map +1 -0
- package/dist/src/utils/approvalModeUtils.d.ts +14 -0
- package/dist/src/utils/approvalModeUtils.js +35 -0
- package/dist/src/utils/approvalModeUtils.js.map +1 -0
- package/dist/src/utils/approvalModeUtils.test.d.ts +6 -0
- package/dist/src/utils/approvalModeUtils.test.js +36 -0
- package/dist/src/utils/approvalModeUtils.test.js.map +1 -0
- package/dist/src/utils/cache.d.ts +63 -0
- package/dist/src/utils/cache.js +103 -0
- package/dist/src/utils/cache.js.map +1 -0
- package/dist/src/utils/cache.test.d.ts +6 -0
- package/dist/src/utils/cache.test.js +158 -0
- package/dist/src/utils/cache.test.js.map +1 -0
- package/dist/src/utils/compatibility.d.ts +41 -0
- package/dist/src/utils/compatibility.js +112 -0
- package/dist/src/utils/compatibility.js.map +1 -0
- package/dist/src/utils/compatibility.test.d.ts +6 -0
- package/dist/src/utils/compatibility.test.js +233 -0
- package/dist/src/utils/compatibility.test.js.map +1 -0
- package/dist/src/utils/envExpansion.d.ts +18 -0
- package/dist/src/utils/envExpansion.js +46 -0
- package/dist/src/utils/envExpansion.js.map +1 -0
- package/dist/src/utils/envExpansion.test.d.ts +6 -0
- package/dist/src/utils/envExpansion.test.js +110 -0
- package/dist/src/utils/envExpansion.test.js.map +1 -0
- package/dist/src/utils/errors_timeout.test.d.ts +6 -0
- package/dist/src/utils/errors_timeout.test.js +40 -0
- package/dist/src/utils/errors_timeout.test.js.map +1 -0
- package/dist/src/utils/fetch.test.d.ts +6 -0
- package/dist/src/utils/fetch.test.js +206 -0
- package/dist/src/utils/fetch.test.js.map +1 -0
- package/dist/src/utils/markdownUtils.d.ts +22 -0
- package/dist/src/utils/markdownUtils.js +126 -0
- package/dist/src/utils/markdownUtils.js.map +1 -0
- package/dist/src/utils/markdownUtils.test.d.ts +6 -0
- package/dist/src/utils/markdownUtils.test.js +107 -0
- package/dist/src/utils/markdownUtils.test.js.map +1 -0
- package/dist/src/utils/oauth-flow.d.ts +105 -0
- package/dist/src/utils/oauth-flow.js +370 -0
- package/dist/src/utils/oauth-flow.js.map +1 -0
- package/dist/src/utils/oauth-flow.test.d.ts +6 -0
- package/dist/src/utils/oauth-flow.test.js +360 -0
- package/dist/src/utils/oauth-flow.test.js.map +1 -0
- package/dist/src/utils/sessionUtils.d.ts +14 -0
- package/dist/src/utils/sessionUtils.js +122 -0
- package/dist/src/utils/sessionUtils.js.map +1 -0
- package/dist/src/utils/sessionUtils.test.d.ts +1 -0
- package/dist/src/utils/sessionUtils.test.js +171 -0
- package/dist/src/utils/sessionUtils.test.js.map +1 -0
- package/dist/src/voice/responseFormatter.d.ts +38 -0
- package/dist/src/voice/responseFormatter.js +130 -0
- package/dist/src/voice/responseFormatter.js.map +1 -0
- package/dist/src/voice/responseFormatter.test.d.ts +6 -0
- package/dist/src/voice/responseFormatter.test.js +214 -0
- package/dist/src/voice/responseFormatter.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +1 -1
- package/dist/docs/CONTRIBUTING.md +0 -555
- package/dist/src/agents/executor.d.ts +0 -114
- package/dist/src/agents/executor.js +0 -779
- package/dist/src/agents/executor.js.map +0 -1
- package/dist/src/agents/executor.test.js +0 -1362
- package/dist/src/agents/executor.test.js.map +0 -1
- package/dist/src/agents/invocation.d.ts +0 -46
- package/dist/src/agents/invocation.js +0 -102
- package/dist/src/agents/invocation.js.map +0 -1
- package/dist/src/agents/invocation.test.js +0 -215
- package/dist/src/agents/invocation.test.js.map +0 -1
- package/dist/src/core/subagent.d.ts +0 -236
- package/dist/src/core/subagent.js +0 -482
- package/dist/src/core/subagent.js.map +0 -1
- package/dist/src/core/subagent.test.js +0 -530
- package/dist/src/core/subagent.test.js.map +0 -1
- package/dist/src/tools/smart-edit.d.ts +0 -78
- package/dist/src/tools/smart-edit.js +0 -717
- package/dist/src/tools/smart-edit.js.map +0 -1
- package/dist/src/tools/smart-edit.test.js +0 -592
- package/dist/src/tools/smart-edit.test.js.map +0 -1
- /package/dist/src/{agents/executor.test.d.ts → policy/workspace-policy.test.d.ts} +0 -0
- /package/dist/src/{agents/invocation.test.d.ts → safety/conseca/conseca.test.d.ts} +0 -0
- /package/dist/src/{core/subagent.test.d.ts → safety/conseca/integration.test.d.ts} +0 -0
- /package/dist/src/{tools/smart-edit.test.d.ts → safety/conseca/policy-enforcer.test.d.ts} +0 -0
|
@@ -0,0 +1,181 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2026 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { getCommonAttributes } from './telemetryAttributes.js';
|
|
7
|
+
// ============================================================================
|
|
8
|
+
// Event: Overage Menu Shown
|
|
9
|
+
// ============================================================================
|
|
10
|
+
export const EVENT_OVERAGE_MENU_SHOWN = 'gemini_cli.overage_menu_shown';
|
|
11
|
+
export class OverageMenuShownEvent {
|
|
12
|
+
'event.name';
|
|
13
|
+
'event.timestamp';
|
|
14
|
+
model;
|
|
15
|
+
credit_balance;
|
|
16
|
+
overage_strategy;
|
|
17
|
+
constructor(model, creditBalance, overageStrategy) {
|
|
18
|
+
this['event.name'] = 'overage_menu_shown';
|
|
19
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
20
|
+
this.model = model;
|
|
21
|
+
this.credit_balance = creditBalance;
|
|
22
|
+
this.overage_strategy = overageStrategy;
|
|
23
|
+
}
|
|
24
|
+
toOpenTelemetryAttributes(config) {
|
|
25
|
+
return {
|
|
26
|
+
...getCommonAttributes(config),
|
|
27
|
+
'event.name': EVENT_OVERAGE_MENU_SHOWN,
|
|
28
|
+
'event.timestamp': this['event.timestamp'],
|
|
29
|
+
model: this.model,
|
|
30
|
+
credit_balance: this.credit_balance,
|
|
31
|
+
overage_strategy: this.overage_strategy,
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
toLogBody() {
|
|
35
|
+
return `Overage menu shown for model ${this.model} with ${this.credit_balance} credits available.`;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
// ============================================================================
|
|
39
|
+
// Event: Overage Option Selected
|
|
40
|
+
// ============================================================================
|
|
41
|
+
export const EVENT_OVERAGE_OPTION_SELECTED = 'gemini_cli.overage_option_selected';
|
|
42
|
+
export class OverageOptionSelectedEvent {
|
|
43
|
+
'event.name';
|
|
44
|
+
'event.timestamp';
|
|
45
|
+
model;
|
|
46
|
+
selected_option;
|
|
47
|
+
credit_balance;
|
|
48
|
+
constructor(model, selectedOption, creditBalance) {
|
|
49
|
+
this['event.name'] = 'overage_option_selected';
|
|
50
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
51
|
+
this.model = model;
|
|
52
|
+
this.selected_option = selectedOption;
|
|
53
|
+
this.credit_balance = creditBalance;
|
|
54
|
+
}
|
|
55
|
+
toOpenTelemetryAttributes(config) {
|
|
56
|
+
return {
|
|
57
|
+
...getCommonAttributes(config),
|
|
58
|
+
'event.name': EVENT_OVERAGE_OPTION_SELECTED,
|
|
59
|
+
'event.timestamp': this['event.timestamp'],
|
|
60
|
+
model: this.model,
|
|
61
|
+
selected_option: this.selected_option,
|
|
62
|
+
credit_balance: this.credit_balance,
|
|
63
|
+
};
|
|
64
|
+
}
|
|
65
|
+
toLogBody() {
|
|
66
|
+
return `Overage option '${this.selected_option}' selected for model ${this.model}.`;
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
// ============================================================================
|
|
70
|
+
// Event: Empty Wallet Menu Shown
|
|
71
|
+
// ============================================================================
|
|
72
|
+
export const EVENT_EMPTY_WALLET_MENU_SHOWN = 'gemini_cli.empty_wallet_menu_shown';
|
|
73
|
+
export class EmptyWalletMenuShownEvent {
|
|
74
|
+
'event.name';
|
|
75
|
+
'event.timestamp';
|
|
76
|
+
model;
|
|
77
|
+
constructor(model) {
|
|
78
|
+
this['event.name'] = 'empty_wallet_menu_shown';
|
|
79
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
80
|
+
this.model = model;
|
|
81
|
+
}
|
|
82
|
+
toOpenTelemetryAttributes(config) {
|
|
83
|
+
return {
|
|
84
|
+
...getCommonAttributes(config),
|
|
85
|
+
'event.name': EVENT_EMPTY_WALLET_MENU_SHOWN,
|
|
86
|
+
'event.timestamp': this['event.timestamp'],
|
|
87
|
+
model: this.model,
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
toLogBody() {
|
|
91
|
+
return `Empty wallet menu shown for model ${this.model}.`;
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
// ============================================================================
|
|
95
|
+
// Event: Credit Purchase Click
|
|
96
|
+
// ============================================================================
|
|
97
|
+
export const EVENT_CREDIT_PURCHASE_CLICK = 'gemini_cli.credit_purchase_click';
|
|
98
|
+
export class CreditPurchaseClickEvent {
|
|
99
|
+
'event.name';
|
|
100
|
+
'event.timestamp';
|
|
101
|
+
source;
|
|
102
|
+
model;
|
|
103
|
+
constructor(source, model) {
|
|
104
|
+
this['event.name'] = 'credit_purchase_click';
|
|
105
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
106
|
+
this.source = source;
|
|
107
|
+
this.model = model;
|
|
108
|
+
}
|
|
109
|
+
toOpenTelemetryAttributes(config) {
|
|
110
|
+
return {
|
|
111
|
+
...getCommonAttributes(config),
|
|
112
|
+
'event.name': EVENT_CREDIT_PURCHASE_CLICK,
|
|
113
|
+
'event.timestamp': this['event.timestamp'],
|
|
114
|
+
source: this.source,
|
|
115
|
+
model: this.model,
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
toLogBody() {
|
|
119
|
+
return `Credit purchase clicked from ${this.source} for model ${this.model}.`;
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
// ============================================================================
|
|
123
|
+
// Event: Credits Used
|
|
124
|
+
// ============================================================================
|
|
125
|
+
export const EVENT_CREDITS_USED = 'gemini_cli.credits_used';
|
|
126
|
+
export class CreditsUsedEvent {
|
|
127
|
+
'event.name';
|
|
128
|
+
'event.timestamp';
|
|
129
|
+
model;
|
|
130
|
+
credits_consumed;
|
|
131
|
+
credits_remaining;
|
|
132
|
+
constructor(model, creditsConsumed, creditsRemaining) {
|
|
133
|
+
this['event.name'] = 'credits_used';
|
|
134
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
135
|
+
this.model = model;
|
|
136
|
+
this.credits_consumed = creditsConsumed;
|
|
137
|
+
this.credits_remaining = creditsRemaining;
|
|
138
|
+
}
|
|
139
|
+
toOpenTelemetryAttributes(config) {
|
|
140
|
+
return {
|
|
141
|
+
...getCommonAttributes(config),
|
|
142
|
+
'event.name': EVENT_CREDITS_USED,
|
|
143
|
+
'event.timestamp': this['event.timestamp'],
|
|
144
|
+
model: this.model,
|
|
145
|
+
credits_consumed: this.credits_consumed,
|
|
146
|
+
credits_remaining: this.credits_remaining,
|
|
147
|
+
};
|
|
148
|
+
}
|
|
149
|
+
toLogBody() {
|
|
150
|
+
return `${this.credits_consumed} credits consumed for model ${this.model}. ${this.credits_remaining} remaining.`;
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
// ============================================================================
|
|
154
|
+
// Event: API Key Updated (Auth Type Changed)
|
|
155
|
+
// ============================================================================
|
|
156
|
+
export const EVENT_API_KEY_UPDATED = 'gemini_cli.api_key_updated';
|
|
157
|
+
export class ApiKeyUpdatedEvent {
|
|
158
|
+
'event.name';
|
|
159
|
+
'event.timestamp';
|
|
160
|
+
previous_auth_type;
|
|
161
|
+
new_auth_type;
|
|
162
|
+
constructor(previousAuthType, newAuthType) {
|
|
163
|
+
this['event.name'] = 'api_key_updated';
|
|
164
|
+
this['event.timestamp'] = new Date().toISOString();
|
|
165
|
+
this.previous_auth_type = previousAuthType;
|
|
166
|
+
this.new_auth_type = newAuthType;
|
|
167
|
+
}
|
|
168
|
+
toOpenTelemetryAttributes(config) {
|
|
169
|
+
return {
|
|
170
|
+
...getCommonAttributes(config),
|
|
171
|
+
'event.name': EVENT_API_KEY_UPDATED,
|
|
172
|
+
'event.timestamp': this['event.timestamp'],
|
|
173
|
+
previous_auth_type: this.previous_auth_type,
|
|
174
|
+
new_auth_type: this.new_auth_type,
|
|
175
|
+
};
|
|
176
|
+
}
|
|
177
|
+
toLogBody() {
|
|
178
|
+
return `Auth type changed from ${this.previous_auth_type} to ${this.new_auth_type}.`;
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
//# sourceMappingURL=billingEvents.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"billingEvents.js","sourceRoot":"","sources":["../../../src/telemetry/billingEvents.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAW/D,+EAA+E;AAC/E,4BAA4B;AAC5B,+EAA+E;AAE/E,MAAM,CAAC,MAAM,wBAAwB,GAAG,+BAA+B,CAAC;AAExE,MAAM,OAAO,qBAAqB;IAChC,YAAY,CAAuB;IACnC,iBAAiB,CAAS;IAC1B,KAAK,CAAS;IACd,cAAc,CAAS;IACvB,gBAAgB,CAAkB;IAElC,YACE,KAAa,EACb,aAAqB,EACrB,eAAgC;QAEhC,IAAI,CAAC,YAAY,CAAC,GAAG,oBAAoB,CAAC;QAC1C,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;IAC1C,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,wBAAwB;YACtC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,gCAAgC,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,cAAc,qBAAqB,CAAC;IACrG,CAAC;CACF;AAED,+EAA+E;AAC/E,iCAAiC;AACjC,+EAA+E;AAE/E,MAAM,CAAC,MAAM,6BAA6B,GACxC,oCAAoC,CAAC;AAEvC,MAAM,OAAO,0BAA0B;IACrC,YAAY,CAA4B;IACxC,iBAAiB,CAAS;IAC1B,KAAK,CAAS;IACd,eAAe,CAAgB;IAC/B,cAAc,CAAS;IAEvB,YACE,KAAa,EACb,cAA6B,EAC7B,aAAqB;QAErB,IAAI,CAAC,YAAY,CAAC,GAAG,yBAAyB,CAAC;QAC/C,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;IACtC,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,6BAA6B;YAC3C,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,mBAAmB,IAAI,CAAC,eAAe,wBAAwB,IAAI,CAAC,KAAK,GAAG,CAAC;IACtF,CAAC;CACF;AAED,+EAA+E;AAC/E,iCAAiC;AACjC,+EAA+E;AAE/E,MAAM,CAAC,MAAM,6BAA6B,GACxC,oCAAoC,CAAC;AAEvC,MAAM,OAAO,yBAAyB;IACpC,YAAY,CAA4B;IACxC,iBAAiB,CAAS;IAC1B,KAAK,CAAS;IAEd,YAAY,KAAa;QACvB,IAAI,CAAC,YAAY,CAAC,GAAG,yBAAyB,CAAC;QAC/C,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,6BAA6B;YAC3C,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,qCAAqC,IAAI,CAAC,KAAK,GAAG,CAAC;IAC5D,CAAC;CACF;AAED,+EAA+E;AAC/E,+BAA+B;AAC/B,+EAA+E;AAE/E,MAAM,CAAC,MAAM,2BAA2B,GAAG,kCAAkC,CAAC;AAE9E,MAAM,OAAO,wBAAwB;IACnC,YAAY,CAA0B;IACtC,iBAAiB,CAAS;IAC1B,MAAM,CAAkD;IACxD,KAAK,CAAS;IAEd,YACE,MAAuD,EACvD,KAAa;QAEb,IAAI,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC;QAC7C,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,2BAA2B;YACzC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,gCAAgC,IAAI,CAAC,MAAM,cAAc,IAAI,CAAC,KAAK,GAAG,CAAC;IAChF,CAAC;CACF;AAED,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAE/E,MAAM,CAAC,MAAM,kBAAkB,GAAG,yBAAyB,CAAC;AAE5D,MAAM,OAAO,gBAAgB;IAC3B,YAAY,CAAiB;IAC7B,iBAAiB,CAAS;IAC1B,KAAK,CAAS;IACd,gBAAgB,CAAS;IACzB,iBAAiB,CAAS;IAE1B,YACE,KAAa,EACb,eAAuB,EACvB,gBAAwB;QAExB,IAAI,CAAC,YAAY,CAAC,GAAG,cAAc,CAAC;QACpC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QACxC,IAAI,CAAC,iBAAiB,GAAG,gBAAgB,CAAC;IAC5C,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,kBAAkB;YAChC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;SAC1C,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,GAAG,IAAI,CAAC,gBAAgB,+BAA+B,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,iBAAiB,aAAa,CAAC;IACnH,CAAC;CACF;AAED,+EAA+E;AAC/E,6CAA6C;AAC7C,+EAA+E;AAE/E,MAAM,CAAC,MAAM,qBAAqB,GAAG,4BAA4B,CAAC;AAElE,MAAM,OAAO,kBAAkB;IAC7B,YAAY,CAAoB;IAChC,iBAAiB,CAAS;IAC1B,kBAAkB,CAAS;IAC3B,aAAa,CAAS;IAEtB,YAAY,gBAAwB,EAAE,WAAmB;QACvD,IAAI,CAAC,YAAY,CAAC,GAAG,iBAAiB,CAAC;QACvC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,kBAAkB,GAAG,gBAAgB,CAAC;QAC3C,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC;IACnC,CAAC;IAED,yBAAyB,CAAC,MAAc;QACtC,OAAO;YACL,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC9B,YAAY,EAAE,qBAAqB;YACnC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC;YAC1C,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,aAAa,EAAE,IAAI,CAAC,aAAa;SAClC,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,0BAA0B,IAAI,CAAC,kBAAkB,OAAO,IAAI,CAAC,aAAa,GAAG,CAAC;IACvF,CAAC;CACF"}
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2026 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
7
|
+
import { makeFakeConfig } from '../test-utils/config.js';
|
|
8
|
+
import { OverageMenuShownEvent, OverageOptionSelectedEvent, EmptyWalletMenuShownEvent, CreditPurchaseClickEvent, CreditsUsedEvent, ApiKeyUpdatedEvent, EVENT_OVERAGE_MENU_SHOWN, EVENT_OVERAGE_OPTION_SELECTED, EVENT_EMPTY_WALLET_MENU_SHOWN, EVENT_CREDIT_PURCHASE_CLICK, EVENT_CREDITS_USED, EVENT_API_KEY_UPDATED, } from './billingEvents.js';
|
|
9
|
+
describe('billingEvents', () => {
|
|
10
|
+
const fakeConfig = makeFakeConfig();
|
|
11
|
+
beforeEach(() => {
|
|
12
|
+
vi.useFakeTimers();
|
|
13
|
+
vi.setSystemTime(new Date('2026-01-15T10:30:00.000Z'));
|
|
14
|
+
});
|
|
15
|
+
afterEach(() => {
|
|
16
|
+
vi.useRealTimers();
|
|
17
|
+
});
|
|
18
|
+
describe('OverageMenuShownEvent', () => {
|
|
19
|
+
it('should construct with correct properties', () => {
|
|
20
|
+
const event = new OverageMenuShownEvent('gemini-3-pro-preview', 500, 'ask');
|
|
21
|
+
expect(event['event.name']).toBe('overage_menu_shown');
|
|
22
|
+
expect(event.model).toBe('gemini-3-pro-preview');
|
|
23
|
+
expect(event.credit_balance).toBe(500);
|
|
24
|
+
expect(event.overage_strategy).toBe('ask');
|
|
25
|
+
});
|
|
26
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
27
|
+
const event = new OverageMenuShownEvent('gemini-3-pro-preview', 500, 'ask');
|
|
28
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
29
|
+
expect(attrs['event.name']).toBe(EVENT_OVERAGE_MENU_SHOWN);
|
|
30
|
+
expect(attrs['model']).toBe('gemini-3-pro-preview');
|
|
31
|
+
expect(attrs['credit_balance']).toBe(500);
|
|
32
|
+
expect(attrs['overage_strategy']).toBe('ask');
|
|
33
|
+
});
|
|
34
|
+
it('should produce a human-readable log body', () => {
|
|
35
|
+
const event = new OverageMenuShownEvent('gemini-3-pro-preview', 500, 'ask');
|
|
36
|
+
expect(event.toLogBody()).toContain('gemini-3-pro-preview');
|
|
37
|
+
expect(event.toLogBody()).toContain('500');
|
|
38
|
+
});
|
|
39
|
+
});
|
|
40
|
+
describe('OverageOptionSelectedEvent', () => {
|
|
41
|
+
it('should construct with correct properties', () => {
|
|
42
|
+
const event = new OverageOptionSelectedEvent('gemini-3-pro-preview', 'use_credits', 100);
|
|
43
|
+
expect(event['event.name']).toBe('overage_option_selected');
|
|
44
|
+
expect(event.selected_option).toBe('use_credits');
|
|
45
|
+
expect(event.credit_balance).toBe(100);
|
|
46
|
+
});
|
|
47
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
48
|
+
const event = new OverageOptionSelectedEvent('gemini-3-pro-preview', 'use_fallback', 200);
|
|
49
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
50
|
+
expect(attrs['event.name']).toBe(EVENT_OVERAGE_OPTION_SELECTED);
|
|
51
|
+
expect(attrs['selected_option']).toBe('use_fallback');
|
|
52
|
+
});
|
|
53
|
+
it('should produce a human-readable log body', () => {
|
|
54
|
+
const event = new OverageOptionSelectedEvent('gemini-3-pro-preview', 'manage', 100);
|
|
55
|
+
expect(event.toLogBody()).toContain('manage');
|
|
56
|
+
expect(event.toLogBody()).toContain('gemini-3-pro-preview');
|
|
57
|
+
});
|
|
58
|
+
});
|
|
59
|
+
describe('EmptyWalletMenuShownEvent', () => {
|
|
60
|
+
it('should construct with correct properties', () => {
|
|
61
|
+
const event = new EmptyWalletMenuShownEvent('gemini-3-pro-preview');
|
|
62
|
+
expect(event['event.name']).toBe('empty_wallet_menu_shown');
|
|
63
|
+
expect(event.model).toBe('gemini-3-pro-preview');
|
|
64
|
+
});
|
|
65
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
66
|
+
const event = new EmptyWalletMenuShownEvent('gemini-3-pro-preview');
|
|
67
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
68
|
+
expect(attrs['event.name']).toBe(EVENT_EMPTY_WALLET_MENU_SHOWN);
|
|
69
|
+
expect(attrs['model']).toBe('gemini-3-pro-preview');
|
|
70
|
+
});
|
|
71
|
+
it('should produce a human-readable log body', () => {
|
|
72
|
+
const event = new EmptyWalletMenuShownEvent('gemini-3-pro-preview');
|
|
73
|
+
expect(event.toLogBody()).toContain('gemini-3-pro-preview');
|
|
74
|
+
});
|
|
75
|
+
});
|
|
76
|
+
describe('CreditPurchaseClickEvent', () => {
|
|
77
|
+
it('should construct with correct properties', () => {
|
|
78
|
+
const event = new CreditPurchaseClickEvent('empty_wallet_menu', 'gemini-3-pro-preview');
|
|
79
|
+
expect(event['event.name']).toBe('credit_purchase_click');
|
|
80
|
+
expect(event.source).toBe('empty_wallet_menu');
|
|
81
|
+
expect(event.model).toBe('gemini-3-pro-preview');
|
|
82
|
+
});
|
|
83
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
84
|
+
const event = new CreditPurchaseClickEvent('overage_menu', 'gemini-3-pro-preview');
|
|
85
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
86
|
+
expect(attrs['event.name']).toBe(EVENT_CREDIT_PURCHASE_CLICK);
|
|
87
|
+
expect(attrs['source']).toBe('overage_menu');
|
|
88
|
+
});
|
|
89
|
+
it('should produce a human-readable log body', () => {
|
|
90
|
+
const event = new CreditPurchaseClickEvent('manage', 'gemini-3-pro-preview');
|
|
91
|
+
expect(event.toLogBody()).toContain('manage');
|
|
92
|
+
expect(event.toLogBody()).toContain('gemini-3-pro-preview');
|
|
93
|
+
});
|
|
94
|
+
});
|
|
95
|
+
describe('CreditsUsedEvent', () => {
|
|
96
|
+
it('should construct with correct properties', () => {
|
|
97
|
+
const event = new CreditsUsedEvent('gemini-3-pro-preview', 10, 490);
|
|
98
|
+
expect(event['event.name']).toBe('credits_used');
|
|
99
|
+
expect(event.credits_consumed).toBe(10);
|
|
100
|
+
expect(event.credits_remaining).toBe(490);
|
|
101
|
+
});
|
|
102
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
103
|
+
const event = new CreditsUsedEvent('gemini-3-pro-preview', 10, 490);
|
|
104
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
105
|
+
expect(attrs['event.name']).toBe(EVENT_CREDITS_USED);
|
|
106
|
+
expect(attrs['credits_consumed']).toBe(10);
|
|
107
|
+
expect(attrs['credits_remaining']).toBe(490);
|
|
108
|
+
});
|
|
109
|
+
it('should produce a human-readable log body', () => {
|
|
110
|
+
const event = new CreditsUsedEvent('gemini-3-pro-preview', 10, 490);
|
|
111
|
+
const body = event.toLogBody();
|
|
112
|
+
expect(body).toContain('10');
|
|
113
|
+
expect(body).toContain('490');
|
|
114
|
+
expect(body).toContain('gemini-3-pro-preview');
|
|
115
|
+
});
|
|
116
|
+
});
|
|
117
|
+
describe('ApiKeyUpdatedEvent', () => {
|
|
118
|
+
it('should construct with correct properties', () => {
|
|
119
|
+
const event = new ApiKeyUpdatedEvent('google_login', 'api_key');
|
|
120
|
+
expect(event['event.name']).toBe('api_key_updated');
|
|
121
|
+
expect(event.previous_auth_type).toBe('google_login');
|
|
122
|
+
expect(event.new_auth_type).toBe('api_key');
|
|
123
|
+
});
|
|
124
|
+
it('should produce correct OpenTelemetry attributes', () => {
|
|
125
|
+
const event = new ApiKeyUpdatedEvent('google_login', 'api_key');
|
|
126
|
+
const attrs = event.toOpenTelemetryAttributes(fakeConfig);
|
|
127
|
+
expect(attrs['event.name']).toBe(EVENT_API_KEY_UPDATED);
|
|
128
|
+
expect(attrs['previous_auth_type']).toBe('google_login');
|
|
129
|
+
expect(attrs['new_auth_type']).toBe('api_key');
|
|
130
|
+
});
|
|
131
|
+
it('should produce a human-readable log body', () => {
|
|
132
|
+
const event = new ApiKeyUpdatedEvent('google_login', 'api_key');
|
|
133
|
+
const body = event.toLogBody();
|
|
134
|
+
expect(body).toContain('google_login');
|
|
135
|
+
expect(body).toContain('api_key');
|
|
136
|
+
});
|
|
137
|
+
});
|
|
138
|
+
});
|
|
139
|
+
//# sourceMappingURL=billingEvents.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"billingEvents.test.js","sourceRoot":"","sources":["../../../src/telemetry/billingEvents.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EACL,qBAAqB,EACrB,0BAA0B,EAC1B,yBAAyB,EACzB,wBAAwB,EACxB,gBAAgB,EAChB,kBAAkB,EAClB,wBAAwB,EACxB,6BAA6B,EAC7B,6BAA6B,EAC7B,2BAA2B,EAC3B,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,oBAAoB,CAAC;AAE5B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;IAEpC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,0BAA0B,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,qBAAqB,CACrC,sBAAsB,EACtB,GAAG,EACH,KAAK,CACN,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACvD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,qBAAqB,CACrC,sBAAsB,EACtB,GAAG,EACH,KAAK,CACN,CAAC;YACF,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC3D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,qBAAqB,CACrC,sBAAsB,EACtB,GAAG,EACH,KAAK,CACN,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,0BAA0B,CAC1C,sBAAsB,EACtB,aAAa,EACb,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,0BAA0B,CAC1C,sBAAsB,EACtB,cAAc,EACd,GAAG,CACJ,CAAC;YACF,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,0BAA0B,CAC1C,sBAAsB,EACtB,QAAQ,EACR,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,yBAAyB,CAAC,sBAAsB,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,yBAAyB,CAAC,sBAAsB,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,yBAAyB,CAAC,sBAAsB,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,wBAAwB,CACxC,mBAAmB,EACnB,sBAAsB,CACvB,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,wBAAwB,CACxC,cAAc,EACd,sBAAsB,CACvB,CAAC;YACF,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAC9D,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,wBAAwB,CACxC,QAAQ,EACR,sBAAsB,CACvB,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,sBAAsB,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxC,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,sBAAsB,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,gBAAgB,CAAC,sBAAsB,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;YACpE,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACtD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,KAAK,GAAG,KAAK,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACxD,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACzD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import type { Config } from '../config/config.js';
|
|
7
|
+
import type { ConsecaPolicyGenerationEvent, ConsecaVerdictEvent } from './types.js';
|
|
8
|
+
export declare function logConsecaPolicyGeneration(config: Config, event: ConsecaPolicyGenerationEvent): void;
|
|
9
|
+
export declare function logConsecaVerdict(config: Config, event: ConsecaVerdictEvent): void;
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { logs } from '@opentelemetry/api-logs';
|
|
7
|
+
import { SERVICE_NAME } from './constants.js';
|
|
8
|
+
import { isTelemetrySdkInitialized } from './sdk.js';
|
|
9
|
+
import { ClearcutLogger, EventNames, } from './clearcut-logger/clearcut-logger.js';
|
|
10
|
+
import { EventMetadataKey } from './clearcut-logger/event-metadata-key.js';
|
|
11
|
+
import { safeJsonStringify } from '../utils/safeJsonStringify.js';
|
|
12
|
+
import { debugLogger } from '../utils/debugLogger.js';
|
|
13
|
+
export function logConsecaPolicyGeneration(config, event) {
|
|
14
|
+
debugLogger.debug('Conseca Policy Generation Event:', event);
|
|
15
|
+
const clearcutLogger = ClearcutLogger.getInstance(config);
|
|
16
|
+
if (clearcutLogger) {
|
|
17
|
+
const data = [
|
|
18
|
+
{
|
|
19
|
+
gemini_cli_key: EventMetadataKey.CONSECA_USER_PROMPT,
|
|
20
|
+
value: safeJsonStringify(event.user_prompt),
|
|
21
|
+
},
|
|
22
|
+
{
|
|
23
|
+
gemini_cli_key: EventMetadataKey.CONSECA_TRUSTED_CONTENT,
|
|
24
|
+
value: safeJsonStringify(event.trusted_content),
|
|
25
|
+
},
|
|
26
|
+
{
|
|
27
|
+
gemini_cli_key: EventMetadataKey.CONSECA_GENERATED_POLICY,
|
|
28
|
+
value: safeJsonStringify(event.policy),
|
|
29
|
+
},
|
|
30
|
+
];
|
|
31
|
+
if (event.error) {
|
|
32
|
+
data.push({
|
|
33
|
+
gemini_cli_key: EventMetadataKey.CONSECA_ERROR,
|
|
34
|
+
value: event.error,
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
clearcutLogger.enqueueLogEvent(clearcutLogger.createLogEvent(EventNames.CONSECA_POLICY_GENERATION, data));
|
|
38
|
+
}
|
|
39
|
+
if (!isTelemetrySdkInitialized())
|
|
40
|
+
return;
|
|
41
|
+
const logger = logs.getLogger(SERVICE_NAME);
|
|
42
|
+
const logRecord = {
|
|
43
|
+
body: event.toLogBody(),
|
|
44
|
+
attributes: event.toOpenTelemetryAttributes(config),
|
|
45
|
+
};
|
|
46
|
+
logger.emit(logRecord);
|
|
47
|
+
}
|
|
48
|
+
export function logConsecaVerdict(config, event) {
|
|
49
|
+
debugLogger.debug('Conseca Verdict Event:', event);
|
|
50
|
+
const clearcutLogger = ClearcutLogger.getInstance(config);
|
|
51
|
+
if (clearcutLogger) {
|
|
52
|
+
const data = [
|
|
53
|
+
{
|
|
54
|
+
gemini_cli_key: EventMetadataKey.CONSECA_USER_PROMPT,
|
|
55
|
+
value: safeJsonStringify(event.user_prompt),
|
|
56
|
+
},
|
|
57
|
+
{
|
|
58
|
+
gemini_cli_key: EventMetadataKey.CONSECA_GENERATED_POLICY,
|
|
59
|
+
value: safeJsonStringify(event.policy),
|
|
60
|
+
},
|
|
61
|
+
{
|
|
62
|
+
gemini_cli_key: EventMetadataKey.GEMINI_CLI_TOOL_CALL_NAME,
|
|
63
|
+
value: safeJsonStringify(event.tool_call),
|
|
64
|
+
},
|
|
65
|
+
{
|
|
66
|
+
gemini_cli_key: EventMetadataKey.CONSECA_VERDICT_RESULT,
|
|
67
|
+
value: safeJsonStringify(event.verdict),
|
|
68
|
+
},
|
|
69
|
+
{
|
|
70
|
+
gemini_cli_key: EventMetadataKey.CONSECA_VERDICT_RATIONALE,
|
|
71
|
+
value: event.verdict_rationale,
|
|
72
|
+
},
|
|
73
|
+
];
|
|
74
|
+
if (event.error) {
|
|
75
|
+
data.push({
|
|
76
|
+
gemini_cli_key: EventMetadataKey.CONSECA_ERROR,
|
|
77
|
+
value: event.error,
|
|
78
|
+
});
|
|
79
|
+
}
|
|
80
|
+
clearcutLogger.enqueueLogEvent(clearcutLogger.createLogEvent(EventNames.CONSECA_VERDICT, data));
|
|
81
|
+
}
|
|
82
|
+
if (!isTelemetrySdkInitialized())
|
|
83
|
+
return;
|
|
84
|
+
const logger = logs.getLogger(SERVICE_NAME);
|
|
85
|
+
const logRecord = {
|
|
86
|
+
body: event.toLogBody(),
|
|
87
|
+
attributes: event.toOpenTelemetryAttributes(config),
|
|
88
|
+
};
|
|
89
|
+
logger.emit(logRecord);
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=conseca-logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conseca-logger.js","sourceRoot":"","sources":["../../../src/telemetry/conseca-logger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,IAAI,EAAkB,MAAM,yBAAyB,CAAC;AAE/D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AACrD,OAAO,EACL,cAAc,EACd,UAAU,GACX,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,yCAAyC,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAKlE,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,MAAM,UAAU,0BAA0B,CACxC,MAAc,EACd,KAAmC;IAEnC,WAAW,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;IAC7D,MAAM,cAAc,GAAG,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,IAAI,GAAG;YACX;gBACE,cAAc,EAAE,gBAAgB,CAAC,mBAAmB;gBACpD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,WAAW,CAAC;aAC5C;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,uBAAuB;gBACxD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,eAAe,CAAC;aAChD;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,wBAAwB;gBACzD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC;aACvC;SACF,CAAC;QAEF,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC;gBACR,cAAc,EAAE,gBAAgB,CAAC,aAAa;gBAC9C,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAC;QACL,CAAC;QAED,cAAc,CAAC,eAAe,CAC5B,cAAc,CAAC,cAAc,CAAC,UAAU,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAC1E,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,yBAAyB,EAAE;QAAE,OAAO;IAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,SAAS,GAAc;QAC3B,IAAI,EAAE,KAAK,CAAC,SAAS,EAAE;QACvB,UAAU,EAAE,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC;KACpD,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,MAAc,EACd,KAA0B;IAE1B,WAAW,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;IACnD,MAAM,cAAc,GAAG,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,IAAI,GAAG;YACX;gBACE,cAAc,EAAE,gBAAgB,CAAC,mBAAmB;gBACpD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,WAAW,CAAC;aAC5C;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,wBAAwB;gBACzD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC;aACvC;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,yBAAyB;gBAC1D,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC;aAC1C;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,sBAAsB;gBACvD,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC;aACxC;YACD;gBACE,cAAc,EAAE,gBAAgB,CAAC,yBAAyB;gBAC1D,KAAK,EAAE,KAAK,CAAC,iBAAiB;aAC/B;SACF,CAAC;QAEF,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC;gBACR,cAAc,EAAE,gBAAgB,CAAC,aAAa;gBAC9C,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAC;QACL,CAAC;QAED,cAAc,CAAC,eAAe,CAC5B,cAAc,CAAC,cAAc,CAAC,UAAU,CAAC,eAAe,EAAE,IAAI,CAAC,CAChE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,yBAAyB,EAAE;QAAE,OAAO;IAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,SAAS,GAAc;QAC3B,IAAI,EAAE,KAAK,CAAC,SAAS,EAAE;QACvB,UAAU,EAAE,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC;KACpD,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACzB,CAAC"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
7
|
+
import { logs } from '@opentelemetry/api-logs';
|
|
8
|
+
import { logConsecaPolicyGeneration, logConsecaVerdict, } from './conseca-logger.js';
|
|
9
|
+
import { ConsecaPolicyGenerationEvent, ConsecaVerdictEvent, EVENT_CONSECA_POLICY_GENERATION, EVENT_CONSECA_VERDICT, } from './types.js';
|
|
10
|
+
import * as sdk from './sdk.js';
|
|
11
|
+
import { ClearcutLogger } from './clearcut-logger/clearcut-logger.js';
|
|
12
|
+
vi.mock('@opentelemetry/api-logs');
|
|
13
|
+
vi.mock('./sdk.js');
|
|
14
|
+
vi.mock('./clearcut-logger/clearcut-logger.js');
|
|
15
|
+
describe('conseca-logger', () => {
|
|
16
|
+
let mockConfig;
|
|
17
|
+
let mockLogger;
|
|
18
|
+
let mockClearcutLogger;
|
|
19
|
+
beforeEach(() => {
|
|
20
|
+
mockConfig = {
|
|
21
|
+
getTelemetryEnabled: vi.fn().mockReturnValue(true),
|
|
22
|
+
getSessionId: vi.fn().mockReturnValue('test-session-id'),
|
|
23
|
+
getTelemetryLogPromptsEnabled: vi.fn().mockReturnValue(true),
|
|
24
|
+
isInteractive: vi.fn().mockReturnValue(true),
|
|
25
|
+
getExperiments: vi.fn().mockReturnValue({ experimentIds: [] }),
|
|
26
|
+
getContentGeneratorConfig: vi.fn().mockReturnValue({ authType: 'oauth' }),
|
|
27
|
+
};
|
|
28
|
+
mockLogger = {
|
|
29
|
+
emit: vi.fn(),
|
|
30
|
+
};
|
|
31
|
+
vi.mocked(logs.getLogger).mockReturnValue(mockLogger);
|
|
32
|
+
vi.mocked(sdk.isTelemetrySdkInitialized).mockReturnValue(true);
|
|
33
|
+
mockClearcutLogger = {
|
|
34
|
+
enqueueLogEvent: vi.fn(),
|
|
35
|
+
createLogEvent: vi.fn().mockReturnValue({ event_name: 'test' }),
|
|
36
|
+
};
|
|
37
|
+
vi.mocked(ClearcutLogger.getInstance).mockReturnValue(mockClearcutLogger);
|
|
38
|
+
});
|
|
39
|
+
afterEach(() => {
|
|
40
|
+
vi.clearAllMocks();
|
|
41
|
+
});
|
|
42
|
+
it('should log policy generation event to OTEL and Clearcut', () => {
|
|
43
|
+
const event = new ConsecaPolicyGenerationEvent('user prompt', 'trusted content', 'generated policy');
|
|
44
|
+
logConsecaPolicyGeneration(mockConfig, event);
|
|
45
|
+
// Verify OTEL
|
|
46
|
+
expect(logs.getLogger).toHaveBeenCalled();
|
|
47
|
+
expect(mockLogger.emit).toHaveBeenCalledWith(expect.objectContaining({
|
|
48
|
+
body: 'Conseca Policy Generation.',
|
|
49
|
+
attributes: expect.objectContaining({
|
|
50
|
+
'event.name': EVENT_CONSECA_POLICY_GENERATION,
|
|
51
|
+
}),
|
|
52
|
+
}));
|
|
53
|
+
// Verify Clearcut
|
|
54
|
+
expect(ClearcutLogger.getInstance).toHaveBeenCalledWith(mockConfig);
|
|
55
|
+
expect(mockClearcutLogger.createLogEvent).toHaveBeenCalled();
|
|
56
|
+
expect(mockClearcutLogger.enqueueLogEvent).toHaveBeenCalled();
|
|
57
|
+
});
|
|
58
|
+
it('should log policy generation error to Clearcut', () => {
|
|
59
|
+
const event = new ConsecaPolicyGenerationEvent('user prompt', 'trusted content', '{}', 'some error');
|
|
60
|
+
logConsecaPolicyGeneration(mockConfig, event);
|
|
61
|
+
expect(mockClearcutLogger.createLogEvent).toHaveBeenCalledWith(expect.anything(), expect.arrayContaining([
|
|
62
|
+
expect.objectContaining({
|
|
63
|
+
value: 'some error',
|
|
64
|
+
}),
|
|
65
|
+
]));
|
|
66
|
+
});
|
|
67
|
+
it('should log verdict event to OTEL and Clearcut', () => {
|
|
68
|
+
const event = new ConsecaVerdictEvent('user prompt', 'policy', 'tool call', 'ALLOW', 'rationale');
|
|
69
|
+
logConsecaVerdict(mockConfig, event);
|
|
70
|
+
// Verify OTEL
|
|
71
|
+
expect(logs.getLogger).toHaveBeenCalled();
|
|
72
|
+
expect(mockLogger.emit).toHaveBeenCalledWith(expect.objectContaining({
|
|
73
|
+
body: 'Conseca Verdict: ALLOW.',
|
|
74
|
+
attributes: expect.objectContaining({
|
|
75
|
+
'event.name': EVENT_CONSECA_VERDICT,
|
|
76
|
+
}),
|
|
77
|
+
}));
|
|
78
|
+
// Verify Clearcut
|
|
79
|
+
expect(ClearcutLogger.getInstance).toHaveBeenCalledWith(mockConfig);
|
|
80
|
+
expect(mockClearcutLogger.createLogEvent).toHaveBeenCalled();
|
|
81
|
+
expect(mockClearcutLogger.enqueueLogEvent).toHaveBeenCalled();
|
|
82
|
+
});
|
|
83
|
+
it('should not log if SDK is not initialized', () => {
|
|
84
|
+
vi.mocked(sdk.isTelemetrySdkInitialized).mockReturnValue(false);
|
|
85
|
+
const event = new ConsecaPolicyGenerationEvent('a', 'b', 'c');
|
|
86
|
+
logConsecaPolicyGeneration(mockConfig, event);
|
|
87
|
+
expect(mockLogger.emit).not.toHaveBeenCalled();
|
|
88
|
+
});
|
|
89
|
+
});
|
|
90
|
+
//# sourceMappingURL=conseca-logger.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conseca-logger.test.js","sourceRoot":"","sources":["../../../src/telemetry/conseca-logger.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,IAAI,EAAe,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EACL,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,4BAA4B,EAC5B,mBAAmB,EACnB,+BAA+B,EAC/B,qBAAqB,GACtB,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,GAAG,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AAEtE,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AACpB,EAAE,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;AAEhD,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAAkB,CAAC;IACvB,IAAI,UAA8C,CAAC;IACnD,IAAI,kBAGH,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAClD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;YACxD,6BAA6B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5D,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;YAC9D,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;SACrD,CAAC;QAEvB,UAAU,GAAG;YACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;SACd,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,UAA+B,CAAC,CAAC;QAC3E,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAE/D,kBAAkB,GAAG;YACnB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;SAChE,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,eAAe,CACnD,kBAA+C,CAChD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,KAAK,GAAG,IAAI,4BAA4B,CAC5C,aAAa,EACb,iBAAiB,EACjB,kBAAkB,CACnB,CAAC;QAEF,0BAA0B,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAE9C,cAAc;QACd,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,4BAA4B;YAClC,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAClC,YAAY,EAAE,+BAA+B;aAC9C,CAAC;SACH,CAAC,CACH,CAAC;QAEF,kBAAkB;QAClB,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACpE,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7D,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,KAAK,GAAG,IAAI,4BAA4B,CAC5C,aAAa,EACb,iBAAiB,EACjB,IAAI,EACJ,YAAY,CACb,CAAC;QAEF,0BAA0B,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAE9C,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAC5D,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,YAAY;aACpB,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG,IAAI,mBAAmB,CACnC,aAAa,EACb,QAAQ,EACR,WAAW,EACX,OAAO,EACP,WAAW,CACZ,CAAC;QAEF,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErC,cAAc;QACd,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,yBAAyB;YAC/B,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAClC,YAAY,EAAE,qBAAqB;aACpC,CAAC;SACH,CAAC,CACH,CAAC;QAEF,kBAAkB;QAClB,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACpE,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7D,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,IAAI,4BAA4B,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE9D,0BAA0B,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAE9C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|