@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,321 @@
|
|
|
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 * as actualNodeFs from 'node:fs'; // For setup/teardown
|
|
8
|
+
import fsPromises from 'node:fs/promises';
|
|
9
|
+
import path from 'node:path';
|
|
10
|
+
import os from 'node:os';
|
|
11
|
+
import mime from 'mime-types';
|
|
12
|
+
import { isWithinRoot, isBinaryFile, detectFileType, processSingleFileContent, } from './fileUtils.js';
|
|
13
|
+
vi.mock('mime-types', () => ({
|
|
14
|
+
default: { lookup: vi.fn() },
|
|
15
|
+
lookup: vi.fn(),
|
|
16
|
+
}));
|
|
17
|
+
const mockMimeLookup = mime.lookup;
|
|
18
|
+
describe('fileUtils', () => {
|
|
19
|
+
let tempRootDir;
|
|
20
|
+
const originalProcessCwd = process.cwd;
|
|
21
|
+
let testTextFilePath;
|
|
22
|
+
let testImageFilePath;
|
|
23
|
+
let testPdfFilePath;
|
|
24
|
+
let testBinaryFilePath;
|
|
25
|
+
let nonExistentFilePath;
|
|
26
|
+
let directoryPath;
|
|
27
|
+
beforeEach(() => {
|
|
28
|
+
vi.resetAllMocks(); // Reset all mocks, including mime.lookup
|
|
29
|
+
tempRootDir = actualNodeFs.mkdtempSync(path.join(os.tmpdir(), 'fileUtils-test-'));
|
|
30
|
+
process.cwd = vi.fn(() => tempRootDir); // Mock cwd if necessary for relative path logic within tests
|
|
31
|
+
testTextFilePath = path.join(tempRootDir, 'test.txt');
|
|
32
|
+
testImageFilePath = path.join(tempRootDir, 'image.png');
|
|
33
|
+
testPdfFilePath = path.join(tempRootDir, 'document.pdf');
|
|
34
|
+
testBinaryFilePath = path.join(tempRootDir, 'app.exe');
|
|
35
|
+
nonExistentFilePath = path.join(tempRootDir, 'notfound.txt');
|
|
36
|
+
directoryPath = path.join(tempRootDir, 'subdir');
|
|
37
|
+
actualNodeFs.mkdirSync(directoryPath, { recursive: true }); // Ensure subdir exists
|
|
38
|
+
});
|
|
39
|
+
afterEach(() => {
|
|
40
|
+
if (actualNodeFs.existsSync(tempRootDir)) {
|
|
41
|
+
actualNodeFs.rmSync(tempRootDir, { recursive: true, force: true });
|
|
42
|
+
}
|
|
43
|
+
process.cwd = originalProcessCwd;
|
|
44
|
+
vi.restoreAllMocks(); // Restore any spies
|
|
45
|
+
});
|
|
46
|
+
describe('isWithinRoot', () => {
|
|
47
|
+
const root = path.resolve('/project/root');
|
|
48
|
+
it('should return true for paths directly within the root', () => {
|
|
49
|
+
expect(isWithinRoot(path.join(root, 'file.txt'), root)).toBe(true);
|
|
50
|
+
expect(isWithinRoot(path.join(root, 'subdir', 'file.txt'), root)).toBe(true);
|
|
51
|
+
});
|
|
52
|
+
it('should return true for the root path itself', () => {
|
|
53
|
+
expect(isWithinRoot(root, root)).toBe(true);
|
|
54
|
+
});
|
|
55
|
+
it('should return false for paths outside the root', () => {
|
|
56
|
+
expect(isWithinRoot(path.resolve('/project/other', 'file.txt'), root)).toBe(false);
|
|
57
|
+
expect(isWithinRoot(path.resolve('/unrelated', 'file.txt'), root)).toBe(false);
|
|
58
|
+
});
|
|
59
|
+
it('should return false for paths that only partially match the root prefix', () => {
|
|
60
|
+
expect(isWithinRoot(path.resolve('/project/root-but-actually-different'), root)).toBe(false);
|
|
61
|
+
});
|
|
62
|
+
it('should handle paths with trailing slashes correctly', () => {
|
|
63
|
+
expect(isWithinRoot(path.join(root, 'file.txt') + path.sep, root)).toBe(true);
|
|
64
|
+
expect(isWithinRoot(root + path.sep, root)).toBe(true);
|
|
65
|
+
});
|
|
66
|
+
it('should handle different path separators (POSIX vs Windows)', () => {
|
|
67
|
+
const posixRoot = '/project/root';
|
|
68
|
+
const posixPathInside = '/project/root/file.txt';
|
|
69
|
+
const posixPathOutside = '/project/other/file.txt';
|
|
70
|
+
expect(isWithinRoot(posixPathInside, posixRoot)).toBe(true);
|
|
71
|
+
expect(isWithinRoot(posixPathOutside, posixRoot)).toBe(false);
|
|
72
|
+
});
|
|
73
|
+
it('should return false for a root path that is a sub-path of the path to check', () => {
|
|
74
|
+
const pathToCheck = path.resolve('/project/root/sub');
|
|
75
|
+
const rootSub = path.resolve('/project/root');
|
|
76
|
+
expect(isWithinRoot(pathToCheck, rootSub)).toBe(true);
|
|
77
|
+
const pathToCheckSuper = path.resolve('/project/root');
|
|
78
|
+
const rootSuper = path.resolve('/project/root/sub');
|
|
79
|
+
expect(isWithinRoot(pathToCheckSuper, rootSuper)).toBe(false);
|
|
80
|
+
});
|
|
81
|
+
});
|
|
82
|
+
describe('isBinaryFile', () => {
|
|
83
|
+
let filePathForBinaryTest;
|
|
84
|
+
beforeEach(() => {
|
|
85
|
+
filePathForBinaryTest = path.join(tempRootDir, 'binaryCheck.tmp');
|
|
86
|
+
});
|
|
87
|
+
afterEach(() => {
|
|
88
|
+
if (actualNodeFs.existsSync(filePathForBinaryTest)) {
|
|
89
|
+
actualNodeFs.unlinkSync(filePathForBinaryTest);
|
|
90
|
+
}
|
|
91
|
+
});
|
|
92
|
+
it('should return false for an empty file', () => {
|
|
93
|
+
actualNodeFs.writeFileSync(filePathForBinaryTest, '');
|
|
94
|
+
expect(isBinaryFile(filePathForBinaryTest)).toBe(false);
|
|
95
|
+
});
|
|
96
|
+
it('should return false for a typical text file', () => {
|
|
97
|
+
actualNodeFs.writeFileSync(filePathForBinaryTest, 'Hello, world!\nThis is a test file with normal text content.');
|
|
98
|
+
expect(isBinaryFile(filePathForBinaryTest)).toBe(false);
|
|
99
|
+
});
|
|
100
|
+
it('should return true for a file with many null bytes', () => {
|
|
101
|
+
const binaryContent = Buffer.from([
|
|
102
|
+
0x48, 0x65, 0x00, 0x6c, 0x6f, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
103
|
+
]); // "He\0llo\0\0\0\0\0"
|
|
104
|
+
actualNodeFs.writeFileSync(filePathForBinaryTest, binaryContent);
|
|
105
|
+
expect(isBinaryFile(filePathForBinaryTest)).toBe(true);
|
|
106
|
+
});
|
|
107
|
+
it('should return true for a file with high percentage of non-printable ASCII', () => {
|
|
108
|
+
const binaryContent = Buffer.from([
|
|
109
|
+
0x41, 0x42, 0x01, 0x02, 0x03, 0x04, 0x05, 0x43, 0x44, 0x06,
|
|
110
|
+
]); // AB\x01\x02\x03\x04\x05CD\x06
|
|
111
|
+
actualNodeFs.writeFileSync(filePathForBinaryTest, binaryContent);
|
|
112
|
+
expect(isBinaryFile(filePathForBinaryTest)).toBe(true);
|
|
113
|
+
});
|
|
114
|
+
it('should return false if file access fails (e.g., ENOENT)', () => {
|
|
115
|
+
// Ensure the file does not exist
|
|
116
|
+
if (actualNodeFs.existsSync(filePathForBinaryTest)) {
|
|
117
|
+
actualNodeFs.unlinkSync(filePathForBinaryTest);
|
|
118
|
+
}
|
|
119
|
+
expect(isBinaryFile(filePathForBinaryTest)).toBe(false);
|
|
120
|
+
});
|
|
121
|
+
});
|
|
122
|
+
describe('detectFileType', () => {
|
|
123
|
+
let filePathForDetectTest;
|
|
124
|
+
beforeEach(() => {
|
|
125
|
+
filePathForDetectTest = path.join(tempRootDir, 'detectType.tmp');
|
|
126
|
+
// Default: create as a text file for isBinaryFile fallback
|
|
127
|
+
actualNodeFs.writeFileSync(filePathForDetectTest, 'Plain text content');
|
|
128
|
+
});
|
|
129
|
+
afterEach(() => {
|
|
130
|
+
if (actualNodeFs.existsSync(filePathForDetectTest)) {
|
|
131
|
+
actualNodeFs.unlinkSync(filePathForDetectTest);
|
|
132
|
+
}
|
|
133
|
+
vi.restoreAllMocks(); // Restore spies on actualNodeFs
|
|
134
|
+
});
|
|
135
|
+
it('should detect typescript type by extension (ts)', () => {
|
|
136
|
+
expect(detectFileType('file.ts')).toBe('text');
|
|
137
|
+
expect(detectFileType('file.test.ts')).toBe('text');
|
|
138
|
+
});
|
|
139
|
+
it('should detect image type by extension (png)', () => {
|
|
140
|
+
mockMimeLookup.mockReturnValueOnce('image/png');
|
|
141
|
+
expect(detectFileType('file.png')).toBe('image');
|
|
142
|
+
});
|
|
143
|
+
it('should detect image type by extension (jpeg)', () => {
|
|
144
|
+
mockMimeLookup.mockReturnValueOnce('image/jpeg');
|
|
145
|
+
expect(detectFileType('file.jpg')).toBe('image');
|
|
146
|
+
});
|
|
147
|
+
it('should detect svg type by extension', () => {
|
|
148
|
+
expect(detectFileType('image.svg')).toBe('svg');
|
|
149
|
+
expect(detectFileType('image.icon.svg')).toBe('svg');
|
|
150
|
+
});
|
|
151
|
+
it('should detect pdf type by extension', () => {
|
|
152
|
+
mockMimeLookup.mockReturnValueOnce('application/pdf');
|
|
153
|
+
expect(detectFileType('file.pdf')).toBe('pdf');
|
|
154
|
+
});
|
|
155
|
+
it('should detect audio type by extension', () => {
|
|
156
|
+
mockMimeLookup.mockReturnValueOnce('audio/mpeg');
|
|
157
|
+
expect(detectFileType('song.mp3')).toBe('audio');
|
|
158
|
+
});
|
|
159
|
+
it('should detect video type by extension', () => {
|
|
160
|
+
mockMimeLookup.mockReturnValueOnce('video/mp4');
|
|
161
|
+
expect(detectFileType('movie.mp4')).toBe('video');
|
|
162
|
+
});
|
|
163
|
+
it('should detect known binary extensions as binary (e.g. .zip)', () => {
|
|
164
|
+
mockMimeLookup.mockReturnValueOnce('application/zip');
|
|
165
|
+
expect(detectFileType('archive.zip')).toBe('binary');
|
|
166
|
+
});
|
|
167
|
+
it('should detect known binary extensions as binary (e.g. .exe)', () => {
|
|
168
|
+
mockMimeLookup.mockReturnValueOnce('application/octet-stream'); // Common for .exe
|
|
169
|
+
expect(detectFileType('app.exe')).toBe('binary');
|
|
170
|
+
});
|
|
171
|
+
it('should use isBinaryFile for unknown extensions and detect as binary', () => {
|
|
172
|
+
mockMimeLookup.mockReturnValueOnce(false); // Unknown mime type
|
|
173
|
+
// Create a file that isBinaryFile will identify as binary
|
|
174
|
+
const binaryContent = Buffer.from([
|
|
175
|
+
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a,
|
|
176
|
+
]);
|
|
177
|
+
actualNodeFs.writeFileSync(filePathForDetectTest, binaryContent);
|
|
178
|
+
expect(detectFileType(filePathForDetectTest)).toBe('binary');
|
|
179
|
+
});
|
|
180
|
+
it('should default to text if mime type is unknown and content is not binary', () => {
|
|
181
|
+
mockMimeLookup.mockReturnValueOnce(false); // Unknown mime type
|
|
182
|
+
// filePathForDetectTest is already a text file by default from beforeEach
|
|
183
|
+
expect(detectFileType(filePathForDetectTest)).toBe('text');
|
|
184
|
+
});
|
|
185
|
+
});
|
|
186
|
+
describe('processSingleFileContent', () => {
|
|
187
|
+
beforeEach(() => {
|
|
188
|
+
// Ensure files exist for statSync checks before readFile might be mocked
|
|
189
|
+
if (actualNodeFs.existsSync(testTextFilePath))
|
|
190
|
+
actualNodeFs.unlinkSync(testTextFilePath);
|
|
191
|
+
if (actualNodeFs.existsSync(testImageFilePath))
|
|
192
|
+
actualNodeFs.unlinkSync(testImageFilePath);
|
|
193
|
+
if (actualNodeFs.existsSync(testPdfFilePath))
|
|
194
|
+
actualNodeFs.unlinkSync(testPdfFilePath);
|
|
195
|
+
if (actualNodeFs.existsSync(testBinaryFilePath))
|
|
196
|
+
actualNodeFs.unlinkSync(testBinaryFilePath);
|
|
197
|
+
});
|
|
198
|
+
it('should read a text file successfully', async () => {
|
|
199
|
+
const content = 'Line 1\\nLine 2\\nLine 3';
|
|
200
|
+
actualNodeFs.writeFileSync(testTextFilePath, content);
|
|
201
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir);
|
|
202
|
+
expect(result.llmContent).toBe(content);
|
|
203
|
+
expect(result.returnDisplay).toBe('');
|
|
204
|
+
expect(result.error).toBeUndefined();
|
|
205
|
+
});
|
|
206
|
+
it('should handle file not found', async () => {
|
|
207
|
+
const result = await processSingleFileContent(nonExistentFilePath, tempRootDir);
|
|
208
|
+
expect(result.error).toContain('File not found');
|
|
209
|
+
expect(result.returnDisplay).toContain('File not found');
|
|
210
|
+
});
|
|
211
|
+
it('should handle read errors for text files', async () => {
|
|
212
|
+
actualNodeFs.writeFileSync(testTextFilePath, 'content'); // File must exist for initial statSync
|
|
213
|
+
const readError = new Error('Simulated read error');
|
|
214
|
+
vi.spyOn(fsPromises, 'readFile').mockRejectedValueOnce(readError);
|
|
215
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir);
|
|
216
|
+
expect(result.error).toContain('Simulated read error');
|
|
217
|
+
expect(result.returnDisplay).toContain('Simulated read error');
|
|
218
|
+
});
|
|
219
|
+
it('should handle read errors for image/pdf files', async () => {
|
|
220
|
+
actualNodeFs.writeFileSync(testImageFilePath, 'content'); // File must exist
|
|
221
|
+
mockMimeLookup.mockReturnValue('image/png');
|
|
222
|
+
const readError = new Error('Simulated image read error');
|
|
223
|
+
vi.spyOn(fsPromises, 'readFile').mockRejectedValueOnce(readError);
|
|
224
|
+
const result = await processSingleFileContent(testImageFilePath, tempRootDir);
|
|
225
|
+
expect(result.error).toContain('Simulated image read error');
|
|
226
|
+
expect(result.returnDisplay).toContain('Simulated image read error');
|
|
227
|
+
});
|
|
228
|
+
it('should process an image file', async () => {
|
|
229
|
+
const fakePngData = Buffer.from('fake png data');
|
|
230
|
+
actualNodeFs.writeFileSync(testImageFilePath, fakePngData);
|
|
231
|
+
mockMimeLookup.mockReturnValue('image/png');
|
|
232
|
+
const result = await processSingleFileContent(testImageFilePath, tempRootDir);
|
|
233
|
+
expect(result.llmContent.inlineData).toBeDefined();
|
|
234
|
+
expect(result.llmContent.inlineData
|
|
235
|
+
.mimeType).toBe('image/png');
|
|
236
|
+
expect(result.llmContent.inlineData.data).toBe(fakePngData.toString('base64'));
|
|
237
|
+
expect(result.returnDisplay).toContain('Read image file: image.png');
|
|
238
|
+
});
|
|
239
|
+
it('should process a PDF file', async () => {
|
|
240
|
+
const fakePdfData = Buffer.from('fake pdf data');
|
|
241
|
+
actualNodeFs.writeFileSync(testPdfFilePath, fakePdfData);
|
|
242
|
+
mockMimeLookup.mockReturnValue('application/pdf');
|
|
243
|
+
const result = await processSingleFileContent(testPdfFilePath, tempRootDir);
|
|
244
|
+
expect(result.llmContent.inlineData).toBeDefined();
|
|
245
|
+
expect(result.llmContent.inlineData
|
|
246
|
+
.mimeType).toBe('application/pdf');
|
|
247
|
+
expect(result.llmContent.inlineData.data).toBe(fakePdfData.toString('base64'));
|
|
248
|
+
expect(result.returnDisplay).toContain('Read pdf file: document.pdf');
|
|
249
|
+
});
|
|
250
|
+
it('should read an SVG file as text when under 1MB', async () => {
|
|
251
|
+
const svgContent = `
|
|
252
|
+
<svg xmlns="http://www.w3.org/2000/svg" width="100" height="100">
|
|
253
|
+
<rect width="100" height="100" fill="blue" />
|
|
254
|
+
</svg>
|
|
255
|
+
`;
|
|
256
|
+
const testSvgFilePath = path.join(tempRootDir, 'test.svg');
|
|
257
|
+
actualNodeFs.writeFileSync(testSvgFilePath, svgContent, 'utf-8');
|
|
258
|
+
mockMimeLookup.mockReturnValue('image/svg+xml');
|
|
259
|
+
const result = await processSingleFileContent(testSvgFilePath, tempRootDir);
|
|
260
|
+
expect(result.llmContent).toBe(svgContent);
|
|
261
|
+
expect(result.returnDisplay).toContain('Read SVG as text');
|
|
262
|
+
});
|
|
263
|
+
it('should skip binary files', async () => {
|
|
264
|
+
actualNodeFs.writeFileSync(testBinaryFilePath, Buffer.from([0x00, 0x01, 0x02]));
|
|
265
|
+
mockMimeLookup.mockReturnValueOnce('application/octet-stream');
|
|
266
|
+
// isBinaryFile will operate on the real file.
|
|
267
|
+
const result = await processSingleFileContent(testBinaryFilePath, tempRootDir);
|
|
268
|
+
expect(result.llmContent).toContain('Cannot display content of binary file');
|
|
269
|
+
expect(result.returnDisplay).toContain('Skipped binary file: app.exe');
|
|
270
|
+
});
|
|
271
|
+
it('should handle path being a directory', async () => {
|
|
272
|
+
const result = await processSingleFileContent(directoryPath, tempRootDir);
|
|
273
|
+
expect(result.error).toContain('Path is a directory');
|
|
274
|
+
expect(result.returnDisplay).toContain('Path is a directory');
|
|
275
|
+
});
|
|
276
|
+
it('should paginate text files correctly (offset and limit)', async () => {
|
|
277
|
+
const lines = Array.from({ length: 20 }, (_, i) => `Line ${i + 1}`);
|
|
278
|
+
actualNodeFs.writeFileSync(testTextFilePath, lines.join('\n'));
|
|
279
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir, 5, 5); // Read lines 6-10
|
|
280
|
+
const expectedContent = lines.slice(5, 10).join('\n');
|
|
281
|
+
expect(result.llmContent).toContain(expectedContent);
|
|
282
|
+
expect(result.llmContent).toContain('[File content truncated. Showing lines 6-10 of 20. To read the next chunk, use offset: 10, limit: 5.]');
|
|
283
|
+
expect(result.returnDisplay).toBe('(truncated)');
|
|
284
|
+
expect(result.isTruncated).toBe(true);
|
|
285
|
+
expect(result.originalLineCount).toBe(20);
|
|
286
|
+
expect(result.linesShown).toEqual([6, 10]);
|
|
287
|
+
});
|
|
288
|
+
it('should handle limit exceeding file length', async () => {
|
|
289
|
+
const lines = ['Line 1', 'Line 2'];
|
|
290
|
+
actualNodeFs.writeFileSync(testTextFilePath, lines.join('\n'));
|
|
291
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir, 0, 10);
|
|
292
|
+
const expectedContent = lines.join('\n');
|
|
293
|
+
expect(result.llmContent).toBe(expectedContent);
|
|
294
|
+
expect(result.returnDisplay).toBe('');
|
|
295
|
+
expect(result.isTruncated).toBe(false);
|
|
296
|
+
expect(result.originalLineCount).toBe(2);
|
|
297
|
+
expect(result.linesShown).toEqual([1, 2]);
|
|
298
|
+
});
|
|
299
|
+
it('should truncate long lines in text files', async () => {
|
|
300
|
+
const longLine = 'a'.repeat(2500);
|
|
301
|
+
actualNodeFs.writeFileSync(testTextFilePath, `Short line\n${longLine}\nAnother short line`);
|
|
302
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir);
|
|
303
|
+
expect(result.llmContent).toContain('Short line');
|
|
304
|
+
expect(result.llmContent).toContain(longLine.substring(0, 2000) + '... [truncated]');
|
|
305
|
+
expect(result.llmContent).toContain('Another short line');
|
|
306
|
+
expect(result.llmContent).toContain('[File content partially truncated: some lines exceeded maximum length of 2000 characters.]');
|
|
307
|
+
expect(result.isTruncated).toBe(true);
|
|
308
|
+
});
|
|
309
|
+
it('should return an error if the file size exceeds 20MB', async () => {
|
|
310
|
+
// Create a file just over 20MB
|
|
311
|
+
const twentyOneMB = 21 * 1024 * 1024;
|
|
312
|
+
const buffer = Buffer.alloc(twentyOneMB, 0x61); // Fill with 'a'
|
|
313
|
+
actualNodeFs.writeFileSync(testTextFilePath, buffer);
|
|
314
|
+
const result = await processSingleFileContent(testTextFilePath, tempRootDir);
|
|
315
|
+
expect(result.error).toContain('File size exceeds the 20MB limit');
|
|
316
|
+
expect(result.returnDisplay).toContain('File size exceeds the 20MB limit');
|
|
317
|
+
expect(result.llmContent).toContain('File size exceeds the 20MB limit');
|
|
318
|
+
});
|
|
319
|
+
});
|
|
320
|
+
});
|
|
321
|
+
//# sourceMappingURL=fileUtils.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fileUtils.test.js","sourceRoot":"","sources":["../../../src/utils/fileUtils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAEhB,OAAO,KAAK,YAAY,MAAM,SAAS,CAAC,CAAC,qBAAqB;AAC9D,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAC1C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,YAAY,CAAC;AAE9B,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,cAAc,EACd,wBAAwB,GACzB,MAAM,gBAAgB,CAAC;AAExB,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3B,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;IAC5B,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;CAChB,CAAC,CAAC,CAAC;AAEJ,MAAM,cAAc,GAAG,IAAI,CAAC,MAAc,CAAC;AAE3C,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,WAAmB,CAAC;IACxB,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC;IAEvC,IAAI,gBAAwB,CAAC;IAC7B,IAAI,iBAAyB,CAAC;IAC9B,IAAI,eAAuB,CAAC;IAC5B,IAAI,kBAA0B,CAAC;IAC/B,IAAI,mBAA2B,CAAC;IAChC,IAAI,aAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,yCAAyC;QAE7D,WAAW,GAAG,YAAY,CAAC,WAAW,CACpC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAC1C,CAAC;QACF,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,6DAA6D;QAErG,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACtD,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;QACxD,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QACzD,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACvD,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QAC7D,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAEjD,YAAY,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,uBAAuB;IACrF,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YACzC,YAAY,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrE,CAAC;QACD,OAAO,CAAC,GAAG,GAAG,kBAAkB,CAAC;QACjC,EAAE,CAAC,eAAe,EAAE,CAAC,CAAC,oBAAoB;IAC5C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAE3C,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnE,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CACpE,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,CAC/D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACd,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CACrE,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,MAAM,CACJ,YAAY,CACV,IAAI,CAAC,OAAO,CAAC,sCAAsC,CAAC,EACpD,IAAI,CACL,CACF,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CACrE,IAAI,CACL,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,SAAS,GAAG,eAAe,CAAC;YAClC,MAAM,eAAe,GAAG,wBAAwB,CAAC;YACjD,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;YACnD,MAAM,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,CAAC,YAAY,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEtD,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;YACvD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;YACpD,MAAM,CAAC,YAAY,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,qBAA6B,CAAC;QAElC,UAAU,CAAC,GAAG,EAAE;YACd,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,iBAAiB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBACnD,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC;YACjD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,YAAY,CAAC,aAAa,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,YAAY,CAAC,aAAa,CACxB,qBAAqB,EACrB,8DAA8D,CAC/D,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;aAC3D,CAAC,CAAC,CAAC,sBAAsB;YAC1B,YAAY,CAAC,aAAa,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YACjE,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;aAC3D,CAAC,CAAC,CAAC,+BAA+B;YACnC,YAAY,CAAC,aAAa,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YACjE,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,iCAAiC;YACjC,IAAI,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBACnD,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,IAAI,qBAA6B,CAAC;QAElC,UAAU,CAAC,GAAG,EAAE;YACd,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC;YACjE,2DAA2D;YAC3D,YAAY,CAAC,aAAa,CAAC,qBAAqB,EAAE,oBAAoB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBACnD,YAAY,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC;YACjD,CAAC;YACD,EAAE,CAAC,eAAe,EAAE,CAAC,CAAC,gCAAgC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,cAAc,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,cAAc,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACjD,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,cAAc,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;YACtD,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,cAAc,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACjD,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,cAAc,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,cAAc,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;YACtD,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,cAAc,CAAC,mBAAmB,CAAC,0BAA0B,CAAC,CAAC,CAAC,kBAAkB;YAClF,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,cAAc,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAoB;YAC/D,0DAA0D;YAC1D,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;aAC3D,CAAC,CAAC;YACH,YAAY,CAAC,aAAa,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YACjE,MAAM,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,cAAc,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAoB;YAC/D,0EAA0E;YAC1E,MAAM,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,UAAU,CAAC,GAAG,EAAE;YACd,yEAAyE;YACzE,IAAI,YAAY,CAAC,UAAU,CAAC,gBAAgB,CAAC;gBAC3C,YAAY,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;YAC5C,IAAI,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC;gBAC5C,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;YAC7C,IAAI,YAAY,CAAC,UAAU,CAAC,eAAe,CAAC;gBAC1C,YAAY,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC3C,IAAI,YAAY,CAAC,UAAU,CAAC,kBAAkB,CAAC;gBAC7C,YAAY,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,OAAO,GAAG,0BAA0B,CAAC;YAC3C,YAAY,CAAC,aAAa,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,mBAAmB,EACnB,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,YAAY,CAAC,aAAa,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC,CAAC,uCAAuC;YAChG,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YACpD,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAElE,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,YAAY,CAAC,aAAa,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC,CAAC,kBAAkB;YAC5E,cAAc,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAC1D,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAElE,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,iBAAiB,EACjB,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACjD,YAAY,CAAC,aAAa,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;YAC3D,cAAc,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,iBAAiB,EACjB,WAAW,CACZ,CAAC;YACF,MAAM,CACH,MAAM,CAAC,UAAsC,CAAC,UAAU,CAC1D,CAAC,WAAW,EAAE,CAAC;YAChB,MAAM,CACH,MAAM,CAAC,UAAmD,CAAC,UAAU;iBACnE,QAAQ,CACZ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpB,MAAM,CACH,MAAM,CAAC,UAA+C,CAAC,UAAU,CAAC,IAAI,CACxE,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACjD,YAAY,CAAC,aAAa,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;YACzD,cAAc,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,eAAe,EACf,WAAW,CACZ,CAAC;YACF,MAAM,CACH,MAAM,CAAC,UAAsC,CAAC,UAAU,CAC1D,CAAC,WAAW,EAAE,CAAC;YAChB,MAAM,CACH,MAAM,CAAC,UAAmD,CAAC,UAAU;iBACnE,QAAQ,CACZ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC1B,MAAM,CACH,MAAM,CAAC,UAA+C,CAAC,UAAU,CAAC,IAAI,CACxE,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG;;;;GAItB,CAAC;YACE,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YAC3D,YAAY,CAAC,aAAa,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YAEjE,cAAc,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAEhD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,eAAe,EACf,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,YAAY,CAAC,aAAa,CACxB,kBAAkB,EAClB,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAChC,CAAC;YACF,cAAc,CAAC,mBAAmB,CAAC,0BAA0B,CAAC,CAAC;YAC/D,8CAA8C;YAE9C,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,kBAAkB,EAClB,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CACjC,uCAAuC,CACxC,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpE,YAAY,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,EACX,CAAC,EACD,CAAC,CACF,CAAC,CAAC,kBAAkB;YACrB,MAAM,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEtD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CACjC,uGAAuG,CACxG,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,KAAK,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACnC,YAAY,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,EACX,CAAC,EACD,EAAE,CACH,CAAC;YACF,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClC,YAAY,CAAC,aAAa,CACxB,gBAAgB,EAChB,eAAe,QAAQ,sBAAsB,CAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CACjC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,iBAAiB,CAChD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CACjC,4FAA4F,CAC7F,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,+BAA+B;YAC/B,MAAM,WAAW,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;YACrC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,gBAAgB;YAChE,YAAY,CAAC,aAAa,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,gBAAgB,EAChB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CACpC,kCAAkC,CACnC,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
7
|
+
import { Config } from '../config/config.js';
|
|
8
|
+
import { setSimulate429, disableSimulationAfterFallback, shouldSimulate429, createSimulated429Error, resetRequestCounter, } from './testUtils.js';
|
|
9
|
+
import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js';
|
|
10
|
+
import { retryWithBackoff } from './retry.js';
|
|
11
|
+
import { AuthType } from '../core/contentGenerator.js';
|
|
12
|
+
describe('Flash Fallback Integration', () => {
|
|
13
|
+
let config;
|
|
14
|
+
beforeEach(() => {
|
|
15
|
+
config = new Config({
|
|
16
|
+
sessionId: 'test-session',
|
|
17
|
+
targetDir: '/test',
|
|
18
|
+
debugMode: false,
|
|
19
|
+
cwd: '/test',
|
|
20
|
+
model: 'gemini-2.5-pro',
|
|
21
|
+
});
|
|
22
|
+
// Reset simulation state for each test
|
|
23
|
+
setSimulate429(false);
|
|
24
|
+
resetRequestCounter();
|
|
25
|
+
});
|
|
26
|
+
it('should automatically accept fallback', async () => {
|
|
27
|
+
// Set up a minimal flash fallback handler for testing
|
|
28
|
+
const flashFallbackHandler = async () => true;
|
|
29
|
+
config.setFlashFallbackHandler(flashFallbackHandler);
|
|
30
|
+
// Call the handler directly to test
|
|
31
|
+
const result = await config.flashFallbackHandler('gemini-2.5-pro', DEFAULT_GEMINI_FLASH_MODEL);
|
|
32
|
+
// Verify it automatically accepts
|
|
33
|
+
expect(result).toBe(true);
|
|
34
|
+
});
|
|
35
|
+
it('should trigger fallback after 2 consecutive 429 errors for OAuth users', async () => {
|
|
36
|
+
let fallbackCalled = false;
|
|
37
|
+
let fallbackModel = '';
|
|
38
|
+
// Mock function that simulates exactly 2 429 errors, then succeeds after fallback
|
|
39
|
+
const mockApiCall = vi
|
|
40
|
+
.fn()
|
|
41
|
+
.mockRejectedValueOnce(createSimulated429Error())
|
|
42
|
+
.mockRejectedValueOnce(createSimulated429Error())
|
|
43
|
+
.mockResolvedValueOnce('success after fallback');
|
|
44
|
+
// Mock fallback handler
|
|
45
|
+
const mockFallbackHandler = vi.fn(async (_authType) => {
|
|
46
|
+
fallbackCalled = true;
|
|
47
|
+
fallbackModel = DEFAULT_GEMINI_FLASH_MODEL;
|
|
48
|
+
return fallbackModel;
|
|
49
|
+
});
|
|
50
|
+
// Test with OAuth personal auth type, with maxAttempts = 2 to ensure fallback triggers
|
|
51
|
+
const result = await retryWithBackoff(mockApiCall, {
|
|
52
|
+
maxAttempts: 2,
|
|
53
|
+
initialDelayMs: 1,
|
|
54
|
+
maxDelayMs: 10,
|
|
55
|
+
shouldRetry: (error) => {
|
|
56
|
+
const status = error.status;
|
|
57
|
+
return status === 429;
|
|
58
|
+
},
|
|
59
|
+
onPersistent429: mockFallbackHandler,
|
|
60
|
+
authType: AuthType.LOGIN_WITH_GOOGLE,
|
|
61
|
+
});
|
|
62
|
+
// Verify fallback was triggered
|
|
63
|
+
expect(fallbackCalled).toBe(true);
|
|
64
|
+
expect(fallbackModel).toBe(DEFAULT_GEMINI_FLASH_MODEL);
|
|
65
|
+
expect(mockFallbackHandler).toHaveBeenCalledWith(AuthType.LOGIN_WITH_GOOGLE, expect.any(Error));
|
|
66
|
+
expect(result).toBe('success after fallback');
|
|
67
|
+
// Should have: 2 failures, then fallback triggered, then 1 success after retry reset
|
|
68
|
+
expect(mockApiCall).toHaveBeenCalledTimes(3);
|
|
69
|
+
});
|
|
70
|
+
it('should not trigger fallback for API key users', async () => {
|
|
71
|
+
let fallbackCalled = false;
|
|
72
|
+
// Mock function that simulates 429 errors
|
|
73
|
+
const mockApiCall = vi.fn().mockRejectedValue(createSimulated429Error());
|
|
74
|
+
// Mock fallback handler
|
|
75
|
+
const mockFallbackHandler = vi.fn(async () => {
|
|
76
|
+
fallbackCalled = true;
|
|
77
|
+
return DEFAULT_GEMINI_FLASH_MODEL;
|
|
78
|
+
});
|
|
79
|
+
// Test with API key auth type - should not trigger fallback
|
|
80
|
+
try {
|
|
81
|
+
await retryWithBackoff(mockApiCall, {
|
|
82
|
+
maxAttempts: 5,
|
|
83
|
+
initialDelayMs: 10,
|
|
84
|
+
maxDelayMs: 100,
|
|
85
|
+
shouldRetry: (error) => {
|
|
86
|
+
const status = error.status;
|
|
87
|
+
return status === 429;
|
|
88
|
+
},
|
|
89
|
+
onPersistent429: mockFallbackHandler,
|
|
90
|
+
authType: AuthType.USE_GEMINI, // API key auth type
|
|
91
|
+
});
|
|
92
|
+
}
|
|
93
|
+
catch (error) {
|
|
94
|
+
// Expected to throw after max attempts
|
|
95
|
+
expect(error.message).toContain('Rate limit exceeded');
|
|
96
|
+
}
|
|
97
|
+
// Verify fallback was NOT triggered for API key users
|
|
98
|
+
expect(fallbackCalled).toBe(false);
|
|
99
|
+
expect(mockFallbackHandler).not.toHaveBeenCalled();
|
|
100
|
+
});
|
|
101
|
+
it('should properly disable simulation state after fallback', () => {
|
|
102
|
+
// Enable simulation
|
|
103
|
+
setSimulate429(true);
|
|
104
|
+
// Verify simulation is enabled
|
|
105
|
+
expect(shouldSimulate429()).toBe(true);
|
|
106
|
+
// Disable simulation after fallback
|
|
107
|
+
disableSimulationAfterFallback();
|
|
108
|
+
// Verify simulation is now disabled
|
|
109
|
+
expect(shouldSimulate429()).toBe(false);
|
|
110
|
+
});
|
|
111
|
+
});
|
|
112
|
+
//# sourceMappingURL=flashFallback.integration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"flashFallback.integration.test.js","sourceRoot":"","sources":["../../../src/utils/flashFallback.integration.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EACL,cAAc,EACd,8BAA8B,EAC9B,iBAAiB,EACjB,uBAAuB,EACvB,mBAAmB,GACpB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAEvD,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,MAAM,CAAC;YAClB,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,OAAO;YAClB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,OAAO;YACZ,KAAK,EAAE,gBAAgB;SACxB,CAAC,CAAC;QAEH,uCAAuC;QACvC,cAAc,CAAC,KAAK,CAAC,CAAC;QACtB,mBAAmB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,sDAAsD;QACtD,MAAM,oBAAoB,GAAG,KAAK,IAAsB,EAAE,CAAC,IAAI,CAAC;QAEhE,MAAM,CAAC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;QAErD,oCAAoC;QACpC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,oBAAqB,CAC/C,gBAAgB,EAChB,0BAA0B,CAC3B,CAAC;QAEF,kCAAkC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,aAAa,GAAG,EAAE,CAAC;QAEvB,kFAAkF;QAClF,MAAM,WAAW,GAAG,EAAE;aACnB,EAAE,EAAE;aACJ,qBAAqB,CAAC,uBAAuB,EAAE,CAAC;aAChD,qBAAqB,CAAC,uBAAuB,EAAE,CAAC;aAChD,qBAAqB,CAAC,wBAAwB,CAAC,CAAC;QAEnD,wBAAwB;QACxB,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,SAAkB,EAAE,EAAE;YAC7D,cAAc,GAAG,IAAI,CAAC;YACtB,aAAa,GAAG,0BAA0B,CAAC;YAC3C,OAAO,aAAa,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,uFAAuF;QACvF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,WAAW,EAAE;YACjD,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,CAAC;YACjB,UAAU,EAAE,EAAE;YACd,WAAW,EAAE,CAAC,KAAY,EAAE,EAAE;gBAC5B,MAAM,MAAM,GAAI,KAAqC,CAAC,MAAM,CAAC;gBAC7D,OAAO,MAAM,KAAK,GAAG,CAAC;YACxB,CAAC;YACD,eAAe,EAAE,mBAAmB;YACpC,QAAQ,EAAE,QAAQ,CAAC,iBAAiB;SACrC,CAAC,CAAC;QAEH,gCAAgC;QAChC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QACvD,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,QAAQ,CAAC,iBAAiB,EAC1B,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC9C,qFAAqF;QACrF,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,IAAI,cAAc,GAAG,KAAK,CAAC;QAE3B,0CAA0C;QAC1C,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAEzE,wBAAwB;QACxB,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAC3C,cAAc,GAAG,IAAI,CAAC;YACtB,OAAO,0BAA0B,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,4DAA4D;QAC5D,IAAI,CAAC;YACH,MAAM,gBAAgB,CAAC,WAAW,EAAE;gBAClC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,EAAE;gBAClB,UAAU,EAAE,GAAG;gBACf,WAAW,EAAE,CAAC,KAAY,EAAE,EAAE;oBAC5B,MAAM,MAAM,GAAI,KAAqC,CAAC,MAAM,CAAC;oBAC7D,OAAO,MAAM,KAAK,GAAG,CAAC;gBACxB,CAAC;gBACD,eAAe,EAAE,mBAAmB;gBACpC,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,oBAAoB;aACpD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,uCAAuC;YACvC,MAAM,CAAE,KAAe,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QACpE,CAAC;QAED,sDAAsD;QACtD,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,oBAAoB;QACpB,cAAc,CAAC,IAAI,CAAC,CAAC;QAErB,+BAA+B;QAC/B,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEvC,oCAAoC;QACpC,8BAA8B,EAAE,CAAC;QAEjC,oCAAoC;QACpC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { GenerateContentResponse, Part, FunctionCall } from '@google/genai';
|
|
7
|
+
export declare function getResponseText(response: GenerateContentResponse): string | undefined;
|
|
8
|
+
export declare function getResponseTextFromParts(parts: Part[]): string | undefined;
|
|
9
|
+
export declare function getFunctionCalls(response: GenerateContentResponse): FunctionCall[] | undefined;
|
|
10
|
+
export declare function getFunctionCallsFromParts(parts: Part[]): FunctionCall[] | undefined;
|
|
11
|
+
export declare function getFunctionCallsAsJson(response: GenerateContentResponse): string | undefined;
|
|
12
|
+
export declare function getFunctionCallsFromPartsAsJson(parts: Part[]): string | undefined;
|
|
13
|
+
export declare function getStructuredResponse(response: GenerateContentResponse): string | undefined;
|
|
14
|
+
export declare function getStructuredResponseFromParts(parts: Part[]): string | undefined;
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
export function getResponseText(response) {
|
|
7
|
+
const parts = response.candidates?.[0]?.content?.parts;
|
|
8
|
+
if (!parts) {
|
|
9
|
+
return undefined;
|
|
10
|
+
}
|
|
11
|
+
const textSegments = parts
|
|
12
|
+
.map((part) => part.text)
|
|
13
|
+
.filter((text) => typeof text === 'string');
|
|
14
|
+
if (textSegments.length === 0) {
|
|
15
|
+
return undefined;
|
|
16
|
+
}
|
|
17
|
+
return textSegments.join('');
|
|
18
|
+
}
|
|
19
|
+
export function getResponseTextFromParts(parts) {
|
|
20
|
+
if (!parts) {
|
|
21
|
+
return undefined;
|
|
22
|
+
}
|
|
23
|
+
const textSegments = parts
|
|
24
|
+
.map((part) => part.text)
|
|
25
|
+
.filter((text) => typeof text === 'string');
|
|
26
|
+
if (textSegments.length === 0) {
|
|
27
|
+
return undefined;
|
|
28
|
+
}
|
|
29
|
+
return textSegments.join('');
|
|
30
|
+
}
|
|
31
|
+
export function getFunctionCalls(response) {
|
|
32
|
+
const parts = response.candidates?.[0]?.content?.parts;
|
|
33
|
+
if (!parts) {
|
|
34
|
+
return undefined;
|
|
35
|
+
}
|
|
36
|
+
const functionCallParts = parts
|
|
37
|
+
.filter((part) => !!part.functionCall)
|
|
38
|
+
.map((part) => part.functionCall);
|
|
39
|
+
return functionCallParts.length > 0 ? functionCallParts : undefined;
|
|
40
|
+
}
|
|
41
|
+
export function getFunctionCallsFromParts(parts) {
|
|
42
|
+
if (!parts) {
|
|
43
|
+
return undefined;
|
|
44
|
+
}
|
|
45
|
+
const functionCallParts = parts
|
|
46
|
+
.filter((part) => !!part.functionCall)
|
|
47
|
+
.map((part) => part.functionCall);
|
|
48
|
+
return functionCallParts.length > 0 ? functionCallParts : undefined;
|
|
49
|
+
}
|
|
50
|
+
export function getFunctionCallsAsJson(response) {
|
|
51
|
+
const functionCalls = getFunctionCalls(response);
|
|
52
|
+
if (!functionCalls) {
|
|
53
|
+
return undefined;
|
|
54
|
+
}
|
|
55
|
+
return JSON.stringify(functionCalls, null, 2);
|
|
56
|
+
}
|
|
57
|
+
export function getFunctionCallsFromPartsAsJson(parts) {
|
|
58
|
+
const functionCalls = getFunctionCallsFromParts(parts);
|
|
59
|
+
if (!functionCalls) {
|
|
60
|
+
return undefined;
|
|
61
|
+
}
|
|
62
|
+
return JSON.stringify(functionCalls, null, 2);
|
|
63
|
+
}
|
|
64
|
+
export function getStructuredResponse(response) {
|
|
65
|
+
const textContent = getResponseText(response);
|
|
66
|
+
const functionCallsJson = getFunctionCallsAsJson(response);
|
|
67
|
+
if (textContent && functionCallsJson) {
|
|
68
|
+
return `${textContent}\n${functionCallsJson}`;
|
|
69
|
+
}
|
|
70
|
+
if (textContent) {
|
|
71
|
+
return textContent;
|
|
72
|
+
}
|
|
73
|
+
if (functionCallsJson) {
|
|
74
|
+
return functionCallsJson;
|
|
75
|
+
}
|
|
76
|
+
return undefined;
|
|
77
|
+
}
|
|
78
|
+
export function getStructuredResponseFromParts(parts) {
|
|
79
|
+
const textContent = getResponseTextFromParts(parts);
|
|
80
|
+
const functionCallsJson = getFunctionCallsFromPartsAsJson(parts);
|
|
81
|
+
if (textContent && functionCallsJson) {
|
|
82
|
+
return `${textContent}\n${functionCallsJson}`;
|
|
83
|
+
}
|
|
84
|
+
if (textContent) {
|
|
85
|
+
return textContent;
|
|
86
|
+
}
|
|
87
|
+
if (functionCallsJson) {
|
|
88
|
+
return functionCallsJson;
|
|
89
|
+
}
|
|
90
|
+
return undefined;
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=generateContentResponseUtilities.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generateContentResponseUtilities.js","sourceRoot":"","sources":["../../../src/utils/generateContentResponseUtilities.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,UAAU,eAAe,CAC7B,QAAiC;IAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,KAAK;SACvB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;IAE9D,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,KAAa;IACpD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,KAAK;SACvB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;IAE9D,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,QAAiC;IAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,iBAAiB,GAAG,KAAK;SAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;SACrC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAA4B,CAAC,CAAC;IACpD,OAAO,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC;AACtE,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,KAAa;IAEb,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,iBAAiB,GAAG,KAAK;SAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;SACrC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAA4B,CAAC,CAAC;IACpD,OAAO,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC;AACtE,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,QAAiC;IAEjC,MAAM,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACjD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,KAAa;IAEb,MAAM,aAAa,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;IACvD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,QAAiC;IAEjC,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9C,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAE3D,IAAI,WAAW,IAAI,iBAAiB,EAAE,CAAC;QACrC,OAAO,GAAG,WAAW,KAAK,iBAAiB,EAAE,CAAC;IAChD,CAAC;IACD,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAAa;IAEb,MAAM,WAAW,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACpD,MAAM,iBAAiB,GAAG,+BAA+B,CAAC,KAAK,CAAC,CAAC;IAEjE,IAAI,WAAW,IAAI,iBAAiB,EAAE,CAAC;QACrC,OAAO,GAAG,WAAW,KAAK,iBAAiB,EAAE,CAAC;IAChD,CAAC;IACD,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}
|