@google/gemini-cli-core 0.15.0-preview.3 → 0.16.0-nightly.20251113.ad1f0d99
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/google-gemini-cli-core-0.16.0-nightly.20251112.c961f274.tgz +0 -0
- package/dist/src/agents/codebase-investigator.test.d.ts +6 -0
- package/dist/src/agents/codebase-investigator.test.js +35 -0
- package/dist/src/agents/codebase-investigator.test.js.map +1 -0
- package/dist/src/agents/executor.test.js +181 -1
- package/dist/src/agents/executor.test.js.map +1 -1
- package/dist/src/code_assist/codeAssist.test.d.ts +6 -0
- package/dist/src/code_assist/codeAssist.test.js +99 -0
- package/dist/src/code_assist/codeAssist.test.js.map +1 -0
- package/dist/src/code_assist/experiments/client_metadata.js +2 -1
- package/dist/src/code_assist/experiments/client_metadata.js.map +1 -1
- package/dist/src/code_assist/experiments/client_metadata.test.d.ts +6 -0
- package/dist/src/code_assist/experiments/client_metadata.test.js +99 -0
- package/dist/src/code_assist/experiments/client_metadata.test.js.map +1 -0
- package/dist/src/code_assist/experiments/experiments.test.d.ts +6 -0
- package/dist/src/code_assist/experiments/experiments.test.js +92 -0
- package/dist/src/code_assist/experiments/experiments.test.js.map +1 -0
- package/dist/src/code_assist/oauth-credential-storage.test.js +49 -0
- package/dist/src/code_assist/oauth-credential-storage.test.js.map +1 -1
- package/dist/src/code_assist/server.js +5 -8
- package/dist/src/code_assist/server.js.map +1 -1
- package/dist/src/code_assist/server.test.js +109 -28
- package/dist/src/code_assist/server.test.js.map +1 -1
- package/dist/src/config/defaultModelConfigs.js +6 -0
- package/dist/src/config/defaultModelConfigs.js.map +1 -1
- package/dist/src/confirmation-bus/message-bus.d.ts +1 -1
- package/dist/src/confirmation-bus/message-bus.js +2 -2
- package/dist/src/confirmation-bus/message-bus.js.map +1 -1
- package/dist/src/confirmation-bus/message-bus.test.js +30 -24
- package/dist/src/confirmation-bus/message-bus.test.js.map +1 -1
- package/dist/src/core/loggingContentGenerator.test.d.ts +6 -0
- package/dist/src/core/loggingContentGenerator.test.js +180 -0
- package/dist/src/core/loggingContentGenerator.test.js.map +1 -0
- package/dist/src/core/tokenLimits.test.d.ts +6 -0
- package/dist/src/core/tokenLimits.test.js +26 -0
- package/dist/src/core/tokenLimits.test.js.map +1 -0
- package/dist/src/generated/git-commit.d.ts +2 -2
- package/dist/src/generated/git-commit.js +2 -2
- package/dist/src/generated/git-commit.js.map +1 -1
- package/dist/src/hooks/hookAggregator.d.ts +68 -0
- package/dist/src/hooks/hookAggregator.js +262 -0
- package/dist/src/hooks/hookAggregator.js.map +1 -0
- package/dist/src/hooks/hookAggregator.test.d.ts +6 -0
- package/dist/src/hooks/hookAggregator.test.js +387 -0
- package/dist/src/hooks/hookAggregator.test.js.map +1 -0
- package/dist/src/hooks/types.js +1 -1
- package/dist/src/hooks/types.js.map +1 -1
- package/dist/src/hooks/types.test.js +280 -2
- package/dist/src/hooks/types.test.js.map +1 -1
- package/dist/src/ide/ide-client.test.js +159 -0
- package/dist/src/ide/ide-client.test.js.map +1 -1
- package/dist/src/mcp/oauth-provider.test.js +177 -0
- package/dist/src/mcp/oauth-provider.test.js.map +1 -1
- package/dist/src/policy/config.js +3 -1
- package/dist/src/policy/config.js.map +1 -1
- package/dist/src/policy/config.test.js +118 -1
- package/dist/src/policy/config.test.js.map +1 -1
- package/dist/src/policy/policies/write.toml +10 -0
- package/dist/src/policy/policy-engine.d.ts +12 -3
- package/dist/src/policy/policy-engine.js +61 -7
- package/dist/src/policy/policy-engine.js.map +1 -1
- package/dist/src/policy/policy-engine.test.js +422 -86
- package/dist/src/policy/policy-engine.test.js.map +1 -1
- package/dist/src/policy/toml-loader.d.ts +2 -1
- package/dist/src/policy/toml-loader.js +103 -6
- package/dist/src/policy/toml-loader.js.map +1 -1
- package/dist/src/policy/toml-loader.test.js +32 -88
- package/dist/src/policy/toml-loader.test.js.map +1 -1
- package/dist/src/policy/types.d.ts +65 -0
- package/dist/src/policy/types.js +4 -0
- package/dist/src/policy/types.js.map +1 -1
- package/dist/src/prompts/mcp-prompts.test.d.ts +6 -0
- package/dist/src/prompts/mcp-prompts.test.js +40 -0
- package/dist/src/prompts/mcp-prompts.test.js.map +1 -0
- package/dist/src/prompts/prompt-registry.test.d.ts +6 -0
- package/dist/src/prompts/prompt-registry.test.js +111 -0
- package/dist/src/prompts/prompt-registry.test.js.map +1 -0
- package/dist/src/safety/built-in.d.ts +21 -0
- package/dist/src/safety/built-in.js +106 -0
- package/dist/src/safety/built-in.js.map +1 -0
- package/dist/src/safety/built-in.test.d.ts +6 -0
- package/dist/src/safety/built-in.test.js +199 -0
- package/dist/src/safety/built-in.test.js.map +1 -0
- package/dist/src/safety/checker-runner.d.ts +48 -0
- package/dist/src/safety/checker-runner.js +208 -0
- package/dist/src/safety/checker-runner.js.map +1 -0
- package/dist/src/safety/checker-runner.test.d.ts +6 -0
- package/dist/src/safety/checker-runner.test.js +238 -0
- package/dist/src/safety/checker-runner.test.js.map +1 -0
- package/dist/src/safety/context-builder.d.ts +23 -0
- package/dist/src/safety/context-builder.js +47 -0
- package/dist/src/safety/context-builder.js.map +1 -0
- package/dist/src/safety/context-builder.test.d.ts +6 -0
- package/dist/src/safety/context-builder.test.js +49 -0
- package/dist/src/safety/context-builder.test.js.map +1 -0
- package/dist/src/safety/protocol.d.ts +88 -0
- package/dist/src/safety/protocol.js +15 -0
- package/dist/src/safety/protocol.js.map +1 -0
- package/dist/src/safety/registry.d.ts +26 -0
- package/dist/src/safety/registry.js +65 -0
- package/dist/src/safety/registry.js.map +1 -0
- package/dist/src/safety/registry.test.d.ts +6 -0
- package/dist/src/safety/registry.test.js +31 -0
- package/dist/src/safety/registry.test.js.map +1 -0
- package/dist/src/services/loopDetectionService.d.ts +3 -0
- package/dist/src/services/loopDetectionService.js +81 -41
- package/dist/src/services/loopDetectionService.js.map +1 -1
- package/dist/src/services/loopDetectionService.test.js +96 -4
- package/dist/src/services/loopDetectionService.test.js.map +1 -1
- package/dist/src/services/test-data/resolved-aliases.golden.json +7 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts +4 -2
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js +29 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js.map +1 -1
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.test.js +31 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.test.js.map +1 -1
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.d.ts +5 -1
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js +12 -1
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js.map +1 -1
- package/dist/src/telemetry/loggers.d.ts +2 -1
- package/dist/src/telemetry/loggers.js +11 -0
- package/dist/src/telemetry/loggers.js.map +1 -1
- package/dist/src/telemetry/types.d.ts +15 -2
- package/dist/src/telemetry/types.js +42 -3
- package/dist/src/telemetry/types.js.map +1 -1
- package/dist/src/tools/base-tool-invocation.test.js +2 -2
- package/dist/src/tools/base-tool-invocation.test.js.map +1 -1
- package/dist/src/tools/memoryTool.js +1 -1
- package/dist/src/tools/memoryTool.js.map +1 -1
- package/dist/src/tools/memoryTool.test.js +1 -1
- package/dist/src/tools/memoryTool.test.js.map +1 -1
- package/dist/src/tools/ripGrep.d.ts +24 -7
- package/dist/src/tools/ripGrep.js +125 -145
- package/dist/src/tools/ripGrep.js.map +1 -1
- package/dist/src/tools/ripGrep.test.js +144 -20
- package/dist/src/tools/ripGrep.test.js.map +1 -1
- package/dist/src/tools/tools.js +1 -1
- package/dist/src/tools/tools.js.map +1 -1
- package/dist/src/tools/write-todos.js +1 -1
- package/dist/src/tools/write-todos.js.map +1 -1
- package/dist/src/utils/llm-edit-fixer.test.js +8 -1
- package/dist/src/utils/llm-edit-fixer.test.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +1 -1
- package/dist/google-gemini-cli-core-0.15.0-preview.2.tgz +0 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hookAggregator.test.js","sourceRoot":"","sources":["../../../src/hooks/hookAggregator.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAOrD,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAErD,+DAA+D;AAC/D,SAAS,yBAAyB,CAChC,MAAmB,EACnB,OAAO,GAAG,IAAI,EACd,QAAQ,GAAG,GAAG,EACd,KAAa;IAEb,OAAO;QACL,OAAO;QACP,MAAM;QACN,QAAQ;QACR,KAAK;QACL,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ,CAAC,OAAO;YACtB,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,KAAK;SACf;QACD,SAAS,EAAE,aAAa,CAAC,UAAU;KACpC,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,OAAO,GAA0B,EAAE,CAAC;YAE1C,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,UAAU,CACzB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAA0B;gBACrC,yBAAyB,CACvB,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,EAChD,IAAI,EACJ,GAAG,CACJ;gBACD,yBAAyB,CACvB,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,EAChD,IAAI,EACJ,GAAG,CACJ;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,UAAU,CACzB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,IAAI,CACzC,kCAAkC,CACnC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,IAAI,KAAK,CAAC,aAAa,CAAC;oBAC/B,QAAQ,EAAE,EAAE;iBACb;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE;oBAC7B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,UAAU,CACzB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE;oBACvD,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE;oBACvD,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,UAAU,CACzB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,IAAI,CACzC,gCAAgC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE;oBAC7C,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;oBACnC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,QAAQ,EAAE,OAAO;wBACjB,QAAQ,EAAE,KAAK;wBACf,UAAU,EAAE,gBAAgB;qBAC7B;oBACD,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,UAAU,CACzB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,KAAK;gCACX,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;6BACzC;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,MAAM;gCACZ,oBAAoB,EAAE,EAAE;6BACzB;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,mBAAmB,CAClC,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,UAAU,CAAC,WAAwC,CAAC;YACnE,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,EAAE,UAAU,CAAC;YACzD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,MAAM;gCACZ,oBAAoB,EAAE,CAAC,OAAO,CAAC;6BAChC;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,KAAK;gCACX,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;6BACzC;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,mBAAmB,CAClC,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,UAAU,CAAC,WAAwC,CAAC;YACnE,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,EAAE,UAAU,CAAC;YACzD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,OAAO,CAAC;gBAC/C,OAAO;gBACP,OAAO;gBACP,OAAO;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,MAAM;gCACZ,oBAAoB,EAAE,CAAC,OAAO,CAAC;6BAChC;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,mBAAmB;oBAC5C,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,qBAAqB;4BACpC,UAAU,EAAE;gCACV,IAAI,EAAE,MAAM;gCACZ,oBAAoB,EAAE,CAAC,OAAO,CAAC;6BAChC;yBACF;qBAC2B;oBAC9B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,mBAAmB,CAClC,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,UAAU,CAAC,WAAwC,CAAC;YACnE,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,EAAE,UAAU,CAAC;YACzD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;QACrD,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,WAAW;oBACpC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,QAAQ,EAAE,OAAO;wBACjB,kBAAkB,EAAE;4BAClB,aAAa,EAAE,aAAa;4BAC5B,WAAW,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;yBAC3D;qBACF;oBACD,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,WAAW;oBACpC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,QAAQ,EAAE,OAAO;wBACjB,kBAAkB,EAAE;4BAClB,aAAa,EAAE,aAAa;4BAC5B,WAAW,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;yBAC3D;qBACF;oBACD,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,WAAW,CAC1B,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB;YAC3E,MAAM,MAAM,GAAG,UAAU,CAAC,WAAgC,CAAC;YAC3D,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,EAAE,WAAW,CAAC;YAC1D,MAAM,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,mBAAmB;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,OAAO,GAA0B;gBACrC;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,SAAS;oBAClC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,WAAW;4BAC1B,iBAAiB,EAAE,qBAAqB;yBACzC;qBACF;oBACD,QAAQ,EAAE,GAAG;iBACd;gBACD;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,cAAc;wBACvB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,SAAS;oBAClC,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE;wBACN,kBAAkB,EAAE;4BAClB,aAAa,EAAE,WAAW;4BAC1B,iBAAiB,EAAE,qBAAqB;yBACzC;qBACF;oBACD,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,MAAM,UAAU,GAAG,UAAU,CAAC,gBAAgB,CAC5C,OAAO,EACP,aAAa,CAAC,SAAS,CACxB,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CACJ,UAAU,CAAC,WAAW,EAAE,kBAAkB,EAAE,CAAC,mBAAmB,CAAC,CAClE,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/src/hooks/types.js
CHANGED
|
@@ -222,7 +222,7 @@ export class AfterModelHookOutput extends DefaultHookOutput {
|
|
|
222
222
|
getModifiedResponse() {
|
|
223
223
|
if (this.hookSpecificOutput && 'llm_response' in this.hookSpecificOutput) {
|
|
224
224
|
const hookResponse = this.hookSpecificOutput['llm_response'];
|
|
225
|
-
if (hookResponse?.candidates?.[0]?.content) {
|
|
225
|
+
if (hookResponse?.candidates?.[0]?.content?.parts?.length) {
|
|
226
226
|
// Convert hook format to SDK format
|
|
227
227
|
return defaultHookTranslator.fromHookLLMResponse(hookResponse);
|
|
228
228
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/hooks/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAaH,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAE5D;;GAEG;AACH,MAAM,CAAN,IAAY,aAYX;AAZD,WAAY,aAAa;IACvB,0CAAyB,CAAA;IACzB,wCAAuB,CAAA;IACvB,4CAA2B,CAAA;IAC3B,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,4CAA2B,CAAA;IAC3B,4CAA2B,CAAA;IAC3B,0CAAyB,CAAA;IACzB,4DAA2C,CAAA;AAC7C,CAAC,EAZW,aAAa,KAAb,aAAa,QAYxB;AAsBD;;GAEG;AACH,MAAM,CAAN,IAAY,QAEX;AAFD,WAAY,QAAQ;IAClB,+BAAmB,CAAA;AACrB,CAAC,EAFW,QAAQ,KAAR,QAAQ,QAEnB;AAqCD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,IAAyB;IAEzB,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,aAAa;YAChB,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACzC,KAAK,YAAY;YACf,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,qBAAqB;YACxB,OAAO,IAAI,6BAA6B,CAAC,IAAI,CAAC,CAAC;QACjD;YACE,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,iBAAiB;IAC5B,QAAQ,CAAW;IACnB,UAAU,CAAU;IACpB,cAAc,CAAW;IACzB,aAAa,CAAU;IACvB,QAAQ,CAAgB;IACxB,MAAM,CAAU;IAChB,kBAAkB,CAA2B;IAE7C,YAAY,OAA4B,EAAE;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,IAAI,oBAAoB,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,4BAA4B,CAC1B,MAAiC;QAEjC,4DAA4D;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,4BAA4B,CAAC,MAG5B;QAIC,oEAAoE;QACpE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,IACE,IAAI,CAAC,kBAAkB;YACvB,mBAAmB,IAAI,IAAI,CAAC,kBAAkB,EAC9C,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC7D,OAAO,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAC9B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,kBAAkB,EAAE;aAClC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACM,kBAAkB;QACzB,uCAAuC;QACvC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,IAAI,0BAA0B,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC1D,MAAM,YAAY,GAChB,IAAI,CAAC,kBAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtD,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO,YAAY,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACpC,CAAC;IAED;;OAEG;IACM,kBAAkB;QACzB,kCAAkC;QAClC,IACE,IAAI,CAAC,kBAAkB;YACvB,oBAAoB,IAAI,IAAI,CAAC,kBAAkB,EAC/C,CAAC;YACD,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC;YACrE,IAAI,cAAc,KAAK,OAAO,IAAI,cAAc,KAAK,MAAM,EAAE,CAAC;gBAC5D,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACpC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,qBAAsB,SAAQ,iBAAiB;IAC1D;;OAEG;IACH,oBAAoB;QAClB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACA,CAAC;YACjB,IAAI,YAAY,EAAE,CAAC;gBACjB,oCAAoC;gBACpC,OAAO,qBAAqB,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACM,4BAA4B,CACnC,MAAiC;QAEjC,IAAI,IAAI,CAAC,kBAAkB,IAAI,aAAa,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CACzC,aAAa,CACS,CAAC;YACzB,IAAI,WAAW,EAAE,CAAC;gBAChB,oCAAoC;gBACpC,MAAM,UAAU,GAAG,qBAAqB,CAAC,kBAAkB,CACzD,WAAyB,EACzB,MAAM,CACP,CAAC;gBACF,OAAO;oBACL,GAAG,MAAM;oBACT,GAAG,UAAU;iBACd,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,6BAA8B,SAAQ,iBAAiB;IAClE;;OAEG;IACM,4BAA4B,CAAC,MAGrC;QACC,IAAI,IAAI,CAAC,kBAAkB,IAAI,YAAY,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAC5C,YAAY,CACK,CAAC;YACpB,IAAI,cAAc,EAAE,CAAC;gBACnB,oCAAoC;gBACpC,MAAM,aAAa,GACjB,qBAAqB,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAC3D,OAAO;oBACL,GAAG,MAAM;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;oBACzB,UAAU,EAAE,aAAa;iBAC1B,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACH,mBAAmB;QACjB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACS,CAAC;YAC1B,IAAI,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/hooks/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAaH,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAE5D;;GAEG;AACH,MAAM,CAAN,IAAY,aAYX;AAZD,WAAY,aAAa;IACvB,0CAAyB,CAAA;IACzB,wCAAuB,CAAA;IACvB,4CAA2B,CAAA;IAC3B,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,4CAA2B,CAAA;IAC3B,4CAA2B,CAAA;IAC3B,0CAAyB,CAAA;IACzB,4DAA2C,CAAA;AAC7C,CAAC,EAZW,aAAa,KAAb,aAAa,QAYxB;AAsBD;;GAEG;AACH,MAAM,CAAN,IAAY,QAEX;AAFD,WAAY,QAAQ;IAClB,+BAAmB,CAAA;AACrB,CAAC,EAFW,QAAQ,KAAR,QAAQ,QAEnB;AAqCD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,IAAyB;IAEzB,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,aAAa;YAChB,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACzC,KAAK,YAAY;YACf,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,qBAAqB;YACxB,OAAO,IAAI,6BAA6B,CAAC,IAAI,CAAC,CAAC;QACjD;YACE,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,iBAAiB;IAC5B,QAAQ,CAAW;IACnB,UAAU,CAAU;IACpB,cAAc,CAAW;IACzB,aAAa,CAAU;IACvB,QAAQ,CAAgB;IACxB,MAAM,CAAU;IAChB,kBAAkB,CAA2B;IAE7C,YAAY,OAA4B,EAAE;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,IAAI,oBAAoB,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,4BAA4B,CAC1B,MAAiC;QAEjC,4DAA4D;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,4BAA4B,CAAC,MAG5B;QAIC,oEAAoE;QACpE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,IACE,IAAI,CAAC,kBAAkB;YACvB,mBAAmB,IAAI,IAAI,CAAC,kBAAkB,EAC9C,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC7D,OAAO,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAC9B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,kBAAkB,EAAE;aAClC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACM,kBAAkB;QACzB,uCAAuC;QACvC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,IAAI,0BAA0B,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC1D,MAAM,YAAY,GAChB,IAAI,CAAC,kBAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtD,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO,YAAY,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACpC,CAAC;IAED;;OAEG;IACM,kBAAkB;QACzB,kCAAkC;QAClC,IACE,IAAI,CAAC,kBAAkB;YACvB,oBAAoB,IAAI,IAAI,CAAC,kBAAkB,EAC/C,CAAC;YACD,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC;YACrE,IAAI,cAAc,KAAK,OAAO,IAAI,cAAc,KAAK,MAAM,EAAE,CAAC;gBAC5D,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACpC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,qBAAsB,SAAQ,iBAAiB;IAC1D;;OAEG;IACH,oBAAoB;QAClB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACA,CAAC;YACjB,IAAI,YAAY,EAAE,CAAC;gBACjB,oCAAoC;gBACpC,OAAO,qBAAqB,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACM,4BAA4B,CACnC,MAAiC;QAEjC,IAAI,IAAI,CAAC,kBAAkB,IAAI,aAAa,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CACzC,aAAa,CACS,CAAC;YACzB,IAAI,WAAW,EAAE,CAAC;gBAChB,oCAAoC;gBACpC,MAAM,UAAU,GAAG,qBAAqB,CAAC,kBAAkB,CACzD,WAAyB,EACzB,MAAM,CACP,CAAC;gBACF,OAAO;oBACL,GAAG,MAAM;oBACT,GAAG,UAAU;iBACd,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,6BAA8B,SAAQ,iBAAiB;IAClE;;OAEG;IACM,4BAA4B,CAAC,MAGrC;QACC,IAAI,IAAI,CAAC,kBAAkB,IAAI,YAAY,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAC5C,YAAY,CACK,CAAC;YACpB,IAAI,cAAc,EAAE,CAAC;gBACnB,oCAAoC;gBACpC,MAAM,aAAa,GACjB,qBAAqB,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAC3D,OAAO;oBACL,GAAG,MAAM;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;oBACzB,UAAU,EAAE,aAAa;iBAC1B,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACH,mBAAmB;QACjB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACS,CAAC;YAC1B,IAAI,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBAC1D,oCAAoC;gBACpC,OAAO,qBAAqB,CAAC,mBAAmB,CAC9C,YAA2B,CAC5B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,oEAAoE;QACpE,IAAI,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;YAC/B,MAAM,YAAY,GAAgB;gBAChC,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,2BAA2B,CAAC;yBAClE;wBACD,YAAY,EAAE,MAAM;qBACrB;iBACF;aACF,CAAC;YACF,OAAO,qBAAqB,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;QACjE,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAyDD;;GAEG;AACH,MAAM,CAAN,IAAY,gBAEX;AAFD,WAAY,gBAAgB;IAC1B,qDAAiC,CAAA;AACnC,CAAC,EAFW,gBAAgB,KAAhB,gBAAgB,QAE3B;AA4BD;;GAEG;AACH,MAAM,CAAN,IAAY,kBAKX;AALD,WAAY,kBAAkB;IAC5B,yCAAmB,CAAA;IACnB,uCAAiB,CAAA;IACjB,qCAAe,CAAA;IACf,2CAAqB,CAAA;AACvB,CAAC,EALW,kBAAkB,KAAlB,kBAAkB,QAK7B;AAmBD;;GAEG;AACH,MAAM,CAAN,IAAY,gBAMX;AAND,WAAY,gBAAgB;IAC1B,iCAAa,CAAA;IACb,mCAAe,CAAA;IACf,qCAAiB,CAAA;IACjB,yDAAqC,CAAA;IACrC,mCAAe,CAAA;AACjB,CAAC,EANW,gBAAgB,KAAhB,gBAAgB,QAM3B;AASD;;GAEG;AACH,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,uCAAiB,CAAA;IACjB,mCAAa,CAAA;AACf,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B"}
|
|
@@ -3,8 +3,19 @@
|
|
|
3
3
|
* Copyright 2025 Google LLC
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { describe, it, expect } from 'vitest';
|
|
7
|
-
import { HookEventName, HookType } from './types.js';
|
|
6
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
7
|
+
import { createHookOutput, DefaultHookOutput, BeforeToolHookOutput, BeforeModelHookOutput, BeforeToolSelectionHookOutput, AfterModelHookOutput, HookEventName, HookType, } from './types.js';
|
|
8
|
+
import { defaultHookTranslator } from './hookTranslator.js';
|
|
9
|
+
vi.mock('./hookTranslator.js', () => ({
|
|
10
|
+
defaultHookTranslator: {
|
|
11
|
+
fromHookLLMResponse: vi.fn((response) => response),
|
|
12
|
+
fromHookLLMRequest: vi.fn((request, target) => ({
|
|
13
|
+
...target,
|
|
14
|
+
...request,
|
|
15
|
+
})),
|
|
16
|
+
fromHookToolConfig: vi.fn((config) => config),
|
|
17
|
+
},
|
|
18
|
+
}));
|
|
8
19
|
describe('Hook Types', () => {
|
|
9
20
|
describe('HookEventName', () => {
|
|
10
21
|
it('should contain all required event names', () => {
|
|
@@ -32,4 +43,271 @@ describe('Hook Types', () => {
|
|
|
32
43
|
});
|
|
33
44
|
});
|
|
34
45
|
});
|
|
46
|
+
describe('Hook Output Classes', () => {
|
|
47
|
+
describe('createHookOutput', () => {
|
|
48
|
+
it('should return DefaultHookOutput for unknown event names', () => {
|
|
49
|
+
const output = createHookOutput('UnknownEvent', {});
|
|
50
|
+
expect(output).toBeInstanceOf(DefaultHookOutput);
|
|
51
|
+
expect(output).not.toBeInstanceOf(BeforeModelHookOutput);
|
|
52
|
+
expect(output).not.toBeInstanceOf(AfterModelHookOutput);
|
|
53
|
+
expect(output).not.toBeInstanceOf(BeforeToolSelectionHookOutput);
|
|
54
|
+
});
|
|
55
|
+
it('should return BeforeModelHookOutput for BeforeModel event', () => {
|
|
56
|
+
const output = createHookOutput(HookEventName.BeforeModel, {});
|
|
57
|
+
expect(output).toBeInstanceOf(BeforeModelHookOutput);
|
|
58
|
+
});
|
|
59
|
+
it('should return AfterModelHookOutput for AfterModel event', () => {
|
|
60
|
+
const output = createHookOutput(HookEventName.AfterModel, {});
|
|
61
|
+
expect(output).toBeInstanceOf(AfterModelHookOutput);
|
|
62
|
+
});
|
|
63
|
+
it('should return BeforeToolSelectionHookOutput for BeforeToolSelection event', () => {
|
|
64
|
+
const output = createHookOutput(HookEventName.BeforeToolSelection, {});
|
|
65
|
+
expect(output).toBeInstanceOf(BeforeToolSelectionHookOutput);
|
|
66
|
+
});
|
|
67
|
+
});
|
|
68
|
+
describe('DefaultHookOutput', () => {
|
|
69
|
+
it('should construct with provided data', () => {
|
|
70
|
+
const data = {
|
|
71
|
+
continue: false,
|
|
72
|
+
stopReason: 'test stop',
|
|
73
|
+
suppressOutput: true,
|
|
74
|
+
systemMessage: 'test system message',
|
|
75
|
+
decision: 'block',
|
|
76
|
+
reason: 'test reason',
|
|
77
|
+
hookSpecificOutput: { key: 'value' },
|
|
78
|
+
};
|
|
79
|
+
const output = new DefaultHookOutput(data);
|
|
80
|
+
expect(output.continue).toBe(data.continue);
|
|
81
|
+
expect(output.stopReason).toBe(data.stopReason);
|
|
82
|
+
expect(output.suppressOutput).toBe(data.suppressOutput);
|
|
83
|
+
expect(output.systemMessage).toBe(data.systemMessage);
|
|
84
|
+
expect(output.decision).toBe(data.decision);
|
|
85
|
+
expect(output.reason).toBe(data.reason);
|
|
86
|
+
expect(output.hookSpecificOutput).toEqual(data.hookSpecificOutput);
|
|
87
|
+
});
|
|
88
|
+
it('should return false for isBlockingDecision if decision is not block or deny', () => {
|
|
89
|
+
const output1 = new DefaultHookOutput({ decision: 'approve' });
|
|
90
|
+
expect(output1.isBlockingDecision()).toBe(false);
|
|
91
|
+
const output2 = new DefaultHookOutput({ decision: undefined });
|
|
92
|
+
expect(output2.isBlockingDecision()).toBe(false);
|
|
93
|
+
});
|
|
94
|
+
it('should return true for isBlockingDecision if decision is block or deny', () => {
|
|
95
|
+
const output1 = new DefaultHookOutput({ decision: 'block' });
|
|
96
|
+
expect(output1.isBlockingDecision()).toBe(true);
|
|
97
|
+
const output2 = new DefaultHookOutput({ decision: 'deny' });
|
|
98
|
+
expect(output2.isBlockingDecision()).toBe(true);
|
|
99
|
+
});
|
|
100
|
+
it('should return true for shouldStopExecution if continue is false', () => {
|
|
101
|
+
const output = new DefaultHookOutput({ continue: false });
|
|
102
|
+
expect(output.shouldStopExecution()).toBe(true);
|
|
103
|
+
});
|
|
104
|
+
it('should return false for shouldStopExecution if continue is true or undefined', () => {
|
|
105
|
+
const output1 = new DefaultHookOutput({ continue: true });
|
|
106
|
+
expect(output1.shouldStopExecution()).toBe(false);
|
|
107
|
+
const output2 = new DefaultHookOutput({});
|
|
108
|
+
expect(output2.shouldStopExecution()).toBe(false);
|
|
109
|
+
});
|
|
110
|
+
it('should return reason if available', () => {
|
|
111
|
+
const output = new DefaultHookOutput({ reason: 'specific reason' });
|
|
112
|
+
expect(output.getEffectiveReason()).toBe('specific reason');
|
|
113
|
+
});
|
|
114
|
+
it('should return stopReason if reason is not available', () => {
|
|
115
|
+
const output = new DefaultHookOutput({ stopReason: 'stop reason' });
|
|
116
|
+
expect(output.getEffectiveReason()).toBe('stop reason');
|
|
117
|
+
});
|
|
118
|
+
it('should return "No reason provided" if neither reason nor stopReason are available', () => {
|
|
119
|
+
const output = new DefaultHookOutput({});
|
|
120
|
+
expect(output.getEffectiveReason()).toBe('No reason provided');
|
|
121
|
+
});
|
|
122
|
+
it('applyLLMRequestModifications should return target unchanged', () => {
|
|
123
|
+
const target = {
|
|
124
|
+
model: 'gemini-pro',
|
|
125
|
+
contents: [],
|
|
126
|
+
};
|
|
127
|
+
const output = new DefaultHookOutput({});
|
|
128
|
+
expect(output.applyLLMRequestModifications(target)).toBe(target);
|
|
129
|
+
});
|
|
130
|
+
it('applyToolConfigModifications should return target unchanged', () => {
|
|
131
|
+
const target = { toolConfig: {}, tools: [] };
|
|
132
|
+
const output = new DefaultHookOutput({});
|
|
133
|
+
expect(output.applyToolConfigModifications(target)).toBe(target);
|
|
134
|
+
});
|
|
135
|
+
it('getAdditionalContext should return additionalContext if present', () => {
|
|
136
|
+
const output = new DefaultHookOutput({
|
|
137
|
+
hookSpecificOutput: { additionalContext: 'some context' },
|
|
138
|
+
});
|
|
139
|
+
expect(output.getAdditionalContext()).toBe('some context');
|
|
140
|
+
});
|
|
141
|
+
it('getAdditionalContext should return undefined if additionalContext is not present', () => {
|
|
142
|
+
const output = new DefaultHookOutput({
|
|
143
|
+
hookSpecificOutput: { other: 'value' },
|
|
144
|
+
});
|
|
145
|
+
expect(output.getAdditionalContext()).toBeUndefined();
|
|
146
|
+
});
|
|
147
|
+
it('getAdditionalContext should return undefined if hookSpecificOutput is undefined', () => {
|
|
148
|
+
const output = new DefaultHookOutput({});
|
|
149
|
+
expect(output.getAdditionalContext()).toBeUndefined();
|
|
150
|
+
});
|
|
151
|
+
it('getBlockingError should return blocked: true and reason if blocking decision', () => {
|
|
152
|
+
const output = new DefaultHookOutput({
|
|
153
|
+
decision: 'block',
|
|
154
|
+
reason: 'blocked by hook',
|
|
155
|
+
});
|
|
156
|
+
expect(output.getBlockingError()).toEqual({
|
|
157
|
+
blocked: true,
|
|
158
|
+
reason: 'blocked by hook',
|
|
159
|
+
});
|
|
160
|
+
});
|
|
161
|
+
it('getBlockingError should return blocked: false if not blocking decision', () => {
|
|
162
|
+
const output = new DefaultHookOutput({ decision: 'approve' });
|
|
163
|
+
expect(output.getBlockingError()).toEqual({ blocked: false, reason: '' });
|
|
164
|
+
});
|
|
165
|
+
});
|
|
166
|
+
describe('BeforeToolHookOutput', () => {
|
|
167
|
+
it('isBlockingDecision should use permissionDecision from hookSpecificOutput', () => {
|
|
168
|
+
const output1 = new BeforeToolHookOutput({
|
|
169
|
+
hookSpecificOutput: { permissionDecision: 'block' },
|
|
170
|
+
});
|
|
171
|
+
expect(output1.isBlockingDecision()).toBe(true);
|
|
172
|
+
const output2 = new BeforeToolHookOutput({
|
|
173
|
+
hookSpecificOutput: { permissionDecision: 'approve' },
|
|
174
|
+
});
|
|
175
|
+
expect(output2.isBlockingDecision()).toBe(false);
|
|
176
|
+
});
|
|
177
|
+
it('getEffectiveReason should use permissionDecisionReason from hookSpecificOutput', () => {
|
|
178
|
+
const output1 = new BeforeToolHookOutput({
|
|
179
|
+
hookSpecificOutput: { permissionDecisionReason: 'compat reason' },
|
|
180
|
+
});
|
|
181
|
+
expect(output1.getEffectiveReason()).toBe('compat reason');
|
|
182
|
+
const output2 = new BeforeToolHookOutput({
|
|
183
|
+
reason: 'default reason',
|
|
184
|
+
hookSpecificOutput: { other: 'value' },
|
|
185
|
+
});
|
|
186
|
+
expect(output2.getEffectiveReason()).toBe('default reason');
|
|
187
|
+
});
|
|
188
|
+
});
|
|
189
|
+
describe('BeforeModelHookOutput', () => {
|
|
190
|
+
it('getSyntheticResponse should return synthetic response if llm_response is present', () => {
|
|
191
|
+
const mockResponse = { candidates: [] };
|
|
192
|
+
const output = new BeforeModelHookOutput({
|
|
193
|
+
hookSpecificOutput: { llm_response: mockResponse },
|
|
194
|
+
});
|
|
195
|
+
expect(output.getSyntheticResponse()).toEqual(mockResponse);
|
|
196
|
+
expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
|
|
197
|
+
});
|
|
198
|
+
it('getSyntheticResponse should return undefined if llm_response is not present', () => {
|
|
199
|
+
const output = new BeforeModelHookOutput({});
|
|
200
|
+
expect(output.getSyntheticResponse()).toBeUndefined();
|
|
201
|
+
});
|
|
202
|
+
it('applyLLMRequestModifications should apply modifications if llm_request is present', () => {
|
|
203
|
+
const target = {
|
|
204
|
+
model: 'gemini-pro',
|
|
205
|
+
contents: [{ parts: [{ text: 'original' }] }],
|
|
206
|
+
};
|
|
207
|
+
const mockRequest = {
|
|
208
|
+
messages: [{ role: 'user', content: 'modified' }],
|
|
209
|
+
};
|
|
210
|
+
const output = new BeforeModelHookOutput({
|
|
211
|
+
hookSpecificOutput: { llm_request: mockRequest },
|
|
212
|
+
});
|
|
213
|
+
const result = output.applyLLMRequestModifications(target);
|
|
214
|
+
expect(result).toEqual({ ...target, ...mockRequest });
|
|
215
|
+
expect(defaultHookTranslator.fromHookLLMRequest).toHaveBeenCalledWith(mockRequest, target);
|
|
216
|
+
});
|
|
217
|
+
it('applyLLMRequestModifications should return target unchanged if llm_request is not present', () => {
|
|
218
|
+
const target = {
|
|
219
|
+
model: 'gemini-pro',
|
|
220
|
+
contents: [],
|
|
221
|
+
};
|
|
222
|
+
const output = new BeforeModelHookOutput({});
|
|
223
|
+
expect(output.applyLLMRequestModifications(target)).toBe(target);
|
|
224
|
+
});
|
|
225
|
+
});
|
|
226
|
+
describe('BeforeToolSelectionHookOutput', () => {
|
|
227
|
+
it('applyToolConfigModifications should apply modifications if toolConfig is present', () => {
|
|
228
|
+
const target = { tools: [{ functionDeclarations: [] }] };
|
|
229
|
+
const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
|
|
230
|
+
const output = new BeforeToolSelectionHookOutput({
|
|
231
|
+
hookSpecificOutput: { toolConfig: mockToolConfig },
|
|
232
|
+
});
|
|
233
|
+
const result = output.applyToolConfigModifications(target);
|
|
234
|
+
expect(result).toEqual({ ...target, toolConfig: mockToolConfig });
|
|
235
|
+
expect(defaultHookTranslator.fromHookToolConfig).toHaveBeenCalledWith(mockToolConfig);
|
|
236
|
+
});
|
|
237
|
+
it('applyToolConfigModifications should return target unchanged if toolConfig is not present', () => {
|
|
238
|
+
const target = { toolConfig: {}, tools: [] };
|
|
239
|
+
const output = new BeforeToolSelectionHookOutput({});
|
|
240
|
+
expect(output.applyToolConfigModifications(target)).toBe(target);
|
|
241
|
+
});
|
|
242
|
+
it('applyToolConfigModifications should initialize tools array if not present', () => {
|
|
243
|
+
const target = {};
|
|
244
|
+
const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
|
|
245
|
+
const output = new BeforeToolSelectionHookOutput({
|
|
246
|
+
hookSpecificOutput: { toolConfig: mockToolConfig },
|
|
247
|
+
});
|
|
248
|
+
const result = output.applyToolConfigModifications(target);
|
|
249
|
+
expect(result).toEqual({ tools: [], toolConfig: mockToolConfig });
|
|
250
|
+
});
|
|
251
|
+
});
|
|
252
|
+
describe('AfterModelHookOutput', () => {
|
|
253
|
+
it('getModifiedResponse should return modified response if llm_response is present and has content', () => {
|
|
254
|
+
const mockResponse = {
|
|
255
|
+
candidates: [{ content: { role: 'model', parts: ['modified'] } }],
|
|
256
|
+
};
|
|
257
|
+
const output = new AfterModelHookOutput({
|
|
258
|
+
hookSpecificOutput: { llm_response: mockResponse },
|
|
259
|
+
});
|
|
260
|
+
expect(output.getModifiedResponse()).toEqual(mockResponse);
|
|
261
|
+
expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
|
|
262
|
+
});
|
|
263
|
+
it('getModifiedResponse should return undefined if llm_response is present but no content', () => {
|
|
264
|
+
const mockResponse = {
|
|
265
|
+
candidates: [{ content: { role: 'model', parts: [] } }],
|
|
266
|
+
};
|
|
267
|
+
const output = new AfterModelHookOutput({
|
|
268
|
+
hookSpecificOutput: { llm_response: mockResponse },
|
|
269
|
+
});
|
|
270
|
+
expect(output.getModifiedResponse()).toBeUndefined();
|
|
271
|
+
});
|
|
272
|
+
it('getModifiedResponse should return undefined if llm_response is not present', () => {
|
|
273
|
+
const output = new AfterModelHookOutput({});
|
|
274
|
+
expect(output.getModifiedResponse()).toBeUndefined();
|
|
275
|
+
});
|
|
276
|
+
it('getModifiedResponse should return a synthetic stop response if shouldStopExecution is true', () => {
|
|
277
|
+
const output = new AfterModelHookOutput({
|
|
278
|
+
continue: false,
|
|
279
|
+
stopReason: 'stopped by hook',
|
|
280
|
+
});
|
|
281
|
+
const expectedResponse = {
|
|
282
|
+
candidates: [
|
|
283
|
+
{
|
|
284
|
+
content: {
|
|
285
|
+
role: 'model',
|
|
286
|
+
parts: ['stopped by hook'],
|
|
287
|
+
},
|
|
288
|
+
finishReason: 'STOP',
|
|
289
|
+
},
|
|
290
|
+
],
|
|
291
|
+
};
|
|
292
|
+
expect(output.getModifiedResponse()).toEqual(expectedResponse);
|
|
293
|
+
expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(expectedResponse);
|
|
294
|
+
});
|
|
295
|
+
it('getModifiedResponse should return a synthetic stop response with default reason if shouldStopExecution is true and no stopReason', () => {
|
|
296
|
+
const output = new AfterModelHookOutput({ continue: false });
|
|
297
|
+
const expectedResponse = {
|
|
298
|
+
candidates: [
|
|
299
|
+
{
|
|
300
|
+
content: {
|
|
301
|
+
role: 'model',
|
|
302
|
+
parts: ['No reason provided'],
|
|
303
|
+
},
|
|
304
|
+
finishReason: 'STOP',
|
|
305
|
+
},
|
|
306
|
+
],
|
|
307
|
+
};
|
|
308
|
+
expect(output.getModifiedResponse()).toEqual(expectedResponse);
|
|
309
|
+
expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(expectedResponse);
|
|
310
|
+
});
|
|
311
|
+
});
|
|
312
|
+
});
|
|
35
313
|
//# sourceMappingURL=types.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.test.js","sourceRoot":"","sources":["../../../src/hooks/types.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAErD,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,cAAc,GAAG;gBACrB,YAAY;gBACZ,WAAW;gBACX,aAAa;gBACb,cAAc;gBACd,YAAY;gBACZ,cAAc;gBACd,YAAY;gBACZ,aAAa;gBACb,aAAa;gBACb,YAAY;gBACZ,qBAAqB;aACtB,CAAC;YAEF,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"types.test.js","sourceRoot":"","sources":["../../../src/hooks/types.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EACL,gBAAgB,EAChB,iBAAiB,EACjB,oBAAoB,EACpB,qBAAqB,EACrB,6BAA6B,EAC7B,oBAAoB,EACpB,aAAa,EACb,QAAQ,GACT,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAS5D,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,qBAAqB,EAAE;QACrB,mBAAmB,EAAE,EAAE,CAAC,EAAE,CACxB,CAAC,QAAqB,EAAE,EAAE,CAAC,QAA8C,CAC1E;QACD,kBAAkB,EAAE,EAAE,CAAC,EAAE,CACvB,CAAC,OAAmB,EAAE,MAAiC,EAAE,EAAE,CAAC,CAAC;YAC3D,GAAG,MAAM;YACT,GAAG,OAAO;SACX,CAAC,CACH;QACD,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,MAAkB,EAAE,EAAE,CAAC,MAAM,CAAC;KAC1D;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,cAAc,GAAG;gBACrB,YAAY;gBACZ,WAAW;gBACX,aAAa;gBACb,cAAc;gBACd,YAAY;gBACZ,cAAc;gBACd,YAAY;gBACZ,aAAa;gBACb,aAAa;gBACb,YAAY;gBACZ,qBAAqB;aACtB,CAAC;YAEF,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,6BAA6B,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,6BAA6B,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,IAAI,GAAG;gBACX,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,WAAW;gBACvB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,qBAAqB;gBACpC,QAAQ,EAAE,OAAuB;gBACjC,MAAM,EAAE,aAAa;gBACrB,kBAAkB,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;aACrC,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;YAC3F,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,EAAE;aACb,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,MAAM,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,kBAAkB,EAAE,EAAE,iBAAiB,EAAE,cAAc,EAAE;aAC1D,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,kBAAkB,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACzF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC;gBACxC,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAE;aACpD,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEhD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,kBAAkB,EAAE,SAAS,EAAE;aACtD,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,wBAAwB,EAAE,eAAe,EAAE;aAClE,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAE3D,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,MAAM,EAAE,gBAAgB;gBACxB,kBAAkB,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,YAAY,GAAgB,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC5D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;YAC3F,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,WAAW,GAAwB;gBACvC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;aAClD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE;aACjD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CACnE,WAAW,EACX,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;YACnG,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,EAAE;aACb,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,oBAAoB,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACzD,MAAM,cAAc,GAAG,EAAE,qBAAqB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC;gBAC/C,kBAAkB,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC;YAClE,MAAM,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CACnE,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0FAA0F,EAAE,GAAG,EAAE;YAClG,MAAM,MAAM,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,cAAc,GAAG,EAAE,qBAAqB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC;gBAC/C,kBAAkB,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gGAAgG,EAAE,GAAG,EAAE;YACxG,MAAM,YAAY,GAAgB;gBAChC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;aAClE,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC3D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,GAAG,EAAE;YAC/F,MAAM,YAAY,GAAgB;gBAChC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC;aACxD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4FAA4F,EAAE,GAAG,EAAE;YACpG,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,iBAAiB;aAC9B,CAAC,CAAC;YACH,MAAM,gBAAgB,GAAgB;gBACpC,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,iBAAiB,CAAC;yBAC3B;wBACD,YAAY,EAAE,MAAM;qBACrB;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kIAAkI,EAAE,GAAG,EAAE;YAC1I,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7D,MAAM,gBAAgB,GAAgB;gBACpC,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,oBAAoB,CAAC;yBAC9B;wBACD,YAAY,EAAE,MAAM;qBACrB;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -366,6 +366,165 @@ describe('IdeClient', () => {
|
|
|
366
366
|
expect(ideClient.isDiffingEnabled()).toBe(true);
|
|
367
367
|
});
|
|
368
368
|
});
|
|
369
|
+
describe('resolveDiffFromCli', () => {
|
|
370
|
+
beforeEach(async () => {
|
|
371
|
+
// Ensure client is "connected" for these tests
|
|
372
|
+
const ideClient = await IdeClient.getInstance();
|
|
373
|
+
// We need to set the client property on the instance for openDiff to work
|
|
374
|
+
ideClient.client = mockClient;
|
|
375
|
+
mockClient.request.mockResolvedValue({
|
|
376
|
+
isError: false,
|
|
377
|
+
content: [],
|
|
378
|
+
});
|
|
379
|
+
});
|
|
380
|
+
it("should resolve an open diff as 'accepted' and return the final content", async () => {
|
|
381
|
+
const ideClient = await IdeClient.getInstance();
|
|
382
|
+
const closeDiffSpy = vi
|
|
383
|
+
.spyOn(ideClient, 'closeDiff')
|
|
384
|
+
.mockResolvedValue('final content from ide');
|
|
385
|
+
const diffPromise = ideClient.openDiff('/test.txt', 'new content');
|
|
386
|
+
// Yield to the event loop to allow the openDiff promise executor to run
|
|
387
|
+
await new Promise((resolve) => setImmediate(resolve));
|
|
388
|
+
await ideClient.resolveDiffFromCli('/test.txt', 'accepted');
|
|
389
|
+
const result = await diffPromise;
|
|
390
|
+
expect(result).toEqual({
|
|
391
|
+
status: 'accepted',
|
|
392
|
+
content: 'final content from ide',
|
|
393
|
+
});
|
|
394
|
+
expect(closeDiffSpy).toHaveBeenCalledWith('/test.txt', {
|
|
395
|
+
suppressNotification: true,
|
|
396
|
+
});
|
|
397
|
+
expect(ideClient.diffResponses.has('/test.txt')).toBe(false);
|
|
398
|
+
});
|
|
399
|
+
it("should resolve an open diff as 'rejected'", async () => {
|
|
400
|
+
const ideClient = await IdeClient.getInstance();
|
|
401
|
+
const closeDiffSpy = vi
|
|
402
|
+
.spyOn(ideClient, 'closeDiff')
|
|
403
|
+
.mockResolvedValue(undefined);
|
|
404
|
+
const diffPromise = ideClient.openDiff('/test.txt', 'new content');
|
|
405
|
+
// Yield to the event loop to allow the openDiff promise executor to run
|
|
406
|
+
await new Promise((resolve) => setImmediate(resolve));
|
|
407
|
+
await ideClient.resolveDiffFromCli('/test.txt', 'rejected');
|
|
408
|
+
const result = await diffPromise;
|
|
409
|
+
expect(result).toEqual({
|
|
410
|
+
status: 'rejected',
|
|
411
|
+
content: undefined,
|
|
412
|
+
});
|
|
413
|
+
expect(closeDiffSpy).toHaveBeenCalledWith('/test.txt', {
|
|
414
|
+
suppressNotification: true,
|
|
415
|
+
});
|
|
416
|
+
expect(ideClient.diffResponses.has('/test.txt')).toBe(false);
|
|
417
|
+
});
|
|
418
|
+
it('should do nothing if no diff is open for the given file path', async () => {
|
|
419
|
+
const ideClient = await IdeClient.getInstance();
|
|
420
|
+
const closeDiffSpy = vi
|
|
421
|
+
.spyOn(ideClient, 'closeDiff')
|
|
422
|
+
.mockResolvedValue(undefined);
|
|
423
|
+
// No call to openDiff, so no resolver will exist.
|
|
424
|
+
await ideClient.resolveDiffFromCli('/non-existent.txt', 'accepted');
|
|
425
|
+
expect(closeDiffSpy).toHaveBeenCalledWith('/non-existent.txt', {
|
|
426
|
+
suppressNotification: true,
|
|
427
|
+
});
|
|
428
|
+
// No crash should occur, and nothing should be in the map.
|
|
429
|
+
expect(ideClient.diffResponses.has('/non-existent.txt')).toBe(false);
|
|
430
|
+
});
|
|
431
|
+
});
|
|
432
|
+
describe('closeDiff', () => {
|
|
433
|
+
beforeEach(async () => {
|
|
434
|
+
const ideClient = await IdeClient.getInstance();
|
|
435
|
+
ideClient.client = mockClient;
|
|
436
|
+
});
|
|
437
|
+
it('should return undefined if client is not connected', async () => {
|
|
438
|
+
const ideClient = await IdeClient.getInstance();
|
|
439
|
+
ideClient.client =
|
|
440
|
+
undefined;
|
|
441
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
442
|
+
expect(result).toBeUndefined();
|
|
443
|
+
});
|
|
444
|
+
it('should call client.request with correct arguments', async () => {
|
|
445
|
+
const ideClient = await IdeClient.getInstance();
|
|
446
|
+
// Return a valid, empty response as the return value is not under test here.
|
|
447
|
+
mockClient.request.mockResolvedValue({ isError: false, content: [] });
|
|
448
|
+
await ideClient.closeDiff('/test.txt', { suppressNotification: true });
|
|
449
|
+
expect(mockClient.request).toHaveBeenCalledWith(expect.objectContaining({
|
|
450
|
+
params: {
|
|
451
|
+
name: 'closeDiff',
|
|
452
|
+
arguments: {
|
|
453
|
+
filePath: '/test.txt',
|
|
454
|
+
suppressNotification: true,
|
|
455
|
+
},
|
|
456
|
+
},
|
|
457
|
+
}), expect.any(Object), // Schema
|
|
458
|
+
expect.any(Object));
|
|
459
|
+
});
|
|
460
|
+
it('should return content from a valid JSON response', async () => {
|
|
461
|
+
const ideClient = await IdeClient.getInstance();
|
|
462
|
+
const response = {
|
|
463
|
+
isError: false,
|
|
464
|
+
content: [
|
|
465
|
+
{ type: 'text', text: JSON.stringify({ content: 'file content' }) },
|
|
466
|
+
],
|
|
467
|
+
};
|
|
468
|
+
mockClient.request.mockResolvedValue(response);
|
|
469
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
470
|
+
expect(result).toBe('file content');
|
|
471
|
+
});
|
|
472
|
+
it('should return undefined for a valid JSON response with null content', async () => {
|
|
473
|
+
const ideClient = await IdeClient.getInstance();
|
|
474
|
+
const response = {
|
|
475
|
+
isError: false,
|
|
476
|
+
content: [{ type: 'text', text: JSON.stringify({ content: null }) }],
|
|
477
|
+
};
|
|
478
|
+
mockClient.request.mockResolvedValue(response);
|
|
479
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
480
|
+
expect(result).toBeUndefined();
|
|
481
|
+
});
|
|
482
|
+
it('should return undefined if response is not valid JSON', async () => {
|
|
483
|
+
const ideClient = await IdeClient.getInstance();
|
|
484
|
+
const response = {
|
|
485
|
+
isError: false,
|
|
486
|
+
content: [{ type: 'text', text: 'not json' }],
|
|
487
|
+
};
|
|
488
|
+
mockClient.request.mockResolvedValue(response);
|
|
489
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
490
|
+
expect(result).toBeUndefined();
|
|
491
|
+
});
|
|
492
|
+
it('should return undefined if request result has isError: true', async () => {
|
|
493
|
+
const ideClient = await IdeClient.getInstance();
|
|
494
|
+
const response = {
|
|
495
|
+
isError: true,
|
|
496
|
+
content: [{ type: 'text', text: 'An error occurred' }],
|
|
497
|
+
};
|
|
498
|
+
mockClient.request.mockResolvedValue(response);
|
|
499
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
500
|
+
expect(result).toBeUndefined();
|
|
501
|
+
});
|
|
502
|
+
it('should return undefined if client.request throws', async () => {
|
|
503
|
+
const ideClient = await IdeClient.getInstance();
|
|
504
|
+
mockClient.request.mockRejectedValue(new Error('Request failed'));
|
|
505
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
506
|
+
expect(result).toBeUndefined();
|
|
507
|
+
});
|
|
508
|
+
it('should return undefined if response has no text part', async () => {
|
|
509
|
+
const ideClient = await IdeClient.getInstance();
|
|
510
|
+
const response = {
|
|
511
|
+
isError: false,
|
|
512
|
+
content: [{ type: 'other' }],
|
|
513
|
+
};
|
|
514
|
+
mockClient.request.mockResolvedValue(response);
|
|
515
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
516
|
+
expect(result).toBeUndefined();
|
|
517
|
+
});
|
|
518
|
+
it('should return undefined if response is falsy', async () => {
|
|
519
|
+
const ideClient = await IdeClient.getInstance();
|
|
520
|
+
// Mocking with `null as any` to test the falsy path, as the mock
|
|
521
|
+
// function is strictly typed.
|
|
522
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
523
|
+
mockClient.request.mockResolvedValue(null);
|
|
524
|
+
const result = await ideClient.closeDiff('/test.txt');
|
|
525
|
+
expect(result).toBeUndefined();
|
|
526
|
+
});
|
|
527
|
+
});
|
|
369
528
|
describe('authentication', () => {
|
|
370
529
|
it('should connect with an auth token if provided in the discovery file', async () => {
|
|
371
530
|
const authToken = 'test-auth-token';
|