promptfoo 0.81.2 → 0.81.4

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.
Files changed (108) hide show
  1. package/dist/package.json +11 -11
  2. package/dist/src/assertions.d.ts.map +1 -1
  3. package/dist/src/assertions.js +2 -1
  4. package/dist/src/assertions.js.map +1 -1
  5. package/dist/src/commands/eval.d.ts.map +1 -1
  6. package/dist/src/commands/eval.js +1 -0
  7. package/dist/src/commands/eval.js.map +1 -1
  8. package/dist/src/evaluator.d.ts.map +1 -1
  9. package/dist/src/evaluator.js +1 -0
  10. package/dist/src/evaluator.js.map +1 -1
  11. package/dist/src/matchers.js +2 -2
  12. package/dist/src/matchers.js.map +1 -1
  13. package/dist/src/providers/promptfoo.js +1 -1
  14. package/dist/src/providers/promptfoo.js.map +1 -1
  15. package/dist/src/redteam/index.d.ts.map +1 -1
  16. package/dist/src/redteam/index.js +35 -30
  17. package/dist/src/redteam/index.js.map +1 -1
  18. package/dist/src/redteam/plugins/base.d.ts.map +1 -1
  19. package/dist/src/redteam/plugins/base.js +15 -5
  20. package/dist/src/redteam/plugins/base.js.map +1 -1
  21. package/dist/src/redteam/plugins/harmful.d.ts.map +1 -1
  22. package/dist/src/redteam/plugins/harmful.js +5 -4
  23. package/dist/src/redteam/plugins/harmful.js.map +1 -1
  24. package/dist/src/redteam/providers/crescendo/index.d.ts.map +1 -1
  25. package/dist/src/redteam/providers/crescendo/index.js +1 -0
  26. package/dist/src/redteam/providers/crescendo/index.js.map +1 -1
  27. package/dist/src/redteam/providers/iterative.d.ts.map +1 -1
  28. package/dist/src/redteam/providers/iterative.js +4 -1
  29. package/dist/src/redteam/providers/iterative.js.map +1 -1
  30. package/dist/src/redteam/providers/iterativeImage.d.ts.map +1 -1
  31. package/dist/src/redteam/providers/iterativeImage.js +1 -0
  32. package/dist/src/redteam/providers/iterativeImage.js.map +1 -1
  33. package/dist/src/redteam/providers/iterativeTree.d.ts.map +1 -1
  34. package/dist/src/redteam/providers/iterativeTree.js +1 -0
  35. package/dist/src/redteam/providers/iterativeTree.js.map +1 -1
  36. package/dist/src/redteam/providers/shared.d.ts +2 -1
  37. package/dist/src/redteam/providers/shared.d.ts.map +1 -1
  38. package/dist/src/redteam/providers/shared.js +2 -2
  39. package/dist/src/redteam/providers/shared.js.map +1 -1
  40. package/dist/src/redteam/util.d.ts +0 -21
  41. package/dist/src/redteam/util.d.ts.map +1 -1
  42. package/dist/src/redteam/util.js +0 -53
  43. package/dist/src/redteam/util.js.map +1 -1
  44. package/dist/src/testCases.d.ts +2 -0
  45. package/dist/src/testCases.d.ts.map +1 -1
  46. package/dist/src/testCases.js +107 -97
  47. package/dist/src/testCases.js.map +1 -1
  48. package/dist/src/types/index.d.ts.map +1 -1
  49. package/dist/src/types/index.js.map +1 -1
  50. package/dist/src/util/generation.d.ts +22 -0
  51. package/dist/src/util/generation.d.ts.map +1 -0
  52. package/dist/src/util/generation.js +56 -0
  53. package/dist/src/util/generation.js.map +1 -0
  54. package/dist/src/util/index.d.ts +0 -1
  55. package/dist/src/util/index.d.ts.map +1 -1
  56. package/dist/src/util/index.js +0 -30
  57. package/dist/src/util/index.js.map +1 -1
  58. package/dist/src/util/json.d.ts +1 -0
  59. package/dist/src/util/json.d.ts.map +1 -1
  60. package/dist/src/util/json.js +30 -0
  61. package/dist/src/util/json.js.map +1 -1
  62. package/dist/src/web/nextui/404/index.html +1 -1
  63. package/dist/src/web/nextui/404.html +1 -1
  64. package/dist/src/web/nextui/_next/static/chunks/app/{layout-d74de07e6e80f296.js → layout-ea3efc5da1d80944.js} +1 -1
  65. package/dist/src/web/nextui/_next/static/chunks/main-app-cf76b7c55faa5c9b.js +1 -0
  66. package/dist/src/web/nextui/auth/login/index.html +1 -1
  67. package/dist/src/web/nextui/auth/login/index.txt +3 -3
  68. package/dist/src/web/nextui/auth/signup/index.html +1 -1
  69. package/dist/src/web/nextui/auth/signup/index.txt +3 -3
  70. package/dist/src/web/nextui/datasets/index.html +1 -1
  71. package/dist/src/web/nextui/datasets/index.txt +3 -3
  72. package/dist/src/web/nextui/eval/index.html +1 -1
  73. package/dist/src/web/nextui/eval/index.txt +3 -3
  74. package/dist/src/web/nextui/index.html +1 -1
  75. package/dist/src/web/nextui/index.txt +2 -2
  76. package/dist/src/web/nextui/progress/index.html +1 -1
  77. package/dist/src/web/nextui/progress/index.txt +3 -3
  78. package/dist/src/web/nextui/prompts/index.html +1 -1
  79. package/dist/src/web/nextui/prompts/index.txt +3 -3
  80. package/dist/src/web/nextui/report/index.html +1 -1
  81. package/dist/src/web/nextui/report/index.txt +3 -3
  82. package/dist/src/web/nextui/setup/index.html +1 -1
  83. package/dist/src/web/nextui/setup/index.txt +3 -3
  84. package/dist/test/redteam/index.test.js +4 -27
  85. package/dist/test/redteam/index.test.js.map +1 -1
  86. package/dist/test/redteam/plugins/base.test.js +22 -0
  87. package/dist/test/redteam/plugins/base.test.js.map +1 -1
  88. package/dist/test/redteam/providers/shared.test.js +14 -2
  89. package/dist/test/redteam/providers/shared.test.js.map +1 -1
  90. package/dist/test/testCases.test.js +192 -3
  91. package/dist/test/testCases.test.js.map +1 -1
  92. package/dist/test/util.generation.test.d.ts +2 -0
  93. package/dist/test/util.generation.test.d.ts.map +1 -0
  94. package/dist/test/{redteam/util.test.js → util.generation.test.js} +14 -14
  95. package/dist/test/util.generation.test.js.map +1 -0
  96. package/dist/test/util.json.test.js +27 -0
  97. package/dist/test/util.json.test.js.map +1 -1
  98. package/dist/test/util.test.js +0 -27
  99. package/dist/test/util.test.js.map +1 -1
  100. package/package.json +11 -11
  101. package/dist/src/web/nextui/_next/static/chunks/main-app-0dc90092cd7acaaf.js +0 -1
  102. package/dist/test/redteam/util.test.d.ts +0 -2
  103. package/dist/test/redteam/util.test.d.ts.map +0 -1
  104. package/dist/test/redteam/util.test.js.map +0 -1
  105. /package/dist/src/web/nextui/_next/static/chunks/{148-f6a1a312ab2d360a.js → 148-1c1aa4563e769657.js} +0 -0
  106. /package/dist/src/web/nextui/_next/static/chunks/{258-358dad65a69fa35d.js → 258-666c2b1786df7260.js} +0 -0
  107. /package/dist/src/web/nextui/_next/static/{V2kgKrpK-FTH5Ywn9fzkB → v1B0stUXeP04j29_pdYoD}/_buildManifest.js +0 -0
  108. /package/dist/src/web/nextui/_next/static/{V2kgKrpK-FTH5Ywn9fzkB → v1B0stUXeP04j29_pdYoD}/_ssgManifest.js +0 -0
@@ -1 +1 @@
1
- {"version":3,"file":"iterative.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AASlC,oDAAyD;AACzD,uCAAgG;AAChG,qCAA+C;AAE/C,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,GAQV;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,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,CAAC,CAAC;QAC/D,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,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAG1E,CAAC;QAEF,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,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;QAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,KAAK,SAAS,EAAE,kCAAkC,CAAC,CAAC;QAE9E,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC9D,IAAA,wBAAS,EAAC,UAAU,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;QAC9D,IAAI,cAAc,CAAC;QACnB,IAAI,OAAO,UAAU,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,gBAAgB,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC3D,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/D,cAAc,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC9C,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;aACxB;SACF,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAErD,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;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,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;IACL,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;KACF,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAI5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3E,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;QAC/C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,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,EAAE,QAAQ,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC;YAC9E,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,wBAAwB,CAAC"}
1
+ {"version":3,"file":"iterative.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterative.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AASlC,oDAAyD;AACzD,uCAAgG;AAChG,qCAA+C;AAE/C,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,GAQV;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,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,CAAC,CAAC;QAC/D,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,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAG1E,CAAC;QAEF,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,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;QAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,KAAK,SAAS,EAAE,kCAAkC,CAAC,CAAC;QAE9E,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC9D,IAAA,wBAAS,EAAC,UAAU,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;QAC9D,IAAI,cAAc,CAAC;QACnB,IAAI,OAAO,UAAU,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,gBAAgB,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC3D,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAC/D,cAAc,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC9C,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;aACxB;SACF,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAErD,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;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,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;IACL,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;KACF,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAI5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3E,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;QAC/C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,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;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,wBAAwB,CAAC"}
@@ -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;AAqOrB,cAAM,wBAAyB,YAAW,WAAW;IAIvC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAH5D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA4B;IAC5D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAS5D,EAAE;IAII,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,oBAAoB;;;;;;CAgB7F;AAED,eAAe,wBAAwB,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;AAqOrB,cAAM,wBAAyB,YAAW,WAAW;IAIvC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAH5D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA4B;IAC5D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IAS5D,EAAE;IAII,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,oBAAoB;;;;;;CAiB7F;AAED,eAAe,wBAAwB,CAAC"}
@@ -207,6 +207,7 @@ class RedteamIterativeProvider {
207
207
  redteamProvider: await (0, shared_1.loadRedteamProvider)({
208
208
  provider: this.redteamProvider,
209
209
  preferSmallModel: true,
210
+ jsonOnly: true,
210
211
  }),
211
212
  targetProvider: context.originalProvider,
212
213
  injectVar: this.injectVar,
@@ -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,oDAAyD;AACzD,qCAA+C;AAE/C,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,GAQV;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,GAAG,EAAE,CAAC;IAEtB,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,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,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAG1E,CAAC;QAEF,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,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC;QAC7D,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,wBAAwB,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QAE/E,IAAI,cAAc,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9D,cAAc,GAAG,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,KAAK,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,cAAc,GAAI,KAAe,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,cAAc,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAChF,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,sBAAsB,cAAc,EAAE,CAAC,CAAC;QAEvE,gCAAgC;QAChC,MAAM,QAAQ,GAAG,uBAAuB,CAAC;QACzC,MAAM,GAAG,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC3C,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,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,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvD,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;IACL,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;KACF,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAI5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3E,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;QAC/C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,CAAC;IAC7C,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,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,gBAAgB,EAAE,IAAI;aACvB,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,wBAAwB,CAAC"}
1
+ {"version":3,"file":"iterativeImage.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeImage.ts"],"names":[],"mappings":";;;;;AAAA,oDAA4B;AAC5B,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AASlC,oDAAyD;AACzD,qCAA+C;AAE/C,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,GAQV;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,GAAG,EAAE,CAAC;IAEtB,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,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,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAG1E,CAAC;QAEF,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,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC;QAC7D,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,wBAAwB,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QAE/E,IAAI,cAAc,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9D,cAAc,GAAG,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,KAAK,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,cAAc,GAAI,KAAe,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAA,wBAAS,EAAC,OAAO,cAAc,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAChF,gBAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,sBAAsB,cAAc,EAAE,CAAC,CAAC;QAEvE,gCAAgC;QAChC,MAAM,QAAQ,GAAG,uBAAuB,CAAC;QACzC,MAAM,GAAG,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC3C,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,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,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;QAClF,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvD,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;IACL,CAAC;IAED,OAAO;QACL,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE;YACR,kBAAkB,EAAE,YAAY,IAAI,SAAS;SAC9C;KACF,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAI5B,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3E,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;QAC/C,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,CAAC;IAED,EAAE;QACA,OAAO,mCAAmC,CAAC;IAC7C,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,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,gBAAgB,EAAE,IAAI;gBACtB,QAAQ,EAAE,IAAI;aACf,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,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;AAI5C,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,MAAM,EACN,iBAAiB,EAClB,MAAM,aAAa,CAAC;AA6BrB;;;;;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,CAqBjB;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,CAAA;CAAE,CAAC,CAQlD;AAED;;;;;;GAMG;AACH,wBAAsB,cAAc,CAClC,eAAe,EAAE,WAAW,EAC5B,mBAAmB,EAAE,MAAM,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,OAAO,CAAC,CAgBlB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,cAAc,EAAE,WAAW,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,MAAM,CAAC,CAMjB;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,GACV,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;CACnB,GAAG,OAAO,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC,CA4KzE;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;CAc1E;AAED,eAAe,4BAA4B,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;AAI5C,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,MAAM,EACN,iBAAiB,EAClB,MAAM,aAAa,CAAC;AA6BrB;;;;;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,CAqBjB;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,CAAA;CAAE,CAAC,CAQlD;AAED;;;;;;GAMG;AACH,wBAAsB,cAAc,CAClC,eAAe,EAAE,WAAW,EAC5B,mBAAmB,EAAE,MAAM,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,OAAO,CAAC,CAgBlB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,cAAc,EAAE,WAAW,EAC3B,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,MAAM,CAAC,CAMjB;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,GACV,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;CACnB,GAAG,OAAO,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC,CA4KzE;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;CAe1E;AAED,eAAe,4BAA4B,CAAC"}
@@ -360,6 +360,7 @@ class RedteamIterativeTreeProvider {
360
360
  vars: context.vars,
361
361
  redteamProvider: await (0, shared_1.loadRedteamProvider)({
362
362
  provider: this.config.redteamProvider,
363
+ jsonOnly: true,
363
364
  }),
364
365
  targetProvider: context.originalProvider,
365
366
  injectVar: this.injectVar,
@@ -1 +1 @@
1
- {"version":3,"file":"iterativeTree.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeTree.ts"],"names":[],"mappings":";;;;;AA4DA,kDAaC;AAUD,4CA0BC;AAQD,oCAWC;AASD,wCAoBC;AAQD,8CASC;AAUD,oDAaC;AAmBD,wCAEC;AAQD,8CAoBC;AASD,kDAMC;AAQD,wDAyBC;AAOD,wDA0LC;AAvdD,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AAQlC,oDAAyD;AACzD,2CAAgD;AAChD,uCAAgG;AAChG,qCAA+C;AAE/C,6CAA6C;AAE7C,+DAA+D;AAC/D,MAAM,YAAY,GAAG,EAAE,CAAC;AAExB,qEAAqE;AACrE,MAAM,YAAY,GAAG,EAAE,CAAC;AAExB,uCAAuC;AACvC,MAAM,SAAS,GAAG,CAAC,CAAC;AAEpB,uDAAuD;AACvD,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,oDAAoD;AACpD,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,qDAAqD;AACrD,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,mFAAmF;AACnF,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAE9B;;;;;GAKG;AACH,SAAgB,mBAAmB,CACjC,QAAqB,EACrB,IAAY;IAMZ,OAAO;QACL,mBAAmB,EAAE,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC;QAC5E,mBAAmB,EAAE,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC;QAC5E,iBAAiB,EAAE,QAAQ,CAAC,YAAY,CAAC,6BAAmB,EAAE,EAAE,IAAI,EAAE,CAAC;KACxE,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACI,KAAK,UAAU,gBAAgB,CACpC,eAA4B,EAC5B,iBAAyB,EACzB,cAAsB,EACtB,uBAAgC;IAEhC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAC/B;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,iBAAiB;SAC3B;QACD;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,cAAc;SACxB;KACF,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAClF,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAErD,yCAAyC;IACzC,IAAI,uBAAuB,EAAE,CAAC;QAC5B,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,YAAY,CAChC,eAA4B,EAC5B,cAA4E;IAE5E,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IACnD,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/D,IAAA,wBAAS,EACP,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC,qDAAqD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CACnF,CAAC;IACF,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAA4C,CAAC;AACnF,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,cAAc,CAClC,eAA4B,EAC5B,mBAA2B,EAC3B,YAAoB;IAEpB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;QACnC;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;QACD;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,YAAY;SACtB;KACF,CAAC,CAAC;IACH,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACnE,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IACtF,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IACrD,IAAA,wBAAS,EAAC,OAAO,OAAO,KAAK,SAAS,EAAE,kCAAkC,CAAC,CAAC;IAC5E,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,iBAAiB,CACrC,cAA2B,EAC3B,YAAoB;IAEpB,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9D,IAAA,wBAAS,EAAC,UAAU,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;IAC9D,OAAO,OAAO,UAAU,CAAC,MAAM,KAAK,QAAQ;QAC1C,CAAC,CAAC,UAAU,CAAC,MAAM;QACnB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,oBAAoB,CAClC,cAAsB,EACtB,IAAY,EACZ,KAAa,EACb,SAAkB,EAClB,uBAAgC;IAEhC,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,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;KACnE,CAAC;AACJ,CAAC;AAYD;;;;;;GAMG;AACH,SAAgB,cAAc,CAAC,MAAc,EAAE,KAAa,EAAE,KAAa;IACzE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAChD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAAC,KAAa,EAAE,KAAa;IAC5D,4CAA4C;IAC5C,IAAI,QAAQ,GAAG,aAAa,CAAC;IAE7B,oCAAoC;IACpC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;SAAM,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACtB,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,mCAAmC;IACnC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,2DAA2D;IAC3D,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;IAEpE,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CAAC,OAAe,EAAE,OAAe;IAClE,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;AACxC,CAAC;AAED;;;;;GAKG;AACH,SAAgB,sBAAsB,CAAC,KAAiB,EAAE,WAAmB;IAC3E,0CAA0C;IAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAe,EAAE,CAAC;IACrC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,mDAAmD;QACnD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;YAC5F,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzB,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;YACxC,MAAM;QACR,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,OAAO,aAAa,CAAC,MAAM,GAAG,WAAW,IAAI,aAAa,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QACvF,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,sBAAsB,CAAC,EAC3C,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,GAQV;IACC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,IAAI,GAAW,IAAI,CAAC,SAAS,CAAW,CAAC;IAE/C,IAAI,QAAQ,GAAG,CAAC,CAAC;IAEjB,gBAAM,CAAC,KAAK,CACV,qEAAqE,SAAS,mBAAmB,aAAa,kBAAkB,YAAY,kBAAkB,YAAY,WAAW,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAClN,CAAC;IAEF,MAAM,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,GAAG,mBAAmB,CACzF,QAAQ,EACR,IAAI,CACL,CAAC;IAEF,MAAM,cAAc,GAAiE;QACnF;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;KACF,CAAC;IAEF,IAAI,gBAAgB,GAAe,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhE,MAAM,QAAQ,GAAa,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAE3B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QAC/C,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,6CAA6C,gBAAgB,CAAC,MAAM,uBAAuB,QAAQ,EAAE,CACrH,CAAC;QAEF,MAAM,cAAc,GAAe,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,gBAAgB,EAAE,CAAC;YACpC,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAC7D,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,6BAA6B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,IAAI,CAAC,KAAK,cAAc,eAAe,uBAAuB,QAAQ,EAAE,CAChK,CAAC;YAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBACzC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,YAAY,CAAC,eAAe,EAAE;oBAChF,GAAG,cAAc;oBACjB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC5C,CAAC,CAAC;gBAEH,QAAQ,EAAE,CAAC;gBACX,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,4BAA4B,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,2BAA2B,QAAQ,EAAE,CACrL,CAAC;gBAEF,MAAM,YAAY,GAAG,MAAM,IAAA,+BAAY,EACrC,MAAM,EACN;oBACE,GAAG,IAAI;oBACP,CAAC,SAAS,CAAC,EAAE,YAAY;iBAC1B,EACD,OAAO,EACP,cAAc,CACf,CAAC;gBAEF,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,eAAe,EAAE,mBAAmB,EAAE,YAAY,CAAC,CAAC;gBAC3F,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;gBAC7E,MAAM,uBAAuB,GAAG,6BAAiB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAChE,cAAc,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC9C,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAClC,eAAe,EACf,iBAAiB,EACjB,cAAc,EACd,uBAAuB,CACxB,CAAC;gBAEF,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,uBAAuB,KAAK,cAAc,SAAS,eAAe,uBAAuB,uBAAuB,QAAQ,EAAE,CAC/J,CAAC;gBAEF,MAAM,OAAO,GAAG,cAAc,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC/D,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAE7B,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;oBACrB,QAAQ,GAAG,KAAK,CAAC;oBACjB,gBAAM,CAAC,KAAK,CAAC,UAAU,KAAK,aAAa,QAAQ,oBAAoB,QAAQ,EAAE,CAAC,CAAC;gBACnF,CAAC;qBAAM,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;oBAC7B,SAAS,GAAG,KAAK,CAAC;oBAClB,kBAAkB,GAAG,CAAC,CAAC;oBACvB,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,qBAAqB,KAAK,gBAAgB,QAAQ,EAAE,CACzF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,kBAAkB,EAAE,CAAC;oBACrB,IAAI,kBAAkB,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjC,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,wBAAwB,kBAAkB,uCAAuC,QAAQ,EAAE,CAChI,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,IAAI,KAAK,IAAI,YAAY,EAAE,CAAC;oBAC1B,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,2BAA2B,YAAY,iBAAiB,QAAQ,aAAa,CAClH,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,IAAI,kBAAkB,IAAI,kBAAkB,EAAE,CAAC;oBAC7C,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,wBAAwB,kBAAkB,uCAAuC,QAAQ,aAAa,CAC3I,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,IAAI,QAAQ,IAAI,YAAY,EAAE,CAAC;oBAC7B,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,+BAA+B,YAAY,iBAAiB,QAAQ,aAAa,CACtH,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,cAAc,CAAC,IAAI,CACjB,oBAAoB,CAAC,cAAc,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,uBAAuB,CAAC,CACtF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,gBAAgB,GAAG,sBAAsB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QACzE,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,oCAAoC,gBAAgB,CAAC,MAAM,sDAAsD,SAAS,gBAAgB,QAAQ,EAAE,CACpK,CAAC;IACJ,CAAC;IAED,MAAM,iBAAiB,GAAG,MAAM,IAAA,+BAAY,EAC1C,MAAM,EACN;QACE,GAAG,IAAI;QACP,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC,MAAM;KAC7B,EACD,OAAO,EACP,cAAc,CACf,CAAC;IAEF,MAAM,mBAAmB,GAAG,MAAM,iBAAiB,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;IAEvF,gBAAM,CAAC,KAAK,CACV,qDAAqD,SAAS,gBAAgB,QAAQ,qBAAqB,QAAQ,EAAE,CACtH,CAAC;IAEF,OAAO;QACL,MAAM,EAAE,mBAAmB;QAC3B,QAAQ,EAAE;YACR,kBAAkB,EAAE,iBAAiB;SACtC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,4BAA4B;IAGhC;;;;OAIG;IACH,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,wCAAwC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC/E,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,EAAE;QACA,OAAO,kCAAkC,CAAC;IAC5C,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,OAA8B;QAE9B,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QACpD,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,MAAM,CAAC,eAAe;aACtC,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,4BAA4B,CAAC"}
1
+ {"version":3,"file":"iterativeTree.js","sourceRoot":"","sources":["../../../../src/redteam/providers/iterativeTree.ts"],"names":[],"mappings":";;;;;AA4DA,kDAaC;AAUD,4CA0BC;AAQD,oCAWC;AASD,wCAoBC;AAQD,8CASC;AAUD,oDAaC;AAmBD,wCAEC;AAQD,8CAoBC;AASD,kDAMC;AAQD,wDAyBC;AAOD,wDA0LC;AAvdD,oEAAuC;AACvC,6DAAsD;AACtD,0DAAkC;AAQlC,oDAAyD;AACzD,2CAAgD;AAChD,uCAAgG;AAChG,qCAA+C;AAE/C,6CAA6C;AAE7C,+DAA+D;AAC/D,MAAM,YAAY,GAAG,EAAE,CAAC;AAExB,qEAAqE;AACrE,MAAM,YAAY,GAAG,EAAE,CAAC;AAExB,uCAAuC;AACvC,MAAM,SAAS,GAAG,CAAC,CAAC;AAEpB,uDAAuD;AACvD,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,oDAAoD;AACpD,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,qDAAqD;AACrD,MAAM,YAAY,GAAG,CAAC,CAAC;AAEvB,mFAAmF;AACnF,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAE9B;;;;;GAKG;AACH,SAAgB,mBAAmB,CACjC,QAAqB,EACrB,IAAY;IAMZ,OAAO;QACL,mBAAmB,EAAE,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC;QAC5E,mBAAmB,EAAE,QAAQ,CAAC,YAAY,CAAC,gCAAsB,EAAE,EAAE,IAAI,EAAE,CAAC;QAC5E,iBAAiB,EAAE,QAAQ,CAAC,YAAY,CAAC,6BAAmB,EAAE,EAAE,IAAI,EAAE,CAAC;KACxE,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACI,KAAK,UAAU,gBAAgB,CACpC,eAA4B,EAC5B,iBAAyB,EACzB,cAAsB,EACtB,uBAAgC;IAEhC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAC/B;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,iBAAiB;SAC3B;QACD;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,cAAc;SACxB;KACF,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3D,IAAA,wBAAS,EAAC,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAClF,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAErD,yCAAyC;IACzC,IAAI,uBAAuB,EAAE,CAAC;QAC5B,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,YAAY,CAChC,eAA4B,EAC5B,cAA4E;IAE5E,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IACnD,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/D,IAAA,wBAAS,EACP,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,EACtC,qDAAqD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CACnF,CAAC;IACF,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAA4C,CAAC;AACnF,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,cAAc,CAClC,eAA4B,EAC5B,mBAA2B,EAC3B,YAAoB;IAEpB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;QACnC;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;QACD;YACE,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,YAAY;SACtB;KACF,CAAC,CAAC;IACH,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACnE,IAAA,wBAAS,EAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IACtF,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IACrD,IAAA,wBAAS,EAAC,OAAO,OAAO,KAAK,SAAS,EAAE,kCAAkC,CAAC,CAAC;IAC5E,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,iBAAiB,CACrC,cAA2B,EAC3B,YAAoB;IAEpB,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9D,IAAA,wBAAS,EAAC,UAAU,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC;IAC9D,OAAO,OAAO,UAAU,CAAC,MAAM,KAAK,QAAQ;QAC1C,CAAC,CAAC,UAAU,CAAC,MAAM;QACnB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,oBAAoB,CAClC,cAAsB,EACtB,IAAY,EACZ,KAAa,EACb,SAAkB,EAClB,uBAAgC;IAEhC,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,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;KACnE,CAAC;AACJ,CAAC;AAYD;;;;;;GAMG;AACH,SAAgB,cAAc,CAAC,MAAc,EAAE,KAAa,EAAE,KAAa;IACzE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAChD,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAAC,KAAa,EAAE,KAAa;IAC5D,4CAA4C;IAC5C,IAAI,QAAQ,GAAG,aAAa,CAAC;IAE7B,oCAAoC;IACpC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;SAAM,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACtB,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,mCAAmC;IACnC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACf,QAAQ,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,2DAA2D;IAC3D,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;IAEpE,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,mBAAmB,CAAC,OAAe,EAAE,OAAe;IAClE,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;AACxC,CAAC;AAED;;;;;GAKG;AACH,SAAgB,sBAAsB,CAAC,KAAiB,EAAE,WAAmB;IAC3E,0CAA0C;IAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAe,EAAE,CAAC;IACrC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,mDAAmD;QACnD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;YAC5F,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzB,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;YACxC,MAAM;QACR,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,OAAO,aAAa,CAAC,MAAM,GAAG,WAAW,IAAI,aAAa,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QACvF,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,sBAAsB,CAAC,EAC3C,MAAM,EACN,OAAO,EACP,IAAI,EACJ,eAAe,EACf,cAAc,EACd,SAAS,GAQV;IACC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,IAAI,GAAW,IAAI,CAAC,SAAS,CAAW,CAAC;IAE/C,IAAI,QAAQ,GAAG,CAAC,CAAC;IAEjB,gBAAM,CAAC,KAAK,CACV,qEAAqE,SAAS,mBAAmB,aAAa,kBAAkB,YAAY,kBAAkB,YAAY,WAAW,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAClN,CAAC;IAEF,MAAM,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,GAAG,mBAAmB,CACzF,QAAQ,EACR,IAAI,CACL,CAAC;IAEF,MAAM,cAAc,GAAiE;QACnF;YACE,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mBAAmB;SAC7B;KACF,CAAC;IAEF,IAAI,gBAAgB,GAAe,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEhE,MAAM,QAAQ,GAAa,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAE3B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QAC/C,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,6CAA6C,gBAAgB,CAAC,MAAM,uBAAuB,QAAQ,EAAE,CACrH,CAAC;QAEF,MAAM,cAAc,GAAe,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,gBAAgB,EAAE,CAAC;YACpC,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAC7D,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,6BAA6B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,IAAI,CAAC,KAAK,cAAc,eAAe,uBAAuB,QAAQ,EAAE,CAChK,CAAC;YAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBACzC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,YAAY,CAAC,eAAe,EAAE;oBAChF,GAAG,cAAc;oBACjB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC5C,CAAC,CAAC;gBAEH,QAAQ,EAAE,CAAC;gBACX,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,4BAA4B,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,2BAA2B,QAAQ,EAAE,CACrL,CAAC;gBAEF,MAAM,YAAY,GAAG,MAAM,IAAA,+BAAY,EACrC,MAAM,EACN;oBACE,GAAG,IAAI;oBACP,CAAC,SAAS,CAAC,EAAE,YAAY;iBAC1B,EACD,OAAO,EACP,cAAc,CACf,CAAC;gBAEF,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,eAAe,EAAE,mBAAmB,EAAE,YAAY,CAAC,CAAC;gBAC3F,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;gBAC7E,MAAM,uBAAuB,GAAG,6BAAiB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAChE,cAAc,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC9C,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAClC,eAAe,EACf,iBAAiB,EACjB,cAAc,EACd,uBAAuB,CACxB,CAAC;gBAEF,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,uBAAuB,KAAK,cAAc,SAAS,eAAe,uBAAuB,uBAAuB,QAAQ,EAAE,CAC/J,CAAC;gBAEF,MAAM,OAAO,GAAG,cAAc,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC/D,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAE7B,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;oBACrB,QAAQ,GAAG,KAAK,CAAC;oBACjB,gBAAM,CAAC,KAAK,CAAC,UAAU,KAAK,aAAa,QAAQ,oBAAoB,QAAQ,EAAE,CAAC,CAAC;gBACnF,CAAC;qBAAM,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;oBAC7B,SAAS,GAAG,KAAK,CAAC;oBAClB,kBAAkB,GAAG,CAAC,CAAC;oBACvB,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,qBAAqB,KAAK,gBAAgB,QAAQ,EAAE,CACzF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,kBAAkB,EAAE,CAAC;oBACrB,IAAI,kBAAkB,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjC,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,wBAAwB,kBAAkB,uCAAuC,QAAQ,EAAE,CAChI,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,IAAI,KAAK,IAAI,YAAY,EAAE,CAAC;oBAC1B,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,2BAA2B,YAAY,iBAAiB,QAAQ,aAAa,CAClH,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,IAAI,kBAAkB,IAAI,kBAAkB,EAAE,CAAC;oBAC7C,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,wBAAwB,kBAAkB,uCAAuC,QAAQ,aAAa,CAC3I,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,IAAI,QAAQ,IAAI,YAAY,EAAE,CAAC;oBAC7B,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,aAAa,QAAQ,+BAA+B,YAAY,iBAAiB,QAAQ,aAAa,CACtH,CAAC;oBACF,OAAO;wBACL,MAAM,EAAE,cAAc;wBACtB,QAAQ,EAAE;4BACR,kBAAkB,EAAE,YAAY;yBACjC;qBACF,CAAC;gBACJ,CAAC;gBAED,cAAc,CAAC,IAAI,CACjB,oBAAoB,CAAC,cAAc,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,uBAAuB,CAAC,CACtF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,gBAAgB,GAAG,sBAAsB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QACzE,gBAAM,CAAC,KAAK,CACV,UAAU,KAAK,oCAAoC,gBAAgB,CAAC,MAAM,sDAAsD,SAAS,gBAAgB,QAAQ,EAAE,CACpK,CAAC;IACJ,CAAC;IAED,MAAM,iBAAiB,GAAG,MAAM,IAAA,+BAAY,EAC1C,MAAM,EACN;QACE,GAAG,IAAI;QACP,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC,MAAM;KAC7B,EACD,OAAO,EACP,cAAc,CACf,CAAC;IAEF,MAAM,mBAAmB,GAAG,MAAM,iBAAiB,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;IAEvF,gBAAM,CAAC,KAAK,CACV,qDAAqD,SAAS,gBAAgB,QAAQ,qBAAqB,QAAQ,EAAE,CACtH,CAAC;IAEF,OAAO;QACL,MAAM,EAAE,mBAAmB;QAC3B,QAAQ,EAAE;YACR,kBAAkB,EAAE,iBAAiB;SACtC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,4BAA4B;IAGhC;;;;OAIG;IACH,YAAqB,MAAuC;QAAvC,WAAM,GAAN,MAAM,CAAiC;QAC1D,gBAAM,CAAC,KAAK,CAAC,wCAAwC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC/E,IAAA,wBAAS,EAAC,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,EAAE;QACA,OAAO,kCAAkC,CAAC;IAC5C,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CACX,MAAc,EACd,OAA8B,EAC9B,OAA8B;QAE9B,IAAA,wBAAS,EAAC,OAAO,EAAE,gBAAgB,EAAE,qCAAqC,CAAC,CAAC;QAC5E,IAAA,wBAAS,EAAC,OAAO,EAAE,IAAI,EAAE,yBAAyB,CAAC,CAAC;QACpD,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,MAAM,CAAC,eAAe;gBACrC,QAAQ,EAAE,IAAI;aACf,CAAC;YACF,cAAc,EAAE,OAAO,CAAC,gBAAgB;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;CACF;AAED,kBAAe,4BAA4B,CAAC"}
@@ -1,6 +1,7 @@
1
1
  import { type RedteamConfig } from '../../types';
2
- export declare function loadRedteamProvider({ provider, preferSmallModel, }?: {
2
+ export declare function loadRedteamProvider({ provider, jsonOnly, preferSmallModel, }?: {
3
3
  provider?: RedteamConfig['provider'];
4
+ jsonOnly?: boolean;
4
5
  preferSmallModel?: boolean;
5
6
  }): Promise<import("../../types").ApiProvider | import("../../providers/openai").OpenAiChatCompletionProvider>;
6
7
  //# sourceMappingURL=shared.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"shared.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/shared.ts"],"names":[],"mappings":"AAEA,OAAO,EAAoC,KAAK,aAAa,EAAE,MAAM,aAAa,CAAC;AAGnF,wBAAsB,mBAAmB,CAAC,EACxC,QAAQ,EACR,gBAAwB,GACzB,GAAE;IACD,QAAQ,CAAC,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;IACrC,gBAAgB,CAAC,EAAE,OAAO,CAAC;CACvB,8GA0BL"}
1
+ {"version":3,"file":"shared.d.ts","sourceRoot":"","sources":["../../../../src/redteam/providers/shared.ts"],"names":[],"mappings":"AAEA,OAAO,EAAoC,KAAK,aAAa,EAAE,MAAM,aAAa,CAAC;AAGnF,wBAAsB,mBAAmB,CAAC,EACxC,QAAQ,EACR,QAAgB,EAChB,gBAAwB,GACzB,GAAE;IACD,QAAQ,CAAC,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;IACrC,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CACvB,8GA0BL"}
@@ -31,7 +31,7 @@ const cliState_1 = __importDefault(require("../../cliState"));
31
31
  const logger_1 = __importDefault(require("../../logger"));
32
32
  const types_1 = require("../../types");
33
33
  const constants_1 = require("./constants");
34
- async function loadRedteamProvider({ provider, preferSmallModel = false, } = {}) {
34
+ async function loadRedteamProvider({ provider, jsonOnly = false, preferSmallModel = false, } = {}) {
35
35
  // FIXME(ian): This approach only works on CLI, it doesn't work when running via node module.
36
36
  // That's ok for now because we only officially support redteams from CLI.
37
37
  let ret;
@@ -54,7 +54,7 @@ async function loadRedteamProvider({ provider, preferSmallModel = false, } = {})
54
54
  ret = new OpenAiChatCompletionProviderModule.OpenAiChatCompletionProvider(defaultModel, {
55
55
  config: {
56
56
  temperature: constants_1.TEMPERATURE,
57
- response_format: { type: 'json_object' },
57
+ response_format: jsonOnly ? { type: 'json_object' } : undefined,
58
58
  },
59
59
  });
60
60
  }
@@ -1 +1 @@
1
- {"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../../src/redteam/providers/shared.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKA,kDAgCC;AArCD,8DAAsC;AACtC,0DAAkC;AAClC,uCAAmF;AACnF,2CAAgF;AAEzE,KAAK,UAAU,mBAAmB,CAAC,EACxC,QAAQ,EACR,gBAAgB,GAAG,KAAK,MAItB,EAAE;IACJ,6FAA6F;IAC7F,0EAA0E;IAC1E,IAAI,GAAG,CAAC;IACR,MAAM,eAAe,GAAG,QAAQ,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC;IACvE,IAAI,IAAA,qBAAa,EAAC,eAAe,CAAC,EAAE,CAAC;QACnC,gBAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;QAC3D,GAAG,GAAG,eAAe,CAAC;IACxB,CAAC;SAAM,IAAI,OAAO,eAAe,KAAK,QAAQ,IAAI,IAAA,yBAAiB,EAAC,eAAe,CAAC,EAAE,CAAC;QACrF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,eAAe,EAAE,CAAC,CAAC;QAC7D,MAAM,sBAAsB,GAAG,wDAAa,iBAAiB,GAAC,CAAC;QAC/D,4CAA4C;QAC5C,GAAG,GAAG,CAAC,MAAM,sBAAsB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,CAAC;SAAM,CAAC;QACN,MAAM,YAAY,GAAG,gBAAgB,CAAC,CAAC,CAAC,gCAAoB,CAAC,CAAC,CAAC,0BAAc,CAAC;QAC9E,gBAAM,CAAC,KAAK,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC;QAChE,4CAA4C;QAC5C,MAAM,kCAAkC,GAAG,wDAAa,wBAAwB,GAAC,CAAC;QAClF,GAAG,GAAG,IAAI,kCAAkC,CAAC,4BAA4B,CAAC,YAAY,EAAE;YACtF,MAAM,EAAE;gBACN,WAAW,EAAE,uBAAW;gBACxB,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;aACzC;SACF,CAAC,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
1
+ {"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../../src/redteam/providers/shared.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKA,kDAkCC;AAvCD,8DAAsC;AACtC,0DAAkC;AAClC,uCAAmF;AACnF,2CAAgF;AAEzE,KAAK,UAAU,mBAAmB,CAAC,EACxC,QAAQ,EACR,QAAQ,GAAG,KAAK,EAChB,gBAAgB,GAAG,KAAK,MAKtB,EAAE;IACJ,6FAA6F;IAC7F,0EAA0E;IAC1E,IAAI,GAAG,CAAC;IACR,MAAM,eAAe,GAAG,QAAQ,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC;IACvE,IAAI,IAAA,qBAAa,EAAC,eAAe,CAAC,EAAE,CAAC;QACnC,gBAAM,CAAC,KAAK,CAAC,2BAA2B,eAAe,EAAE,CAAC,CAAC;QAC3D,GAAG,GAAG,eAAe,CAAC;IACxB,CAAC;SAAM,IAAI,OAAO,eAAe,KAAK,QAAQ,IAAI,IAAA,yBAAiB,EAAC,eAAe,CAAC,EAAE,CAAC;QACrF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,eAAe,EAAE,CAAC,CAAC;QAC7D,MAAM,sBAAsB,GAAG,wDAAa,iBAAiB,GAAC,CAAC;QAC/D,4CAA4C;QAC5C,GAAG,GAAG,CAAC,MAAM,sBAAsB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,CAAC;SAAM,CAAC;QACN,MAAM,YAAY,GAAG,gBAAgB,CAAC,CAAC,CAAC,gCAAoB,CAAC,CAAC,CAAC,0BAAc,CAAC;QAC9E,gBAAM,CAAC,KAAK,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC;QAChE,4CAA4C;QAC5C,MAAM,kCAAkC,GAAG,wDAAa,wBAAwB,GAAC,CAAC;QAClF,GAAG,GAAG,IAAI,kCAAkC,CAAC,4BAA4B,CAAC,YAAY,EAAE;YACtF,MAAM,EAAE;gBACN,WAAW,EAAE,uBAAW;gBACxB,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;aAChE;SACF,CAAC,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
@@ -1,24 +1,3 @@
1
- /**
2
- * Retries an operation with deduplication until the target count is reached or max retries are exhausted.
3
- *
4
- * @param operation - A function that takes the current items and returns a Promise of new items.
5
- * @param targetCount - The desired number of unique items to collect.
6
- * @param maxConsecutiveRetries - Maximum number of consecutive retries allowed when no new items are found. Defaults to 2.
7
- * @param dedupFn - A function to deduplicate items. Defaults to using a Set for uniqueness.
8
- * @returns A Promise that resolves to an array of unique items.
9
- *
10
- * @typeParam T - The type of items being collected.
11
- */
12
- export declare function retryWithDeduplication<T>(operation: (currentItems: T[]) => Promise<T[]>, targetCount: number, maxConsecutiveRetries?: number, dedupFn?: (items: T[]) => T[]): Promise<T[]>;
13
- /**
14
- * Randomly samples n items from an array.
15
- * If n is greater than the length of the array, the entire array is returned.
16
- *
17
- * @param array The array to sample from
18
- * @param n The number of items to sample
19
- * @returns A new array with n randomly sampled items
20
- */
21
- export declare function sampleArray<T>(array: T[], n: number): T[];
22
1
  export declare function isBasicRefusal(response: string): boolean;
23
2
  export declare function removePrefix(str: string, prefix: string): string;
24
3
  //# sourceMappingURL=util.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../../../src/redteam/util.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;GAUG;AACH,wBAAsB,sBAAsB,CAAC,CAAC,EAC5C,SAAS,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC,EAAE,CAAC,EAC9C,WAAW,EAAE,MAAM,EACnB,qBAAqB,GAAE,MAAU,EACjC,OAAO,GAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,EACsE,GAC/F,OAAO,CAAC,CAAC,EAAE,CAAC,CA2Bd;AAED;;;;;;;GAOG;AACH,wBAAgB,WAAW,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,CAAC,EAAE,CAIzD;AAWD,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAGxD;AAED,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,UAKvD"}
1
+ {"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../../../src/redteam/util.ts"],"names":[],"mappings":"AASA,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAGxD;AAED,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,UAKvD"}
@@ -1,60 +1,7 @@
1
1
  "use strict";
2
- var __importDefault = (this && this.__importDefault) || function (mod) {
3
- return (mod && mod.__esModule) ? mod : { "default": mod };
4
- };
5
2
  Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.retryWithDeduplication = retryWithDeduplication;
7
- exports.sampleArray = sampleArray;
8
3
  exports.isBasicRefusal = isBasicRefusal;
9
4
  exports.removePrefix = removePrefix;
10
- const logger_1 = __importDefault(require("../logger"));
11
- /**
12
- * Retries an operation with deduplication until the target count is reached or max retries are exhausted.
13
- *
14
- * @param operation - A function that takes the current items and returns a Promise of new items.
15
- * @param targetCount - The desired number of unique items to collect.
16
- * @param maxConsecutiveRetries - Maximum number of consecutive retries allowed when no new items are found. Defaults to 2.
17
- * @param dedupFn - A function to deduplicate items. Defaults to using a Set for uniqueness.
18
- * @returns A Promise that resolves to an array of unique items.
19
- *
20
- * @typeParam T - The type of items being collected.
21
- */
22
- async function retryWithDeduplication(operation, targetCount, maxConsecutiveRetries = 2, dedupFn = (items) => Array.from(new Set(items.map((item) => JSON.stringify(item)))).map((item) => JSON.parse(item))) {
23
- const allItems = [];
24
- let consecutiveRetries = 0;
25
- while (allItems.length < targetCount && consecutiveRetries <= maxConsecutiveRetries) {
26
- const newItems = await operation(allItems);
27
- if (!Array.isArray(newItems)) {
28
- logger_1.default.warn('Operation returned non-iterable result. Skipping this iteration.');
29
- consecutiveRetries++;
30
- continue;
31
- }
32
- const uniqueNewItems = dedupFn([...allItems, ...newItems]).slice(allItems.length);
33
- allItems.push(...uniqueNewItems);
34
- logger_1.default.debug(`Added ${uniqueNewItems.length} unique items. Total: ${allItems.length}`);
35
- if (uniqueNewItems.length === 0) {
36
- consecutiveRetries++;
37
- logger_1.default.debug(`No new unique items. Consecutive retries: ${consecutiveRetries}`);
38
- }
39
- else {
40
- consecutiveRetries = 0;
41
- }
42
- }
43
- return allItems;
44
- }
45
- /**
46
- * Randomly samples n items from an array.
47
- * If n is greater than the length of the array, the entire array is returned.
48
- *
49
- * @param array The array to sample from
50
- * @param n The number of items to sample
51
- * @returns A new array with n randomly sampled items
52
- */
53
- function sampleArray(array, n) {
54
- logger_1.default.debug(`Sampling ${n} items from array of length ${array.length}`);
55
- const shuffled = array.slice().sort(() => 0.5 - Math.random());
56
- return shuffled.slice(0, Math.min(n, array.length));
57
- }
58
5
  const REFUSAL_PREFIXES = [
59
6
  'Sorry',
60
7
  "I'm sorry",
@@ -1 +1 @@
1
- {"version":3,"file":"util.js","sourceRoot":"","sources":["../../../src/redteam/util.ts"],"names":[],"mappings":";;;;;AAaA,wDAiCC;AAUD,kCAIC;AAWD,wCAGC;AAED,oCAKC;AAjFD,uDAA+B;AAE/B;;;;;;;;;;GAUG;AACI,KAAK,UAAU,sBAAsB,CAC1C,SAA8C,EAC9C,WAAmB,EACnB,wBAAgC,CAAC,EACjC,UAA+B,CAAC,KAAK,EAAE,EAAE,CACvC,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAEhG,MAAM,QAAQ,GAAQ,EAAE,CAAC;IACzB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAE3B,OAAO,QAAQ,CAAC,MAAM,GAAG,WAAW,IAAI,kBAAkB,IAAI,qBAAqB,EAAE,CAAC;QACpF,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,CAAC;QAE3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,gBAAM,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;YAChF,kBAAkB,EAAE,CAAC;YACrB,SAAS;QACX,CAAC;QAED,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,GAAG,QAAQ,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAClF,QAAQ,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QAEjC,gBAAM,CAAC,KAAK,CAAC,SAAS,cAAc,CAAC,MAAM,yBAAyB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAEvF,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,kBAAkB,EAAE,CAAC;YACrB,gBAAM,CAAC,KAAK,CAAC,6CAA6C,kBAAkB,EAAE,CAAC,CAAC;QAClF,CAAC;aAAM,CAAC;YACN,kBAAkB,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,WAAW,CAAI,KAAU,EAAE,CAAS;IAClD,gBAAM,CAAC,KAAK,CAAC,YAAY,CAAC,+BAA+B,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IACzE,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/D,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,MAAM,gBAAgB,GAAG;IACvB,OAAO;IACP,WAAW;IACX,YAAY;IACZ,iBAAiB;IACjB,kBAAkB;IAClB,aAAa;CACd,CAAC;AAEF,SAAgB,cAAc,CAAC,QAAgB;IAC7C,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpD,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,SAAgB,YAAY,CAAC,GAAW,EAAE,MAAc;IACtD,0FAA0F;IAC1F,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;IAC/C,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC5D,OAAO,GAAG,CAAC;AACb,CAAC"}
1
+ {"version":3,"file":"util.js","sourceRoot":"","sources":["../../../src/redteam/util.ts"],"names":[],"mappings":";;AASA,wCAGC;AAED,oCAKC;AAnBD,MAAM,gBAAgB,GAAG;IACvB,OAAO;IACP,WAAW;IACX,YAAY;IACZ,iBAAiB;IACjB,kBAAkB;IAClB,aAAa;CACd,CAAC;AAEF,SAAgB,cAAc,CAAC,QAAgB;IAC7C,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpD,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,SAAgB,YAAY,CAAC,GAAW,EAAE,MAAc;IACtD,0FAA0F;IAC1F,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;IAC/C,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC5D,OAAO,GAAG,CAAC;AACb,CAAC"}
@@ -11,6 +11,8 @@ interface SynthesizeOptions {
11
11
  provider?: string;
12
12
  tests: TestCase[];
13
13
  }
14
+ export declare function generatePersonasPrompt(prompts: string[], numPersonas: number): string;
15
+ export declare function testCasesPrompt(prompts: string[], persona: string, tests: TestCase[], numTestCasesPerPersona: number, variables: string[], instructions?: string): string;
14
16
  export declare function synthesize({ prompts, instructions, tests, numPersonas, numTestCasesPerPersona, provider, }: SynthesizeOptions): Promise<VarMapping[]>;
15
17
  export declare function synthesizeFromTestSuite(testSuite: TestSuite, options: Partial<SynthesizeOptions>): Promise<VarMapping[]>;
16
18
  export {};
@@ -1 +1 @@
1
- {"version":3,"file":"testCases.d.ts","sourceRoot":"","sources":["../../src/testCases.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAEV,QAAQ,EACR,oBAAoB,EACpB,SAAS,EACT,eAAe,EACf,UAAU,EAGX,MAAM,SAAS,CAAC;AAYjB,wBAAsB,aAAa,CACjC,WAAW,EAAE,MAAM,GAAG,MAAM,EAAE,EAC9B,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAmBrD;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAsBrB;AAED,wBAAsB,QAAQ,CAC5B,IAAI,EAAE,MAAM,GAAG,oBAAoB,EACnC,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,CAAC,CAkEnB;AAED,wBAAsB,SAAS,CAC7B,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,EAC/B,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAuFrB;AAED,UAAU,iBAAiB;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,QAAQ,EAAE,CAAC;CACnB;AAED,wBAAsB,UAAU,CAAC,EAC/B,OAAO,EACP,YAAY,EACZ,KAAK,EACL,WAAW,EACX,sBAAsB,EACtB,QAAQ,GACT,EAAE,iBAAiB,yBAwInB;AAED,wBAAsB,uBAAuB,CAC3C,SAAS,EAAE,SAAS,EACpB,OAAO,EAAE,OAAO,CAAC,iBAAiB,CAAC,yBAOpC"}
1
+ {"version":3,"file":"testCases.d.ts","sourceRoot":"","sources":["../../src/testCases.ts"],"names":[],"mappings":"AAgBA,OAAO,KAAK,EAEV,QAAQ,EACR,oBAAoB,EACpB,SAAS,EACT,eAAe,EACf,UAAU,EAGX,MAAM,SAAS,CAAC;AAajB,wBAAsB,aAAa,CACjC,WAAW,EAAE,MAAM,GAAG,MAAM,EAAE,EAC9B,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAmBrD;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAsBrB;AAeD,wBAAsB,QAAQ,CAC5B,IAAI,EAAE,MAAM,GAAG,oBAAoB,EACnC,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,CAAC,CAoCnB;AAED,wBAAsB,SAAS,CAC7B,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,EAC/B,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAuFrB;AAED,UAAU,iBAAiB;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,QAAQ,EAAE,CAAC;CACnB;AAED,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAWrF;AAED,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,EAAE,EACjB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,QAAQ,EAAE,EACjB,sBAAsB,EAAE,MAAM,EAC9B,SAAS,EAAE,MAAM,EAAE,EACnB,YAAY,CAAC,EAAE,MAAM,GACpB,MAAM,CAiDR;AAED,wBAAsB,UAAU,CAAC,EAC/B,OAAO,EACP,YAAY,EACZ,KAAK,EACL,WAAW,EACX,sBAAsB,EACtB,QAAQ,GACT,EAAE,iBAAiB,yBAmHnB;AAED,wBAAsB,uBAAuB,CAC3C,SAAS,EAAE,SAAS,EACpB,OAAO,EAAE,OAAO,CAAC,iBAAiB,CAAC,yBAOpC"}
@@ -30,6 +30,8 @@ exports.readVarsFiles = readVarsFiles;
30
30
  exports.readStandaloneTestsFile = readStandaloneTestsFile;
31
31
  exports.readTest = readTest;
32
32
  exports.readTests = readTests;
33
+ exports.generatePersonasPrompt = generatePersonasPrompt;
34
+ exports.testCasesPrompt = testCasesPrompt;
33
35
  exports.synthesize = synthesize;
34
36
  exports.synthesizeFromTestSuite = synthesizeFromTestSuite;
35
37
  const json_schema_ref_parser_1 = __importDefault(require("@apidevtools/json-schema-ref-parser"));
@@ -47,7 +49,8 @@ const googleSheets_1 = require("./googleSheets");
47
49
  const logger_1 = __importDefault(require("./logger"));
48
50
  const providers_1 = require("./providers");
49
51
  const defaults_1 = require("./providers/defaults");
50
- const util_1 = require("./util");
52
+ const generation_1 = require("./util/generation");
53
+ const json_1 = require("./util/json");
51
54
  const templates_1 = require("./util/templates");
52
55
  function parseJson(json) {
53
56
  try {
@@ -98,34 +101,17 @@ async function readStandaloneTestsFile(varsPath, basePath = '') {
98
101
  return test;
99
102
  });
100
103
  }
104
+ async function loadTestWithVars(testCase, testBasePath) {
105
+ const ret = { ...testCase, vars: undefined };
106
+ if (typeof testCase.vars === 'string' || Array.isArray(testCase.vars)) {
107
+ ret.vars = await readVarsFiles(testCase.vars, testBasePath);
108
+ }
109
+ else {
110
+ ret.vars = testCase.vars;
111
+ }
112
+ return ret;
113
+ }
101
114
  async function readTest(test, basePath = '') {
102
- const loadTestWithVars = async (testCase, testBasePath) => {
103
- const ret = { ...testCase, vars: undefined };
104
- if (typeof testCase.vars === 'string' || Array.isArray(testCase.vars)) {
105
- ret.vars = await readVarsFiles(testCase.vars, testBasePath);
106
- }
107
- else {
108
- ret.vars = testCase.vars;
109
- } /*else if (typeof testCase.vars === 'object') {
110
- const vars: Record<string, string | string[] | object> = {};
111
- for (const [key, value] of Object.entries(testCase.vars)) {
112
- if (typeof value === 'string' && value.startsWith('file://')) {
113
- // Load file from disk.
114
- const filePath = path.resolve(testBasePath, value.slice('file://'.length));
115
- if (filePath.endsWith('.yaml') || filePath.endsWith('.yml')) {
116
- vars[key] = (yaml.load(fs.readFileSync(filePath, 'utf-8')) as string).trim();
117
- } else {
118
- vars[key] = fs.readFileSync(filePath, 'utf-8').trim();
119
- }
120
- } else {
121
- // This is a normal key:value.
122
- vars[key] = value;
123
- }
124
- }
125
- ret.vars = vars;
126
- }*/
127
- return ret;
128
- };
129
115
  let testCase;
130
116
  if (typeof test === 'string') {
131
117
  const testFilePath = path.resolve(basePath, test);
@@ -148,9 +134,9 @@ async function readTest(test, basePath = '') {
148
134
  });
149
135
  }
150
136
  }
151
- // Validation of the shape of test
137
+ // Validate the shape of the test case
152
138
  if (!testCase.assert && !testCase.vars && !testCase.options && !testCase.metadata) {
153
- throw new Error(`Test case must have either assert, vars, or options property. Instead got ${JSON.stringify(testCase, null, 2)}`);
139
+ throw new Error(`Test case must have either assert, vars, options, or metadata property. Instead got ${JSON.stringify(testCase, null, 2)}`);
154
140
  }
155
141
  return testCase;
156
142
  }
@@ -165,7 +151,7 @@ async function readTests(tests, basePath = '') {
165
151
  windowsPathsNoEscape: true,
166
152
  });
167
153
  const _deref = async (testCases, file) => {
168
- logger_1.default.debug(`Dereferencing testfile ${file}`);
154
+ logger_1.default.debug(`Dereferencing test file: ${file}`);
169
155
  return (await json_schema_ref_parser_1.default.dereference(testCases));
170
156
  };
171
157
  const ret = [];
@@ -216,7 +202,7 @@ async function readTests(tests, basePath = '') {
216
202
  ret.push(...(await loadTestsFromGlob(globOrTest)));
217
203
  }
218
204
  else {
219
- // It's just a TestCase
205
+ // Load individual TestCase
220
206
  ret.push(await readTest(globOrTest, basePath));
221
207
  }
222
208
  }
@@ -240,6 +226,62 @@ async function readTests(tests, basePath = '') {
240
226
  }
241
227
  return ret;
242
228
  }
229
+ function generatePersonasPrompt(prompts, numPersonas) {
230
+ const promptsString = (0, dedent_1.default) `<Prompts>
231
+ ${prompts.map((prompt) => `<Prompt>\n${prompt}\n</Prompt>`).join('\n')}
232
+ </Prompts>`;
233
+ return (0, dedent_1.default) `
234
+ Consider the following prompt${prompts.length > 1 ? 's' : ''} for an LLM application:
235
+
236
+ ${promptsString}
237
+
238
+ List up to ${numPersonas} user personas that would send ${prompts.length > 1 ? 'these prompts' : 'this prompt'}. Your response should be JSON of the form {personas: string[]}`;
239
+ }
240
+ function testCasesPrompt(prompts, persona, tests, numTestCasesPerPersona, variables, instructions) {
241
+ const promptsString = (0, dedent_1.default) `
242
+ <Prompts>
243
+ ${prompts
244
+ .map((prompt) => (0, dedent_1.default) `
245
+ <Prompt>
246
+ ${prompt}
247
+ </Prompt>`)
248
+ .join('\n')}
249
+ </Prompts>`;
250
+ const existingTests = (0, dedent_1.default) `
251
+ Here are some existing tests:
252
+ ${(0, generation_1.sampleArray)(tests, 100)
253
+ .map((test) => {
254
+ if (!test.vars) {
255
+ return;
256
+ }
257
+ return (0, dedent_1.default) `
258
+ <Test>
259
+ ${JSON.stringify(test.vars, null, 2)}
260
+ </Test>`;
261
+ })
262
+ .filter(Boolean)
263
+ .sort()
264
+ .join('\n')}
265
+ `;
266
+ return (0, dedent_1.default) `
267
+ Consider ${prompts.length > 1 ? 'these prompts' : 'this prompt'}, which contains some {{variables}}:
268
+ ${promptsString}
269
+
270
+ This is your persona:
271
+ <Persona>
272
+ ${persona}
273
+ </Persona>
274
+
275
+ ${existingTests}
276
+
277
+ Fully embody this persona and determine a value for each variable, such that the prompt would be sent by this persona.
278
+
279
+ You are a tester, so try to think of ${numTestCasesPerPersona} sets of values that would be interesting or unusual to test.${instructions ? ` ${instructions}` : ''}
280
+
281
+ Your response should contain a JSON map of variable names to values, of the form {vars: {${Array.from(variables)
282
+ .map((varName) => `${varName}: string`)
283
+ .join(', ')}}[]}`;
284
+ }
243
285
  async function synthesize({ prompts, instructions, tests, numPersonas, numTestCasesPerPersona, provider, }) {
244
286
  if (prompts.length < 1) {
245
287
  throw new Error('Dataset synthesis requires at least one prompt.');
@@ -254,8 +296,7 @@ async function synthesize({ prompts, instructions, tests, numPersonas, numTestCa
254
296
  progressBar.start(totalProgressSteps, 0);
255
297
  }
256
298
  logger_1.default.debug(`Starting dataset synthesis. We'll begin by generating up to ${numPersonas} personas. Each persona will be used to generate ${numTestCasesPerPersona} test cases.`);
257
- // Consider the following prompt for an LLM application: {{prompt}}. List up to 5 user personas that would send this prompt.
258
- logger_1.default.debug(`\nGenerating user personas from ${prompts.length} prompts...`);
299
+ logger_1.default.debug(`Generating user personas from ${prompts.length} prompt${prompts.length > 1 ? 's' : ''}...`);
259
300
  let providerModel;
260
301
  if (typeof provider === 'undefined') {
261
302
  providerModel = (await (0, defaults_1.getDefaultProviders)()).synthesizeProvider;
@@ -263,85 +304,54 @@ async function synthesize({ prompts, instructions, tests, numPersonas, numTestCa
263
304
  else {
264
305
  providerModel = await (0, providers_1.loadApiProvider)(provider);
265
306
  }
266
- const promptsString = (0, dedent_1.default) `<Prompts>
267
- ${prompts.map((prompt) => `<Prompt>\n${prompt}\n</Prompt>`).join('\n')}
268
- </Prompts>`;
269
- const resp = await providerModel.callApi((0, dedent_1.default) `
270
- Consider the following prompt${prompts.length > 1 ? 's' : ''} for an LLM application:
271
-
272
- ${promptsString}
273
-
274
- List up to ${numPersonas} user personas that would send ${prompts.length > 1 ? 'these prompts' : 'this prompt'}. Your response should be JSON of the form {personas: string[]}`);
275
- const respObjects = (0, util_1.extractJsonObjects)(resp.output);
307
+ const personasPrompt = generatePersonasPrompt(prompts, numPersonas);
308
+ logger_1.default.debug(`Generated personas prompt:\n${personasPrompt}`);
309
+ const resp = await providerModel.callApi(personasPrompt);
310
+ logger_1.default.debug(`Received personas response:\n${resp.output}`);
311
+ (0, tiny_invariant_1.default)(typeof resp.output !== 'undefined', 'resp.output must be defined');
312
+ const output = typeof resp.output === 'string' ? resp.output : JSON.stringify(resp.output);
313
+ const respObjects = (0, json_1.extractJsonObjects)(output);
276
314
  (0, tiny_invariant_1.default)(respObjects.length === 1, 'Expected exactly one JSON object in the response');
277
315
  const personas = respObjects[0].personas;
278
- logger_1.default.debug(`\nGenerated ${personas.length} personas:\n${personas.map((p) => ` - ${p}`).join('\n')}`);
316
+ logger_1.default.debug(`Generated ${personas.length} persona${personas.length === 1 ? '' : 's'}:\n${personas.map((p) => ` - ${p}`).join('\n')}`);
279
317
  if (progressBar) {
280
318
  progressBar.increment();
281
319
  }
282
320
  // Extract variable names from the nunjucks template in the prompts
283
321
  const variables = (0, templates_1.extractVariablesFromTemplates)(prompts);
284
- logger_1.default.debug(`\nExtracted ${variables.length} variables from prompts:\n${variables
322
+ logger_1.default.debug(`Extracted ${variables.length} variable${variables.length === 1 ? '' : 's'} from prompt${prompts.length === 1 ? '' : 's'}:\n${variables
285
323
  .map((v) => ` - ${v}`)
286
324
  .join('\n')}`);
287
- const existingTests = (0, dedent_1.default) `Here are some existing tests:` +
288
- tests
289
- .map((test) => {
290
- if (!test.vars) {
291
- return;
292
- }
293
- return (0, dedent_1.default) `<Test>
294
- ${JSON.stringify(test.vars, null, 2)}
295
- </Test>`;
296
- })
297
- .filter(Boolean)
298
- .slice(0, 100)
299
- .join('\n');
300
- // For each user persona, we will generate a map of variable names to values
301
- const testCaseVars = [];
302
- for (let i = 0; i < personas.length; i++) {
303
- const persona = personas[i];
304
- logger_1.default.debug(`\nGenerating test cases for persona ${i + 1}...`);
305
- // Construct the prompt for the LLM to generate variable values
306
- const personaPrompt = (0, dedent_1.default) `
307
-
308
- Consider ${prompts.length > 1 ? 'these prompts' : 'this prompt'}, which contains some {{variables}}:
309
- ${promptsString}
310
-
311
- This is your persona:
312
- <Persona>
313
- ${persona}
314
- </Persona>
315
-
316
- ${existingTests}
317
-
318
- Fully embody this persona and determine a value for each variable, such that the prompt would be sent by this persona.
319
-
320
- You are a tester, so try to think of ${numTestCasesPerPersona} sets of values that would be interesting or unusual to test. ${instructions || ''}
321
-
322
- Your response should contain a JSON map of variable names to values, of the form {vars: {${Array.from(variables)
323
- .map((varName) => `${varName}: string`)
324
- .join(', ')}}[]}`;
325
- // Call the LLM API with the constructed prompt
325
+ const batchSize = 20;
326
+ const totalTestCases = numPersonas * numTestCasesPerPersona;
327
+ const generateTestCasesForPersona = async (currentTestCases) => {
328
+ const remainingCount = totalTestCases - currentTestCases.length;
329
+ const currentBatchSize = Math.min(remainingCount, batchSize);
330
+ const persona = personas[currentTestCases.length % personas.length];
331
+ logger_1.default.debug(`Generating ${currentBatchSize} test cases for persona ${(currentTestCases.length % personas.length) + 1} of ${personas.length}...`);
332
+ const personaPrompt = testCasesPrompt(prompts, persona, tests, currentBatchSize, variables, instructions);
333
+ logger_1.default.debug(`Generated persona prompt:\n${personaPrompt}`);
326
334
  const personaResponse = await providerModel.callApi(personaPrompt);
327
- const personaResponseObjects = (0, util_1.extractJsonObjects)(personaResponse.output);
328
- (0, tiny_invariant_1.default)(personaResponseObjects.length === 1, `Expected exactly one JSON object in the response for persona ${i + 1}`);
335
+ logger_1.default.debug(`Received persona response:\n${personaResponse.output}`);
336
+ const personaResponseObjects = (0, json_1.extractJsonObjects)(personaResponse.output);
337
+ (0, tiny_invariant_1.default)(personaResponseObjects.length === 1, `Expected exactly one JSON object in the response for persona ${persona}`);
329
338
  const parsed = personaResponseObjects[0];
330
- for (const vars of parsed.vars) {
331
- logger_1.default.debug(`${JSON.stringify(vars, null, 2)}`);
332
- testCaseVars.push(vars);
333
- if (progressBar) {
334
- progressBar.increment();
335
- }
339
+ logger_1.default.debug(`Received ${parsed.vars.length} test cases`);
340
+ if (progressBar) {
341
+ progressBar.increment(parsed.vars.length);
336
342
  }
343
+ return parsed.vars || [];
344
+ };
345
+ let testCaseVars = await (0, generation_1.retryWithDeduplication)(generateTestCasesForPersona, totalTestCases);
346
+ logger_1.default.debug(`Generated ${testCaseVars.length} test cases`);
347
+ if (testCaseVars.length > totalTestCases) {
348
+ logger_1.default.debug(`Generated ${testCaseVars.length} test cases, but only ${totalTestCases} were requested. Sampling down to ${totalTestCases}...`);
349
+ testCaseVars = (0, generation_1.sampleArray)(testCaseVars, totalTestCases);
337
350
  }
338
351
  if (progressBar) {
339
352
  progressBar.stop();
340
353
  }
341
- // Dedup test case vars
342
- const uniqueTestCaseStrings = new Set(testCaseVars.map((testCase) => JSON.stringify(testCase)));
343
- const dedupedTestCaseVars = Array.from(uniqueTestCaseStrings).map((testCase) => JSON.parse(testCase));
344
- return dedupedTestCaseVars;
354
+ return testCaseVars;
345
355
  }
346
356
  async function synthesizeFromTestSuite(testSuite, options) {
347
357
  return synthesize({