@google/gemini-cli-core 0.1.0-development.1
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/LICENSE +202 -0
- package/README.md +156 -0
- 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 +8 -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 +26 -0
- package/dist/src/code_assist/oauth2.js +277 -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 +195 -0
- package/dist/src/code_assist/oauth2.test.js.map +1 -0
- package/dist/src/code_assist/server.d.ts +35 -0
- package/dist/src/code_assist/server.js +123 -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/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 +175 -0
- package/dist/src/config/config.js +385 -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 +263 -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 +39 -0
- package/dist/src/core/client.js +386 -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 +630 -0
- package/dist/src/core/client.test.js.map +1 -0
- package/dist/src/core/contentGenerator.d.ts +29 -0
- package/dist/src/core/contentGenerator.js +71 -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 +45 -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 +397 -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 +318 -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 +491 -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 +422 -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 +35 -0
- package/dist/src/core/logger.js +235 -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 +92 -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 +181 -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 +334 -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 +27 -0
- package/dist/src/core/tokenLimits.js.map +1 -0
- package/dist/src/core/turn.d.ts +96 -0
- package/dist/src/core/turn.js +125 -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 +236 -0
- package/dist/src/core/turn.test.js.map +1 -0
- package/dist/src/index.d.ts +46 -0
- package/dist/src/index.js +54 -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/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 +105 -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 +233 -0
- package/dist/src/services/gitService.test.js.map +1 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts +33 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js +364 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js.map +1 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.d.ts +43 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js +109 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js.map +1 -0
- package/dist/src/telemetry/constants.d.ts +19 -0
- package/dist/src/telemetry/constants.js +20 -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/loggers.d.ts +13 -0
- package/dist/src/telemetry/loggers.js +187 -0
- package/dist/src/telemetry/loggers.js.map +1 -0
- package/dist/src/telemetry/loggers.test.d.ts +6 -0
- package/dist/src/telemetry/loggers.test.js +525 -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 +116 -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 +50 -0
- package/dist/src/telemetry/telemetry.test.js.map +1 -0
- package/dist/src/telemetry/types.d.ts +91 -0
- package/dist/src/telemetry/types.js +176 -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 +411 -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 +84 -0
- package/dist/src/tools/edit.js +359 -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 +71 -0
- package/dist/src/tools/glob.js +215 -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 +293 -0
- package/dist/src/tools/glob.test.js.map +1 -0
- package/dist/src/tools/grep.d.ts +81 -0
- package/dist/src/tools/grep.js +427 -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 +185 -0
- package/dist/src/tools/grep.test.js.map +1 -0
- package/dist/src/tools/ls.d.ts +94 -0
- package/dist/src/tools/ls.js +215 -0
- package/dist/src/tools/ls.js.map +1 -0
- package/dist/src/tools/mcp-client.d.ts +56 -0
- package/dist/src/tools/mcp-client.js +302 -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 +608 -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 +171 -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 +36 -0
- package/dist/src/tools/read-file.js +103 -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 +180 -0
- package/dist/src/tools/read-file.test.js.map +1 -0
- package/dist/src/tools/read-many-files.d.ts +65 -0
- package/dist/src/tools/read-many-files.js +344 -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 +300 -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 +421 -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 +321 -0
- package/dist/src/tools/shell.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 +353 -0
- package/dist/src/tools/tool-registry.test.js.map +1 -0
- package/dist/src/tools/tools.d.ts +179 -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 +240 -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 +119 -0
- package/dist/src/tools/web-search.js.map +1 -0
- package/dist/src/tools/write-file.d.ts +50 -0
- package/dist/src/tools/write-file.js +261 -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 +560 -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 +284 -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 +300 -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 +111 -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 +152 -0
- package/dist/src/utils/nextSpeakerChecker.test.js.map +1 -0
- package/dist/src/utils/paths.d.ts +47 -0
- package/dist/src/utils/paths.js +136 -0
- package/dist/src/utils/paths.js.map +1 -0
- package/dist/src/utils/retry.d.ts +21 -0
- package/dist/src/utils/retry.js +220 -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/schemaValidator.d.ts +17 -0
- package/dist/src/utils/schemaValidator.js +50 -0
- package/dist/src/utils/schemaValidator.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/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_id.d.ts +17 -0
- package/dist/src/utils/user_id.js +70 -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 +42 -0
- package/dist/src/utils/user_id.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -0
- package/package.json +60 -0
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { getResponseText } from '../utils/generateContentResponseUtilities.js';
|
|
7
|
+
import { reportError } from '../utils/errorReporting.js';
|
|
8
|
+
import { getErrorMessage, UnauthorizedError, toFriendlyError, } from '../utils/errors.js';
|
|
9
|
+
export var GeminiEventType;
|
|
10
|
+
(function (GeminiEventType) {
|
|
11
|
+
GeminiEventType["Content"] = "content";
|
|
12
|
+
GeminiEventType["ToolCallRequest"] = "tool_call_request";
|
|
13
|
+
GeminiEventType["ToolCallResponse"] = "tool_call_response";
|
|
14
|
+
GeminiEventType["ToolCallConfirmation"] = "tool_call_confirmation";
|
|
15
|
+
GeminiEventType["UserCancelled"] = "user_cancelled";
|
|
16
|
+
GeminiEventType["Error"] = "error";
|
|
17
|
+
GeminiEventType["ChatCompressed"] = "chat_compressed";
|
|
18
|
+
GeminiEventType["Thought"] = "thought";
|
|
19
|
+
})(GeminiEventType || (GeminiEventType = {}));
|
|
20
|
+
// A turn manages the agentic loop turn within the server context.
|
|
21
|
+
export class Turn {
|
|
22
|
+
chat;
|
|
23
|
+
pendingToolCalls;
|
|
24
|
+
debugResponses;
|
|
25
|
+
constructor(chat) {
|
|
26
|
+
this.chat = chat;
|
|
27
|
+
this.pendingToolCalls = [];
|
|
28
|
+
this.debugResponses = [];
|
|
29
|
+
}
|
|
30
|
+
// The run method yields simpler events suitable for server logic
|
|
31
|
+
async *run(req, signal) {
|
|
32
|
+
try {
|
|
33
|
+
const responseStream = await this.chat.sendMessageStream({
|
|
34
|
+
message: req,
|
|
35
|
+
config: {
|
|
36
|
+
abortSignal: signal,
|
|
37
|
+
},
|
|
38
|
+
});
|
|
39
|
+
for await (const resp of responseStream) {
|
|
40
|
+
if (signal?.aborted) {
|
|
41
|
+
yield { type: GeminiEventType.UserCancelled };
|
|
42
|
+
// Do not add resp to debugResponses if aborted before processing
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
this.debugResponses.push(resp);
|
|
46
|
+
const thoughtPart = resp.candidates?.[0]?.content?.parts?.[0];
|
|
47
|
+
if (thoughtPart?.thought) {
|
|
48
|
+
// Thought always has a bold "subject" part enclosed in double asterisks
|
|
49
|
+
// (e.g., **Subject**). The rest of the string is considered the description.
|
|
50
|
+
const rawText = thoughtPart.text ?? '';
|
|
51
|
+
const subjectStringMatches = rawText.match(/\*\*(.*?)\*\*/s);
|
|
52
|
+
const subject = subjectStringMatches
|
|
53
|
+
? subjectStringMatches[1].trim()
|
|
54
|
+
: '';
|
|
55
|
+
const description = rawText.replace(/\*\*(.*?)\*\*/s, '').trim();
|
|
56
|
+
const thought = {
|
|
57
|
+
subject,
|
|
58
|
+
description,
|
|
59
|
+
};
|
|
60
|
+
yield {
|
|
61
|
+
type: GeminiEventType.Thought,
|
|
62
|
+
value: thought,
|
|
63
|
+
};
|
|
64
|
+
continue;
|
|
65
|
+
}
|
|
66
|
+
const text = getResponseText(resp);
|
|
67
|
+
if (text) {
|
|
68
|
+
yield { type: GeminiEventType.Content, value: text };
|
|
69
|
+
}
|
|
70
|
+
// Handle function calls (requesting tool execution)
|
|
71
|
+
const functionCalls = resp.functionCalls ?? [];
|
|
72
|
+
for (const fnCall of functionCalls) {
|
|
73
|
+
const event = this.handlePendingFunctionCall(fnCall);
|
|
74
|
+
if (event) {
|
|
75
|
+
yield event;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
catch (e) {
|
|
81
|
+
const error = toFriendlyError(e);
|
|
82
|
+
if (error instanceof UnauthorizedError) {
|
|
83
|
+
throw error;
|
|
84
|
+
}
|
|
85
|
+
if (signal.aborted) {
|
|
86
|
+
yield { type: GeminiEventType.UserCancelled };
|
|
87
|
+
// Regular cancellation error, fail gracefully.
|
|
88
|
+
return;
|
|
89
|
+
}
|
|
90
|
+
const contextForReport = [...this.chat.getHistory(/*curated*/ true), req];
|
|
91
|
+
await reportError(error, 'Error when talking to Gemini API', contextForReport, 'Turn.run-sendMessageStream');
|
|
92
|
+
const status = typeof error === 'object' &&
|
|
93
|
+
error !== null &&
|
|
94
|
+
'status' in error &&
|
|
95
|
+
typeof error.status === 'number'
|
|
96
|
+
? error.status
|
|
97
|
+
: undefined;
|
|
98
|
+
const structuredError = {
|
|
99
|
+
message: getErrorMessage(error),
|
|
100
|
+
status,
|
|
101
|
+
};
|
|
102
|
+
yield { type: GeminiEventType.Error, value: { error: structuredError } };
|
|
103
|
+
return;
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
handlePendingFunctionCall(fnCall) {
|
|
107
|
+
const callId = fnCall.id ??
|
|
108
|
+
`${fnCall.name}-${Date.now()}-${Math.random().toString(16).slice(2)}`;
|
|
109
|
+
const name = fnCall.name || 'undefined_tool_name';
|
|
110
|
+
const args = (fnCall.args || {});
|
|
111
|
+
const toolCallRequest = {
|
|
112
|
+
callId,
|
|
113
|
+
name,
|
|
114
|
+
args,
|
|
115
|
+
isClientInitiated: false,
|
|
116
|
+
};
|
|
117
|
+
this.pendingToolCalls.push(toolCallRequest);
|
|
118
|
+
// Yield a request for the tool call, not the pending/confirming status
|
|
119
|
+
return { type: GeminiEventType.ToolCallRequest, value: toolCallRequest };
|
|
120
|
+
}
|
|
121
|
+
getDebugResponses() {
|
|
122
|
+
return this.debugResponses;
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
//# sourceMappingURL=turn.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"turn.js","sourceRoot":"","sources":["../../../src/core/turn.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAaH,OAAO,EAAE,eAAe,EAAE,MAAM,8CAA8C,CAAC;AAC/E,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EACL,eAAe,EACf,iBAAiB,EACjB,eAAe,GAChB,MAAM,oBAAoB,CAAC;AAkB5B,MAAM,CAAN,IAAY,eASX;AATD,WAAY,eAAe;IACzB,sCAAmB,CAAA;IACnB,wDAAqC,CAAA;IACrC,0DAAuC,CAAA;IACvC,kEAA+C,CAAA;IAC/C,mDAAgC,CAAA;IAChC,kCAAe,CAAA;IACf,qDAAkC,CAAA;IAClC,sCAAmB,CAAA;AACrB,CAAC,EATW,eAAe,KAAf,eAAe,QAS1B;AA0FD,kEAAkE;AAClE,MAAM,OAAO,IAAI;IAIc;IAHpB,gBAAgB,CAAwB;IACzC,cAAc,CAA4B;IAElD,YAA6B,IAAgB;QAAhB,SAAI,GAAJ,IAAI,CAAY;QAC3C,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;IAC3B,CAAC;IACD,iEAAiE;IACjE,KAAK,CAAC,CAAC,GAAG,CACR,GAAkB,EAClB,MAAmB;QAEnB,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC;gBACvD,OAAO,EAAE,GAAG;gBACZ,MAAM,EAAE;oBACN,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;gBACxC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACpB,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,aAAa,EAAE,CAAC;oBAC9C,iEAAiE;oBACjE,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE/B,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC9D,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;oBACzB,wEAAwE;oBACxE,6EAA6E;oBAC7E,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC;oBACvC,MAAM,oBAAoB,GAAG,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;oBAC7D,MAAM,OAAO,GAAG,oBAAoB;wBAClC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;wBAChC,CAAC,CAAC,EAAE,CAAC;oBACP,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjE,MAAM,OAAO,GAAmB;wBAC9B,OAAO;wBACP,WAAW;qBACZ,CAAC;oBAEF,MAAM;wBACJ,IAAI,EAAE,eAAe,CAAC,OAAO;wBAC7B,KAAK,EAAE,OAAO;qBACf,CAAC;oBACF,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;gBACnC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBACvD,CAAC;gBAED,oDAAoD;gBACpD,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;gBAC/C,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;oBACnC,MAAM,KAAK,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;oBACrD,IAAI,KAAK,EAAE,CAAC;wBACV,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;gBACvC,MAAM,KAAK,CAAC;YACd,CAAC;YACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,aAAa,EAAE,CAAC;gBAC9C,+CAA+C;gBAC/C,OAAO;YACT,CAAC;YAED,MAAM,gBAAgB,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;YAC1E,MAAM,WAAW,CACf,KAAK,EACL,kCAAkC,EAClC,gBAAgB,EAChB,4BAA4B,CAC7B,CAAC;YACF,MAAM,MAAM,GACV,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,KAAK,IAAI;gBACd,QAAQ,IAAI,KAAK;gBACjB,OAAQ,KAA6B,CAAC,MAAM,KAAK,QAAQ;gBACvD,CAAC,CAAE,KAA4B,CAAC,MAAM;gBACtC,CAAC,CAAC,SAAS,CAAC;YAChB,MAAM,eAAe,GAAoB;gBACvC,OAAO,EAAE,eAAe,CAAC,KAAK,CAAC;gBAC/B,MAAM;aACP,CAAC;YACF,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;YACzE,OAAO;QACT,CAAC;IACH,CAAC;IAEO,yBAAyB,CAC/B,MAAoB;QAEpB,MAAM,MAAM,GACV,MAAM,CAAC,EAAE;YACT,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,qBAAqB,CAAC;QAClD,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC;QAE5D,MAAM,eAAe,GAAwB;YAC3C,MAAM;YACN,IAAI;YACJ,IAAI;YACJ,iBAAiB,EAAE,KAAK;SACzB,CAAC;QAEF,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5C,uEAAuE;QACvE,OAAO,EAAE,IAAI,EAAE,eAAe,CAAC,eAAe,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;IAC3E,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;CACF"}
|
|
@@ -0,0 +1,236 @@
|
|
|
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 { Turn, GeminiEventType, } from './turn.js';
|
|
8
|
+
import { reportError } from '../utils/errorReporting.js';
|
|
9
|
+
const mockSendMessageStream = vi.fn();
|
|
10
|
+
const mockGetHistory = vi.fn();
|
|
11
|
+
vi.mock('@google/genai', async (importOriginal) => {
|
|
12
|
+
const actual = await importOriginal();
|
|
13
|
+
const MockChat = vi.fn().mockImplementation(() => ({
|
|
14
|
+
sendMessageStream: mockSendMessageStream,
|
|
15
|
+
getHistory: mockGetHistory,
|
|
16
|
+
}));
|
|
17
|
+
return {
|
|
18
|
+
...actual,
|
|
19
|
+
Chat: MockChat,
|
|
20
|
+
};
|
|
21
|
+
});
|
|
22
|
+
vi.mock('../utils/errorReporting', () => ({
|
|
23
|
+
reportError: vi.fn(),
|
|
24
|
+
}));
|
|
25
|
+
vi.mock('../utils/generateContentResponseUtilities', () => ({
|
|
26
|
+
getResponseText: (resp) => resp.candidates?.[0]?.content?.parts?.map((part) => part.text).join('') ||
|
|
27
|
+
undefined,
|
|
28
|
+
}));
|
|
29
|
+
describe('Turn', () => {
|
|
30
|
+
let turn;
|
|
31
|
+
let mockChatInstance;
|
|
32
|
+
beforeEach(() => {
|
|
33
|
+
vi.resetAllMocks();
|
|
34
|
+
mockChatInstance = {
|
|
35
|
+
sendMessageStream: mockSendMessageStream,
|
|
36
|
+
getHistory: mockGetHistory,
|
|
37
|
+
};
|
|
38
|
+
turn = new Turn(mockChatInstance);
|
|
39
|
+
mockGetHistory.mockReturnValue([]);
|
|
40
|
+
mockSendMessageStream.mockResolvedValue((async function* () { })());
|
|
41
|
+
});
|
|
42
|
+
afterEach(() => {
|
|
43
|
+
vi.restoreAllMocks();
|
|
44
|
+
});
|
|
45
|
+
describe('constructor', () => {
|
|
46
|
+
it('should initialize pendingToolCalls and debugResponses', () => {
|
|
47
|
+
expect(turn.pendingToolCalls).toEqual([]);
|
|
48
|
+
expect(turn.getDebugResponses()).toEqual([]);
|
|
49
|
+
});
|
|
50
|
+
});
|
|
51
|
+
describe('run', () => {
|
|
52
|
+
it('should yield content events for text parts', async () => {
|
|
53
|
+
const mockResponseStream = (async function* () {
|
|
54
|
+
yield {
|
|
55
|
+
candidates: [{ content: { parts: [{ text: 'Hello' }] } }],
|
|
56
|
+
};
|
|
57
|
+
yield {
|
|
58
|
+
candidates: [{ content: { parts: [{ text: ' world' }] } }],
|
|
59
|
+
};
|
|
60
|
+
})();
|
|
61
|
+
mockSendMessageStream.mockResolvedValue(mockResponseStream);
|
|
62
|
+
const events = [];
|
|
63
|
+
const reqParts = [{ text: 'Hi' }];
|
|
64
|
+
for await (const event of turn.run(reqParts, new AbortController().signal)) {
|
|
65
|
+
events.push(event);
|
|
66
|
+
}
|
|
67
|
+
expect(mockSendMessageStream).toHaveBeenCalledWith({
|
|
68
|
+
message: reqParts,
|
|
69
|
+
config: { abortSignal: expect.any(AbortSignal) },
|
|
70
|
+
});
|
|
71
|
+
expect(events).toEqual([
|
|
72
|
+
{ type: GeminiEventType.Content, value: 'Hello' },
|
|
73
|
+
{ type: GeminiEventType.Content, value: ' world' },
|
|
74
|
+
]);
|
|
75
|
+
expect(turn.getDebugResponses().length).toBe(2);
|
|
76
|
+
});
|
|
77
|
+
it('should yield tool_call_request events for function calls', async () => {
|
|
78
|
+
const mockResponseStream = (async function* () {
|
|
79
|
+
yield {
|
|
80
|
+
functionCalls: [
|
|
81
|
+
{
|
|
82
|
+
id: 'fc1',
|
|
83
|
+
name: 'tool1',
|
|
84
|
+
args: { arg1: 'val1' },
|
|
85
|
+
isClientInitiated: false,
|
|
86
|
+
},
|
|
87
|
+
{ name: 'tool2', args: { arg2: 'val2' }, isClientInitiated: false }, // No ID
|
|
88
|
+
],
|
|
89
|
+
};
|
|
90
|
+
})();
|
|
91
|
+
mockSendMessageStream.mockResolvedValue(mockResponseStream);
|
|
92
|
+
const events = [];
|
|
93
|
+
const reqParts = [{ text: 'Use tools' }];
|
|
94
|
+
for await (const event of turn.run(reqParts, new AbortController().signal)) {
|
|
95
|
+
events.push(event);
|
|
96
|
+
}
|
|
97
|
+
expect(events.length).toBe(2);
|
|
98
|
+
const event1 = events[0];
|
|
99
|
+
expect(event1.type).toBe(GeminiEventType.ToolCallRequest);
|
|
100
|
+
expect(event1.value).toEqual(expect.objectContaining({
|
|
101
|
+
callId: 'fc1',
|
|
102
|
+
name: 'tool1',
|
|
103
|
+
args: { arg1: 'val1' },
|
|
104
|
+
isClientInitiated: false,
|
|
105
|
+
}));
|
|
106
|
+
expect(turn.pendingToolCalls[0]).toEqual(event1.value);
|
|
107
|
+
const event2 = events[1];
|
|
108
|
+
expect(event2.type).toBe(GeminiEventType.ToolCallRequest);
|
|
109
|
+
expect(event2.value).toEqual(expect.objectContaining({
|
|
110
|
+
name: 'tool2',
|
|
111
|
+
args: { arg2: 'val2' },
|
|
112
|
+
isClientInitiated: false,
|
|
113
|
+
}));
|
|
114
|
+
expect(event2.value.callId).toEqual(expect.stringMatching(/^tool2-\d{13}-\w{10,}$/));
|
|
115
|
+
expect(turn.pendingToolCalls[1]).toEqual(event2.value);
|
|
116
|
+
expect(turn.getDebugResponses().length).toBe(1);
|
|
117
|
+
});
|
|
118
|
+
it('should yield UserCancelled event if signal is aborted', async () => {
|
|
119
|
+
const abortController = new AbortController();
|
|
120
|
+
const mockResponseStream = (async function* () {
|
|
121
|
+
yield {
|
|
122
|
+
candidates: [{ content: { parts: [{ text: 'First part' }] } }],
|
|
123
|
+
};
|
|
124
|
+
abortController.abort();
|
|
125
|
+
yield {
|
|
126
|
+
candidates: [
|
|
127
|
+
{
|
|
128
|
+
content: {
|
|
129
|
+
parts: [{ text: 'Second part - should not be processed' }],
|
|
130
|
+
},
|
|
131
|
+
},
|
|
132
|
+
],
|
|
133
|
+
};
|
|
134
|
+
})();
|
|
135
|
+
mockSendMessageStream.mockResolvedValue(mockResponseStream);
|
|
136
|
+
const events = [];
|
|
137
|
+
const reqParts = [{ text: 'Test abort' }];
|
|
138
|
+
for await (const event of turn.run(reqParts, abortController.signal)) {
|
|
139
|
+
events.push(event);
|
|
140
|
+
}
|
|
141
|
+
expect(events).toEqual([
|
|
142
|
+
{ type: GeminiEventType.Content, value: 'First part' },
|
|
143
|
+
{ type: GeminiEventType.UserCancelled },
|
|
144
|
+
]);
|
|
145
|
+
expect(turn.getDebugResponses().length).toBe(1);
|
|
146
|
+
});
|
|
147
|
+
it('should yield Error event and report if sendMessageStream throws', async () => {
|
|
148
|
+
const error = new Error('API Error');
|
|
149
|
+
mockSendMessageStream.mockRejectedValue(error);
|
|
150
|
+
const reqParts = [{ text: 'Trigger error' }];
|
|
151
|
+
const historyContent = [
|
|
152
|
+
{ role: 'model', parts: [{ text: 'Previous history' }] },
|
|
153
|
+
];
|
|
154
|
+
mockGetHistory.mockReturnValue(historyContent);
|
|
155
|
+
const events = [];
|
|
156
|
+
for await (const event of turn.run(reqParts, new AbortController().signal)) {
|
|
157
|
+
events.push(event);
|
|
158
|
+
}
|
|
159
|
+
expect(events.length).toBe(1);
|
|
160
|
+
const errorEvent = events[0];
|
|
161
|
+
expect(errorEvent.type).toBe(GeminiEventType.Error);
|
|
162
|
+
expect(errorEvent.value).toEqual({
|
|
163
|
+
error: { message: 'API Error', status: undefined },
|
|
164
|
+
});
|
|
165
|
+
expect(turn.getDebugResponses().length).toBe(0);
|
|
166
|
+
expect(reportError).toHaveBeenCalledWith(error, 'Error when talking to Gemini API', [...historyContent, reqParts], 'Turn.run-sendMessageStream');
|
|
167
|
+
});
|
|
168
|
+
it('should handle function calls with undefined name or args', async () => {
|
|
169
|
+
const mockResponseStream = (async function* () {
|
|
170
|
+
yield {
|
|
171
|
+
functionCalls: [
|
|
172
|
+
{ id: 'fc1', name: undefined, args: { arg1: 'val1' } },
|
|
173
|
+
{ id: 'fc2', name: 'tool2', args: undefined },
|
|
174
|
+
{ id: 'fc3', name: undefined, args: undefined },
|
|
175
|
+
],
|
|
176
|
+
};
|
|
177
|
+
})();
|
|
178
|
+
mockSendMessageStream.mockResolvedValue(mockResponseStream);
|
|
179
|
+
const events = [];
|
|
180
|
+
const reqParts = [{ text: 'Test undefined tool parts' }];
|
|
181
|
+
for await (const event of turn.run(reqParts, new AbortController().signal)) {
|
|
182
|
+
events.push(event);
|
|
183
|
+
}
|
|
184
|
+
expect(events.length).toBe(3);
|
|
185
|
+
const event1 = events[0];
|
|
186
|
+
expect(event1.type).toBe(GeminiEventType.ToolCallRequest);
|
|
187
|
+
expect(event1.value).toEqual(expect.objectContaining({
|
|
188
|
+
callId: 'fc1',
|
|
189
|
+
name: 'undefined_tool_name',
|
|
190
|
+
args: { arg1: 'val1' },
|
|
191
|
+
isClientInitiated: false,
|
|
192
|
+
}));
|
|
193
|
+
expect(turn.pendingToolCalls[0]).toEqual(event1.value);
|
|
194
|
+
const event2 = events[1];
|
|
195
|
+
expect(event2.type).toBe(GeminiEventType.ToolCallRequest);
|
|
196
|
+
expect(event2.value).toEqual(expect.objectContaining({
|
|
197
|
+
callId: 'fc2',
|
|
198
|
+
name: 'tool2',
|
|
199
|
+
args: {},
|
|
200
|
+
isClientInitiated: false,
|
|
201
|
+
}));
|
|
202
|
+
expect(turn.pendingToolCalls[1]).toEqual(event2.value);
|
|
203
|
+
const event3 = events[2];
|
|
204
|
+
expect(event3.type).toBe(GeminiEventType.ToolCallRequest);
|
|
205
|
+
expect(event3.value).toEqual(expect.objectContaining({
|
|
206
|
+
callId: 'fc3',
|
|
207
|
+
name: 'undefined_tool_name',
|
|
208
|
+
args: {},
|
|
209
|
+
isClientInitiated: false,
|
|
210
|
+
}));
|
|
211
|
+
expect(turn.pendingToolCalls[2]).toEqual(event3.value);
|
|
212
|
+
expect(turn.getDebugResponses().length).toBe(1);
|
|
213
|
+
});
|
|
214
|
+
});
|
|
215
|
+
describe('getDebugResponses', () => {
|
|
216
|
+
it('should return collected debug responses', async () => {
|
|
217
|
+
const resp1 = {
|
|
218
|
+
candidates: [{ content: { parts: [{ text: 'Debug 1' }] } }],
|
|
219
|
+
};
|
|
220
|
+
const resp2 = {
|
|
221
|
+
functionCalls: [{ name: 'debugTool' }],
|
|
222
|
+
};
|
|
223
|
+
const mockResponseStream = (async function* () {
|
|
224
|
+
yield resp1;
|
|
225
|
+
yield resp2;
|
|
226
|
+
})();
|
|
227
|
+
mockSendMessageStream.mockResolvedValue(mockResponseStream);
|
|
228
|
+
const reqParts = [{ text: 'Hi' }];
|
|
229
|
+
for await (const _ of turn.run(reqParts, new AbortController().signal)) {
|
|
230
|
+
// consume stream
|
|
231
|
+
}
|
|
232
|
+
expect(turn.getDebugResponses()).toEqual([resp1, resp2]);
|
|
233
|
+
});
|
|
234
|
+
});
|
|
235
|
+
});
|
|
236
|
+
//# sourceMappingURL=turn.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"turn.test.js","sourceRoot":"","sources":["../../../src/core/turn.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,IAAI,EACJ,eAAe,GAGhB,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAGzD,MAAM,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACtC,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE/B,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAChD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAkC,CAAC;IACtE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACjD,iBAAiB,EAAE,qBAAqB;QACxC,UAAU,EAAE,cAAc;KAC3B,CAAC,CAAC,CAAC;IACJ,OAAO;QACL,GAAG,MAAM;QACT,IAAI,EAAE,QAAQ;KACf,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,eAAe,EAAE,CAAC,IAA6B,EAAE,EAAE,CACjD,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QACvE,SAAS;CACZ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,IAAI,IAAU,CAAC;IAMf,IAAI,gBAAoC,CAAC;IAEzC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,gBAAgB,GAAG;YACjB,iBAAiB,EAAE,qBAAqB;YACxC,UAAU,EAAE,cAAc;SAC3B,CAAC;QACF,IAAI,GAAG,IAAI,IAAI,CAAC,gBAAyC,CAAC,CAAC;QAC3D,cAAc,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QACnC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC,KAAK,SAAS,CAAC,MAAK,CAAC,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,kBAAkB,GAAG,CAAC,KAAK,SAAS,CAAC;gBACzC,MAAM;oBACJ,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;iBACpB,CAAC;gBACxC,MAAM;oBACJ,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;iBACrB,CAAC;YAC1C,CAAC,CAAC,EAAE,CAAC;YACL,qBAAqB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,CAChC,QAAQ,EACR,IAAI,eAAe,EAAE,CAAC,MAAM,CAC7B,EAAE,CAAC;gBACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;aACjD,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,EAAE,IAAI,EAAE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE;gBACjD,EAAE,IAAI,EAAE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,kBAAkB,GAAG,CAAC,KAAK,SAAS,CAAC;gBACzC,MAAM;oBACJ,aAAa,EAAE;wBACb;4BACE,EAAE,EAAE,KAAK;4BACT,IAAI,EAAE,OAAO;4BACb,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;4BACtB,iBAAiB,EAAE,KAAK;yBACzB;wBACD,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,QAAQ;qBAC9E;iBACoC,CAAC;YAC1C,CAAC,CAAC,EAAE,CAAC;YACL,qBAAqB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YACjD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,CAChC,QAAQ,EACR,IAAI,eAAe,EAAE,CAAC,MAAM,CAC7B,EAAE,CAAC;gBACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAqC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBACtB,iBAAiB,EAAE,KAAK;aACzB,CAAC,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAqC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBACtB,iBAAiB,EAAE,KAAK;aACzB,CAAC,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CACjC,MAAM,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAChD,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,MAAM,kBAAkB,GAAG,CAAC,KAAK,SAAS,CAAC;gBACzC,MAAM;oBACJ,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;iBACzB,CAAC;gBACxC,eAAe,CAAC,KAAK,EAAE,CAAC;gBACxB,MAAM;oBACJ,UAAU,EAAE;wBACV;4BACE,OAAO,EAAE;gCACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,uCAAuC,EAAE,CAAC;6BAC3D;yBACF;qBACF;iBACoC,CAAC;YAC1C,CAAC,CAAC,EAAE,CAAC;YACL,qBAAqB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YAClD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;gBACrE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,EAAE,IAAI,EAAE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE;gBACtD,EAAE,IAAI,EAAE,eAAe,CAAC,aAAa,EAAE;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACrC,qBAAqB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;YACrD,MAAM,cAAc,GAAc;gBAChC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,EAAE;aACzD,CAAC;YACF,cAAc,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAE/C,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,CAChC,QAAQ,EACR,IAAI,eAAe,EAAE,CAAC,MAAM,CAC7B,EAAE,CAAC;gBACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAA2B,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACpD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBAC/B,KAAK,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,KAAK,EACL,kCAAkC,EAClC,CAAC,GAAG,cAAc,EAAE,QAAQ,CAAC,EAC7B,4BAA4B,CAC7B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,kBAAkB,GAAG,CAAC,KAAK,SAAS,CAAC;gBACzC,MAAM;oBACJ,aAAa,EAAE;wBACb,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;wBACtD,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE;wBAC7C,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE;qBAChD;iBACoC,CAAC;YAC1C,CAAC,CAAC,EAAE,CAAC;YACL,qBAAqB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,2BAA2B,EAAE,CAAC,CAAC;YACjE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,CAChC,QAAQ,EACR,IAAI,eAAe,EAAE,CAAC,MAAM,CAC7B,EAAE,CAAC;gBACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAqC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,qBAAqB;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBACtB,iBAAiB,EAAE,KAAK;aACzB,CAAC,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAqC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;aACzB,CAAC,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAqC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,qBAAqB;gBAC3B,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;aACzB,CAAC,CACH,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,GAAG;gBACZ,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;aACtB,CAAC;YACxC,MAAM,KAAK,GAAG;gBACZ,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;aACD,CAAC;YACxC,MAAM,kBAAkB,GAAG,CAAC,KAAK,SAAS,CAAC;gBACzC,MAAM,KAAK,CAAC;gBACZ,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,EAAE,CAAC;YACL,qBAAqB,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAW,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1C,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvE,iBAAiB;YACnB,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
export * from './config/config.js';
|
|
7
|
+
export * from './core/client.js';
|
|
8
|
+
export * from './core/contentGenerator.js';
|
|
9
|
+
export * from './core/geminiChat.js';
|
|
10
|
+
export * from './core/logger.js';
|
|
11
|
+
export * from './core/prompts.js';
|
|
12
|
+
export * from './core/tokenLimits.js';
|
|
13
|
+
export * from './core/turn.js';
|
|
14
|
+
export * from './core/geminiRequest.js';
|
|
15
|
+
export * from './core/coreToolScheduler.js';
|
|
16
|
+
export * from './core/nonInteractiveToolExecutor.js';
|
|
17
|
+
export * from './code_assist/codeAssist.js';
|
|
18
|
+
export * from './code_assist/oauth2.js';
|
|
19
|
+
export * from './code_assist/server.js';
|
|
20
|
+
export * from './code_assist/types.js';
|
|
21
|
+
export * from './utils/paths.js';
|
|
22
|
+
export * from './utils/schemaValidator.js';
|
|
23
|
+
export * from './utils/errors.js';
|
|
24
|
+
export * from './utils/getFolderStructure.js';
|
|
25
|
+
export * from './utils/memoryDiscovery.js';
|
|
26
|
+
export * from './utils/gitIgnoreParser.js';
|
|
27
|
+
export * from './utils/editor.js';
|
|
28
|
+
export * from './services/fileDiscoveryService.js';
|
|
29
|
+
export * from './services/gitService.js';
|
|
30
|
+
export * from './tools/tools.js';
|
|
31
|
+
export * from './tools/tool-registry.js';
|
|
32
|
+
export * from './tools/read-file.js';
|
|
33
|
+
export * from './tools/ls.js';
|
|
34
|
+
export * from './tools/grep.js';
|
|
35
|
+
export * from './tools/glob.js';
|
|
36
|
+
export * from './tools/edit.js';
|
|
37
|
+
export * from './tools/write-file.js';
|
|
38
|
+
export * from './tools/web-fetch.js';
|
|
39
|
+
export * from './tools/memoryTool.js';
|
|
40
|
+
export * from './tools/shell.js';
|
|
41
|
+
export * from './tools/web-search.js';
|
|
42
|
+
export * from './tools/read-many-files.js';
|
|
43
|
+
export * from './tools/mcp-client.js';
|
|
44
|
+
export * from './tools/mcp-tool.js';
|
|
45
|
+
export * from './telemetry/index.js';
|
|
46
|
+
export { sessionId } from './utils/session.js';
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
// Export config
|
|
7
|
+
export * from './config/config.js';
|
|
8
|
+
// Export Core Logic
|
|
9
|
+
export * from './core/client.js';
|
|
10
|
+
export * from './core/contentGenerator.js';
|
|
11
|
+
export * from './core/geminiChat.js';
|
|
12
|
+
export * from './core/logger.js';
|
|
13
|
+
export * from './core/prompts.js';
|
|
14
|
+
export * from './core/tokenLimits.js';
|
|
15
|
+
export * from './core/turn.js';
|
|
16
|
+
export * from './core/geminiRequest.js';
|
|
17
|
+
export * from './core/coreToolScheduler.js';
|
|
18
|
+
export * from './core/nonInteractiveToolExecutor.js';
|
|
19
|
+
export * from './code_assist/codeAssist.js';
|
|
20
|
+
export * from './code_assist/oauth2.js';
|
|
21
|
+
export * from './code_assist/server.js';
|
|
22
|
+
export * from './code_assist/types.js';
|
|
23
|
+
// Export utilities
|
|
24
|
+
export * from './utils/paths.js';
|
|
25
|
+
export * from './utils/schemaValidator.js';
|
|
26
|
+
export * from './utils/errors.js';
|
|
27
|
+
export * from './utils/getFolderStructure.js';
|
|
28
|
+
export * from './utils/memoryDiscovery.js';
|
|
29
|
+
export * from './utils/gitIgnoreParser.js';
|
|
30
|
+
export * from './utils/editor.js';
|
|
31
|
+
// Export services
|
|
32
|
+
export * from './services/fileDiscoveryService.js';
|
|
33
|
+
export * from './services/gitService.js';
|
|
34
|
+
// Export base tool definitions
|
|
35
|
+
export * from './tools/tools.js';
|
|
36
|
+
export * from './tools/tool-registry.js';
|
|
37
|
+
// Export specific tool logic
|
|
38
|
+
export * from './tools/read-file.js';
|
|
39
|
+
export * from './tools/ls.js';
|
|
40
|
+
export * from './tools/grep.js';
|
|
41
|
+
export * from './tools/glob.js';
|
|
42
|
+
export * from './tools/edit.js';
|
|
43
|
+
export * from './tools/write-file.js';
|
|
44
|
+
export * from './tools/web-fetch.js';
|
|
45
|
+
export * from './tools/memoryTool.js';
|
|
46
|
+
export * from './tools/shell.js';
|
|
47
|
+
export * from './tools/web-search.js';
|
|
48
|
+
export * from './tools/read-many-files.js';
|
|
49
|
+
export * from './tools/mcp-client.js';
|
|
50
|
+
export * from './tools/mcp-tool.js';
|
|
51
|
+
// Export telemetry functions
|
|
52
|
+
export * from './telemetry/index.js';
|
|
53
|
+
export { sessionId } from './utils/session.js';
|
|
54
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,gBAAgB;AAChB,cAAc,oBAAoB,CAAC;AAEnC,oBAAoB;AACpB,cAAc,kBAAkB,CAAC;AACjC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,sBAAsB,CAAC;AACrC,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,sCAAsC,CAAC;AAErD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AAEvC,mBAAmB;AACnB,cAAc,kBAAkB,CAAC;AACjC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAElC,kBAAkB;AAClB,cAAc,oCAAoC,CAAC;AACnD,cAAc,0BAA0B,CAAC;AAEzC,+BAA+B;AAC/B,cAAc,kBAAkB,CAAC;AACjC,cAAc,0BAA0B,CAAC;AAEzC,6BAA6B;AAC7B,cAAc,sBAAsB,CAAC;AACrC,cAAc,eAAe,CAAC;AAC9B,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,uBAAuB,CAAC;AACtC,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,kBAAkB,CAAC;AACjC,cAAc,uBAAuB,CAAC;AACtC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,uBAAuB,CAAC;AACtC,cAAc,qBAAqB,CAAC;AAEpC,6BAA6B;AAC7B,cAAc,sBAAsB,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect } from 'vitest';
|
|
7
|
+
describe('placeholder tests', () => {
|
|
8
|
+
it('should pass', () => {
|
|
9
|
+
expect(true).toBe(true);
|
|
10
|
+
});
|
|
11
|
+
});
|
|
12
|
+
//# sourceMappingURL=index.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/index.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;QACrB,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
export interface FilterFilesOptions {
|
|
7
|
+
respectGitIgnore?: boolean;
|
|
8
|
+
respectGeminiIgnore?: boolean;
|
|
9
|
+
}
|
|
10
|
+
export declare class FileDiscoveryService {
|
|
11
|
+
private gitIgnoreFilter;
|
|
12
|
+
private geminiIgnoreFilter;
|
|
13
|
+
private projectRoot;
|
|
14
|
+
constructor(projectRoot: string);
|
|
15
|
+
/**
|
|
16
|
+
* Filters a list of file paths based on git ignore rules
|
|
17
|
+
*/
|
|
18
|
+
filterFiles(filePaths: string[], options?: FilterFilesOptions): string[];
|
|
19
|
+
/**
|
|
20
|
+
* Checks if a single file should be git-ignored
|
|
21
|
+
*/
|
|
22
|
+
shouldGitIgnoreFile(filePath: string): boolean;
|
|
23
|
+
/**
|
|
24
|
+
* Checks if a single file should be gemini-ignored
|
|
25
|
+
*/
|
|
26
|
+
shouldGeminiIgnoreFile(filePath: string): boolean;
|
|
27
|
+
/**
|
|
28
|
+
* Unified method to check if a file should be ignored based on filtering options
|
|
29
|
+
*/
|
|
30
|
+
shouldIgnoreFile(filePath: string, options?: FilterFilesOptions): boolean;
|
|
31
|
+
/**
|
|
32
|
+
* Returns loaded patterns from .geminiignore
|
|
33
|
+
*/
|
|
34
|
+
getGeminiIgnorePatterns(): string[];
|
|
35
|
+
}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { GitIgnoreParser } from '../utils/gitIgnoreParser.js';
|
|
7
|
+
import { isGitRepository } from '../utils/gitUtils.js';
|
|
8
|
+
import * as path from 'path';
|
|
9
|
+
const GEMINI_IGNORE_FILE_NAME = '.geminiignore';
|
|
10
|
+
export class FileDiscoveryService {
|
|
11
|
+
gitIgnoreFilter = null;
|
|
12
|
+
geminiIgnoreFilter = null;
|
|
13
|
+
projectRoot;
|
|
14
|
+
constructor(projectRoot) {
|
|
15
|
+
this.projectRoot = path.resolve(projectRoot);
|
|
16
|
+
if (isGitRepository(this.projectRoot)) {
|
|
17
|
+
const parser = new GitIgnoreParser(this.projectRoot);
|
|
18
|
+
try {
|
|
19
|
+
parser.loadGitRepoPatterns();
|
|
20
|
+
}
|
|
21
|
+
catch (_error) {
|
|
22
|
+
// ignore file not found
|
|
23
|
+
}
|
|
24
|
+
this.gitIgnoreFilter = parser;
|
|
25
|
+
}
|
|
26
|
+
const gParser = new GitIgnoreParser(this.projectRoot);
|
|
27
|
+
try {
|
|
28
|
+
gParser.loadPatterns(GEMINI_IGNORE_FILE_NAME);
|
|
29
|
+
}
|
|
30
|
+
catch (_error) {
|
|
31
|
+
// ignore file not found
|
|
32
|
+
}
|
|
33
|
+
this.geminiIgnoreFilter = gParser;
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* Filters a list of file paths based on git ignore rules
|
|
37
|
+
*/
|
|
38
|
+
filterFiles(filePaths, options = {
|
|
39
|
+
respectGitIgnore: true,
|
|
40
|
+
respectGeminiIgnore: true,
|
|
41
|
+
}) {
|
|
42
|
+
return filePaths.filter((filePath) => {
|
|
43
|
+
if (options.respectGitIgnore && this.shouldGitIgnoreFile(filePath)) {
|
|
44
|
+
return false;
|
|
45
|
+
}
|
|
46
|
+
if (options.respectGeminiIgnore &&
|
|
47
|
+
this.shouldGeminiIgnoreFile(filePath)) {
|
|
48
|
+
return false;
|
|
49
|
+
}
|
|
50
|
+
return true;
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* Checks if a single file should be git-ignored
|
|
55
|
+
*/
|
|
56
|
+
shouldGitIgnoreFile(filePath) {
|
|
57
|
+
if (this.gitIgnoreFilter) {
|
|
58
|
+
return this.gitIgnoreFilter.isIgnored(filePath);
|
|
59
|
+
}
|
|
60
|
+
return false;
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Checks if a single file should be gemini-ignored
|
|
64
|
+
*/
|
|
65
|
+
shouldGeminiIgnoreFile(filePath) {
|
|
66
|
+
if (this.geminiIgnoreFilter) {
|
|
67
|
+
return this.geminiIgnoreFilter.isIgnored(filePath);
|
|
68
|
+
}
|
|
69
|
+
return false;
|
|
70
|
+
}
|
|
71
|
+
/**
|
|
72
|
+
* Unified method to check if a file should be ignored based on filtering options
|
|
73
|
+
*/
|
|
74
|
+
shouldIgnoreFile(filePath, options = {}) {
|
|
75
|
+
const { respectGitIgnore = true, respectGeminiIgnore = true } = options;
|
|
76
|
+
if (respectGitIgnore && this.shouldGitIgnoreFile(filePath)) {
|
|
77
|
+
return true;
|
|
78
|
+
}
|
|
79
|
+
if (respectGeminiIgnore && this.shouldGeminiIgnoreFile(filePath)) {
|
|
80
|
+
return true;
|
|
81
|
+
}
|
|
82
|
+
return false;
|
|
83
|
+
}
|
|
84
|
+
/**
|
|
85
|
+
* Returns loaded patterns from .geminiignore
|
|
86
|
+
*/
|
|
87
|
+
getGeminiIgnorePatterns() {
|
|
88
|
+
return this.geminiIgnoreFilter?.getPatterns() ?? [];
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=fileDiscoveryService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fileDiscoveryService.js","sourceRoot":"","sources":["../../../src/services/fileDiscoveryService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,eAAe,EAAmB,MAAM,6BAA6B,CAAC;AAC/E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAE7B,MAAM,uBAAuB,GAAG,eAAe,CAAC;AAOhD,MAAM,OAAO,oBAAoB;IACvB,eAAe,GAA2B,IAAI,CAAC;IAC/C,kBAAkB,GAA2B,IAAI,CAAC;IAClD,WAAW,CAAS;IAE5B,YAAY,WAAmB;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC7C,IAAI,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACrD,IAAI,CAAC;gBACH,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC/B,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC;gBAChB,wBAAwB;YAC1B,CAAC;YACD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC;QAChC,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACtD,IAAI,CAAC;YACH,OAAO,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YAChB,wBAAwB;QAC1B,CAAC;QACD,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,WAAW,CACT,SAAmB,EACnB,UAA8B;QAC5B,gBAAgB,EAAE,IAAI;QACtB,mBAAmB,EAAE,IAAI;KAC1B;QAED,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE;YACnC,IAAI,OAAO,CAAC,gBAAgB,IAAI,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnE,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IACE,OAAO,CAAC,mBAAmB;gBAC3B,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,EACrC,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,QAAgB;QAClC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,QAAgB;QACrC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,gBAAgB,CACd,QAAgB,EAChB,UAA8B,EAAE;QAEhC,MAAM,EAAE,gBAAgB,GAAG,IAAI,EAAE,mBAAmB,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QAExE,IAAI,gBAAgB,IAAI,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3D,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,mBAAmB,IAAI,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,uBAAuB;QACrB,OAAO,IAAI,CAAC,kBAAkB,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IACtD,CAAC;CACF"}
|