@vybestack/llxprt-code-core 0.1.12
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/.last_build +0 -0
- package/dist/index.d.ts +7 -0
- package/dist/index.js +8 -0
- package/dist/index.js.map +1 -0
- package/dist/src/__mocks__/fs/promises.d.ts +11 -0
- package/dist/src/__mocks__/fs/promises.js +17 -0
- package/dist/src/__mocks__/fs/promises.js.map +1 -0
- package/dist/src/code_assist/codeAssist.d.ts +9 -0
- package/dist/src/code_assist/codeAssist.js +19 -0
- package/dist/src/code_assist/codeAssist.js.map +1 -0
- package/dist/src/code_assist/converter.d.ts +68 -0
- package/dist/src/code_assist/converter.js +125 -0
- package/dist/src/code_assist/converter.js.map +1 -0
- package/dist/src/code_assist/converter.test.d.ts +6 -0
- package/dist/src/code_assist/converter.test.js +229 -0
- package/dist/src/code_assist/converter.test.js.map +1 -0
- package/dist/src/code_assist/oauth2.d.ts +20 -0
- package/dist/src/code_assist/oauth2.js +339 -0
- package/dist/src/code_assist/oauth2.js.map +1 -0
- package/dist/src/code_assist/oauth2.test.d.ts +6 -0
- package/dist/src/code_assist/oauth2.test.js +249 -0
- package/dist/src/code_assist/oauth2.test.js.map +1 -0
- package/dist/src/code_assist/server.d.ts +39 -0
- package/dist/src/code_assist/server.js +191 -0
- package/dist/src/code_assist/server.js.map +1 -0
- package/dist/src/code_assist/server.test.d.ts +6 -0
- package/dist/src/code_assist/server.test.js +123 -0
- package/dist/src/code_assist/server.test.js.map +1 -0
- package/dist/src/code_assist/setup.d.ts +15 -0
- package/dist/src/code_assist/setup.js +67 -0
- package/dist/src/code_assist/setup.js.map +1 -0
- package/dist/src/code_assist/setup.test.d.ts +6 -0
- package/dist/src/code_assist/setup.test.js +62 -0
- package/dist/src/code_assist/setup.test.js.map +1 -0
- package/dist/src/code_assist/types.d.ts +148 -0
- package/dist/src/code_assist/types.js +46 -0
- package/dist/src/code_assist/types.js.map +1 -0
- package/dist/src/config/config.d.ts +212 -0
- package/dist/src/config/config.js +454 -0
- package/dist/src/config/config.js.map +1 -0
- package/dist/src/config/config.test.d.ts +6 -0
- package/dist/src/config/config.test.js +291 -0
- package/dist/src/config/config.test.js.map +1 -0
- package/dist/src/config/flashFallback.test.d.ts +6 -0
- package/dist/src/config/flashFallback.test.js +115 -0
- package/dist/src/config/flashFallback.test.js.map +1 -0
- package/dist/src/config/models.d.ts +8 -0
- package/dist/src/config/models.js +9 -0
- package/dist/src/config/models.js.map +1 -0
- package/dist/src/core/client.d.ts +59 -0
- package/dist/src/core/client.js +503 -0
- package/dist/src/core/client.js.map +1 -0
- package/dist/src/core/client.test.d.ts +6 -0
- package/dist/src/core/client.test.js +1006 -0
- package/dist/src/core/client.test.js.map +1 -0
- package/dist/src/core/contentGenerator.d.ts +36 -0
- package/dist/src/core/contentGenerator.js +79 -0
- package/dist/src/core/contentGenerator.js.map +1 -0
- package/dist/src/core/contentGenerator.test.d.ts +6 -0
- package/dist/src/core/contentGenerator.test.js +92 -0
- package/dist/src/core/contentGenerator.test.js.map +1 -0
- package/dist/src/core/coreToolScheduler.d.ts +104 -0
- package/dist/src/core/coreToolScheduler.js +413 -0
- package/dist/src/core/coreToolScheduler.js.map +1 -0
- package/dist/src/core/coreToolScheduler.test.d.ts +6 -0
- package/dist/src/core/coreToolScheduler.test.js +347 -0
- package/dist/src/core/coreToolScheduler.test.js.map +1 -0
- package/dist/src/core/geminiChat.d.ts +116 -0
- package/dist/src/core/geminiChat.js +513 -0
- package/dist/src/core/geminiChat.js.map +1 -0
- package/dist/src/core/geminiChat.test.d.ts +6 -0
- package/dist/src/core/geminiChat.test.js +424 -0
- package/dist/src/core/geminiChat.test.js.map +1 -0
- package/dist/src/core/geminiRequest.d.ts +13 -0
- package/dist/src/core/geminiRequest.js +45 -0
- package/dist/src/core/geminiRequest.js.map +1 -0
- package/dist/src/core/geminiRequest.test.d.ts +6 -0
- package/dist/src/core/geminiRequest.test.js +72 -0
- package/dist/src/core/geminiRequest.test.js.map +1 -0
- package/dist/src/core/logger.d.ts +38 -0
- package/dist/src/core/logger.js +237 -0
- package/dist/src/core/logger.js.map +1 -0
- package/dist/src/core/logger.test.d.ts +6 -0
- package/dist/src/core/logger.test.js +387 -0
- package/dist/src/core/logger.test.js.map +1 -0
- package/dist/src/core/modelCheck.d.ts +14 -0
- package/dist/src/core/modelCheck.js +55 -0
- package/dist/src/core/modelCheck.js.map +1 -0
- package/dist/src/core/nonInteractiveToolExecutor.d.ts +12 -0
- package/dist/src/core/nonInteractiveToolExecutor.js +93 -0
- package/dist/src/core/nonInteractiveToolExecutor.js.map +1 -0
- package/dist/src/core/nonInteractiveToolExecutor.test.d.ts +6 -0
- package/dist/src/core/nonInteractiveToolExecutor.test.js +180 -0
- package/dist/src/core/nonInteractiveToolExecutor.test.js.map +1 -0
- package/dist/src/core/prompts.d.ts +12 -0
- package/dist/src/core/prompts.js +418 -0
- package/dist/src/core/prompts.js.map +1 -0
- package/dist/src/core/prompts.test.d.ts +6 -0
- package/dist/src/core/prompts.test.js +97 -0
- package/dist/src/core/prompts.test.js.map +1 -0
- package/dist/src/core/tokenLimits.d.ts +10 -0
- package/dist/src/core/tokenLimits.js +75 -0
- package/dist/src/core/tokenLimits.js.map +1 -0
- package/dist/src/core/tokenLimits.test.d.ts +6 -0
- package/dist/src/core/tokenLimits.test.js +66 -0
- package/dist/src/core/tokenLimits.test.js.map +1 -0
- package/dist/src/core/turn.d.ts +116 -0
- package/dist/src/core/turn.js +131 -0
- package/dist/src/core/turn.js.map +1 -0
- package/dist/src/core/turn.test.d.ts +6 -0
- package/dist/src/core/turn.test.js +237 -0
- package/dist/src/core/turn.test.js.map +1 -0
- package/dist/src/index.d.ts +57 -0
- package/dist/src/index.js +64 -0
- package/dist/src/index.js.map +1 -0
- package/dist/src/index.test.d.ts +6 -0
- package/dist/src/index.test.js +12 -0
- package/dist/src/index.test.js.map +1 -0
- package/dist/src/providers/ProviderContentGenerator.d.ts +21 -0
- package/dist/src/providers/ProviderContentGenerator.js +52 -0
- package/dist/src/providers/ProviderContentGenerator.js.map +1 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.d.ts +68 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.js +549 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.js.map +1 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.test.d.ts +6 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.test.js +253 -0
- package/dist/src/providers/adapters/GeminiCompatibleWrapper.test.js.map +1 -0
- package/dist/src/providers/types.d.ts +82 -0
- package/dist/src/providers/types.js +7 -0
- package/dist/src/providers/types.js.map +1 -0
- package/dist/src/services/fileDiscoveryService.d.ts +35 -0
- package/dist/src/services/fileDiscoveryService.js +91 -0
- package/dist/src/services/fileDiscoveryService.js.map +1 -0
- package/dist/src/services/fileDiscoveryService.test.d.ts +6 -0
- package/dist/src/services/fileDiscoveryService.test.js +102 -0
- package/dist/src/services/fileDiscoveryService.test.js.map +1 -0
- package/dist/src/services/gitService.d.ts +21 -0
- package/dist/src/services/gitService.js +102 -0
- package/dist/src/services/gitService.js.map +1 -0
- package/dist/src/services/gitService.test.d.ts +6 -0
- package/dist/src/services/gitService.test.js +246 -0
- package/dist/src/services/gitService.test.js.map +1 -0
- package/dist/src/services/ideContext.d.ts +124 -0
- package/dist/src/services/ideContext.js +90 -0
- package/dist/src/services/ideContext.js.map +1 -0
- package/dist/src/services/ideContext.test.d.ts +6 -0
- package/dist/src/services/ideContext.test.js +101 -0
- package/dist/src/services/ideContext.test.js.map +1 -0
- package/dist/src/services/loopDetectionService.d.ts +35 -0
- package/dist/src/services/loopDetectionService.js +117 -0
- package/dist/src/services/loopDetectionService.js.map +1 -0
- package/dist/src/services/loopDetectionService.test.d.ts +6 -0
- package/dist/src/services/loopDetectionService.test.js +232 -0
- package/dist/src/services/loopDetectionService.test.js.map +1 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts +35 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js +424 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js.map +1 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.d.ts +47 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js +123 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js.map +1 -0
- package/dist/src/telemetry/constants.d.ts +20 -0
- package/dist/src/telemetry/constants.js +21 -0
- package/dist/src/telemetry/constants.js.map +1 -0
- package/dist/src/telemetry/index.d.ts +18 -0
- package/dist/src/telemetry/index.js +20 -0
- package/dist/src/telemetry/index.js.map +1 -0
- package/dist/src/telemetry/integration.test.circular.d.ts +6 -0
- package/dist/src/telemetry/integration.test.circular.js +53 -0
- package/dist/src/telemetry/integration.test.circular.js.map +1 -0
- package/dist/src/telemetry/loggers.d.ts +15 -0
- package/dist/src/telemetry/loggers.js +229 -0
- package/dist/src/telemetry/loggers.js.map +1 -0
- package/dist/src/telemetry/loggers.test.circular.d.ts +6 -0
- package/dist/src/telemetry/loggers.test.circular.js +100 -0
- package/dist/src/telemetry/loggers.test.circular.js.map +1 -0
- package/dist/src/telemetry/loggers.test.d.ts +6 -0
- package/dist/src/telemetry/loggers.test.js +558 -0
- package/dist/src/telemetry/loggers.test.js.map +1 -0
- package/dist/src/telemetry/metrics.d.ts +19 -0
- package/dist/src/telemetry/metrics.js +144 -0
- package/dist/src/telemetry/metrics.js.map +1 -0
- package/dist/src/telemetry/metrics.test.d.ts +6 -0
- package/dist/src/telemetry/metrics.test.js +162 -0
- package/dist/src/telemetry/metrics.test.js.map +1 -0
- package/dist/src/telemetry/sdk.d.ts +9 -0
- package/dist/src/telemetry/sdk.js +120 -0
- package/dist/src/telemetry/sdk.js.map +1 -0
- package/dist/src/telemetry/telemetry.test.d.ts +6 -0
- package/dist/src/telemetry/telemetry.test.js +52 -0
- package/dist/src/telemetry/telemetry.test.js.map +1 -0
- package/dist/src/telemetry/types.d.ts +115 -0
- package/dist/src/telemetry/types.js +217 -0
- package/dist/src/telemetry/types.js.map +1 -0
- package/dist/src/telemetry/uiTelemetry.d.ts +67 -0
- package/dist/src/telemetry/uiTelemetry.js +131 -0
- package/dist/src/telemetry/uiTelemetry.js.map +1 -0
- package/dist/src/telemetry/uiTelemetry.test.d.ts +6 -0
- package/dist/src/telemetry/uiTelemetry.test.js +412 -0
- package/dist/src/telemetry/uiTelemetry.test.js.map +1 -0
- package/dist/src/tools/diffOptions.d.ts +7 -0
- package/dist/src/tools/diffOptions.js +10 -0
- package/dist/src/tools/diffOptions.js.map +1 -0
- package/dist/src/tools/edit.d.ts +73 -0
- package/dist/src/tools/edit.js +343 -0
- package/dist/src/tools/edit.js.map +1 -0
- package/dist/src/tools/edit.test.d.ts +6 -0
- package/dist/src/tools/edit.test.js +500 -0
- package/dist/src/tools/edit.test.js.map +1 -0
- package/dist/src/tools/glob.d.ts +58 -0
- package/dist/src/tools/glob.js +193 -0
- package/dist/src/tools/glob.js.map +1 -0
- package/dist/src/tools/glob.test.d.ts +6 -0
- package/dist/src/tools/glob.test.js +294 -0
- package/dist/src/tools/glob.test.js.map +1 -0
- package/dist/src/tools/grep.d.ts +78 -0
- package/dist/src/tools/grep.js +422 -0
- package/dist/src/tools/grep.js.map +1 -0
- package/dist/src/tools/grep.test.d.ts +6 -0
- package/dist/src/tools/grep.test.js +188 -0
- package/dist/src/tools/grep.test.js.map +1 -0
- package/dist/src/tools/ls.d.ts +83 -0
- package/dist/src/tools/ls.js +194 -0
- package/dist/src/tools/ls.js.map +1 -0
- package/dist/src/tools/mcp-client.d.ts +112 -0
- package/dist/src/tools/mcp-client.js +332 -0
- package/dist/src/tools/mcp-client.js.map +1 -0
- package/dist/src/tools/mcp-client.test.d.ts +6 -0
- package/dist/src/tools/mcp-client.test.js +227 -0
- package/dist/src/tools/mcp-client.test.js.map +1 -0
- package/dist/src/tools/mcp-tool.d.ts +23 -0
- package/dist/src/tools/mcp-tool.js +119 -0
- package/dist/src/tools/mcp-tool.js.map +1 -0
- package/dist/src/tools/mcp-tool.test.d.ts +6 -0
- package/dist/src/tools/mcp-tool.test.js +173 -0
- package/dist/src/tools/mcp-tool.test.js.map +1 -0
- package/dist/src/tools/memoryTool.d.ts +28 -0
- package/dist/src/tools/memoryTool.js +172 -0
- package/dist/src/tools/memoryTool.js.map +1 -0
- package/dist/src/tools/memoryTool.test.d.ts +6 -0
- package/dist/src/tools/memoryTool.test.js +181 -0
- package/dist/src/tools/memoryTool.test.js.map +1 -0
- package/dist/src/tools/modifiable-tool.d.ts +29 -0
- package/dist/src/tools/modifiable-tool.js +85 -0
- package/dist/src/tools/modifiable-tool.js.map +1 -0
- package/dist/src/tools/modifiable-tool.test.d.ts +6 -0
- package/dist/src/tools/modifiable-tool.test.js +204 -0
- package/dist/src/tools/modifiable-tool.test.js.map +1 -0
- package/dist/src/tools/read-file.d.ts +35 -0
- package/dist/src/tools/read-file.js +106 -0
- package/dist/src/tools/read-file.js.map +1 -0
- package/dist/src/tools/read-file.test.d.ts +6 -0
- package/dist/src/tools/read-file.test.js +181 -0
- package/dist/src/tools/read-file.test.js.map +1 -0
- package/dist/src/tools/read-many-files.d.ts +59 -0
- package/dist/src/tools/read-many-files.js +321 -0
- package/dist/src/tools/read-many-files.js.map +1 -0
- package/dist/src/tools/read-many-files.test.d.ts +6 -0
- package/dist/src/tools/read-many-files.test.js +312 -0
- package/dist/src/tools/read-many-files.test.js.map +1 -0
- package/dist/src/tools/shell.d.ts +43 -0
- package/dist/src/tools/shell.js +435 -0
- package/dist/src/tools/shell.js.map +1 -0
- package/dist/src/tools/shell.test.d.ts +6 -0
- package/dist/src/tools/shell.test.js +407 -0
- package/dist/src/tools/shell.test.js.map +1 -0
- package/dist/src/tools/todo-read.d.ts +15 -0
- package/dist/src/tools/todo-read.js +66 -0
- package/dist/src/tools/todo-read.js.map +1 -0
- package/dist/src/tools/todo-read.test.d.ts +6 -0
- package/dist/src/tools/todo-read.test.js +158 -0
- package/dist/src/tools/todo-read.test.js.map +1 -0
- package/dist/src/tools/todo-schemas.d.ts +43 -0
- package/dist/src/tools/todo-schemas.js +16 -0
- package/dist/src/tools/todo-schemas.js.map +1 -0
- package/dist/src/tools/todo-schemas.test.d.ts +6 -0
- package/dist/src/tools/todo-schemas.test.js +152 -0
- package/dist/src/tools/todo-schemas.test.js.map +1 -0
- package/dist/src/tools/todo-store.d.ts +15 -0
- package/dist/src/tools/todo-store.js +59 -0
- package/dist/src/tools/todo-store.js.map +1 -0
- package/dist/src/tools/todo-store.test.d.ts +6 -0
- package/dist/src/tools/todo-store.test.js +175 -0
- package/dist/src/tools/todo-store.test.js.map +1 -0
- package/dist/src/tools/todo-write.d.ts +17 -0
- package/dist/src/tools/todo-write.js +98 -0
- package/dist/src/tools/todo-write.js.map +1 -0
- package/dist/src/tools/todo-write.test.d.ts +6 -0
- package/dist/src/tools/todo-write.test.js +208 -0
- package/dist/src/tools/todo-write.test.js.map +1 -0
- package/dist/src/tools/tool-registry.d.ts +67 -0
- package/dist/src/tools/tool-registry.js +326 -0
- package/dist/src/tools/tool-registry.js.map +1 -0
- package/dist/src/tools/tool-registry.test.d.ts +6 -0
- package/dist/src/tools/tool-registry.test.js +398 -0
- package/dist/src/tools/tool-registry.test.js.map +1 -0
- package/dist/src/tools/tools.d.ts +184 -0
- package/dist/src/tools/tools.js +88 -0
- package/dist/src/tools/tools.js.map +1 -0
- package/dist/src/tools/web-fetch.d.ts +29 -0
- package/dist/src/tools/web-fetch.js +243 -0
- package/dist/src/tools/web-fetch.js.map +1 -0
- package/dist/src/tools/web-fetch.test.d.ts +6 -0
- package/dist/src/tools/web-fetch.test.js +70 -0
- package/dist/src/tools/web-fetch.test.js.map +1 -0
- package/dist/src/tools/web-search.d.ts +49 -0
- package/dist/src/tools/web-search.js +120 -0
- package/dist/src/tools/web-search.js.map +1 -0
- package/dist/src/tools/write-file.d.ts +42 -0
- package/dist/src/tools/write-file.js +248 -0
- package/dist/src/tools/write-file.js.map +1 -0
- package/dist/src/tools/write-file.test.d.ts +6 -0
- package/dist/src/tools/write-file.test.js +413 -0
- package/dist/src/tools/write-file.test.js.map +1 -0
- package/dist/src/utils/LruCache.d.ts +13 -0
- package/dist/src/utils/LruCache.js +38 -0
- package/dist/src/utils/LruCache.js.map +1 -0
- package/dist/src/utils/bfsFileSearch.d.ts +22 -0
- package/dist/src/utils/bfsFileSearch.js +62 -0
- package/dist/src/utils/bfsFileSearch.js.map +1 -0
- package/dist/src/utils/bfsFileSearch.test.d.ts +6 -0
- package/dist/src/utils/bfsFileSearch.test.js +129 -0
- package/dist/src/utils/bfsFileSearch.test.js.map +1 -0
- package/dist/src/utils/editCorrector.d.ts +53 -0
- package/dist/src/utils/editCorrector.js +546 -0
- package/dist/src/utils/editCorrector.js.map +1 -0
- package/dist/src/utils/editCorrector.test.d.ts +6 -0
- package/dist/src/utils/editCorrector.test.js +564 -0
- package/dist/src/utils/editCorrector.test.js.map +1 -0
- package/dist/src/utils/editor.d.ts +28 -0
- package/dist/src/utils/editor.js +163 -0
- package/dist/src/utils/editor.js.map +1 -0
- package/dist/src/utils/editor.test.d.ts +6 -0
- package/dist/src/utils/editor.test.js +293 -0
- package/dist/src/utils/editor.test.js.map +1 -0
- package/dist/src/utils/errorReporting.d.ts +14 -0
- package/dist/src/utils/errorReporting.js +88 -0
- package/dist/src/utils/errorReporting.js.map +1 -0
- package/dist/src/utils/errorReporting.test.d.ts +6 -0
- package/dist/src/utils/errorReporting.test.js +124 -0
- package/dist/src/utils/errorReporting.test.js.map +1 -0
- package/dist/src/utils/errors.d.ts +14 -0
- package/dist/src/utils/errors.js +54 -0
- package/dist/src/utils/errors.js.map +1 -0
- package/dist/src/utils/fetch.d.ts +11 -0
- package/dist/src/utils/fetch.js +51 -0
- package/dist/src/utils/fetch.js.map +1 -0
- package/dist/src/utils/fileUtils.d.ts +49 -0
- package/dist/src/utils/fileUtils.js +286 -0
- package/dist/src/utils/fileUtils.js.map +1 -0
- package/dist/src/utils/fileUtils.test.d.ts +6 -0
- package/dist/src/utils/fileUtils.test.js +321 -0
- package/dist/src/utils/fileUtils.test.js.map +1 -0
- package/dist/src/utils/flashFallback.integration.test.d.ts +6 -0
- package/dist/src/utils/flashFallback.integration.test.js +112 -0
- package/dist/src/utils/flashFallback.integration.test.js.map +1 -0
- package/dist/src/utils/generateContentResponseUtilities.d.ts +14 -0
- package/dist/src/utils/generateContentResponseUtilities.js +92 -0
- package/dist/src/utils/generateContentResponseUtilities.js.map +1 -0
- package/dist/src/utils/generateContentResponseUtilities.test.d.ts +6 -0
- package/dist/src/utils/generateContentResponseUtilities.test.js +273 -0
- package/dist/src/utils/generateContentResponseUtilities.test.js.map +1 -0
- package/dist/src/utils/getFolderStructure.d.ts +30 -0
- package/dist/src/utils/getFolderStructure.js +247 -0
- package/dist/src/utils/getFolderStructure.js.map +1 -0
- package/dist/src/utils/getFolderStructure.test.d.ts +6 -0
- package/dist/src/utils/getFolderStructure.test.js +298 -0
- package/dist/src/utils/getFolderStructure.test.js.map +1 -0
- package/dist/src/utils/gitIgnoreParser.d.ts +20 -0
- package/dist/src/utils/gitIgnoreParser.js +64 -0
- package/dist/src/utils/gitIgnoreParser.js.map +1 -0
- package/dist/src/utils/gitIgnoreParser.test.d.ts +6 -0
- package/dist/src/utils/gitIgnoreParser.test.js +145 -0
- package/dist/src/utils/gitIgnoreParser.test.js.map +1 -0
- package/dist/src/utils/gitUtils.d.ts +17 -0
- package/dist/src/utils/gitUtils.js +61 -0
- package/dist/src/utils/gitUtils.js.map +1 -0
- package/dist/src/utils/memoryDiscovery.d.ts +14 -0
- package/dist/src/utils/memoryDiscovery.js +219 -0
- package/dist/src/utils/memoryDiscovery.js.map +1 -0
- package/dist/src/utils/memoryDiscovery.test.d.ts +6 -0
- package/dist/src/utils/memoryDiscovery.test.js +432 -0
- package/dist/src/utils/memoryDiscovery.test.js.map +1 -0
- package/dist/src/utils/memoryImportProcessor.d.ts +35 -0
- package/dist/src/utils/memoryImportProcessor.js +141 -0
- package/dist/src/utils/memoryImportProcessor.js.map +1 -0
- package/dist/src/utils/memoryImportProcessor.test.d.ts +6 -0
- package/dist/src/utils/memoryImportProcessor.test.js +170 -0
- package/dist/src/utils/memoryImportProcessor.test.js.map +1 -0
- package/dist/src/utils/messageInspectors.d.ts +8 -0
- package/dist/src/utils/messageInspectors.js +16 -0
- package/dist/src/utils/messageInspectors.js.map +1 -0
- package/dist/src/utils/nextSpeakerChecker.d.ts +12 -0
- package/dist/src/utils/nextSpeakerChecker.js +113 -0
- package/dist/src/utils/nextSpeakerChecker.js.map +1 -0
- package/dist/src/utils/nextSpeakerChecker.test.d.ts +6 -0
- package/dist/src/utils/nextSpeakerChecker.test.js +168 -0
- package/dist/src/utils/nextSpeakerChecker.test.js.map +1 -0
- package/dist/src/utils/paths.d.ts +49 -0
- package/dist/src/utils/paths.js +145 -0
- package/dist/src/utils/paths.js.map +1 -0
- package/dist/src/utils/quotaErrorDetection.d.ts +22 -0
- package/dist/src/utils/quotaErrorDetection.js +67 -0
- package/dist/src/utils/quotaErrorDetection.js.map +1 -0
- package/dist/src/utils/retry.d.ts +21 -0
- package/dist/src/utils/retry.js +276 -0
- package/dist/src/utils/retry.js.map +1 -0
- package/dist/src/utils/retry.test.d.ts +6 -0
- package/dist/src/utils/retry.test.js +322 -0
- package/dist/src/utils/retry.test.js.map +1 -0
- package/dist/src/utils/safeJsonStringify.d.ts +13 -0
- package/dist/src/utils/safeJsonStringify.js +25 -0
- package/dist/src/utils/safeJsonStringify.js.map +1 -0
- package/dist/src/utils/safeJsonStringify.test.d.ts +6 -0
- package/dist/src/utils/safeJsonStringify.test.js +61 -0
- package/dist/src/utils/safeJsonStringify.test.js.map +1 -0
- package/dist/src/utils/schemaValidator.d.ts +23 -0
- package/dist/src/utils/schemaValidator.js +142 -0
- package/dist/src/utils/schemaValidator.js.map +1 -0
- package/dist/src/utils/schemaValidator.test.d.ts +6 -0
- package/dist/src/utils/schemaValidator.test.js +146 -0
- package/dist/src/utils/schemaValidator.test.js.map +1 -0
- package/dist/src/utils/session.d.ts +6 -0
- package/dist/src/utils/session.js +8 -0
- package/dist/src/utils/session.js.map +1 -0
- package/dist/src/utils/summarizer.d.ts +25 -0
- package/dist/src/utils/summarizer.js +80 -0
- package/dist/src/utils/summarizer.js.map +1 -0
- package/dist/src/utils/summarizer.test.d.ts +6 -0
- package/dist/src/utils/summarizer.test.js +131 -0
- package/dist/src/utils/summarizer.test.js.map +1 -0
- package/dist/src/utils/testUtils.d.ts +29 -0
- package/dist/src/utils/testUtils.js +70 -0
- package/dist/src/utils/testUtils.js.map +1 -0
- package/dist/src/utils/user_account.d.ts +9 -0
- package/dist/src/utils/user_account.js +99 -0
- package/dist/src/utils/user_account.js.map +1 -0
- package/dist/src/utils/user_account.test.d.ts +6 -0
- package/dist/src/utils/user_account.test.js +153 -0
- package/dist/src/utils/user_account.test.js.map +1 -0
- package/dist/src/utils/user_id.d.ts +11 -0
- package/dist/src/utils/user_id.js +49 -0
- package/dist/src/utils/user_id.js.map +1 -0
- package/dist/src/utils/user_id.test.d.ts +6 -0
- package/dist/src/utils/user_id.test.js +21 -0
- package/dist/src/utils/user_id.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -0
- package/dist/vybestack-llxprt-code-core-0.1.12.tgz +0 -0
- package/package.json +62 -0
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { metrics, ValueType, } from '@opentelemetry/api';
|
|
7
|
+
import { SERVICE_NAME, METRIC_TOOL_CALL_COUNT, METRIC_TOOL_CALL_LATENCY, METRIC_API_REQUEST_COUNT, METRIC_API_REQUEST_LATENCY, METRIC_TOKEN_USAGE, METRIC_SESSION_COUNT, METRIC_FILE_OPERATION_COUNT, } from './constants.js';
|
|
8
|
+
export var FileOperation;
|
|
9
|
+
(function (FileOperation) {
|
|
10
|
+
FileOperation["CREATE"] = "create";
|
|
11
|
+
FileOperation["READ"] = "read";
|
|
12
|
+
FileOperation["UPDATE"] = "update";
|
|
13
|
+
})(FileOperation || (FileOperation = {}));
|
|
14
|
+
let cliMeter;
|
|
15
|
+
let toolCallCounter;
|
|
16
|
+
let toolCallLatencyHistogram;
|
|
17
|
+
let apiRequestCounter;
|
|
18
|
+
let apiRequestLatencyHistogram;
|
|
19
|
+
let tokenUsageCounter;
|
|
20
|
+
let fileOperationCounter;
|
|
21
|
+
let isMetricsInitialized = false;
|
|
22
|
+
function getCommonAttributes(config) {
|
|
23
|
+
return {
|
|
24
|
+
'session.id': config.getSessionId(),
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
export function getMeter() {
|
|
28
|
+
if (!cliMeter) {
|
|
29
|
+
cliMeter = metrics.getMeter(SERVICE_NAME);
|
|
30
|
+
}
|
|
31
|
+
return cliMeter;
|
|
32
|
+
}
|
|
33
|
+
export function initializeMetrics(config) {
|
|
34
|
+
if (isMetricsInitialized)
|
|
35
|
+
return;
|
|
36
|
+
const meter = getMeter();
|
|
37
|
+
if (!meter)
|
|
38
|
+
return;
|
|
39
|
+
toolCallCounter = meter.createCounter(METRIC_TOOL_CALL_COUNT, {
|
|
40
|
+
description: 'Counts tool calls, tagged by function name and success.',
|
|
41
|
+
valueType: ValueType.INT,
|
|
42
|
+
});
|
|
43
|
+
toolCallLatencyHistogram = meter.createHistogram(METRIC_TOOL_CALL_LATENCY, {
|
|
44
|
+
description: 'Latency of tool calls in milliseconds.',
|
|
45
|
+
unit: 'ms',
|
|
46
|
+
valueType: ValueType.INT,
|
|
47
|
+
});
|
|
48
|
+
apiRequestCounter = meter.createCounter(METRIC_API_REQUEST_COUNT, {
|
|
49
|
+
description: 'Counts API requests, tagged by model and status.',
|
|
50
|
+
valueType: ValueType.INT,
|
|
51
|
+
});
|
|
52
|
+
apiRequestLatencyHistogram = meter.createHistogram(METRIC_API_REQUEST_LATENCY, {
|
|
53
|
+
description: 'Latency of API requests in milliseconds.',
|
|
54
|
+
unit: 'ms',
|
|
55
|
+
valueType: ValueType.INT,
|
|
56
|
+
});
|
|
57
|
+
tokenUsageCounter = meter.createCounter(METRIC_TOKEN_USAGE, {
|
|
58
|
+
description: 'Counts the total number of tokens used.',
|
|
59
|
+
valueType: ValueType.INT,
|
|
60
|
+
});
|
|
61
|
+
fileOperationCounter = meter.createCounter(METRIC_FILE_OPERATION_COUNT, {
|
|
62
|
+
description: 'Counts file operations (create, read, update).',
|
|
63
|
+
valueType: ValueType.INT,
|
|
64
|
+
});
|
|
65
|
+
const sessionCounter = meter.createCounter(METRIC_SESSION_COUNT, {
|
|
66
|
+
description: 'Count of CLI sessions started.',
|
|
67
|
+
valueType: ValueType.INT,
|
|
68
|
+
});
|
|
69
|
+
sessionCounter.add(1, getCommonAttributes(config));
|
|
70
|
+
isMetricsInitialized = true;
|
|
71
|
+
}
|
|
72
|
+
export function recordToolCallMetrics(config, functionName, durationMs, success, decision) {
|
|
73
|
+
if (!toolCallCounter || !toolCallLatencyHistogram || !isMetricsInitialized)
|
|
74
|
+
return;
|
|
75
|
+
const metricAttributes = {
|
|
76
|
+
...getCommonAttributes(config),
|
|
77
|
+
function_name: functionName,
|
|
78
|
+
success,
|
|
79
|
+
decision,
|
|
80
|
+
};
|
|
81
|
+
toolCallCounter.add(1, metricAttributes);
|
|
82
|
+
toolCallLatencyHistogram.record(durationMs, {
|
|
83
|
+
...getCommonAttributes(config),
|
|
84
|
+
function_name: functionName,
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
export function recordTokenUsageMetrics(config, model, tokenCount, type) {
|
|
88
|
+
if (!tokenUsageCounter || !isMetricsInitialized)
|
|
89
|
+
return;
|
|
90
|
+
tokenUsageCounter.add(tokenCount, {
|
|
91
|
+
...getCommonAttributes(config),
|
|
92
|
+
model,
|
|
93
|
+
type,
|
|
94
|
+
});
|
|
95
|
+
}
|
|
96
|
+
export function recordApiResponseMetrics(config, model, durationMs, statusCode, error) {
|
|
97
|
+
if (!apiRequestCounter ||
|
|
98
|
+
!apiRequestLatencyHistogram ||
|
|
99
|
+
!isMetricsInitialized)
|
|
100
|
+
return;
|
|
101
|
+
const metricAttributes = {
|
|
102
|
+
...getCommonAttributes(config),
|
|
103
|
+
model,
|
|
104
|
+
status_code: statusCode ?? (error ? 'error' : 'ok'),
|
|
105
|
+
};
|
|
106
|
+
apiRequestCounter.add(1, metricAttributes);
|
|
107
|
+
apiRequestLatencyHistogram.record(durationMs, {
|
|
108
|
+
...getCommonAttributes(config),
|
|
109
|
+
model,
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
export function recordApiErrorMetrics(config, model, durationMs, statusCode, errorType) {
|
|
113
|
+
if (!apiRequestCounter ||
|
|
114
|
+
!apiRequestLatencyHistogram ||
|
|
115
|
+
!isMetricsInitialized)
|
|
116
|
+
return;
|
|
117
|
+
const metricAttributes = {
|
|
118
|
+
...getCommonAttributes(config),
|
|
119
|
+
model,
|
|
120
|
+
status_code: statusCode ?? 'error',
|
|
121
|
+
error_type: errorType ?? 'unknown',
|
|
122
|
+
};
|
|
123
|
+
apiRequestCounter.add(1, metricAttributes);
|
|
124
|
+
apiRequestLatencyHistogram.record(durationMs, {
|
|
125
|
+
...getCommonAttributes(config),
|
|
126
|
+
model,
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
export function recordFileOperationMetric(config, operation, lines, mimetype, extension) {
|
|
130
|
+
if (!fileOperationCounter || !isMetricsInitialized)
|
|
131
|
+
return;
|
|
132
|
+
const attributes = {
|
|
133
|
+
...getCommonAttributes(config),
|
|
134
|
+
operation,
|
|
135
|
+
};
|
|
136
|
+
if (lines !== undefined)
|
|
137
|
+
attributes.lines = lines;
|
|
138
|
+
if (mimetype !== undefined)
|
|
139
|
+
attributes.mimetype = mimetype;
|
|
140
|
+
if (extension !== undefined)
|
|
141
|
+
attributes.extension = extension;
|
|
142
|
+
fileOperationCounter.add(1, attributes);
|
|
143
|
+
}
|
|
144
|
+
//# sourceMappingURL=metrics.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../../../src/telemetry/metrics.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,OAAO,EAEP,SAAS,GAIV,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,YAAY,EACZ,sBAAsB,EACtB,wBAAwB,EACxB,wBAAwB,EACxB,0BAA0B,EAC1B,kBAAkB,EAClB,oBAAoB,EACpB,2BAA2B,GAC5B,MAAM,gBAAgB,CAAC;AAGxB,MAAM,CAAN,IAAY,aAIX;AAJD,WAAY,aAAa;IACvB,kCAAiB,CAAA;IACjB,8BAAa,CAAA;IACb,kCAAiB,CAAA;AACnB,CAAC,EAJW,aAAa,KAAb,aAAa,QAIxB;AAED,IAAI,QAA2B,CAAC;AAChC,IAAI,eAAoC,CAAC;AACzC,IAAI,wBAA+C,CAAC;AACpD,IAAI,iBAAsC,CAAC;AAC3C,IAAI,0BAAiD,CAAC;AACtD,IAAI,iBAAsC,CAAC;AAC3C,IAAI,oBAAyC,CAAC;AAC9C,IAAI,oBAAoB,GAAG,KAAK,CAAC;AAEjC,SAAS,mBAAmB,CAAC,MAAc;IACzC,OAAO;QACL,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,QAAQ;IACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,MAAc;IAC9C,IAAI,oBAAoB;QAAE,OAAO;IAEjC,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;IACzB,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,eAAe,GAAG,KAAK,CAAC,aAAa,CAAC,sBAAsB,EAAE;QAC5D,WAAW,EAAE,yDAAyD;QACtE,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,wBAAwB,GAAG,KAAK,CAAC,eAAe,CAAC,wBAAwB,EAAE;QACzE,WAAW,EAAE,wCAAwC;QACrD,IAAI,EAAE,IAAI;QACV,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAAC,wBAAwB,EAAE;QAChE,WAAW,EAAE,kDAAkD;QAC/D,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,0BAA0B,GAAG,KAAK,CAAC,eAAe,CAChD,0BAA0B,EAC1B;QACE,WAAW,EAAE,0CAA0C;QACvD,IAAI,EAAE,IAAI;QACV,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CACF,CAAC;IACF,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAAC,kBAAkB,EAAE;QAC1D,WAAW,EAAE,yCAAyC;QACtD,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,oBAAoB,GAAG,KAAK,CAAC,aAAa,CAAC,2BAA2B,EAAE;QACtE,WAAW,EAAE,gDAAgD;QAC7D,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC,oBAAoB,EAAE;QAC/D,WAAW,EAAE,gCAAgC;QAC7C,SAAS,EAAE,SAAS,CAAC,GAAG;KACzB,CAAC,CAAC;IACH,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC;IACnD,oBAAoB,GAAG,IAAI,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAAc,EACd,YAAoB,EACpB,UAAkB,EAClB,OAAgB,EAChB,QAAyC;IAEzC,IAAI,CAAC,eAAe,IAAI,CAAC,wBAAwB,IAAI,CAAC,oBAAoB;QACxE,OAAO;IAET,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,aAAa,EAAE,YAAY;QAC3B,OAAO;QACP,QAAQ;KACT,CAAC;IACF,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IACzC,wBAAwB,CAAC,MAAM,CAAC,UAAU,EAAE;QAC1C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,aAAa,EAAE,YAAY;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,IAAuD;IAEvD,IAAI,CAAC,iBAAiB,IAAI,CAAC,oBAAoB;QAAE,OAAO;IACxD,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE;QAChC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,UAA4B,EAC5B,KAAc;IAEd,IACE,CAAC,iBAAiB;QAClB,CAAC,0BAA0B;QAC3B,CAAC,oBAAoB;QAErB,OAAO;IACT,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,WAAW,EAAE,UAAU,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;KACpD,CAAC;IACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3C,0BAA0B,CAAC,MAAM,CAAC,UAAU,EAAE;QAC5C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,MAAc,EACd,KAAa,EACb,UAAkB,EAClB,UAA4B,EAC5B,SAAkB;IAElB,IACE,CAAC,iBAAiB;QAClB,CAAC,0BAA0B;QAC3B,CAAC,oBAAoB;QAErB,OAAO;IACT,MAAM,gBAAgB,GAAe;QACnC,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;QACL,WAAW,EAAE,UAAU,IAAI,OAAO;QAClC,UAAU,EAAE,SAAS,IAAI,SAAS;KACnC,CAAC;IACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC3C,0BAA0B,CAAC,MAAM,CAAC,UAAU,EAAE;QAC5C,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,KAAK;KACN,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,MAAc,EACd,SAAwB,EACxB,KAAc,EACd,QAAiB,EACjB,SAAkB;IAElB,IAAI,CAAC,oBAAoB,IAAI,CAAC,oBAAoB;QAAE,OAAO;IAC3D,MAAM,UAAU,GAAe;QAC7B,GAAG,mBAAmB,CAAC,MAAM,CAAC;QAC9B,SAAS;KACV,CAAC;IACF,IAAI,KAAK,KAAK,SAAS;QAAE,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;IAClD,IAAI,QAAQ,KAAK,SAAS;QAAE,UAAU,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3D,IAAI,SAAS,KAAK,SAAS;QAAE,UAAU,CAAC,SAAS,GAAG,SAAS,CAAC;IAC9D,oBAAoB,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;AAC1C,CAAC"}
|
|
@@ -0,0 +1,162 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
7
|
+
import { FileOperation } from './metrics.js';
|
|
8
|
+
const mockCounterAddFn = vi.fn();
|
|
9
|
+
const mockHistogramRecordFn = vi.fn();
|
|
10
|
+
const mockCreateCounterFn = vi.fn();
|
|
11
|
+
const mockCreateHistogramFn = vi.fn();
|
|
12
|
+
const mockCounterInstance = {
|
|
13
|
+
add: mockCounterAddFn,
|
|
14
|
+
};
|
|
15
|
+
const mockHistogramInstance = {
|
|
16
|
+
record: mockHistogramRecordFn,
|
|
17
|
+
};
|
|
18
|
+
const mockMeterInstance = {
|
|
19
|
+
createCounter: mockCreateCounterFn.mockReturnValue(mockCounterInstance),
|
|
20
|
+
createHistogram: mockCreateHistogramFn.mockReturnValue(mockHistogramInstance),
|
|
21
|
+
};
|
|
22
|
+
function originalOtelMockFactory() {
|
|
23
|
+
return {
|
|
24
|
+
metrics: {
|
|
25
|
+
getMeter: vi.fn(),
|
|
26
|
+
},
|
|
27
|
+
ValueType: {
|
|
28
|
+
INT: 1,
|
|
29
|
+
},
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
vi.mock('@opentelemetry/api', originalOtelMockFactory);
|
|
33
|
+
describe('Telemetry Metrics', () => {
|
|
34
|
+
let initializeMetricsModule;
|
|
35
|
+
let recordTokenUsageMetricsModule;
|
|
36
|
+
let recordFileOperationMetricModule;
|
|
37
|
+
beforeEach(async () => {
|
|
38
|
+
vi.resetModules();
|
|
39
|
+
vi.doMock('@opentelemetry/api', () => {
|
|
40
|
+
const actualApi = originalOtelMockFactory();
|
|
41
|
+
actualApi.metrics.getMeter.mockReturnValue(mockMeterInstance);
|
|
42
|
+
return actualApi;
|
|
43
|
+
});
|
|
44
|
+
const metricsJsModule = await import('./metrics.js');
|
|
45
|
+
initializeMetricsModule = metricsJsModule.initializeMetrics;
|
|
46
|
+
recordTokenUsageMetricsModule = metricsJsModule.recordTokenUsageMetrics;
|
|
47
|
+
recordFileOperationMetricModule = metricsJsModule.recordFileOperationMetric;
|
|
48
|
+
const otelApiModule = await import('@opentelemetry/api');
|
|
49
|
+
mockCounterAddFn.mockClear();
|
|
50
|
+
mockCreateCounterFn.mockClear();
|
|
51
|
+
mockCreateHistogramFn.mockClear();
|
|
52
|
+
mockHistogramRecordFn.mockClear();
|
|
53
|
+
otelApiModule.metrics.getMeter.mockClear();
|
|
54
|
+
otelApiModule.metrics.getMeter.mockReturnValue(mockMeterInstance);
|
|
55
|
+
mockCreateCounterFn.mockReturnValue(mockCounterInstance);
|
|
56
|
+
mockCreateHistogramFn.mockReturnValue(mockHistogramInstance);
|
|
57
|
+
});
|
|
58
|
+
describe('recordTokenUsageMetrics', () => {
|
|
59
|
+
const mockConfig = {
|
|
60
|
+
getSessionId: () => 'test-session-id',
|
|
61
|
+
};
|
|
62
|
+
it('should not record metrics if not initialized', () => {
|
|
63
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 100, 'input');
|
|
64
|
+
expect(mockCounterAddFn).not.toHaveBeenCalled();
|
|
65
|
+
});
|
|
66
|
+
it('should record token usage with the correct attributes', () => {
|
|
67
|
+
initializeMetricsModule(mockConfig);
|
|
68
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 100, 'input');
|
|
69
|
+
expect(mockCounterAddFn).toHaveBeenCalledTimes(2);
|
|
70
|
+
expect(mockCounterAddFn).toHaveBeenNthCalledWith(1, 1, {
|
|
71
|
+
'session.id': 'test-session-id',
|
|
72
|
+
});
|
|
73
|
+
expect(mockCounterAddFn).toHaveBeenNthCalledWith(2, 100, {
|
|
74
|
+
'session.id': 'test-session-id',
|
|
75
|
+
model: 'gemini-pro',
|
|
76
|
+
type: 'input',
|
|
77
|
+
});
|
|
78
|
+
});
|
|
79
|
+
it('should record token usage for different types', () => {
|
|
80
|
+
initializeMetricsModule(mockConfig);
|
|
81
|
+
mockCounterAddFn.mockClear();
|
|
82
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 50, 'output');
|
|
83
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(50, {
|
|
84
|
+
'session.id': 'test-session-id',
|
|
85
|
+
model: 'gemini-pro',
|
|
86
|
+
type: 'output',
|
|
87
|
+
});
|
|
88
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 25, 'thought');
|
|
89
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(25, {
|
|
90
|
+
'session.id': 'test-session-id',
|
|
91
|
+
model: 'gemini-pro',
|
|
92
|
+
type: 'thought',
|
|
93
|
+
});
|
|
94
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 75, 'cache');
|
|
95
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(75, {
|
|
96
|
+
'session.id': 'test-session-id',
|
|
97
|
+
model: 'gemini-pro',
|
|
98
|
+
type: 'cache',
|
|
99
|
+
});
|
|
100
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-pro', 125, 'tool');
|
|
101
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(125, {
|
|
102
|
+
'session.id': 'test-session-id',
|
|
103
|
+
model: 'gemini-pro',
|
|
104
|
+
type: 'tool',
|
|
105
|
+
});
|
|
106
|
+
});
|
|
107
|
+
it('should handle different models', () => {
|
|
108
|
+
initializeMetricsModule(mockConfig);
|
|
109
|
+
mockCounterAddFn.mockClear();
|
|
110
|
+
recordTokenUsageMetricsModule(mockConfig, 'gemini-ultra', 200, 'input');
|
|
111
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(200, {
|
|
112
|
+
'session.id': 'test-session-id',
|
|
113
|
+
model: 'gemini-ultra',
|
|
114
|
+
type: 'input',
|
|
115
|
+
});
|
|
116
|
+
});
|
|
117
|
+
});
|
|
118
|
+
describe('recordFileOperationMetric', () => {
|
|
119
|
+
const mockConfig = {
|
|
120
|
+
getSessionId: () => 'test-session-id',
|
|
121
|
+
};
|
|
122
|
+
it('should not record metrics if not initialized', () => {
|
|
123
|
+
recordFileOperationMetricModule(mockConfig, FileOperation.CREATE, 10, 'text/plain', 'txt');
|
|
124
|
+
expect(mockCounterAddFn).not.toHaveBeenCalled();
|
|
125
|
+
});
|
|
126
|
+
it('should record file creation with all attributes', () => {
|
|
127
|
+
initializeMetricsModule(mockConfig);
|
|
128
|
+
recordFileOperationMetricModule(mockConfig, FileOperation.CREATE, 10, 'text/plain', 'txt');
|
|
129
|
+
expect(mockCounterAddFn).toHaveBeenCalledTimes(2);
|
|
130
|
+
expect(mockCounterAddFn).toHaveBeenNthCalledWith(1, 1, {
|
|
131
|
+
'session.id': 'test-session-id',
|
|
132
|
+
});
|
|
133
|
+
expect(mockCounterAddFn).toHaveBeenNthCalledWith(2, 1, {
|
|
134
|
+
'session.id': 'test-session-id',
|
|
135
|
+
operation: FileOperation.CREATE,
|
|
136
|
+
lines: 10,
|
|
137
|
+
mimetype: 'text/plain',
|
|
138
|
+
extension: 'txt',
|
|
139
|
+
});
|
|
140
|
+
});
|
|
141
|
+
it('should record file read with minimal attributes', () => {
|
|
142
|
+
initializeMetricsModule(mockConfig);
|
|
143
|
+
mockCounterAddFn.mockClear();
|
|
144
|
+
recordFileOperationMetricModule(mockConfig, FileOperation.READ);
|
|
145
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(1, {
|
|
146
|
+
'session.id': 'test-session-id',
|
|
147
|
+
operation: FileOperation.READ,
|
|
148
|
+
});
|
|
149
|
+
});
|
|
150
|
+
it('should record file update with some attributes', () => {
|
|
151
|
+
initializeMetricsModule(mockConfig);
|
|
152
|
+
mockCounterAddFn.mockClear();
|
|
153
|
+
recordFileOperationMetricModule(mockConfig, FileOperation.UPDATE, undefined, 'application/javascript');
|
|
154
|
+
expect(mockCounterAddFn).toHaveBeenCalledWith(1, {
|
|
155
|
+
'session.id': 'test-session-id',
|
|
156
|
+
operation: FileOperation.UPDATE,
|
|
157
|
+
mimetype: 'application/javascript',
|
|
158
|
+
});
|
|
159
|
+
});
|
|
160
|
+
});
|
|
161
|
+
});
|
|
162
|
+
//# sourceMappingURL=metrics.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.test.js","sourceRoot":"","sources":["../../../src/telemetry/metrics.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AASzE,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,MAAM,gBAAgB,GAElB,EAAE,CAAC,EAAE,EAAE,CAAC;AACZ,MAAM,qBAAqB,GAEvB,EAAE,CAAC,EAAE,EAAE,CAAC;AAEZ,MAAM,mBAAmB,GACvB,EAAE,CAAC,EAAE,EAAE,CAAC;AACV,MAAM,qBAAqB,GAEvB,EAAE,CAAC,EAAE,EAAE,CAAC;AAEZ,MAAM,mBAAmB,GAAG;IAC1B,GAAG,EAAE,gBAAgB;CACA,CAAC;AAExB,MAAM,qBAAqB,GAAG;IAC5B,MAAM,EAAE,qBAAqB;CACN,CAAC;AAE1B,MAAM,iBAAiB,GAAG;IACxB,aAAa,EAAE,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC;IACvE,eAAe,EAAE,qBAAqB,CAAC,eAAe,CAAC,qBAAqB,CAAC;CAC1D,CAAC;AAEtB,SAAS,uBAAuB;IAC9B,OAAO;QACL,OAAO,EAAE;YACP,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB;QACD,SAAS,EAAE;YACT,GAAG,EAAE,CAAC;SACP;KACF,CAAC;AACJ,CAAC;AAED,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,CAAC;AAEvD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,uBAAwE,CAAC;IAC7E,IAAI,6BAAoF,CAAC;IACzF,IAAI,+BAAwF,CAAC;IAE7F,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,EAAE,CAAC,MAAM,CAAC,oBAAoB,EAAE,GAAG,EAAE;YACnC,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;YAC3C,SAAS,CAAC,OAAO,CAAC,QAAiB,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;YACxE,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;QACrD,uBAAuB,GAAG,eAAe,CAAC,iBAAiB,CAAC;QAC5D,6BAA6B,GAAG,eAAe,CAAC,uBAAuB,CAAC;QACxE,+BAA+B,GAAG,eAAe,CAAC,yBAAyB,CAAC;QAE5E,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAEzD,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC7B,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAChC,qBAAqB,CAAC,SAAS,EAAE,CAAC;QAClC,qBAAqB,CAAC,SAAS,EAAE,CAAC;QACjC,aAAa,CAAC,OAAO,CAAC,QAAiB,CAAC,SAAS,EAAE,CAAC;QAEpD,aAAa,CAAC,OAAO,CAAC,QAAiB,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QAC5E,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACzD,qBAAqB,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;SACjB,CAAC;QAEvB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,GAAG,EAAE;gBACvD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;YACvE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;YACrE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE;gBAChD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;YAEH,6BAA6B,CAAC,UAAU,EAAE,YAAY,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YACrE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE;gBACjD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,MAAM;aACb,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,6BAA6B,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YACxE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE;gBACjD,YAAY,EAAE,iBAAiB;gBAC/B,KAAK,EAAE,cAAc;gBACrB,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,MAAM,UAAU,GAAG;YACjB,YAAY,EAAE,GAAG,EAAE,CAAC,iBAAiB;SACjB,CAAC;QAEvB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,EAAE,EACF,YAAY,EACZ,KAAK,CACN,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,EAAE,EACF,YAAY,EACZ,KAAK,CACN,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE;gBACrD,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,MAAM;gBAC/B,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,+BAA+B,CAAC,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YAChE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE;gBAC/C,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,IAAI;aAC9B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,uBAAuB,CAAC,UAAU,CAAC,CAAC;YACpC,gBAAgB,CAAC,SAAS,EAAE,CAAC;YAE7B,+BAA+B,CAC7B,UAAU,EACV,aAAa,CAAC,MAAM,EACpB,SAAS,EACT,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE;gBAC/C,YAAY,EAAE,iBAAiB;gBAC/B,SAAS,EAAE,aAAa,CAAC,MAAM;gBAC/B,QAAQ,EAAE,wBAAwB;aACnC,CAAC,CAAC;QACL,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 { Config } from '../config/config.js';
|
|
7
|
+
export declare function isTelemetrySdkInitialized(): boolean;
|
|
8
|
+
export declare function initializeTelemetry(_config: Config): void;
|
|
9
|
+
export declare function shutdownTelemetry(): Promise<void>;
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
// TELEMETRY REMOVED: Commented out unused imports for telemetry
|
|
7
|
+
import { DiagConsoleLogger, DiagLogLevel, diag } from '@opentelemetry/api';
|
|
8
|
+
// import { SERVICE_NAME } from './constants.js';
|
|
9
|
+
// import { initializeMetrics } from './metrics.js';
|
|
10
|
+
// TELEMETRY REMOVED: ClearcutLogger import disabled
|
|
11
|
+
// import { ClearcutLogger } from './clearcut-logger/clearcut-logger.js';
|
|
12
|
+
// For troubleshooting, set the log level to DiagLogLevel.DEBUG
|
|
13
|
+
diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.INFO);
|
|
14
|
+
// let sdk: NodeSDK | undefined;
|
|
15
|
+
const telemetryInitialized = false;
|
|
16
|
+
export function isTelemetrySdkInitialized() {
|
|
17
|
+
return telemetryInitialized;
|
|
18
|
+
}
|
|
19
|
+
// TELEMETRY REMOVED: Commented out unused function
|
|
20
|
+
// function parseGrpcEndpoint(
|
|
21
|
+
// otlpEndpointSetting: string | undefined,
|
|
22
|
+
// ): string | undefined {
|
|
23
|
+
// if (!otlpEndpointSetting) {
|
|
24
|
+
// return undefined;
|
|
25
|
+
// }
|
|
26
|
+
// // Trim leading/trailing quotes that might come from env variables
|
|
27
|
+
// const trimmedEndpoint = otlpEndpointSetting.replace(/^["']|["']$/g, '');
|
|
28
|
+
// try {
|
|
29
|
+
// const url = new URL(trimmedEndpoint);
|
|
30
|
+
// // OTLP gRPC exporters expect an endpoint in the format scheme://host:port
|
|
31
|
+
// // The `origin` property provides this, stripping any path, query, or hash.
|
|
32
|
+
// return url.origin;
|
|
33
|
+
// } catch (error) {
|
|
34
|
+
// diag.error('Invalid OTLP endpoint URL provided:', trimmedEndpoint, error);
|
|
35
|
+
// return undefined;
|
|
36
|
+
// }
|
|
37
|
+
// }
|
|
38
|
+
export function initializeTelemetry(_config) {
|
|
39
|
+
// TELEMETRY REMOVED: Disabled telemetry initialization to prevent Google data collection
|
|
40
|
+
return;
|
|
41
|
+
/*
|
|
42
|
+
if (telemetryInitialized || !config.getTelemetryEnabled()) {
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
const resource = new Resource({
|
|
47
|
+
[SemanticResourceAttributes.SERVICE_NAME]: SERVICE_NAME,
|
|
48
|
+
[SemanticResourceAttributes.SERVICE_VERSION]: process.version,
|
|
49
|
+
'session.id': config.getSessionId(),
|
|
50
|
+
});
|
|
51
|
+
|
|
52
|
+
const otlpEndpoint = config.getTelemetryOtlpEndpoint();
|
|
53
|
+
const grpcParsedEndpoint = parseGrpcEndpoint(otlpEndpoint);
|
|
54
|
+
const useOtlp = !!grpcParsedEndpoint;
|
|
55
|
+
|
|
56
|
+
const spanExporter = useOtlp
|
|
57
|
+
? new OTLPTraceExporter({
|
|
58
|
+
url: grpcParsedEndpoint,
|
|
59
|
+
compression: CompressionAlgorithm.GZIP,
|
|
60
|
+
})
|
|
61
|
+
: new ConsoleSpanExporter();
|
|
62
|
+
const logExporter = useOtlp
|
|
63
|
+
? new OTLPLogExporter({
|
|
64
|
+
url: grpcParsedEndpoint,
|
|
65
|
+
compression: CompressionAlgorithm.GZIP,
|
|
66
|
+
})
|
|
67
|
+
: new ConsoleLogRecordExporter();
|
|
68
|
+
const metricReader = useOtlp
|
|
69
|
+
? new PeriodicExportingMetricReader({
|
|
70
|
+
exporter: new OTLPMetricExporter({
|
|
71
|
+
url: grpcParsedEndpoint,
|
|
72
|
+
compression: CompressionAlgorithm.GZIP,
|
|
73
|
+
}),
|
|
74
|
+
exportIntervalMillis: 10000,
|
|
75
|
+
})
|
|
76
|
+
: new PeriodicExportingMetricReader({
|
|
77
|
+
exporter: new ConsoleMetricExporter(),
|
|
78
|
+
exportIntervalMillis: 10000,
|
|
79
|
+
});
|
|
80
|
+
|
|
81
|
+
sdk = new NodeSDK({
|
|
82
|
+
resource,
|
|
83
|
+
spanProcessors: [new BatchSpanProcessor(spanExporter)],
|
|
84
|
+
logRecordProcessor: new BatchLogRecordProcessor(logExporter),
|
|
85
|
+
metricReader,
|
|
86
|
+
instrumentations: [new HttpInstrumentation()],
|
|
87
|
+
});
|
|
88
|
+
|
|
89
|
+
try {
|
|
90
|
+
sdk.start();
|
|
91
|
+
console.log('OpenTelemetry SDK started successfully.');
|
|
92
|
+
telemetryInitialized = true;
|
|
93
|
+
initializeMetrics(config);
|
|
94
|
+
} catch (error) {
|
|
95
|
+
console.error('Error starting OpenTelemetry SDK:', error);
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
process.on('SIGTERM', shutdownTelemetry);
|
|
99
|
+
process.on('SIGINT', shutdownTelemetry);
|
|
100
|
+
*/
|
|
101
|
+
}
|
|
102
|
+
export async function shutdownTelemetry() {
|
|
103
|
+
// TELEMETRY REMOVED: Disabled telemetry shutdown to prevent Google data collection
|
|
104
|
+
return;
|
|
105
|
+
/*
|
|
106
|
+
if (!telemetryInitialized || !sdk) {
|
|
107
|
+
return;
|
|
108
|
+
}
|
|
109
|
+
try {
|
|
110
|
+
ClearcutLogger.getInstance()?.shutdown();
|
|
111
|
+
await sdk.shutdown();
|
|
112
|
+
console.log('OpenTelemetry SDK shut down successfully.');
|
|
113
|
+
} catch (error) {
|
|
114
|
+
console.error('Error shutting down SDK:', error);
|
|
115
|
+
} finally {
|
|
116
|
+
telemetryInitialized = false;
|
|
117
|
+
}
|
|
118
|
+
*/
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=sdk.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sdk.js","sourceRoot":"","sources":["../../../src/telemetry/sdk.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,gEAAgE;AAChE,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAsB3E,iDAAiD;AACjD,oDAAoD;AACpD,oDAAoD;AACpD,yEAAyE;AAEzE,+DAA+D;AAC/D,IAAI,CAAC,SAAS,CAAC,IAAI,iBAAiB,EAAE,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;AAE3D,gCAAgC;AAChC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AAEnC,MAAM,UAAU,yBAAyB;IACvC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,mDAAmD;AACnD,8BAA8B;AAC9B,6CAA6C;AAC7C,0BAA0B;AAC1B,gCAAgC;AAChC,wBAAwB;AACxB,MAAM;AACN,uEAAuE;AACvE,6EAA6E;AAE7E,UAAU;AACV,4CAA4C;AAC5C,iFAAiF;AACjF,kFAAkF;AAClF,yBAAyB;AACzB,sBAAsB;AACtB,iFAAiF;AACjF,wBAAwB;AACxB,MAAM;AACN,IAAI;AAEJ,MAAM,UAAU,mBAAmB,CAAC,OAAe;IACjD,yFAAyF;IACzF,OAAO;IACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;MA2DE;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,mFAAmF;IACnF,OAAO;IACP;;;;;;;;;;;;;MAaE;AACJ,CAAC"}
|
|
@@ -0,0 +1,52 @@
|
|
|
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 { initializeTelemetry, shutdownTelemetry, isTelemetrySdkInitialized, } from './sdk.js';
|
|
8
|
+
import { Config } from '../config/config.js';
|
|
9
|
+
import { NodeSDK } from '@opentelemetry/sdk-node';
|
|
10
|
+
vi.mock('@opentelemetry/sdk-node');
|
|
11
|
+
vi.mock('../config/config.js');
|
|
12
|
+
describe('telemetry', () => {
|
|
13
|
+
let mockConfig;
|
|
14
|
+
let mockNodeSdk;
|
|
15
|
+
beforeEach(() => {
|
|
16
|
+
vi.resetAllMocks();
|
|
17
|
+
mockConfig = new Config({
|
|
18
|
+
sessionId: 'test-session-id',
|
|
19
|
+
model: 'test-model',
|
|
20
|
+
targetDir: '/test/dir',
|
|
21
|
+
debugMode: false,
|
|
22
|
+
cwd: '/test/dir',
|
|
23
|
+
});
|
|
24
|
+
vi.spyOn(mockConfig, 'getTelemetryEnabled').mockReturnValue(true);
|
|
25
|
+
vi.spyOn(mockConfig, 'getTelemetryOtlpEndpoint').mockReturnValue('http://localhost:4317');
|
|
26
|
+
vi.spyOn(mockConfig, 'getSessionId').mockReturnValue('test-session-id');
|
|
27
|
+
mockNodeSdk = {
|
|
28
|
+
start: vi.fn(),
|
|
29
|
+
shutdown: vi.fn().mockResolvedValue(undefined),
|
|
30
|
+
};
|
|
31
|
+
vi.mocked(NodeSDK).mockImplementation(() => mockNodeSdk);
|
|
32
|
+
});
|
|
33
|
+
afterEach(async () => {
|
|
34
|
+
// Ensure we shut down telemetry even if a test fails.
|
|
35
|
+
if (isTelemetrySdkInitialized()) {
|
|
36
|
+
await shutdownTelemetry();
|
|
37
|
+
}
|
|
38
|
+
});
|
|
39
|
+
it('should initialize the telemetry service', () => {
|
|
40
|
+
initializeTelemetry(mockConfig);
|
|
41
|
+
// Telemetry is disabled, so NodeSDK should not be called
|
|
42
|
+
expect(NodeSDK).not.toHaveBeenCalled();
|
|
43
|
+
expect(mockNodeSdk.start).not.toHaveBeenCalled();
|
|
44
|
+
});
|
|
45
|
+
it('should shutdown the telemetry service', async () => {
|
|
46
|
+
initializeTelemetry(mockConfig);
|
|
47
|
+
await shutdownTelemetry();
|
|
48
|
+
// Telemetry is disabled, so shutdown should not be called
|
|
49
|
+
expect(mockNodeSdk.shutdown).not.toHaveBeenCalled();
|
|
50
|
+
});
|
|
51
|
+
});
|
|
52
|
+
//# sourceMappingURL=telemetry.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry.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,EACL,mBAAmB,EACnB,iBAAiB,EACjB,yBAAyB,GAC1B,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG,IAAI,MAAM,CAAC;YACtB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,WAAW;YACtB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAClE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC,eAAe,CAC9D,uBAAuB,CACxB,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QACxE,WAAW,GAAG;YACZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;SACzB,CAAC;QACxB,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,sDAAsD;QACtD,IAAI,yBAAyB,EAAE,EAAE,CAAC;YAChC,MAAM,iBAAiB,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,yDAAyD;QACzD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,iBAAiB,EAAE,CAAC;QAE1B,0DAA0D;QAC1D,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { GenerateContentResponseUsageMetadata } from '@google/genai';
|
|
7
|
+
import { Config } from '../config/config.js';
|
|
8
|
+
import { CompletedToolCall } from '../core/coreToolScheduler.js';
|
|
9
|
+
import { ToolConfirmationOutcome } from '../tools/tools.js';
|
|
10
|
+
export declare enum ToolCallDecision {
|
|
11
|
+
ACCEPT = "accept",
|
|
12
|
+
REJECT = "reject",
|
|
13
|
+
MODIFY = "modify"
|
|
14
|
+
}
|
|
15
|
+
export declare function getDecisionFromOutcome(outcome: ToolConfirmationOutcome): ToolCallDecision;
|
|
16
|
+
export declare class StartSessionEvent {
|
|
17
|
+
'event.name': 'cli_config';
|
|
18
|
+
'event.timestamp': string;
|
|
19
|
+
model: string;
|
|
20
|
+
embedding_model: string;
|
|
21
|
+
sandbox_enabled: boolean;
|
|
22
|
+
core_tools_enabled: string;
|
|
23
|
+
approval_mode: string;
|
|
24
|
+
api_key_enabled: boolean;
|
|
25
|
+
vertex_ai_enabled: boolean;
|
|
26
|
+
debug_enabled: boolean;
|
|
27
|
+
mcp_servers: string;
|
|
28
|
+
telemetry_enabled: boolean;
|
|
29
|
+
telemetry_log_user_prompts_enabled: boolean;
|
|
30
|
+
file_filtering_respect_git_ignore: boolean;
|
|
31
|
+
constructor(config: Config);
|
|
32
|
+
}
|
|
33
|
+
export declare class EndSessionEvent {
|
|
34
|
+
'event.name': 'end_session';
|
|
35
|
+
'event.timestamp': string;
|
|
36
|
+
session_id?: string;
|
|
37
|
+
constructor(config?: Config);
|
|
38
|
+
}
|
|
39
|
+
export declare class UserPromptEvent {
|
|
40
|
+
'event.name': 'user_prompt';
|
|
41
|
+
'event.timestamp': string;
|
|
42
|
+
prompt_length: number;
|
|
43
|
+
prompt_id: string;
|
|
44
|
+
auth_type?: string;
|
|
45
|
+
prompt?: string;
|
|
46
|
+
constructor(prompt_length: number, prompt_Id: string, auth_type?: string, prompt?: string);
|
|
47
|
+
}
|
|
48
|
+
export declare class ToolCallEvent {
|
|
49
|
+
'event.name': 'tool_call';
|
|
50
|
+
'event.timestamp': string;
|
|
51
|
+
function_name: string;
|
|
52
|
+
function_args: Record<string, unknown>;
|
|
53
|
+
duration_ms: number;
|
|
54
|
+
success: boolean;
|
|
55
|
+
decision?: ToolCallDecision;
|
|
56
|
+
error?: string;
|
|
57
|
+
error_type?: string;
|
|
58
|
+
prompt_id: string;
|
|
59
|
+
constructor(call: CompletedToolCall);
|
|
60
|
+
}
|
|
61
|
+
export declare class ApiRequestEvent {
|
|
62
|
+
'event.name': 'api_request';
|
|
63
|
+
'event.timestamp': string;
|
|
64
|
+
model: string;
|
|
65
|
+
prompt_id: string;
|
|
66
|
+
request_text?: string;
|
|
67
|
+
constructor(model: string, prompt_id: string, request_text?: string);
|
|
68
|
+
}
|
|
69
|
+
export declare class ApiErrorEvent {
|
|
70
|
+
'event.name': 'api_error';
|
|
71
|
+
'event.timestamp': string;
|
|
72
|
+
model: string;
|
|
73
|
+
error: string;
|
|
74
|
+
error_type?: string;
|
|
75
|
+
status_code?: number | string;
|
|
76
|
+
duration_ms: number;
|
|
77
|
+
prompt_id: string;
|
|
78
|
+
auth_type?: string;
|
|
79
|
+
constructor(model: string, error: string, duration_ms: number, prompt_id: string, auth_type?: string, error_type?: string, status_code?: number | string);
|
|
80
|
+
}
|
|
81
|
+
export declare class ApiResponseEvent {
|
|
82
|
+
'event.name': 'api_response';
|
|
83
|
+
'event.timestamp': string;
|
|
84
|
+
model: string;
|
|
85
|
+
status_code?: number | string;
|
|
86
|
+
duration_ms: number;
|
|
87
|
+
error?: string;
|
|
88
|
+
input_token_count: number;
|
|
89
|
+
output_token_count: number;
|
|
90
|
+
cached_content_token_count: number;
|
|
91
|
+
thoughts_token_count: number;
|
|
92
|
+
tool_token_count: number;
|
|
93
|
+
total_token_count: number;
|
|
94
|
+
response_text?: string;
|
|
95
|
+
prompt_id: string;
|
|
96
|
+
auth_type?: string;
|
|
97
|
+
constructor(model: string, duration_ms: number, prompt_id: string, auth_type?: string, usage_data?: GenerateContentResponseUsageMetadata, response_text?: string, error?: string);
|
|
98
|
+
}
|
|
99
|
+
export declare class FlashFallbackEvent {
|
|
100
|
+
'event.name': 'flash_fallback';
|
|
101
|
+
'event.timestamp': string;
|
|
102
|
+
auth_type: string;
|
|
103
|
+
constructor(auth_type: string);
|
|
104
|
+
}
|
|
105
|
+
export declare enum LoopType {
|
|
106
|
+
CONSECUTIVE_IDENTICAL_TOOL_CALLS = "consecutive_identical_tool_calls",
|
|
107
|
+
CHANTING_IDENTICAL_SENTENCES = "chanting_identical_sentences"
|
|
108
|
+
}
|
|
109
|
+
export declare class LoopDetectedEvent {
|
|
110
|
+
'event.name': 'loop_detected';
|
|
111
|
+
'event.timestamp': string;
|
|
112
|
+
loop_type: LoopType;
|
|
113
|
+
constructor(loop_type: LoopType);
|
|
114
|
+
}
|
|
115
|
+
export type TelemetryEvent = StartSessionEvent | EndSessionEvent | UserPromptEvent | ToolCallEvent | ApiRequestEvent | ApiErrorEvent | ApiResponseEvent | FlashFallbackEvent | LoopDetectedEvent;
|