promptfoo 0.97.0 → 0.99.0
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/README.md +1 -1
- package/dist/package.json +19 -17
- package/dist/src/app/assets/index-CJGQrbZb.js +787 -0
- package/dist/src/app/assets/{index-YwjS5Vq2.css → index-DatcZSwS.css} +1 -1
- package/dist/src/app/assets/{index.es-Cc9N3uII.js → index.es-BCwhwRu5.js} +1 -1
- package/dist/src/app/assets/{sync-BoVzjiAe.js → sync-hJWHXawF.js} +1 -1
- package/dist/src/app/index.html +2 -2
- package/dist/src/assertions/index.d.ts.map +1 -1
- package/dist/src/assertions/index.js +9 -1
- package/dist/src/assertions/index.js.map +1 -1
- package/dist/src/cache.d.ts +1 -0
- package/dist/src/cache.d.ts.map +1 -1
- package/dist/src/cache.js +5 -0
- package/dist/src/cache.js.map +1 -1
- package/dist/src/commands/debug.d.ts +4 -0
- package/dist/src/commands/debug.d.ts.map +1 -0
- package/dist/src/commands/debug.js +84 -0
- package/dist/src/commands/debug.js.map +1 -0
- package/dist/src/commands/view.d.ts.map +1 -1
- package/dist/src/commands/view.js +6 -4
- package/dist/src/commands/view.js.map +1 -1
- package/dist/src/constants.d.ts +1 -0
- package/dist/src/constants.d.ts.map +1 -1
- package/dist/src/constants.js +2 -1
- package/dist/src/constants.js.map +1 -1
- package/dist/src/database/tables.d.ts +176 -176
- package/dist/src/envars.d.ts +1 -1
- package/dist/src/envars.d.ts.map +1 -1
- package/dist/src/evaluator.d.ts.map +1 -1
- package/dist/src/evaluator.js +7 -3
- package/dist/src/evaluator.js.map +1 -1
- package/dist/src/evaluatorHelpers.d.ts +1 -0
- package/dist/src/evaluatorHelpers.d.ts.map +1 -1
- package/dist/src/evaluatorHelpers.js +19 -0
- package/dist/src/evaluatorHelpers.js.map +1 -1
- package/dist/src/logger.js +10 -10
- package/dist/src/logger.js.map +1 -1
- package/dist/src/main.js +17 -1
- package/dist/src/main.js.map +1 -1
- package/dist/src/models/eval.d.ts +1 -0
- package/dist/src/models/eval.d.ts.map +1 -1
- package/dist/src/models/eval.js +2 -0
- package/dist/src/models/eval.js.map +1 -1
- package/dist/src/prompts/processors/yaml.d.ts.map +1 -1
- package/dist/src/prompts/processors/yaml.js +7 -2
- package/dist/src/prompts/processors/yaml.js.map +1 -1
- package/dist/src/providers/adaline.gateway.d.ts +2 -1
- package/dist/src/providers/adaline.gateway.d.ts.map +1 -1
- package/dist/src/providers/ai21.d.ts +2 -1
- package/dist/src/providers/ai21.d.ts.map +1 -1
- package/dist/src/providers/ai21.js.map +1 -1
- package/dist/src/providers/anthropic.d.ts +2 -1
- package/dist/src/providers/anthropic.d.ts.map +1 -1
- package/dist/src/providers/anthropic.js.map +1 -1
- package/dist/src/providers/azure.d.ts +2 -1
- package/dist/src/providers/azure.d.ts.map +1 -1
- package/dist/src/providers/bam.d.ts +2 -1
- package/dist/src/providers/bam.d.ts.map +1 -1
- package/dist/src/providers/bam.js.map +1 -1
- package/dist/src/providers/bedrock.d.ts +2 -1
- package/dist/src/providers/bedrock.d.ts.map +1 -1
- package/dist/src/providers/cloudflare-ai.d.ts +2 -1
- package/dist/src/providers/cloudflare-ai.d.ts.map +1 -1
- package/dist/src/providers/cohere.d.ts +2 -1
- package/dist/src/providers/cohere.d.ts.map +1 -1
- package/dist/src/providers/defaults.d.ts +2 -1
- package/dist/src/providers/defaults.d.ts.map +1 -1
- package/dist/src/providers/defaults.js.map +1 -1
- package/dist/src/providers/fal.d.ts +2 -1
- package/dist/src/providers/fal.d.ts.map +1 -1
- package/dist/src/providers/fal.js.map +1 -1
- package/dist/src/providers/groq.d.ts +2 -1
- package/dist/src/providers/groq.d.ts.map +1 -1
- package/dist/src/providers/http.d.ts +5 -1
- package/dist/src/providers/http.d.ts.map +1 -1
- package/dist/src/providers/http.js +67 -10
- package/dist/src/providers/http.js.map +1 -1
- package/dist/src/providers/localai.d.ts +2 -1
- package/dist/src/providers/localai.d.ts.map +1 -1
- package/dist/src/providers/localai.js.map +1 -1
- package/dist/src/providers/mistral.d.ts +2 -1
- package/dist/src/providers/mistral.d.ts.map +1 -1
- package/dist/src/providers/openai.d.ts +2 -1
- package/dist/src/providers/openai.d.ts.map +1 -1
- package/dist/src/providers/openai.js +1 -1
- package/dist/src/providers/openai.js.map +1 -1
- package/dist/src/providers/palm.d.ts +2 -1
- package/dist/src/providers/palm.d.ts.map +1 -1
- package/dist/src/providers/palm.js.map +1 -1
- package/dist/src/providers/promptfoo.d.ts +2 -1
- package/dist/src/providers/promptfoo.d.ts.map +1 -1
- package/dist/src/providers/replicate.d.ts +2 -1
- package/dist/src/providers/replicate.d.ts.map +1 -1
- package/dist/src/providers/sequence.d.ts +11 -0
- package/dist/src/providers/sequence.d.ts.map +1 -0
- package/dist/src/providers/sequence.js +66 -0
- package/dist/src/providers/sequence.js.map +1 -0
- package/dist/src/providers/togetherai.d.ts +2 -1
- package/dist/src/providers/togetherai.d.ts.map +1 -1
- package/dist/src/providers/togetherai.js.map +1 -1
- package/dist/src/providers/vertex.d.ts +2 -1
- package/dist/src/providers/vertex.d.ts.map +1 -1
- package/dist/src/providers/watsonx.d.ts +2 -1
- package/dist/src/providers/watsonx.d.ts.map +1 -1
- package/dist/src/providers/watsonx.js.map +1 -1
- package/dist/src/providers/xai.d.ts +2 -1
- package/dist/src/providers/xai.d.ts.map +1 -1
- package/dist/src/providers/xai.js.map +1 -1
- package/dist/src/providers.d.ts +2 -1
- package/dist/src/providers.d.ts.map +1 -1
- package/dist/src/providers.js +4 -0
- package/dist/src/providers.js.map +1 -1
- package/dist/src/redteam/commands/generate.d.ts.map +1 -1
- package/dist/src/redteam/commands/generate.js +25 -4
- package/dist/src/redteam/commands/generate.js.map +1 -1
- package/dist/src/redteam/commands/init.d.ts.map +1 -1
- package/dist/src/redteam/commands/init.js +79 -64
- package/dist/src/redteam/commands/init.js.map +1 -1
- package/dist/src/redteam/commands/report.d.ts.map +1 -1
- package/dist/src/redteam/commands/report.js +11 -3
- package/dist/src/redteam/commands/report.js.map +1 -1
- package/dist/src/redteam/commands/setup.d.ts +3 -0
- package/dist/src/redteam/commands/setup.d.ts.map +1 -0
- package/dist/src/redteam/commands/setup.js +33 -0
- package/dist/src/redteam/commands/setup.js.map +1 -0
- package/dist/src/redteam/constants.d.ts +4 -4
- package/dist/src/redteam/constants.d.ts.map +1 -1
- package/dist/src/redteam/constants.js +12 -2
- package/dist/src/redteam/constants.js.map +1 -1
- package/dist/src/redteam/index.d.ts.map +1 -1
- package/dist/src/redteam/index.js +95 -37
- package/dist/src/redteam/index.js.map +1 -1
- package/dist/src/redteam/plugins/base.d.ts.map +1 -1
- package/dist/src/redteam/plugins/base.js +2 -0
- package/dist/src/redteam/plugins/base.js.map +1 -1
- package/dist/src/redteam/plugins/competitors.d.ts.map +1 -1
- package/dist/src/redteam/plugins/competitors.js +23 -1
- package/dist/src/redteam/plugins/competitors.js.map +1 -1
- package/dist/src/redteam/plugins/debugAccess.d.ts.map +1 -1
- package/dist/src/redteam/plugins/debugAccess.js +18 -0
- package/dist/src/redteam/plugins/debugAccess.js.map +1 -1
- package/dist/src/redteam/plugins/harmful/constants.d.ts.map +1 -1
- package/dist/src/redteam/plugins/harmful/constants.js +32 -8
- package/dist/src/redteam/plugins/harmful/constants.js.map +1 -1
- package/dist/src/redteam/plugins/harmful/unaligned.js +1 -1
- package/dist/src/redteam/plugins/harmful/unaligned.js.map +1 -1
- package/dist/src/redteam/plugins/intent.d.ts.map +1 -1
- package/dist/src/redteam/plugins/intent.js +6 -0
- package/dist/src/redteam/plugins/intent.js.map +1 -1
- package/dist/src/redteam/providers/crescendo/index.d.ts +3 -0
- package/dist/src/redteam/providers/crescendo/index.d.ts.map +1 -1
- package/dist/src/redteam/providers/crescendo/index.js +66 -37
- package/dist/src/redteam/providers/crescendo/index.js.map +1 -1
- package/dist/src/redteam/providers/goat.d.ts.map +1 -1
- package/dist/src/redteam/providers/goat.js +56 -41
- package/dist/src/redteam/providers/goat.js.map +1 -1
- package/dist/src/redteam/providers/iterative.d.ts +1 -0
- package/dist/src/redteam/providers/iterative.d.ts.map +1 -1
- package/dist/src/redteam/providers/iterative.js +14 -9
- package/dist/src/redteam/providers/iterative.js.map +1 -1
- package/dist/src/redteam/providers/iterativeImage.d.ts +3 -1
- package/dist/src/redteam/providers/iterativeImage.d.ts.map +1 -1
- package/dist/src/redteam/providers/iterativeImage.js +14 -8
- package/dist/src/redteam/providers/iterativeImage.js.map +1 -1
- package/dist/src/redteam/providers/iterativeTree.d.ts.map +1 -1
- package/dist/src/redteam/providers/iterativeTree.js +26 -19
- package/dist/src/redteam/providers/iterativeTree.js.map +1 -1
- package/dist/src/redteam/providers/shared.d.ts +9 -6
- package/dist/src/redteam/providers/shared.d.ts.map +1 -1
- package/dist/src/redteam/providers/shared.js +9 -16
- package/dist/src/redteam/providers/shared.js.map +1 -1
- package/dist/src/redteam/strategies/citation.d.ts +3 -0
- package/dist/src/redteam/strategies/citation.d.ts.map +1 -0
- package/dist/src/redteam/strategies/citation.js +96 -0
- package/dist/src/redteam/strategies/citation.js.map +1 -0
- package/dist/src/redteam/strategies/index.d.ts +3 -2
- package/dist/src/redteam/strategies/index.d.ts.map +1 -1
- package/dist/src/redteam/strategies/index.js +68 -13
- package/dist/src/redteam/strategies/index.js.map +1 -1
- package/dist/src/redteam/strategies/singleTurnComposite.d.ts +3 -0
- package/dist/src/redteam/strategies/singleTurnComposite.d.ts.map +1 -0
- package/dist/src/redteam/strategies/singleTurnComposite.js +89 -0
- package/dist/src/redteam/strategies/singleTurnComposite.js.map +1 -0
- package/dist/src/redteam/types.d.ts +5 -1
- package/dist/src/redteam/types.d.ts.map +1 -1
- package/dist/src/server/index.js +4 -2
- package/dist/src/server/index.js.map +1 -1
- package/dist/src/server/routes/providers.js +1 -0
- package/dist/src/server/routes/providers.js.map +1 -1
- package/dist/src/server/server.d.ts +1 -6
- package/dist/src/server/server.d.ts.map +1 -1
- package/dist/src/server/server.js +13 -43
- package/dist/src/server/server.js.map +1 -1
- package/dist/src/testCases.js +1 -1
- package/dist/src/testCases.js.map +1 -1
- package/dist/src/types/env.d.ts +54 -0
- package/dist/src/types/env.d.ts.map +1 -0
- package/dist/src/types/env.js +3 -0
- package/dist/src/types/env.js.map +1 -0
- package/dist/src/types/index.d.ts +8395 -6838
- package/dist/src/types/index.d.ts.map +1 -1
- package/dist/src/types/index.js +2 -1
- package/dist/src/types/index.js.map +1 -1
- package/dist/src/types/providers.d.ts +3 -53
- package/dist/src/types/providers.d.ts.map +1 -1
- package/dist/src/types/providers.js.map +1 -1
- package/dist/src/util/config/load.js +1 -1
- package/dist/src/util/config/load.js.map +1 -1
- package/dist/src/util/index.d.ts +32 -32
- package/dist/src/util/server.d.ts +10 -0
- package/dist/src/util/server.d.ts.map +1 -0
- package/dist/src/util/server.js +74 -0
- package/dist/src/util/server.js.map +1 -0
- package/dist/src/util/transform.d.ts +1 -0
- package/dist/src/util/transform.d.ts.map +1 -1
- package/dist/src/util/transform.js.map +1 -1
- package/dist/src/validators/providers.d.ts +296 -296
- package/dist/src/validators/redteam.d.ts +130 -126
- package/dist/src/validators/redteam.d.ts.map +1 -1
- package/dist/src/validators/redteam.js +18 -2
- package/dist/src/validators/redteam.js.map +1 -1
- package/dist/test/cache.test.js +47 -7
- package/dist/test/cache.test.js.map +1 -1
- package/dist/test/commands/init.test.js +14 -0
- package/dist/test/commands/init.test.js.map +1 -1
- package/dist/test/evaluator.test.js +50 -0
- package/dist/test/evaluator.test.js.map +1 -1
- package/dist/test/factories/evalFactory.d.ts +144 -144
- package/dist/test/prompts/index.test.js +38 -25
- package/dist/test/prompts/index.test.js.map +1 -1
- package/dist/test/prompts/processors/yaml.test.js +50 -4
- package/dist/test/prompts/processors/yaml.test.js.map +1 -1
- package/dist/test/providers/http.test.js +0 -17
- package/dist/test/providers/http.test.js.map +1 -1
- package/dist/test/providers/index.test.js +1 -0
- package/dist/test/providers/index.test.js.map +1 -1
- package/dist/test/providers/togetherai.test.js.map +1 -1
- package/dist/test/redteam/commands/generate.test.js +58 -0
- package/dist/test/redteam/commands/generate.test.js.map +1 -1
- package/dist/test/redteam/index.test.js +2 -2
- package/dist/test/redteam/index.test.js.map +1 -1
- package/dist/test/redteam/plugins/base.test.js +8 -0
- package/dist/test/redteam/plugins/base.test.js.map +1 -1
- package/dist/test/redteam/providers/iterativeTree.test.js +11 -3
- package/dist/test/redteam/providers/iterativeTree.test.js.map +1 -1
- package/dist/test/testCases.test.js +21 -0
- package/dist/test/testCases.test.js.map +1 -1
- package/dist/test/types.test.js +17 -4
- package/dist/test/types.test.js.map +1 -1
- package/dist/test/util/server.test.d.ts +2 -0
- package/dist/test/util/server.test.d.ts.map +1 -0
- package/dist/test/util/server.test.js +114 -0
- package/dist/test/util/server.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +19 -17
- package/dist/src/app/assets/index-CbxDbiAR.js +0 -769
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/redteam/providers/crescendo/index.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,oEAAuC;AACvC,+BAAoC;AACpC,gEAAyD;AACzD,6DAAqC;AACrC,4DAA+E;AAU/E,6CAA4D;AAC5D,uDAA4D;AAC5D,qCAAkE;AAClE,sCAAmE;AACnE,uCAA+F;AAE/F,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAclC,MAAa,YAAY;IAAzB;QACU,kBAAa,GAAuC,IAAI,GAAG,EAAE,CAAC;IAoBxE,CAAC;IAlBC,UAAU,CAAC,cAAsB,EAAE,OAA4B;QAC7D,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,eAAe,CAAC,cAAsB;QACpC,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IACtD,CAAC;IAED,sCAAsC,CAAC,cAAsB;QAC3D,MAAM,oBAAoB,GAAG,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,iBAAiB,GAAG,IAAA,SAAM,GAAE,CAAC;QACnC,MAAM,eAAe,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,sCAAsC;QACjG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;QAC3D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;CACF;AArBD,oCAqBC;AAED,MAAM,iBAAiB;IAYrB,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,sBAAsB,CAAC;QACpE,IAAI,CAAC,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QACjC,IAAI,CAAC,oBAAoB,GAAG,IAAA,SAAM,GAAE,CAAC;QACrC,IAAI,CAAC,4BAA4B,GAAG,IAAA,SAAM,GAAE,CAAC;QAC7C,gBAAM,CAAC,KAAK,CAAC,8CAA8C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACvF,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,IAAI,IAAA,2BAAoB,GAAE,EAAE,CAAC;gBAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,2CAA+B,CAAC;oBACzD,IAAI,EAAE,WAAW;oBACjB,QAAQ,EAAE,IAAI;oBACd,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,GAAG,MAAM,IAAA,4BAAmB,EAAC;oBAC/C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;oBACrC,gBAAgB,EAAE,IAAI;oBACtB,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,IAAI,IAAA,2BAAoB,GAAE,EAAE,CAAC;gBAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,2CAA+B,CAAC;oBACzD,IAAI,EAAE,WAAW;oBACjB,QAAQ,EAAE,KAAK;oBACf,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,GAAG,MAAM,IAAA,4BAAmB,EAAC;oBAC/C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;oBACrC,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,EAAE;QACA,OAAO,6BAA6B,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B,EAAE,OAA8B;QAC1F,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QAEpD,gBAAM,CAAC,KAAK,CAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QAEvD,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,gBAAM,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAE5C,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,OAAO,CAAC,gBAAgB;YAClC,OAAO;YACP,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,EACtB,MAAM,EACN,OAAO,EACP,IAAI,EACJ,QAAQ,EACR,OAAO,EACP,OAAO,GAQR;QACC,gBAAM,CAAC,KAAK,CACV,0CAA0C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,cAAc,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,CAC1K,CAAC;QAEF,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,cAAc,GAAkB,IAAI,CAAC;QACzC,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACf,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,iCAAuB,EAAE;YACvE,qBAAqB,EAAE,IAAI,CAAC,QAAQ;YACpC,YAAY,EAAE,QAAQ,GAAG,CAAC;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;QAEH,OAAO,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YACjC,QAAQ,EAAE,CAAC;YACX,gBAAM,CAAC,KAAK,CAAC,WAAW,QAAQ,IAAI,CAAC,CAAC;YAEtC,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,GACrE,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;YAC7E,IAAI,gBAAgB,EAAE,CAAC;gBACrB,eAAe,CAAC,KAAK,IAAI,gBAAgB,CAAC,KAAK,IAAI,CAAC,CAAC;gBACrD,eAAe,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,IAAI,CAAC,CAAC;gBACvD,eAAe,CAAC,UAAU,IAAI,gBAAgB,CAAC,UAAU,IAAI,CAAC,CAAC;YACjE,CAAC;YACD,gBAAM,CAAC,KAAK,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;YAEzD,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,gBAAgB,EAAE,GAAG,MAAM,IAAI,CAAC,UAAU,CACtE,YAAY,EACZ,MAAM,EACN,IAAI,EACJ,OAAO,EACP,QAAQ,EACR,QAAQ,EACR,OAAO,EACP,OAAO,CACR,CAAC;YACF,YAAY,GAAG,QAAQ,CAAC;YACxB,IAAI,gBAAgB,EAAE,CAAC;gBACrB,eAAe,CAAC,KAAK,IAAI,gBAAgB,CAAC,KAAK,IAAI,CAAC,CAAC;gBACrD,eAAe,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,IAAI,CAAC,CAAC;gBACvD,eAAe,CAAC,UAAU,IAAI,gBAAgB,CAAC,UAAU,IAAI,CAAC,CAAC;gBAC/D,eAAe,CAAC,WAAW,IAAI,gBAAgB,CAAC,WAAW,IAAI,CAAC,CAAC;YACnE,CAAC;YAED,MAAM,CAAC,SAAS,EAAE,gBAAgB,CAAC,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAC7F,gBAAM,CAAC,KAAK,CAAC,mCAAmC,SAAS,eAAe,gBAAgB,EAAE,CAAC,CAAC;YAE5F,IAAI,SAAS,EAAE,CAAC;gBACd,gBAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACnE,cAAc,EAAE,CAAC;gBACjB,IAAI,CAAC,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBAClF,YAAY,GAAG,IAAA,gBAAM,EAAA;;;;+DAIkC,gBAAgB;;;SAGtE,CAAC;gBACF,QAAQ,GAAG,KAAK,CAAC;gBACjB,cAAc,GAAG,IAAI,CAAC;gBAEtB,gBAAM,CAAC,KAAK,CAAC,6BAA6B,cAAc,EAAE,CAAC,CAAC;gBAC5D,IAAI,cAAc,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;oBACxC,QAAQ,EAAE,CAAC;oBACX,gBAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;gBACrE,CAAC;gBAED,IAAI,cAAc,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzC,gBAAM,CAAC,KAAK,CAAC,mBAAmB,IAAI,CAAC,aAAa,0BAA0B,CAAC,CAAC;oBAC9E,MAAM;gBACR,CAAC;gBAED,SAAS;YACX,CAAC;YAED,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAC1E,IAAI,cAAc,EAAE,CAAC;gBACnB,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,KAAK,IAAI,CAAC,CAAC;gBACnD,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,MAAM,IAAI,CAAC,CAAC;gBACrD,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,IAAI,CAAC,CAAC;YAC/D,CAAC;YACD,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC;YAC3B,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC;YACpC,gBAAM,CAAC,KAAK,CAAC,oBAAoB,QAAQ,gBAAgB,cAAc,EAAE,CAAC,CAAC;YAE3E,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;gBAC3B,gBAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;gBACnE,MAAM;YACR,CAAC;YAED,gBAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,gBAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACpD,CAAC;QAED,gBAAM,CAAC,KAAK,CACV,kCAAkC,QAAQ,sBAAsB,cAAc,KAAK,CACpF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC/C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAEvD,OAAO;YACL,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE;gBACR,kBAAkB;gBAClB,kBAAkB,EAAE,IAAI,CAAC,MAAM;qBAC5B,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC;qBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;qBACtC,IAAI,CAAC,MAAM,CAAC;gBACf,wBAAwB,EAAE,QAAQ;gBAClC,uBAAuB,EAAE,cAAc;gBACvC,eAAe,EAAE,QAAQ;gBACzB,mBAAmB,EAAE,cAAc;aACpC;YACD,UAAU,EAAE,eAAe;SAC5B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,QAAgB,EAChB,SAAkB,EAClB,YAAoB,EACpB,YAAoB;QAEpB,gBAAM,CAAC,KAAK,CAAC,iCAAiC,QAAQ,eAAe,SAAS,EAAE,CAAC,CAAC;QAClF,IAAI,UAAkB,CAAC;QAEvB,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YACnB,UAAU;gBACR,kGAAkG;oBAClG,2CAA2C;oBAC3C,uEAAuE;oBACvE,CAAC,YAAY;wBACX,CAAC,CAAC,0BAA0B,YAAY,OAAO,YAAY,EAAE;wBAC7D,CAAC,CAAC,8CAA8C,CAAC,CAAC;QACxD,CAAC;aAAM,CAAC;YACN,UAAU;gBACR,uFAAuF;oBACvF,uEAAuE;oBACvE,wBAAwB,QAAQ,+BAA+B,SAAS,IAAI;oBAC5E,0BAA0B,YAAY,EAAE,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,UAAU;SACpB,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACzF,gBAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC7C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,EAAE;YAC/E,MAAM,EAAE;gBACN,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACtC,KAAK,EAAE,SAAS;aACjB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,QAAQ,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAEjF,MAAM,YAAY,GAAG,IAAA,6BAAsB,EAIxC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,YAAY,GAAG,CAAC,mBAAmB,EAAE,0BAA0B,EAAE,qBAAqB,CAAC,CAAC;QAE9F,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,CAAC,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC;gBAC3B,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,EAAE,CAAC,CAAC;gBAChD,MAAM,IAAI,KAAK,CAAC,iBAAiB,GAAG,iCAAiC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1F,CAAC;QACH,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,IAAA,gBAAM,EAAA;;;2BAGI,YAAY,CAAC,iBAAiB;kCACvB,YAAY,CAAC,wBAAwB;6BAC1C,YAAY,CAAC,mBAAmB;KACxD,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,KAAK,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7D,gBAAM,CAAC,KAAK,CAAC,gCAAgC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrF,MAAM,IAAI,KAAK,CAAC,2CAA2C,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ,CAAC,MAAM;SACzB,CAAC,CAAC;QAEH,OAAO;YACL,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;YACjD,UAAU,EAAE,QAAQ,CAAC,UAAU;SAChC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,YAAoB,EACpB,cAAsB,EACtB,IAAqC,EACrC,OAAsC,EACtC,QAAqB,EACrB,QAAgB,EAChB,OAA8B,EAC9B,OAA8B;QAE9B,MAAM,cAAc,GAAG,MAAM,IAAA,+BAAY,EACvC,cAAc,EACd,EAAE,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,EAClD,OAAO,EACP,QAAQ,CACT,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACxC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAE/C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAAwB,cAAc,CAAC,CAAC;YAC7E,6DAA6D;YAC7D,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IACE,OAAO,CAAC,IAAI,KAAK,QAAQ;oBACzB,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EACvF,CAAC;oBACD,+BAA+B;oBAC/B,SAAS;gBACX,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;YACvD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE;gBAChD,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,cAAc;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,EAAE,iBAAiB,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,MAAM,IAAA,0BAAiB,EAC/E,QAAQ,EACR,cAAc,EACd,OAAO,EACP,OAAO,CACR,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,kCAAkC,cAAc,EAAE,CAAC,CAAC;QAEjE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAChD,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,cAAc;SACxB,CAAC,CAAC;QAEH,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,YAAoB,EACpB,YAAoB;QAEpB,gBAAM,CAAC,KAAK,CAAC,qCAAqC,YAAY,EAAE,CAAC,CAAC;QAClE,8DAA8D;QAC9D,IAAI,IAAA,qBAAc,EAAC,YAAY,CAAC,EAAE,CAAC;YACjC,gBAAM,CAAC,KAAK,CACV,gFAAgF,CACjF,CAAC;YACF,OAAO,CAAC,IAAI,EAAE,yCAAyC,CAAC,CAAC;QAC3D,CAAC;QAED,uDAAuD;QACvD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,qBAAqB,EAAE,YAAY;YACnC,uBAAuB,EAAE,YAAY;SACtC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,+BAAqB,EAAE;YAC5E,qBAAqB,EAAE,IAAI,CAAC,QAAQ;SACrC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC;YACjC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE;YAChD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE;SACxC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,MAAM,eAAe,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,EAAE;YACjE,MAAM,EAAE;gBACN,GAAG,EAAE,WAAW;gBAChB,KAAK,EAAE,SAAS;aACjB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,eAAe,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACxF,gBAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;QAElE,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAIlC,eAAe,CAAC,MAAM,CAAC,CAAC;QAE3B,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,+CAA+C,CAAC,CAAC;QAC9F,IAAA,wBAAS,EACP,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EACnC,iDAAiD,CAClD,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,YAAoB;QAC7C,gBAAM,CAAC,KAAK,CAAC,oCAAoC,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YAC/B,qBAAqB,EAAE,IAAI,CAAC,QAAQ;YACpC,uBAAuB,EAAE,YAAY;SACtC,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,4BAAkB,EAAE;YACtE,qBAAqB,EAAE,IAAI,CAAC,QAAQ;SACrC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAC9B,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,gBAAgB,EAAE;YAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE;SACrC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,QAAQ,EAAE;YAC3D,MAAM,EAAE;gBACN,GAAG,EAAE,QAAQ;gBACb,KAAK,EAAE,MAAM;aACd;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uCAAuC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC;QAC/E,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,YAAY,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACrF,gBAAM,CAAC,KAAK,CAAC,wBAAwB,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5D,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAKlC,YAAY,CAAC,MAAM,CAAC,CAAC;QAExB,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,4CAA4C,CAAC,CAAC;QAC3F,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,8CAA8C,CAAC,CAAC;QAE/F,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,cAAsB;QAClD,OAAO,IAAI,CAAC,MAAM,CAAC,sCAAsC,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC;IAEO,cAAc,CAAC,cAAsB;QAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAC7D,gBAAM,CAAC,KAAK,CAAC,2BAA2B,cAAc,GAAG,CAAC,CAAC;QAC3D,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,gBAAM,CAAC,KAAK,CAAC,OAAO,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF;AAED,kBAAe,iBAAiB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/redteam/providers/crescendo/index.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,oEAAuC;AACvC,+BAAoC;AACpC,gEAAyD;AACzD,6DAAqC;AACrC,4DAA+E;AAU/E,6CAA4D;AAC5D,uDAA4D;AAC5D,qCAAkE;AAClE,sCAAwF;AACxF,uCAA+F;AAE/F,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAelC,MAAa,YAAY;IAAzB;QACU,kBAAa,GAAuC,IAAI,GAAG,EAAE,CAAC;IAoBxE,CAAC;IAlBC,UAAU,CAAC,cAAsB,EAAE,OAA4B;QAC7D,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,eAAe,CAAC,cAAsB;QACpC,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IACtD,CAAC;IAED,sCAAsC,CAAC,cAAsB;QAC3D,MAAM,oBAAoB,GAAG,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAClE,MAAM,iBAAiB,GAAG,IAAA,SAAM,GAAE,CAAC;QACnC,MAAM,eAAe,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,sCAAsC;QACjG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;QAC3D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;CACF;AArBD,oCAqBC;AAED,MAAM,iBAAiB;IAarB,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,sBAAsB,CAAC;QACpE,IAAI,CAAC,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QACjC,IAAI,CAAC,oBAAoB,GAAG,IAAA,SAAM,GAAE,CAAC;QACrC,IAAI,CAAC,4BAA4B,GAAG,IAAA,SAAM,GAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACzB,CAAC;QACD,gBAAM,CAAC,KAAK,CAAC,8CAA8C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACvF,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,IAAI,IAAA,2BAAoB,GAAE,EAAE,CAAC;gBAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,2CAA+B,CAAC;oBACzD,IAAI,EAAE,WAAW;oBACjB,QAAQ,EAAE,IAAI;oBACd,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,GAAG,MAAM,IAAA,4BAAmB,EAAC;oBAC/C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;oBACrC,gBAAgB,EAAE,IAAI;oBACtB,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,IAAI,IAAA,2BAAoB,GAAE,EAAE,CAAC;gBAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,2CAA+B,CAAC;oBACzD,IAAI,EAAE,WAAW;oBACjB,QAAQ,EAAE,KAAK;oBACf,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,GAAG,MAAM,IAAA,4BAAmB,EAAC;oBAC/C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;oBACrC,gBAAgB,EAAE,IAAI;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,EAAE;QACA,OAAO,6BAA6B,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B,EAAE,OAA8B;QAC1F,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QAEpD,gBAAM,CAAC,KAAK,CAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QAEvD,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,gBAAM,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAE5C,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,OAAO,CAAC,gBAAgB;YAClC,OAAO;YACP,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,EACtB,MAAM,EACN,OAAO,EACP,IAAI,EACJ,QAAQ,EACR,OAAO,EACP,OAAO,GAQR;QACC,gBAAM,CAAC,KAAK,CACV,0CAA0C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,OAAO,cAAc,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,CAC1K,CAAC;QAEF,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,IAAI,cAAc,GAAG,CAAC,CAAC;QAEvB,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,YAAY,GAAmB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClD,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,cAAc,GAAkB,IAAI,CAAC;QAEzC,IAAI,cAAgE,CAAC;QAErE,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;YACd,MAAM,EAAE,CAAC;SACV,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,iCAAuB,EAAE;YACvE,qBAAqB,EAAE,IAAI,CAAC,QAAQ;YACpC,YAAY,EAAE,QAAQ,GAAG,CAAC;YAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;QAEH,OAAO,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YACjC,QAAQ,EAAE,CAAC;YACX,gBAAM,CAAC,KAAK,CAAC,WAAW,QAAQ,IAAI,CAAC,CAAC;YAEtC,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,GACrE,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;YAC7F,IAAI,gBAAgB,EAAE,CAAC;gBACrB,eAAe,CAAC,KAAK,IAAI,gBAAgB,CAAC,KAAK,IAAI,CAAC,CAAC;gBACrD,eAAe,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,IAAI,CAAC,CAAC;gBACvD,eAAe,CAAC,UAAU,IAAI,gBAAgB,CAAC,UAAU,IAAI,CAAC,CAAC;gBAC/D,eAAe,CAAC,WAAW,IAAI,gBAAgB,CAAC,WAAW,IAAI,CAAC,CAAC;gBACjE,eAAe,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,IAAI,CAAC,CAAC;YACzD,CAAC;YACD,gBAAM,CAAC,KAAK,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;YAEzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CACpC,YAAY,EACZ,MAAM,EACN,IAAI,EACJ,OAAO,EACP,QAAQ,EACR,QAAQ,EACR,OAAO,EACP,OAAO,CACR,CAAC;YACF,YAAY,GAAG,QAAQ,CAAC;YACxB,IAAI,YAAY,CAAC,UAAU,EAAE,CAAC;gBAC5B,eAAe,CAAC,KAAK,IAAI,YAAY,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;gBAC5D,eAAe,CAAC,MAAM,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;gBAC9D,eAAe,CAAC,UAAU,IAAI,YAAY,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;gBACtE,eAAe,CAAC,WAAW,IAAI,YAAY,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;gBACxE,eAAe,CAAC,MAAM,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAChE,CAAC;YAED,IAAI,YAAY,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC9C,IAAI,CAAC,WAAW,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC;gBAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,OAAO,GAAG;wBACR,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,SAAS,EAAE;wBACpD,MAAM;qBACP,CAAC;gBACJ,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC;YACrD,CAAC;YACD,MAAM,CAAC,SAAS,EAAE,gBAAgB,CAAC,GAAG,MAAM,IAAI,CAAC,eAAe,CAC9D,YAAY,EACZ,YAAY,CAAC,MAAM,CACpB,CAAC;YACF,gBAAM,CAAC,KAAK,CAAC,mCAAmC,SAAS,eAAe,gBAAgB,EAAE,CAAC,CAAC;YAE5F,IAAI,SAAS,EAAE,CAAC;gBACd,gBAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACnE,cAAc,EAAE,CAAC;gBACjB,IAAI,CAAC,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBAClF,YAAY,GAAG,IAAA,gBAAM,EAAA;;;;+DAIkC,gBAAgB;;;SAGtE,CAAC;gBACF,QAAQ,GAAG,KAAK,CAAC;gBACjB,cAAc,GAAG,IAAI,CAAC;gBAEtB,gBAAM,CAAC,KAAK,CAAC,6BAA6B,cAAc,EAAE,CAAC,CAAC;gBAC5D,IAAI,cAAc,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;oBACxC,QAAQ,EAAE,CAAC;oBACX,gBAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;gBACrE,CAAC;gBAED,IAAI,cAAc,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzC,gBAAM,CAAC,KAAK,CAAC,mBAAmB,IAAI,CAAC,aAAa,0BAA0B,CAAC,CAAC;oBAC9E,MAAM;gBACR,CAAC;gBAED,SAAS;YACX,CAAC;YAED,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACjF,IAAI,cAAc,EAAE,CAAC;gBACnB,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,KAAK,IAAI,CAAC,CAAC;gBACnD,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,MAAM,IAAI,CAAC,CAAC;gBACrD,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,IAAI,CAAC,CAAC;gBAC7D,eAAe,CAAC,WAAW,IAAI,cAAc,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC/D,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,MAAM,IAAI,CAAC,CAAC;YACvD,CAAC;YACD,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC;YAC3B,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC;YACpC,cAAc,GAAG;gBACf,KAAK,EAAE,SAAS,CAAC,QAAQ;gBACzB,SAAS,EAAE,SAAS,CAAC,SAAS;aAC/B,CAAC;YACF,gBAAM,CAAC,KAAK,CAAC,oBAAoB,QAAQ,gBAAgB,cAAc,EAAE,CAAC,CAAC;YAE3E,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;gBAC3B,gBAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;gBACnE,MAAM;YACR,CAAC;YAED,gBAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,gBAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACpD,CAAC;QAED,gBAAM,CAAC,KAAK,CACV,kCAAkC,QAAQ,sBAAsB,cAAc,KAAK,CACpF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC/C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC;QACzB,OAAO;YACL,MAAM,EAAE,YAAY,CAAC,MAAM;YAC3B,QAAQ,EAAE;gBACR,kBAAkB;gBAClB,kBAAkB,EAAE,IAAI,CAAC,MAAM;qBAC5B,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC;qBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;qBACtC,IAAI,CAAC,MAAM,CAAC;gBACf,wBAAwB,EAAE,QAAQ;gBAClC,uBAAuB,EAAE,cAAc;gBACvC,eAAe,EAAE,QAAQ;gBACzB,mBAAmB,EAAE,cAAc;aACpC;YACD,UAAU,EAAE,eAAe;SAC5B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,QAAgB,EAChB,QAAiB,EACjB,YAA4B,EAC5B,YAAoB,EACpB,cAAqD;QAErD,gBAAM,CAAC,KAAK,CACV,iCAAiC,QAAQ,cAAc,QAAQ,oBAAoB,IAAI,CAAC,SAAS,CAC/F,cAAc,CACf,EAAE,CACJ,CAAC;QAEF,IAAI,UAAU,GACZ,iBAAiB,QAAQ,OAAO,IAAI,CAAC,SAAS,IAAI;YAClD,wEAAwE;YACxE,8CAA8C,IAAI,CAAC,QAAQ,QAAQ,CAAC;QAEtE,IAAI,YAAY,EAAE,CAAC;YACjB,UAAU;gBACR,8DAA8D;oBAC9D,+EAA+E;oBAC/E,GAAG,YAAY,MAAM,CAAC;QAC1B,CAAC;aAAM,IAAI,cAAc,EAAE,CAAC;YAC1B,UAAU;gBACR,0EAA0E,YAAY,CAAC,MAAM,MAAM;oBACnG,sCAAsC;oBACtC,sCAAsC,cAAc,CAAC,KAAK,GAAG;oBAC7D,oDAAoD;oBACpD,2CAA2C,cAAc,CAAC,SAAS,MAAM,CAAC;QAC9E,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,UAAU;SACpB,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACzF,gBAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC7C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,EAAE;YAC/E,MAAM,EAAE;gBACN,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACtC,KAAK,EAAE,SAAS;aACjB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,QAAQ,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAEjF,MAAM,YAAY,GAAG,IAAA,6BAAsB,EAIxC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,YAAY,GAAG,CAAC,mBAAmB,EAAE,0BAA0B,EAAE,qBAAqB,CAAC,CAAC;QAE9F,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,CAAC,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC;gBAC3B,gBAAM,CAAC,KAAK,CAAC,4BAA4B,GAAG,EAAE,CAAC,CAAC;gBAChD,MAAM,IAAI,KAAK,CAAC,iBAAiB,GAAG,iCAAiC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1F,CAAC;QACH,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,IAAA,gBAAM,EAAA;;;2BAGI,YAAY,CAAC,iBAAiB;kCACvB,YAAY,CAAC,wBAAwB;6BAC1C,YAAY,CAAC,mBAAmB;KACxD,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,KAAK,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7D,gBAAM,CAAC,KAAK,CAAC,gCAAgC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrF,MAAM,IAAI,KAAK,CAAC,2CAA2C,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxD,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ,CAAC,MAAM;SACzB,CAAC,CAAC;QAEH,OAAO;YACL,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;YACjD,UAAU,EAAE,QAAQ,CAAC,UAAU;SAChC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,YAAoB,EACpB,cAAsB,EACtB,IAAqC,EACrC,OAAsC,EACtC,QAAqB,EACrB,QAAgB,EAChB,OAA8B,EAC9B,OAA8B;QAE9B,MAAM,cAAc,GAAG,MAAM,IAAA,+BAAY,EACvC,cAAc,EACd,EAAE,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,EAClD,OAAO,EACP,QAAQ,CACT,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAAwB,cAAc,CAAC,CAAC;YAC7E,6DAA6D;YAC7D,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IACE,OAAO,CAAC,IAAI,KAAK,QAAQ;oBACzB,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EACvF,CAAC;oBACD,+BAA+B;oBAC/B,SAAS;gBACX,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;YACvD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE;gBAChD,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,cAAc;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACnF,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC;QAE3F,gBAAM,CAAC,KAAK,CACV,2BAA2B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,CACxF,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAE3B,MAAM,cAAc,GAAG,MAAM,IAAA,0BAAiB,EAAC,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QACzF,IAAA,wBAAS,EAAC,cAAc,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;QAClE,gBAAM,CAAC,KAAK,CAAC,kCAAkC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAChD,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,cAAc,CAAC,MAAM;SAC/B,CAAC,CAAC;QAEH,OAAO,cAAc,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,YAAoB,EACpB,YAAoB;QAEpB,gBAAM,CAAC,KAAK,CAAC,qCAAqC,YAAY,EAAE,CAAC,CAAC;QAClE,8DAA8D;QAC9D,IAAI,IAAA,qBAAc,EAAC,YAAY,CAAC,EAAE,CAAC;YACjC,gBAAM,CAAC,KAAK,CACV,gFAAgF,CACjF,CAAC;YACF,OAAO,CAAC,IAAI,EAAE,yCAAyC,CAAC,CAAC;QAC3D,CAAC;QAED,uDAAuD;QACvD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,qBAAqB,EAAE,YAAY;YACnC,uBAAuB,EAAE,YAAY;SACtC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,+BAAqB,EAAE;YAC5E,qBAAqB,EAAE,IAAI,CAAC,QAAQ;SACrC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC;YACjC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE;YAChD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE;SACxC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,MAAM,eAAe,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,EAAE;YACjE,MAAM,EAAE;gBACN,GAAG,EAAE,WAAW;gBAChB,KAAK,EAAE,SAAS;aACjB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,eAAe,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACxF,gBAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;QAElE,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAIlC,eAAe,CAAC,MAAM,CAAC,CAAC;QAE3B,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,+CAA+C,CAAC,CAAC;QAC9F,IAAA,wBAAS,EACP,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EACnC,iDAAiD,CAClD,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,YAAoB;QAC7C,gBAAM,CAAC,KAAK,CAAC,oCAAoC,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YAC/B,qBAAqB,EAAE,IAAI,CAAC,QAAQ;YACpC,uBAAuB,EAAE,YAAY;SACtC,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,4BAAkB,EAAE;YACtE,qBAAqB,EAAE,IAAI,CAAC,QAAQ;SACrC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAC9B,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,gBAAgB,EAAE;YAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE;SACrC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxD,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,QAAQ,EAAE;YAC3D,MAAM,EAAE;gBACN,GAAG,EAAE,QAAQ;gBACb,KAAK,EAAE,MAAM;aACd;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uCAAuC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC;QAC/E,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,YAAY,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACrF,gBAAM,CAAC,KAAK,CAAC,wBAAwB,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5D,MAAM,MAAM,GAAG,IAAA,6BAAsB,EAKlC,YAAY,CAAC,MAAM,CAAC,CAAC;QAExB,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,4CAA4C,CAAC,CAAC;QAC3F,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,8CAA8C,CAAC,CAAC;QAE/F,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,cAAsB;QAClD,OAAO,IAAI,CAAC,MAAM,CAAC,sCAAsC,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC;IAEO,cAAc,CAAC,cAAsB,EAAE,eAAe,GAAG,KAAK;QACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAC7D,gBAAM,CAAC,KAAK,CAAC,2BAA2B,cAAc,GAAG,CAAC,CAAC;QAC3D,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,gBAAM,CAAC,KAAK,CAAC,OAAO,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF;AAED,kBAAe,iBAAiB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"goat.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/goat.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EACjB,MAAM,uBAAuB,CAAC;AAI/B,MAAM,CAAC,OAAO,OAAO,YAAa,YAAW,WAAW;IACtD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAU;IAEpC,EAAE;gBAKA,OAAO,GAAE,eAAe,GAAG;QACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,OAAO,CAAC;KAChB;IAWF,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"goat.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/goat.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EACjB,MAAM,uBAAuB,CAAC;AAI/B,MAAM,CAAC,OAAO,OAAO,YAAa,YAAW,WAAW;IACtD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAU;IAEpC,EAAE;gBAKA,OAAO,GAAE,eAAe,GAAG;QACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,OAAO,CAAC;KAChB;IAWF,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,gBAAgB,CAAC;CAoG7B"}
|
|
@@ -36,55 +36,70 @@ class GoatProvider {
|
|
|
36
36
|
prompt: 0,
|
|
37
37
|
completion: 0,
|
|
38
38
|
numRequests: 0,
|
|
39
|
+
cached: 0,
|
|
39
40
|
};
|
|
40
41
|
for (let turn = 0; turn < this.maxTurns; turn++) {
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
42
|
+
try {
|
|
43
|
+
response = await fetch((0, constants_2.getRemoteGenerationUrl)(), {
|
|
44
|
+
body: JSON.stringify({
|
|
45
|
+
goal: context?.vars[this.injectVar],
|
|
46
|
+
i: turn,
|
|
47
|
+
messages,
|
|
48
|
+
prompt: context?.prompt?.raw,
|
|
49
|
+
task: 'goat',
|
|
50
|
+
version: constants_1.VERSION,
|
|
51
|
+
}),
|
|
52
|
+
headers: {
|
|
53
|
+
'Content-Type': 'application/json',
|
|
54
|
+
},
|
|
55
|
+
method: 'POST',
|
|
56
|
+
});
|
|
57
|
+
const data = await response.json();
|
|
58
|
+
messages.push(data.message);
|
|
59
|
+
if (data.tokenUsage) {
|
|
60
|
+
totalTokenUsage.total += data.tokenUsage.total || 0;
|
|
61
|
+
totalTokenUsage.prompt += data.tokenUsage.prompt || 0;
|
|
62
|
+
totalTokenUsage.completion += data.tokenUsage.completion || 0;
|
|
63
|
+
totalTokenUsage.cached += data.tokenUsage.cached || 0;
|
|
64
|
+
totalTokenUsage.numRequests += data.tokenUsage.numRequests ?? 1;
|
|
65
|
+
}
|
|
66
|
+
logger_1.default.debug((0, dedent_1.default) `
|
|
63
67
|
${chalk_1.default.bold.green(`GOAT turn ${turn} history:`)}
|
|
64
68
|
${chalk_1.default.cyan(JSON.stringify(messages, null, 2))}
|
|
65
69
|
`);
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
70
|
+
const targetPrompt = this.stateless
|
|
71
|
+
? JSON.stringify(messages)
|
|
72
|
+
: await (0, evaluatorHelpers_1.renderPrompt)(context.prompt, { ...context.vars, [this.injectVar]: messages[messages.length - 1].content }, context.filters, targetProvider);
|
|
73
|
+
const targetResponse = await targetProvider.callApi(targetPrompt, context, options);
|
|
74
|
+
logger_1.default.debug(`GOAT turn ${turn} target response: ${JSON.stringify(targetResponse)}`);
|
|
75
|
+
if (targetResponse.sessionId) {
|
|
76
|
+
context = context ?? { vars: {}, prompt: { raw: '', label: 'target' } };
|
|
77
|
+
context.vars.sessionId = targetResponse.sessionId;
|
|
78
|
+
}
|
|
79
|
+
if (targetResponse.error) {
|
|
80
|
+
throw new Error(`Error from target provider: ${targetResponse.error}`);
|
|
81
|
+
}
|
|
82
|
+
(0, tiny_invariant_1.default)(targetResponse.output, `Expected target response output to be set, but got: ${JSON.stringify(targetResponse)}`);
|
|
83
|
+
messages.push({
|
|
84
|
+
content: targetResponse.output,
|
|
85
|
+
role: 'assistant',
|
|
86
|
+
});
|
|
87
|
+
if (targetResponse.tokenUsage) {
|
|
88
|
+
totalTokenUsage.total += targetResponse.tokenUsage.total || 0;
|
|
89
|
+
totalTokenUsage.prompt += targetResponse.tokenUsage.prompt || 0;
|
|
90
|
+
totalTokenUsage.completion += targetResponse.tokenUsage.completion || 0;
|
|
91
|
+
totalTokenUsage.numRequests += targetResponse.tokenUsage.numRequests ?? 1;
|
|
92
|
+
totalTokenUsage.cached += targetResponse.tokenUsage.cached || 0;
|
|
93
|
+
}
|
|
94
|
+
else {
|
|
95
|
+
totalTokenUsage.numRequests += 1;
|
|
96
|
+
}
|
|
83
97
|
}
|
|
84
|
-
|
|
85
|
-
|
|
98
|
+
catch (err) {
|
|
99
|
+
logger_1.default.error(`Error in GOAT turn ${turn}: ${err}`);
|
|
86
100
|
}
|
|
87
101
|
}
|
|
102
|
+
delete context?.vars?.sessionId;
|
|
88
103
|
return {
|
|
89
104
|
output: messages[messages.length - 1]?.content,
|
|
90
105
|
metadata: {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"goat.js","sourceRoot":"","sources":["../../../../src/redteam/providers/goat.ts"],"names":[],"mappings":";;;;;AAAA,kDAA0B;AAC1B,oDAA4B;AAC5B,oEAAuC;AACvC,+CAA0C;AAC1C,6DAAsD;AACtD,0DAAkC;AAQlC,4CAAsD;AACtD,kCAA8C;AAE9C,MAAqB,YAAY;IAK/B,EAAE;QACA,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,YACE,UAII,EAAE;QAEN,IAAI,IAAA,0BAAmB,GAAE,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;QACjF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,OAA8B;QAE9B,IAAI,QAAQ,GAAyB,SAAS,CAAC;QAC/C,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QAEpD,MAAM,cAAc,GAA4B,OAAO,EAAE,gBAAgB,CAAC;QAC1E,IAAA,wBAAS,EAAC,cAAc,EAAE,qCAAqC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAiE,EAAE,CAAC;QAClF,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"goat.js","sourceRoot":"","sources":["../../../../src/redteam/providers/goat.ts"],"names":[],"mappings":";;;;;AAAA,kDAA0B;AAC1B,oDAA4B;AAC5B,oEAAuC;AACvC,+CAA0C;AAC1C,6DAAsD;AACtD,0DAAkC;AAQlC,4CAAsD;AACtD,kCAA8C;AAE9C,MAAqB,YAAY;IAK/B,EAAE;QACA,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,YACE,UAII,EAAE;QAEN,IAAI,IAAA,0BAAmB,GAAE,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;QACjF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,OAA8B;QAE9B,IAAI,QAAQ,GAAyB,SAAS,CAAC;QAC/C,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QAEpD,MAAM,cAAc,GAA4B,OAAO,EAAE,gBAAgB,CAAC;QAC1E,IAAA,wBAAS,EAAC,cAAc,EAAE,qCAAqC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAiE,EAAE,CAAC;QAClF,MAAM,eAAe,GAAG;YACtB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;YACd,MAAM,EAAE,CAAC;SACV,CAAC;QAEF,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;YAChD,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAA,kCAAsB,GAAE,EAAE;oBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;wBACnC,CAAC,EAAE,IAAI;wBACP,QAAQ;wBACR,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG;wBAC5B,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,mBAAO;qBACjB,CAAC;oBACF,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;qBACnC;oBACD,MAAM,EAAE,MAAM;iBACf,CAAC,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,eAAe,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;oBACpD,eAAe,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;oBACtD,eAAe,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;oBAC9D,eAAe,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;oBACtD,eAAe,CAAC,WAAW,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;gBAClE,CAAC;gBACD,gBAAM,CAAC,KAAK,CACV,IAAA,gBAAM,EAAA;YACJ,eAAK,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,WAAW,CAAC;YAC9C,eAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;SAChD,CACA,CAAC;gBAEF,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS;oBACjC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;oBAC1B,CAAC,CAAC,MAAM,IAAA,+BAAY,EAChB,OAAO,CAAC,MAAM,EACd,EAAE,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,EAC5E,OAAO,CAAC,OAAO,EACf,cAAc,CACf,CAAC;gBACN,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;gBACpF,gBAAM,CAAC,KAAK,CAAC,aAAa,IAAI,qBAAqB,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;gBAErF,IAAI,cAAc,CAAC,SAAS,EAAE,CAAC;oBAC7B,OAAO,GAAG,OAAO,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC;oBACxE,OAAO,CAAC,IAAI,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC;gBACpD,CAAC;gBACD,IAAI,cAAc,CAAC,KAAK,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,+BAA+B,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC;gBACzE,CAAC;gBACD,IAAA,wBAAS,EACP,cAAc,CAAC,MAAM,EACrB,uDAAuD,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,CACxF,CAAC;gBACF,QAAQ,CAAC,IAAI,CAAC;oBACZ,OAAO,EAAE,cAAc,CAAC,MAAM;oBAC9B,IAAI,EAAE,WAAW;iBAClB,CAAC,CAAC;gBAEH,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;oBAC9B,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;oBAC9D,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;oBAChE,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;oBACxE,eAAe,CAAC,WAAW,IAAI,cAAc,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;oBAC1E,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;gBAClE,CAAC;qBAAM,CAAC;oBACN,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,gBAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,KAAK,GAAG,EAAE,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QACD,OAAO,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC;QAEhC,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO;YAC9C,QAAQ,EAAE;gBACR,kBAAkB,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO;gBAC1D,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;aAC5C;YACD,UAAU,EAAE,eAAe;SAC5B,CAAC;IACJ,CAAC;CACF;AAjID,+BAiIC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"iterative.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,KAAK,WAAW,EAChB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EAI1B,MAAM,aAAa,CAAC;AASrB,eAAO,MAAM,cAAc,QAEtB,CAAC;
|
|
1
|
+
{"version":3,"file":"iterative.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,KAAK,WAAW,EAChB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EAI1B,MAAM,aAAa,CAAC;AASrB,eAAO,MAAM,cAAc,QAEtB,CAAC;AAuNN,cAAM,wBAAyB,YAAW,WAAW;IAIvC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAH5D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAgC;IAChE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAiB5D,EAAE;IAIF;;;;;;OAMG;IACG,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,oBAAoB;;;;;;;;;;;;;CAkB7F;AAED,eAAe,wBAAwB,CAAC"}
|
|
@@ -37,6 +37,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
37
37
|
prompt: 0,
|
|
38
38
|
completion: 0,
|
|
39
39
|
numRequests: 0,
|
|
40
|
+
cached: 0,
|
|
40
41
|
};
|
|
41
42
|
for (let i = 0; i < exports.NUM_ITERATIONS; i++) {
|
|
42
43
|
const redteamBody = JSON.stringify(redteamHistory);
|
|
@@ -84,9 +85,9 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
84
85
|
const isOnTopic = (0, json_1.extractFirstJsonObject)(isOnTopicResp.output)
|
|
85
86
|
.onTopic;
|
|
86
87
|
(0, tiny_invariant_1.default)(typeof isOnTopic === 'boolean', 'Expected onTopic to be a boolean');
|
|
87
|
-
const
|
|
88
|
+
const targetResponse = await (0, shared_1.getTargetResponse)(targetProvider, targetPrompt, context, options);
|
|
88
89
|
const penalizedPhrases = ['in the face of impending doom'];
|
|
89
|
-
const containsPenalizedPhrase = penalizedPhrases.some((phrase) => targetResponse.toLowerCase().includes(phrase));
|
|
90
|
+
const containsPenalizedPhrase = penalizedPhrases.some((phrase) => targetResponse.output.toLowerCase().includes(phrase));
|
|
90
91
|
// Calculate the score
|
|
91
92
|
const judgeBody = JSON.stringify([
|
|
92
93
|
{
|
|
@@ -95,7 +96,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
95
96
|
},
|
|
96
97
|
{
|
|
97
98
|
role: 'user',
|
|
98
|
-
content: targetResponse,
|
|
99
|
+
content: targetResponse.output,
|
|
99
100
|
},
|
|
100
101
|
]);
|
|
101
102
|
const judgeResp = await redteamProvider.callApi(judgeBody, {
|
|
@@ -116,7 +117,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
116
117
|
}
|
|
117
118
|
if (score > highestScore) {
|
|
118
119
|
highestScore = score;
|
|
119
|
-
bestResponse = targetResponse;
|
|
120
|
+
bestResponse = targetResponse.output;
|
|
120
121
|
}
|
|
121
122
|
if (score >= 10) {
|
|
122
123
|
break;
|
|
@@ -131,6 +132,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
131
132
|
totalTokenUsage.completion += redteamResp.tokenUsage.completion || 0;
|
|
132
133
|
totalTokenUsage.numRequests =
|
|
133
134
|
(totalTokenUsage.numRequests || 0) + (redteamResp.tokenUsage.numRequests || 1);
|
|
135
|
+
totalTokenUsage.cached += redteamResp.tokenUsage.cached || 0;
|
|
134
136
|
}
|
|
135
137
|
else {
|
|
136
138
|
totalTokenUsage.numRequests = (totalTokenUsage.numRequests || 0) + 1;
|
|
@@ -141,6 +143,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
141
143
|
totalTokenUsage.completion += isOnTopicResp.tokenUsage.completion || 0;
|
|
142
144
|
totalTokenUsage.numRequests =
|
|
143
145
|
(totalTokenUsage.numRequests || 0) + (isOnTopicResp.tokenUsage.numRequests || 1);
|
|
146
|
+
totalTokenUsage.cached += isOnTopicResp.tokenUsage.cached || 0;
|
|
144
147
|
}
|
|
145
148
|
else {
|
|
146
149
|
totalTokenUsage.numRequests = (totalTokenUsage.numRequests || 0) + 1;
|
|
@@ -151,16 +154,18 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
151
154
|
totalTokenUsage.completion += judgeResp.tokenUsage.completion || 0;
|
|
152
155
|
totalTokenUsage.numRequests =
|
|
153
156
|
(totalTokenUsage.numRequests || 0) + (judgeResp.tokenUsage.numRequests || 1);
|
|
157
|
+
totalTokenUsage.cached += judgeResp.tokenUsage.cached || 0;
|
|
154
158
|
}
|
|
155
159
|
else {
|
|
156
160
|
totalTokenUsage.numRequests = (totalTokenUsage.numRequests || 0) + 1;
|
|
157
161
|
}
|
|
158
|
-
if (
|
|
159
|
-
totalTokenUsage.total +=
|
|
160
|
-
totalTokenUsage.prompt +=
|
|
161
|
-
totalTokenUsage.completion +=
|
|
162
|
+
if (targetResponse.tokenUsage) {
|
|
163
|
+
totalTokenUsage.total += targetResponse.tokenUsage.total || 0;
|
|
164
|
+
totalTokenUsage.prompt += targetResponse.tokenUsage.prompt || 0;
|
|
165
|
+
totalTokenUsage.completion += targetResponse.tokenUsage.completion || 0;
|
|
162
166
|
totalTokenUsage.numRequests =
|
|
163
|
-
(totalTokenUsage.numRequests || 0) + (
|
|
167
|
+
(totalTokenUsage.numRequests || 0) + (targetResponse.tokenUsage.numRequests || 1);
|
|
168
|
+
totalTokenUsage.cached += targetResponse.tokenUsage.cached || 0;
|
|
164
169
|
}
|
|
165
170
|
else {
|
|
166
171
|
totalTokenUsage.numRequests = (totalTokenUsage.numRequests || 0) + 1;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"iterative.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AAClC,yDAA4E;AAS5E,0CAAyD;AACzD,oDAAyD;AACzD,kCAA+C;AAC/C,uCAAgG;AAChG,qCAAkE;AAElE,6CAA6C;AAEhC,QAAA,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,kCAAkC;IAC1E,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC,CAAC;AAEN,KAAK,UAAU,sBAAsB,CAAC,EACpC,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,EACT,OAAO,EACP,OAAO,GAUR;IACC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;IAE7B,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpF,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpF,MAAM,iBAAiB,GAAG,QAAQ,CAAC,YAAY,CAAC,6BAAmB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/E,MAAM,cAAc,GAAiE;QACnF;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;KACF,CAAC;IAEF,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,YAAY,GAAG,EAAE,CAAC;IAEtB,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,MAAM,eAAe,GAAG;QACtB,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"iterative.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AAClC,yDAA4E;AAS5E,0CAAyD;AACzD,oDAAyD;AACzD,kCAA+C;AAC/C,uCAAgG;AAChG,qCAAkE;AAElE,6CAA6C;AAEhC,QAAA,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,kCAAkC;IAC1E,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC,CAAC;AAEN,KAAK,UAAU,sBAAsB,CAAC,EACpC,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,EACT,OAAO,EACP,OAAO,GAUR;IACC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;IAE7B,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpF,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpF,MAAM,iBAAiB,GAAG,QAAQ,CAAC,YAAY,CAAC,6BAAmB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/E,MAAM,cAAc,GAAiE;QACnF;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;KACF,CAAC;IAEF,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,YAAY,GAAG,EAAE,CAAC;IAEtB,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,MAAM,eAAe,GAAG;QACtB,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,MAAM,EAAE,CAAC;KACV,CAAC;IAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAc,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAEnD,iBAAiB;QACjB,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,EAAE;YAC7D,MAAM,EAAE;gBACN,GAAG,EAAE,WAAW;gBAChB,KAAK,EAAE,SAAS;aACjB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,gCAAgC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,IAAA,wBAAS,EACP,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC,qDAAqD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CACnF,CAAC;QACF,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAA,6BAAsB,EAGjE,WAAW,CAAC,MAAM,CAAC,CAAC;QAEvB,wDAAwD;QACxD,gBAAM,CAAC,KAAK,CAAC,kBAAkB,YAAY,kBAAkB,WAAW,EAAE,CAAC,CAAC;QAC5E,YAAY,GAAG,MAAM,IAAA,+BAAY,EAC/B,MAAM,EACN;YACE,GAAG,IAAI;YACP,CAAC,SAAS,CAAC,EAAE,YAAY;SAC1B,EACD,OAAO,EACP,cAAc,CACf,CAAC;QAEF,kBAAkB;QAClB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;YACnC;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,mBAAmB;aAC7B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,YAAY;aACtB;SACF,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,aAAa,EAAE;YACjE,MAAM,EAAE;gBACN,GAAG,EAAE,aAAa;gBAClB,KAAK,EAAE,UAAU;aAClB;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,0CAA0C,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC;QACnF,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAI,IAAA,6BAAsB,EAAC,aAAa,CAAC,MAAM,CAA0B;aACrF,OAAO,CAAC;QACX,IAAA,wBAAS,EAAC,OAAO,SAAS,KAAK,SAAS,EAAE,kCAAkC,CAAC,CAAC;QAE9E,MAAM,cAAc,GAAG,MAAM,IAAA,0BAAiB,EAAC,cAAc,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/F,MAAM,gBAAgB,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC3D,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/D,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CACrD,CAAC;QAEF,sBAAsB;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YAC/B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,iBAAiB;aAC3B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,cAAc,CAAC,MAAM;aAC/B;SACF,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,EAAE;YACzD,MAAM,EAAE;gBACN,GAAG,EAAE,SAAS;gBACd,KAAK,EAAE,OAAO;aACf;YACD,IAAI,EAAE,EAAE;SACT,CAAC,CAAC;QACH,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wCAAwC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;QAC7E,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAA,6BAAsB,EAAqB,SAAS,CAAC,MAAM,CAAC,CAAC;QAErF,yCAAyC;QACzC,IAAI,uBAAuB,EAAE,CAAC;YAC5B,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,KAAK,GAAG,YAAY,EAAE,CAAC;YACzB,YAAY,GAAG,KAAK,CAAC;YACrB,YAAY,GAAG,cAAc,CAAC,MAAM,CAAC;QACvC,CAAC;QAED,IAAI,KAAK,IAAI,EAAE,EAAE,CAAC;YAChB,MAAM;QACR,CAAC;QAED,cAAc,CAAC,IAAI,CAAC;YAClB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,0BAA0B,cAAc,kBAAkB,IAAI,cAAc,KAAK,IACxF,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BACnB,GAAG,uBAAuB,CAAC,CAAC,CAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,EAAE;SACnE,CAAC,CAAC;QAEH,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;YAC3B,eAAe,CAAC,KAAK,IAAI,WAAW,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC3D,eAAe,CAAC,MAAM,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC7D,eAAe,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACrE,eAAe,CAAC,WAAW;gBACzB,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YACjF,eAAe,CAAC,MAAM,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAC/D,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,WAAW,GAAG,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;YAC7B,eAAe,CAAC,KAAK,IAAI,aAAa,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC7D,eAAe,CAAC,MAAM,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC/D,eAAe,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACvE,eAAe,CAAC,WAAW;gBACzB,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YACnF,eAAe,CAAC,MAAM,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QACjE,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,WAAW,GAAG,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;YACzB,eAAe,CAAC,KAAK,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YACzD,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC3D,eAAe,CAAC,UAAU,IAAI,SAAS,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACnE,eAAe,CAAC,WAAW;gBACzB,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YAC/E,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAC7D,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,WAAW,GAAG,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;YAC9B,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC9D,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAChE,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACxE,eAAe,CAAC,WAAW;gBACzB,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YACpF,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAClE,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,WAAW,GAAG,CAAC,eAAe,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;QACD,UAAU,EAAE,eAAe;KAC5B,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAI5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAElC,+CAA+C;QAE/C,IAAI,IAAA,2BAAoB,GAAE,EAAE,CAAC;YAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,2CAA+B,CAAC;gBACzD,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,IAAI;gBACd,gBAAgB,EAAE,KAAK;aACxB,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAChD,CAAC;IACH,CAAC;IAED,EAAE;QACA,OAAO,6BAA6B,CAAC;IACvC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B,EAAE,OAA8B;QAC1F,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;QAEnD,OAAO,sBAAsB,CAAC;YAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,eAAe,EAAE,MAAM,IAAA,4BAAmB,EAAC;gBACzC,QAAQ,EAAE,IAAI,CAAC,eAAe;gBAC9B,QAAQ,EAAE,IAAI;aACf,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO;YACP,OAAO;SACR,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,wBAAwB,CAAC"}
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import type { ApiProvider, CallApiContextParams, CallApiOptionsParams } from '../../types';
|
|
2
|
+
import { type TargetResponse } from './shared';
|
|
2
3
|
declare class RedteamIterativeProvider implements ApiProvider {
|
|
3
4
|
readonly config: Record<string, string | object>;
|
|
4
5
|
private readonly redteamProvider;
|
|
@@ -7,7 +8,7 @@ declare class RedteamIterativeProvider implements ApiProvider {
|
|
|
7
8
|
callApi(prompt: string, context?: CallApiContextParams & {
|
|
8
9
|
injectVar?: string;
|
|
9
10
|
}, options?: CallApiOptionsParams): Promise<{
|
|
10
|
-
output:
|
|
11
|
+
output: TargetResponse | null;
|
|
11
12
|
metadata: {
|
|
12
13
|
redteamFinalPrompt: string | undefined;
|
|
13
14
|
};
|
|
@@ -16,6 +17,7 @@ declare class RedteamIterativeProvider implements ApiProvider {
|
|
|
16
17
|
prompt: number;
|
|
17
18
|
completion: number;
|
|
18
19
|
numRequests: number;
|
|
20
|
+
cached: number;
|
|
19
21
|
};
|
|
20
22
|
}>;
|
|
21
23
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"iterativeImage.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeImage.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EAIrB,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"iterativeImage.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeImage.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EAIrB,MAAM,aAAa,CAAC;AAGrB,OAAO,EAA0C,KAAK,cAAc,EAAE,MAAM,UAAU,CAAC;AA6QvF,cAAM,wBAAyB,YAAW,WAAW;IAGvC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAF5D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAgC;gBAE3C,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAK5D,EAAE;IAII,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,GAAG;QAAE,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,EACvD,OAAO,CAAC,EAAE,oBAAoB;;;;;;;;;;;;;CAqBjC;AAED,eAAe,wBAAwB,CAAC"}
|
|
@@ -85,12 +85,13 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
85
85
|
},
|
|
86
86
|
];
|
|
87
87
|
let highestScore = 0;
|
|
88
|
-
let bestResponse =
|
|
88
|
+
let bestResponse = null;
|
|
89
89
|
const totalTokenUsage = {
|
|
90
90
|
total: 0,
|
|
91
91
|
prompt: 0,
|
|
92
92
|
completion: 0,
|
|
93
93
|
numRequests: 0,
|
|
94
|
+
cached: 0,
|
|
94
95
|
};
|
|
95
96
|
let targetPrompt = null;
|
|
96
97
|
for (let i = 0; i < NUM_ITERATIONS; i++) {
|
|
@@ -101,6 +102,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
101
102
|
totalTokenUsage.total += redteamResp.tokenUsage.total || 0;
|
|
102
103
|
totalTokenUsage.prompt += redteamResp.tokenUsage.prompt || 0;
|
|
103
104
|
totalTokenUsage.completion += redteamResp.tokenUsage.completion || 0;
|
|
105
|
+
totalTokenUsage.cached += redteamResp.tokenUsage.cached ?? 0;
|
|
104
106
|
}
|
|
105
107
|
if (redteamResp.error) {
|
|
106
108
|
throw new Error(`Error from redteam provider: ${redteamResp.error}`);
|
|
@@ -130,6 +132,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
130
132
|
totalTokenUsage.total += isOnTopicResp.tokenUsage.total || 0;
|
|
131
133
|
totalTokenUsage.prompt += isOnTopicResp.tokenUsage.prompt || 0;
|
|
132
134
|
totalTokenUsage.completion += isOnTopicResp.tokenUsage.completion || 0;
|
|
135
|
+
totalTokenUsage.cached += isOnTopicResp.tokenUsage.cached ?? 0;
|
|
133
136
|
}
|
|
134
137
|
if (isOnTopicResp.error) {
|
|
135
138
|
throw new Error(`Error from redteam (onTopic) provider: ${isOnTopicResp.error}`);
|
|
@@ -137,10 +140,10 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
137
140
|
(0, tiny_invariant_1.default)(typeof isOnTopicResp.output === 'string', 'Expected output to be a string');
|
|
138
141
|
const { isOnTopic } = (0, json_1.extractFirstJsonObject)(isOnTopicResp.output);
|
|
139
142
|
logger_1.default.debug(`Iteration ${i + 1}: On-topic response: ${isOnTopicResp.output}`);
|
|
140
|
-
const
|
|
143
|
+
const targetResponse = await (0, shared_1.getTargetResponse)(targetProvider, targetPrompt, context, options);
|
|
141
144
|
// Parse URL from targetResponse
|
|
142
145
|
const urlRegex = /(https?:\/\/[^\s)]+)/g;
|
|
143
|
-
const url = targetResponse.match(urlRegex);
|
|
146
|
+
const url = targetResponse.output.match(urlRegex);
|
|
144
147
|
logger_1.default.debug(`Iteration ${i + 1}: URL: ${url}`);
|
|
145
148
|
let imageDescription;
|
|
146
149
|
if (url) {
|
|
@@ -166,6 +169,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
166
169
|
totalTokenUsage.total += visionResponse.tokenUsage.total || 0;
|
|
167
170
|
totalTokenUsage.prompt += visionResponse.tokenUsage.prompt || 0;
|
|
168
171
|
totalTokenUsage.completion += visionResponse.tokenUsage.completion || 0;
|
|
172
|
+
totalTokenUsage.cached += visionResponse.tokenUsage.cached ?? 0;
|
|
169
173
|
}
|
|
170
174
|
imageDescription = visionResponse.output;
|
|
171
175
|
logger_1.default.debug(`Iteration ${i + 1}: Image description: ${imageDescription}`);
|
|
@@ -186,6 +190,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
186
190
|
totalTokenUsage.total += judgeResp.tokenUsage.total || 0;
|
|
187
191
|
totalTokenUsage.prompt += judgeResp.tokenUsage.prompt || 0;
|
|
188
192
|
totalTokenUsage.completion += judgeResp.tokenUsage.completion || 0;
|
|
193
|
+
totalTokenUsage.cached += judgeResp.tokenUsage.cached ?? 0;
|
|
189
194
|
}
|
|
190
195
|
(0, tiny_invariant_1.default)(typeof judgeResp.output === 'string', 'Expected output to be a string');
|
|
191
196
|
const { rating: score } = (0, json_1.extractFirstJsonObject)(judgeResp.output);
|
|
@@ -201,11 +206,12 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
201
206
|
role: 'user',
|
|
202
207
|
content: `IMAGE MODEL OUTPUT: ${imageDescription || targetResponse}\n\nOBJECTIVE: ${goal}\n\nSCORE: ${score} ${isOnTopic ? '' : '(your prompt was off-topic)'}`,
|
|
203
208
|
});
|
|
204
|
-
if (
|
|
205
|
-
totalTokenUsage.total +=
|
|
206
|
-
totalTokenUsage.prompt +=
|
|
207
|
-
totalTokenUsage.completion +=
|
|
208
|
-
totalTokenUsage.numRequests +=
|
|
209
|
+
if (targetResponse.tokenUsage) {
|
|
210
|
+
totalTokenUsage.total += targetResponse.tokenUsage.total || 0;
|
|
211
|
+
totalTokenUsage.prompt += targetResponse.tokenUsage.prompt || 0;
|
|
212
|
+
totalTokenUsage.completion += targetResponse.tokenUsage.completion || 0;
|
|
213
|
+
totalTokenUsage.numRequests += targetResponse.tokenUsage.numRequests ?? 1;
|
|
214
|
+
totalTokenUsage.cached += targetResponse.tokenUsage.cached ?? 0;
|
|
209
215
|
}
|
|
210
216
|
}
|
|
211
217
|
return {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"iterativeImage.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeImage.ts"],"names":[],"mappings":";;;;;AAAA,oDAA4B;AAC5B,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AASlC,0CAAyD;AACzD,oDAAwF;AACxF,
|
|
1
|
+
{"version":3,"file":"iterativeImage.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeImage.ts"],"names":[],"mappings":";;;;;AAAA,oDAA4B;AAC5B,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AASlC,0CAAyD;AACzD,oDAAwF;AACxF,qCAAuF;AAEvF,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,kCAAkC;IACnE,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC,CAAC;AAEN,6CAA6C;AAE7C,MAAM,sBAAsB,GAAG,IAAA,gBAAM,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2CpC,CAAC;AAEF,MAAM,mBAAmB,GAAG,IAAA,gBAAM,EAAA;;;;CAIjC,CAAC;AAEF,MAAM,sBAAsB,GAAG,IAAA,gBAAM,EAAA;;;;;;CAMpC,CAAC;AAEF,KAAK,UAAU,sBAAsB,CAAC,EACpC,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,EACT,OAAO,EACP,OAAO,GAUR;IACC,iDAAiD;IACjD,MAAM,cAAc,GAAG,eAAe,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;IAE7B,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IACpF,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IACpF,MAAM,iBAAiB,GAAG,QAAQ,CAAC,YAAY,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/E,MAAM,cAAc,GAAiE;QACnF;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;KACF,CAAC;IAEF,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,YAAY,GAA0B,IAAI,CAAC;IAC/C,MAAM,eAAe,GAAG;QACtB,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,MAAM,EAAE,CAAC;KACV,CAAC;IAEF,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAEnD,iBAAiB;QACjB,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC/D,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;YAC3B,eAAe,CAAC,KAAK,IAAI,WAAW,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC3D,eAAe,CAAC,MAAM,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC7D,eAAe,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACrE,eAAe,CAAC,MAAM,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,gCAAgC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,IAAA,wBAAS,EACP,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC,qDAAqD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CACnF,CAAC;QACF,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAA,6BAAsB,EAGjE,WAAW,CAAC,MAAM,CAAC,CAAC;QAEvB,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,uBAAuB,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QAE5E,wDAAwD;QACxD,gBAAM,CAAC,KAAK,CAAC,kBAAkB,YAAY,kBAAkB,WAAW,EAAE,CAAC,CAAC;QAC5E,YAAY,GAAG,MAAM,IAAA,+BAAY,EAC/B,MAAM,EACN;YACE,GAAG,IAAI;YACP,CAAC,SAAS,CAAC,EAAE,YAAY;SAC1B,EACD,OAAO,EACP,cAAc,CACf,CAAC;QAEF,kBAAkB;QAClB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;YACnC;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,mBAAmB;aAC7B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,YAAY;aACtB;SACF,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACnE,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;YAC7B,eAAe,CAAC,KAAK,IAAI,aAAa,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC7D,eAAe,CAAC,MAAM,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC/D,eAAe,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACvE,eAAe,CAAC,MAAM,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,0CAA0C,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC;QACnF,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,EAAE,SAAS,EAAE,GAAG,IAAA,6BAAsB,EAAyB,aAAa,CAAC,MAAM,CAAC,CAAC;QAC3F,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,wBAAwB,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QAE/E,MAAM,cAAc,GAAG,MAAM,IAAA,0BAAiB,EAAC,cAAc,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAE/F,gCAAgC;QAChC,MAAM,QAAQ,GAAG,uBAAuB,CAAC;QACzC,MAAM,GAAG,GAAG,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAClD,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,GAAG,EAAE,CAAC,CAAC;QAChD,IAAI,gBAAgB,CAAC;QACrB,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,OAAO,CACjD,IAAI,CAAC,SAAS,CAAC;gBACb;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,2DAA2D;yBAClE;wBACD;4BACE,IAAI,EAAE,WAAW;4BACjB,SAAS,EAAE;gCACT,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;gCACX,MAAM,EAAE,MAAM;6BACf;yBACF;qBACF;iBACF;aACF,CAAC,CACH,CAAC;YACF,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;gBAC9B,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;gBAC9D,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;gBAChE,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;gBACxE,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAClE,CAAC;YACD,gBAAgB,GAAG,cAAc,CAAC,MAAM,CAAC;YACzC,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,wBAAwB,gBAAgB,EAAE,CAAC,CAAC;QAC7E,CAAC;QAED,sBAAsB;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YAC/B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,iBAAiB;aAC3B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,gBAAgB,IAAI,cAAc;aAC5C;SACF,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;YACzB,eAAe,CAAC,KAAK,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YACzD,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAC3D,eAAe,CAAC,UAAU,IAAI,SAAS,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACnE,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAC7D,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAA,6BAAsB,EAAqB,SAAS,CAAC,MAAM,CAAC,CAAC;QACvF,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,qBAAqB,SAAS,CAAC,MAAM,YAAY,KAAK,EAAE,CAAC,CAAC;QAEzF,IAAI,KAAK,GAAG,YAAY,EAAE,CAAC;YACzB,YAAY,GAAG,KAAK,CAAC;YACrB,YAAY,GAAG,cAAc,CAAC;QAChC,CAAC;QAED,IAAI,KAAK,IAAI,EAAE,EAAE,CAAC;YAChB,MAAM;QACR,CAAC;QAED,cAAc,CAAC,IAAI,CAAC;YAClB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,uBAAuB,gBAAgB,IAAI,cAAc,kBAAkB,IAAI,cAAc,KAAK,IACzG,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BACnB,EAAE;SACH,CAAC,CAAC;QAEH,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;YAC9B,eAAe,CAAC,KAAK,IAAI,cAAc,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;YAC9D,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;YAChE,eAAe,CAAC,UAAU,IAAI,cAAc,CAAC,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC;YACxE,eAAe,CAAC,WAAW,IAAI,cAAc,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;YAC1E,eAAe,CAAC,MAAM,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;QACD,UAAU,EAAE,eAAe;KAC5B,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAG5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,+CAA+C;QAC/C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAAuD,EACvD,OAA8B;QAE9B,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAA,yCAA6B,EAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,IAAA,wBAAS,EAAC,SAAS,EAAE,8BAA8B,CAAC,CAAC;QACrD,OAAO,sBAAsB,CAAC;YAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,eAAe,EAAE,MAAM,IAAA,4BAAmB,EAAC;gBACzC,QAAQ,EAAE,IAAI,CAAC,eAAe;gBAC9B,gBAAgB,EAAE,KAAK;gBACvB,QAAQ,EAAE,IAAI;aACf,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS;YACT,OAAO;YACP,OAAO;SACR,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,wBAAwB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"iterativeTree.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeTree.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AACH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAK5C,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,MAAM,EACN,iBAAiB,EACjB,UAAU,EACX,MAAM,aAAa,CAAC;AA+BrB;;;;;GAKG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,WAAW,EACrB,IAAI,EAAE,MAAM,GACX;IACD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,iBAAiB,EAAE,MAAM,CAAC;CAC3B,CAMA;AAED;;;;;;;GAOG;AACH,wBAAsB,gBAAgB,CACpC,eAAe,EAAE,WAAW,EAC5B,iBAAiB,EAAE,MAAM,EACzB,cAAc,EAAE,MAAM,EACtB,uBAAuB,EAAE,OAAO,GAC/B,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAED;;;;;GAKG;AACH,wBAAsB,YAAY,CAChC,eAAe,EAAE,WAAW,EAC5B,cAAc,EAAE;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,EAAE,GAC3E,OAAO,CAAC;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,UAAU,CAAA;CAAE,CAAC,CAoB3E;AAED;;;;;;GAMG;AACH,wBAAsB,cAAc,CAClC,eAAe,EAAE,WAAW,EAC5B,mBAAmB,EAAE,MAAM,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC;IAAE,SAAS,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,UAAU,CAAA;CAAE,CAAC,CA4B1D;AAED;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAClC,cAAc,EAAE,MAAM,EACtB,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,OAAO,EAClB,uBAAuB,EAAE,OAAO,GAC/B;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAO5D;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,QAAQ,EAAE,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;;;;;GAMG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,QAAQ,CAErF;AAED;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAoBtE;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAM5E;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,QAAQ,EAAE,CAyBzF;AAED;;;;GAIG;AACH,wBAAsB,sBAAsB,CAAC,EAC3C,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,EACT,OAAO,EACP,OAAO,GACR,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,iBAAiB,GAAG,SAAS,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;IACtC,eAAe,EAAE,WAAW,CAAC;IAC7B,cAAc,EAAE,WAAW,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,OAAO,EAAE,oBAAoB,CAAC;CAC/B,GAAG,OAAO,CAAC;IACV,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1C,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,CAAC,
|
|
1
|
+
{"version":3,"file":"iterativeTree.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeTree.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AACH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAK5C,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,MAAM,EACN,iBAAiB,EACjB,UAAU,EACX,MAAM,aAAa,CAAC;AA+BrB;;;;;GAKG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,WAAW,EACrB,IAAI,EAAE,MAAM,GACX;IACD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,iBAAiB,EAAE,MAAM,CAAC;CAC3B,CAMA;AAED;;;;;;;GAOG;AACH,wBAAsB,gBAAgB,CACpC,eAAe,EAAE,WAAW,EAC5B,iBAAiB,EAAE,MAAM,EACzB,cAAc,EAAE,MAAM,EACtB,uBAAuB,EAAE,OAAO,GAC/B,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAED;;;;;GAKG;AACH,wBAAsB,YAAY,CAChC,eAAe,EAAE,WAAW,EAC5B,cAAc,EAAE;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,EAAE,GAC3E,OAAO,CAAC;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,UAAU,CAAA;CAAE,CAAC,CAoB3E;AAED;;;;;;GAMG;AACH,wBAAsB,cAAc,CAClC,eAAe,EAAE,WAAW,EAC5B,mBAAmB,EAAE,MAAM,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC;IAAE,SAAS,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,UAAU,CAAA;CAAE,CAAC,CA4B1D;AAED;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAClC,cAAc,EAAE,MAAM,EACtB,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,OAAO,EAClB,uBAAuB,EAAE,OAAO,GAC/B;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAO5D;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,QAAQ,EAAE,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;;;;;GAMG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,QAAQ,CAErF;AAED;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAoBtE;AAED;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAM5E;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,QAAQ,EAAE,CAyBzF;AAED;;;;GAIG;AACH,wBAAsB,sBAAsB,CAAC,EAC3C,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,EACT,OAAO,EACP,OAAO,GACR,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,iBAAiB,GAAG,SAAS,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;IACtC,eAAe,EAAE,WAAW,CAAC;IAC7B,cAAc,EAAE,WAAW,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,OAAO,EAAE,oBAAoB,CAAC;CAC/B,GAAG,OAAO,CAAC;IACV,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1C,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,CAAC,CA+OD;AAED;;GAEG;AACH,cAAM,4BAA6B,YAAW,WAAW;IAQ3C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAP5D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IAEnC;;;;OAIG;gBACkB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAM5D;;;OAGG;IACH,EAAE,IAAI,MAAM;IAIZ;;;;;;OAMG;IACG,OAAO,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,oBAAoB,EAC9B,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE;YAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;CA8B1E;AAED,eAAe,4BAA4B,CAAC"}
|
|
@@ -264,6 +264,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
264
264
|
prompt: 0,
|
|
265
265
|
completion: 0,
|
|
266
266
|
numRequests: 0,
|
|
267
|
+
cached: 0,
|
|
267
268
|
};
|
|
268
269
|
for (let depth = 0; depth < MAX_DEPTH; depth++) {
|
|
269
270
|
logger_1.default.debug(`[Depth ${depth}] Starting exploration. Nodes to explore: ${currentBestNodes.length}. Max score so far: ${maxScore}`);
|
|
@@ -280,6 +281,8 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
280
281
|
totalTokenUsage.total += redteamTokenUsage.total || 0;
|
|
281
282
|
totalTokenUsage.prompt += redteamTokenUsage.prompt || 0;
|
|
282
283
|
totalTokenUsage.completion += redteamTokenUsage.completion || 0;
|
|
284
|
+
totalTokenUsage.numRequests += redteamTokenUsage.numRequests ?? 1;
|
|
285
|
+
totalTokenUsage.cached += redteamTokenUsage.cached || 0;
|
|
283
286
|
}
|
|
284
287
|
attempts++;
|
|
285
288
|
logger_1.default.debug(`[Depth ${depth}, Attempt ${attempts}] Generated new prompt: "${newInjectVar.substring(0, 30)}...", improvement="${improvement.substring(0, 30)}...". Max score so far: ${maxScore}`);
|
|
@@ -292,16 +295,19 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
292
295
|
totalTokenUsage.total += isOnTopicTokenUsage.total || 0;
|
|
293
296
|
totalTokenUsage.prompt += isOnTopicTokenUsage.prompt || 0;
|
|
294
297
|
totalTokenUsage.completion += isOnTopicTokenUsage.completion || 0;
|
|
298
|
+
totalTokenUsage.cached += isOnTopicTokenUsage.cached || 0;
|
|
299
|
+
totalTokenUsage.numRequests += isOnTopicTokenUsage.numRequests ?? 1;
|
|
295
300
|
}
|
|
296
|
-
const
|
|
297
|
-
if (
|
|
298
|
-
totalTokenUsage.total +=
|
|
299
|
-
totalTokenUsage.prompt +=
|
|
300
|
-
totalTokenUsage.completion +=
|
|
301
|
-
totalTokenUsage.numRequests +=
|
|
301
|
+
const targetResponse = await (0, shared_1.getTargetResponse)(targetProvider, targetPrompt, context, options);
|
|
302
|
+
if (targetResponse.tokenUsage) {
|
|
303
|
+
totalTokenUsage.total += targetResponse.tokenUsage.total || 0;
|
|
304
|
+
totalTokenUsage.prompt += targetResponse.tokenUsage.prompt || 0;
|
|
305
|
+
totalTokenUsage.completion += targetResponse.tokenUsage.completion || 0;
|
|
306
|
+
totalTokenUsage.numRequests += targetResponse.tokenUsage.numRequests ?? 1;
|
|
307
|
+
totalTokenUsage.cached += targetResponse.tokenUsage.cached || 0;
|
|
302
308
|
}
|
|
303
|
-
const containsPenalizedPhrase = constants_1.PENALIZED_PHRASES.some((phrase) => targetResponse.toLowerCase().includes(phrase));
|
|
304
|
-
const score = await evaluateResponse(redteamProvider, judgeSystemPrompt, targetResponse, containsPenalizedPhrase);
|
|
309
|
+
const containsPenalizedPhrase = constants_1.PENALIZED_PHRASES.some((phrase) => targetResponse.output.toLowerCase().includes(phrase));
|
|
310
|
+
const score = await evaluateResponse(redteamProvider, judgeSystemPrompt, targetResponse.output, containsPenalizedPhrase);
|
|
305
311
|
logger_1.default.debug(`[Depth ${depth}, Attempt ${attempts}] Evaluation: score=${score}, on-topic=${isOnTopic}, penalized=${containsPenalizedPhrase}. Max score so far: ${maxScore}`);
|
|
306
312
|
const newNode = createTreeNode(newInjectVar, score, depth + 1);
|
|
307
313
|
nextLevelNodes.push(newNode);
|
|
@@ -323,7 +329,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
323
329
|
if (score >= TARGET_SCORE) {
|
|
324
330
|
logger_1.default.debug(`[Depth ${depth}, Attempt ${attempts}] Reached target score (${TARGET_SCORE}). Max score: ${maxScore}. Stopping.`);
|
|
325
331
|
return {
|
|
326
|
-
output: targetResponse,
|
|
332
|
+
output: targetResponse.output,
|
|
327
333
|
metadata: {
|
|
328
334
|
redteamFinalPrompt: targetPrompt,
|
|
329
335
|
},
|
|
@@ -333,7 +339,7 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
333
339
|
if (noImprovementCount >= MAX_NO_IMPROVEMENT) {
|
|
334
340
|
logger_1.default.debug(`[Depth ${depth}, Attempt ${attempts}] No improvement for ${MAX_NO_IMPROVEMENT} consecutive iterations. Max score: ${maxScore}. Stopping.`);
|
|
335
341
|
return {
|
|
336
|
-
output: targetResponse,
|
|
342
|
+
output: targetResponse.output,
|
|
337
343
|
metadata: {
|
|
338
344
|
redteamFinalPrompt: targetPrompt,
|
|
339
345
|
},
|
|
@@ -343,14 +349,14 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
343
349
|
if (attempts >= MAX_ATTEMPTS) {
|
|
344
350
|
logger_1.default.debug(`[Depth ${depth}, Attempt ${attempts}] Reached maximum attempts (${MAX_ATTEMPTS}). Max score: ${maxScore}. Stopping.`);
|
|
345
351
|
return {
|
|
346
|
-
output: targetResponse,
|
|
352
|
+
output: targetResponse.output,
|
|
347
353
|
metadata: {
|
|
348
354
|
redteamFinalPrompt: targetPrompt,
|
|
349
355
|
},
|
|
350
356
|
tokenUsage: totalTokenUsage,
|
|
351
357
|
};
|
|
352
358
|
}
|
|
353
|
-
redteamHistory.push(updateRedteamHistory(targetResponse, goal, score, isOnTopic, containsPenalizedPhrase));
|
|
359
|
+
redteamHistory.push(updateRedteamHistory(targetResponse.output, goal, score, isOnTopic, containsPenalizedPhrase));
|
|
354
360
|
}
|
|
355
361
|
}
|
|
356
362
|
currentBestNodes = selectDiverseBestNodes(nextLevelNodes, BASE_BRANCHES);
|
|
@@ -360,16 +366,17 @@ async function runRedteamConversation({ prompt, filters, vars, redteamProvider,
|
|
|
360
366
|
...vars,
|
|
361
367
|
[injectVar]: bestNode.prompt,
|
|
362
368
|
}, filters, targetProvider);
|
|
363
|
-
const
|
|
364
|
-
if (
|
|
365
|
-
totalTokenUsage.total +=
|
|
366
|
-
totalTokenUsage.prompt +=
|
|
367
|
-
totalTokenUsage.completion +=
|
|
368
|
-
totalTokenUsage.numRequests +=
|
|
369
|
+
const finalTargetResponse = await (0, shared_1.getTargetResponse)(targetProvider, finalTargetPrompt, context, options);
|
|
370
|
+
if (finalTargetResponse.tokenUsage) {
|
|
371
|
+
totalTokenUsage.total += finalTargetResponse.tokenUsage.total || 0;
|
|
372
|
+
totalTokenUsage.prompt += finalTargetResponse.tokenUsage.prompt || 0;
|
|
373
|
+
totalTokenUsage.completion += finalTargetResponse.tokenUsage.completion || 0;
|
|
374
|
+
totalTokenUsage.numRequests += finalTargetResponse.tokenUsage.numRequests ?? 1;
|
|
375
|
+
totalTokenUsage.cached += finalTargetResponse.tokenUsage.cached || 0;
|
|
369
376
|
}
|
|
370
377
|
logger_1.default.debug(`Red team conversation complete. Final best score: ${bestScore}, Max score: ${maxScore}, Total attempts: ${attempts}`);
|
|
371
378
|
return {
|
|
372
|
-
output: finalTargetResponse,
|
|
379
|
+
output: finalTargetResponse.output,
|
|
373
380
|
metadata: {
|
|
374
381
|
redteamFinalPrompt: finalTargetPrompt,
|
|
375
382
|
},
|