promptfoo 0.103.3 → 0.103.5

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 (160) hide show
  1. package/LICENSE +10 -1
  2. package/dist/package.json +13 -13
  3. package/dist/src/app/assets/index-BR1tgrAf.css +1 -0
  4. package/dist/src/app/assets/{index-XXoiz61D.js → index-CmPQAxfe.js} +276 -276
  5. package/dist/src/app/assets/{index.es-DTKpmNcZ.js → index.es-DfqJ7zdu.js} +1 -1
  6. package/dist/src/app/assets/{sync-ClbHj3jr.js → sync-C-aW1Mpw.js} +1 -1
  7. package/dist/src/app/index.html +2 -2
  8. package/dist/src/assertions/index.d.ts +3 -2
  9. package/dist/src/assertions/index.d.ts.map +1 -1
  10. package/dist/src/assertions/index.js +21 -6
  11. package/dist/src/assertions/index.js.map +1 -1
  12. package/dist/src/assertions/utils.d.ts +6 -2
  13. package/dist/src/assertions/utils.d.ts.map +1 -1
  14. package/dist/src/commands/eval/filterErrorTests.d.ts +5 -0
  15. package/dist/src/commands/eval/filterErrorTests.d.ts.map +1 -0
  16. package/dist/src/commands/eval/filterErrorTests.js +19 -0
  17. package/dist/src/commands/eval/filterErrorTests.js.map +1 -0
  18. package/dist/src/commands/eval/filterTests.d.ts +1 -0
  19. package/dist/src/commands/eval/filterTests.d.ts.map +1 -1
  20. package/dist/src/commands/eval/filterTests.js +4 -0
  21. package/dist/src/commands/eval/filterTests.js.map +1 -1
  22. package/dist/src/commands/eval.d.ts.map +1 -1
  23. package/dist/src/commands/eval.js +1 -0
  24. package/dist/src/commands/eval.js.map +1 -1
  25. package/dist/src/database/tables.d.ts +51 -12
  26. package/dist/src/database/tables.d.ts.map +1 -1
  27. package/dist/src/envars.d.ts +1 -0
  28. package/dist/src/envars.d.ts.map +1 -1
  29. package/dist/src/envars.js.map +1 -1
  30. package/dist/src/evaluator.d.ts.map +1 -1
  31. package/dist/src/evaluator.js +1 -0
  32. package/dist/src/evaluator.js.map +1 -1
  33. package/dist/src/fetch.d.ts.map +1 -1
  34. package/dist/src/fetch.js +20 -3
  35. package/dist/src/fetch.js.map +1 -1
  36. package/dist/src/models/evalResult.d.ts.map +1 -1
  37. package/dist/src/models/evalResult.js +9 -1
  38. package/dist/src/models/evalResult.js.map +1 -1
  39. package/dist/src/providers/browser.js +1 -1
  40. package/dist/src/providers/browser.js.map +1 -1
  41. package/dist/src/providers/defaults.d.ts +1 -0
  42. package/dist/src/providers/defaults.d.ts.map +1 -1
  43. package/dist/src/providers/defaults.js +11 -0
  44. package/dist/src/providers/defaults.js.map +1 -1
  45. package/dist/src/providers/http.d.ts.map +1 -1
  46. package/dist/src/providers/http.js +39 -63
  47. package/dist/src/providers/http.js.map +1 -1
  48. package/dist/src/providers/llama.d.ts.map +1 -1
  49. package/dist/src/providers/llama.js +8 -1
  50. package/dist/src/providers/llama.js.map +1 -1
  51. package/dist/src/providers/openai.d.ts.map +1 -1
  52. package/dist/src/providers/openai.js +6 -13
  53. package/dist/src/providers/openai.js.map +1 -1
  54. package/dist/src/providers/watsonx.d.ts.map +1 -1
  55. package/dist/src/providers/watsonx.js +9 -0
  56. package/dist/src/providers/watsonx.js.map +1 -1
  57. package/dist/src/providers.d.ts.map +1 -1
  58. package/dist/src/providers.js +15 -0
  59. package/dist/src/providers.js.map +1 -1
  60. package/dist/src/redteam/commands/generate.d.ts.map +1 -1
  61. package/dist/src/redteam/commands/generate.js +4 -0
  62. package/dist/src/redteam/commands/generate.js.map +1 -1
  63. package/dist/src/redteam/constants.d.ts +4 -2
  64. package/dist/src/redteam/constants.d.ts.map +1 -1
  65. package/dist/src/redteam/constants.js +11 -7
  66. package/dist/src/redteam/constants.js.map +1 -1
  67. package/dist/src/redteam/plugins/base.d.ts.map +1 -1
  68. package/dist/src/redteam/plugins/base.js +3 -0
  69. package/dist/src/redteam/plugins/base.js.map +1 -1
  70. package/dist/src/redteam/plugins/cyberseceval.d.ts.map +1 -1
  71. package/dist/src/redteam/plugins/cyberseceval.js +13 -3
  72. package/dist/src/redteam/plugins/cyberseceval.js.map +1 -1
  73. package/dist/src/redteam/providers/crescendo/index.d.ts +1 -0
  74. package/dist/src/redteam/providers/crescendo/index.d.ts.map +1 -1
  75. package/dist/src/redteam/providers/crescendo/index.js +58 -3
  76. package/dist/src/redteam/providers/crescendo/index.js.map +1 -1
  77. package/dist/src/redteam/providers/iterative.d.ts.map +1 -1
  78. package/dist/src/redteam/providers/iterative.js +59 -5
  79. package/dist/src/redteam/providers/iterative.js.map +1 -1
  80. package/dist/src/redteam/providers/iterativeImage.d.ts +6 -2
  81. package/dist/src/redteam/providers/iterativeImage.d.ts.map +1 -1
  82. package/dist/src/redteam/providers/iterativeImage.js +322 -131
  83. package/dist/src/redteam/providers/iterativeImage.js.map +1 -1
  84. package/dist/src/redteam/providers/iterativeTree.d.ts +37 -26
  85. package/dist/src/redteam/providers/iterativeTree.d.ts.map +1 -1
  86. package/dist/src/redteam/providers/iterativeTree.js +193 -85
  87. package/dist/src/redteam/providers/iterativeTree.js.map +1 -1
  88. package/dist/src/redteam/shared.d.ts.map +1 -1
  89. package/dist/src/redteam/shared.js +4 -1
  90. package/dist/src/redteam/shared.js.map +1 -1
  91. package/dist/src/server/routes/providers.js +11 -6
  92. package/dist/src/server/routes/providers.js.map +1 -1
  93. package/dist/src/types/env.d.ts +3 -0
  94. package/dist/src/types/env.d.ts.map +1 -1
  95. package/dist/src/types/index.d.ts +1376 -351
  96. package/dist/src/types/index.d.ts.map +1 -1
  97. package/dist/src/types/index.js +4 -1
  98. package/dist/src/types/index.js.map +1 -1
  99. package/dist/src/types/providers.d.ts +22 -0
  100. package/dist/src/types/providers.d.ts.map +1 -1
  101. package/dist/src/types/providers.js.map +1 -1
  102. package/dist/src/util/config/manage.d.ts +1 -1
  103. package/dist/src/util/config/manage.d.ts.map +1 -1
  104. package/dist/src/util/config/manage.js.map +1 -1
  105. package/dist/src/util/convertEvalResultsToTable.d.ts.map +1 -1
  106. package/dist/src/util/convertEvalResultsToTable.js +14 -0
  107. package/dist/src/util/convertEvalResultsToTable.js.map +1 -1
  108. package/dist/src/util/index.d.ts +12 -4
  109. package/dist/src/util/index.d.ts.map +1 -1
  110. package/dist/src/validators/providers.d.ts +71 -2
  111. package/dist/src/validators/providers.d.ts.map +1 -1
  112. package/dist/src/validators/providers.js +3 -0
  113. package/dist/src/validators/providers.js.map +1 -1
  114. package/dist/src/validators/redteam.d.ts +24 -0
  115. package/dist/src/validators/redteam.d.ts.map +1 -1
  116. package/dist/test/assertions/index.test.js +26 -475
  117. package/dist/test/assertions/index.test.js.map +1 -1
  118. package/dist/test/assertions/javascript.test.d.ts +2 -0
  119. package/dist/test/assertions/javascript.test.d.ts.map +1 -0
  120. package/dist/test/assertions/javascript.test.js +679 -0
  121. package/dist/test/assertions/javascript.test.js.map +1 -0
  122. package/dist/test/assertions/python.test.d.ts +2 -0
  123. package/dist/test/assertions/python.test.d.ts.map +1 -0
  124. package/dist/test/assertions/python.test.js +377 -0
  125. package/dist/test/assertions/python.test.js.map +1 -0
  126. package/dist/test/cache.test.js +297 -100
  127. package/dist/test/cache.test.js.map +1 -1
  128. package/dist/test/commands/eval/filterErrorTests.test.d.ts +2 -0
  129. package/dist/test/commands/eval/filterErrorTests.test.d.ts.map +1 -0
  130. package/dist/test/commands/eval/filterErrorTests.test.js +110 -0
  131. package/dist/test/commands/eval/filterErrorTests.test.js.map +1 -0
  132. package/dist/test/evaluator.test.js +10 -0
  133. package/dist/test/evaluator.test.js.map +1 -1
  134. package/dist/test/factories/evalFactory.d.ts +39 -8
  135. package/dist/test/factories/evalFactory.d.ts.map +1 -1
  136. package/dist/test/fetch.test.js +147 -19
  137. package/dist/test/fetch.test.js.map +1 -1
  138. package/dist/test/models/eval.test.js +12 -0
  139. package/dist/test/models/eval.test.js.map +1 -1
  140. package/dist/test/providers/defaults.test.d.ts +2 -0
  141. package/dist/test/providers/defaults.test.d.ts.map +1 -0
  142. package/dist/test/providers/defaults.test.js +77 -0
  143. package/dist/test/providers/defaults.test.js.map +1 -0
  144. package/dist/test/providers/http.test.js +65 -9
  145. package/dist/test/providers/http.test.js.map +1 -1
  146. package/dist/test/providers/index.test.js +6 -3
  147. package/dist/test/providers/index.test.js.map +1 -1
  148. package/dist/test/providers/mistral.test.js +28 -19
  149. package/dist/test/providers/mistral.test.js.map +1 -1
  150. package/dist/test/providers/watsonx.test.js +58 -0
  151. package/dist/test/providers/watsonx.test.js.map +1 -1
  152. package/dist/test/redteam/providers/iterativeTree.test.js +329 -98
  153. package/dist/test/redteam/providers/iterativeTree.test.js.map +1 -1
  154. package/dist/test/server/providers.test.js +4 -4
  155. package/dist/test/server/providers.test.js.map +1 -1
  156. package/dist/test/util/config/main.test.js +3 -0
  157. package/dist/test/util/config/main.test.js.map +1 -1
  158. package/dist/tsconfig.tsbuildinfo +1 -1
  159. package/package.json +13 -13
  160. package/dist/src/app/assets/index-DdUNCsxz.css +0 -1
@@ -1 +1 @@
1
- {"version":3,"file":"mistral.test.js","sourceRoot":"","sources":["../../../test/providers/mistral.test.ts"],"names":[],"mappings":";;AAAA,2CAA2D;AAC3D,2CAAiD;AACjD,yDAGqC;AAErC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC;IACxC,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE9B,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,SAAS,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC/B,MAAM,EAAE,eAAe;gBACvB,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;gBACnD,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,QAAQ;aACf,CAAC;YACF,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,KAAK,EAAE;gBACL,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;aACf;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,IAAI,QAAuC,CAAC;QAE5C,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,GAAG,IAAI,uCAA6B,CAAC,cAAc,CAAC,CAAC;YAC7D,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,gBAAgB,EAAE;gBACzE,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAAC;gBAClD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC;iBAClD,CAAC;gBACF,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sDAAsD,CAAC;aACtF,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,aAAa;gBACrB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC;gBACpC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC/B,MAAM,EAAE,eAAe;oBACvB,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;oBACnD,MAAM,EAAE,IAAI;oBACZ,IAAI,EAAE,QAAQ;iBACf,CAAC;gBACF,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE;oBACL,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;oBACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;iBACf;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,eAAe;gBACvB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;oBACb,MAAM,EAAE,EAAE;iBACX;gBACD,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,EAAE,CAAC;gBACnD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACnD,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,cAAc;gBACtB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,KAAK,EAAE,kCAAkC;aAC1C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,cAAc,EAAE;gBACvE,MAAM,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE;aACvD,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAAE,CAAC;gBAC1D,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAE5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,+CAA+C,EAC/C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,cAAc,EAAE;gBACvE,MAAM,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE;aACzC,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAAE,CAAC;gBAC1D,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAE5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,+CAA+C,EAC/C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,IAAI,QAAkC,CAAC;QAEvC,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,GAAG,IAAI,kCAAwB,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC;iBAClD,CAAC;gBACF,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;aACrD,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,yCAAyC;aACtF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE1F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,UAAU,EAAE,CAAC;oBACb,MAAM,EAAE,CAAC;iBACV;aACF,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACnD,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEzD,MAAM,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG,IAAI,kCAAwB,CAAC;gBAClD,MAAM,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE;aACvD,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,cAAc,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,yCAAyC,EACzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"mistral.test.js","sourceRoot":"","sources":["../../../test/providers/mistral.test.ts"],"names":[],"mappings":";;AAAA,2CAA2D;AAC3D,2CAAiD;AACjD,yDAGqC;AAErC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC;IACxC,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE9B,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,SAAS,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;YACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,KAAK,EAAE;gBACL,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;aACf;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,IAAI,QAAuC,CAAC;QAE5C,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,GAAG,IAAI,uCAA6B,CAAC,cAAc,CAAC,CAAC;YAC7D,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,gBAAgB,EAAE;gBACzE,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAAC;gBAClD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,qBAAqB,CAAC;gBAChD,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,KAAK;gBACb,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC;iBAClD,CAAC;gBACF,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sDAAsD,CAAC;aACtF,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,aAAa;gBACrB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC;gBACpC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBAC/B,MAAM,EAAE,eAAe;oBACvB,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;oBACnD,MAAM,EAAE,IAAI;oBACZ,IAAI,EAAE,QAAQ;iBACf,CAAC;gBACF,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE;oBACL,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;oBACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;iBACf;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,eAAe;gBACvB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;oBACb,MAAM,EAAE,EAAE;iBACX;gBACD,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,EAAE,CAAC;gBACnD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,qBAAqB,CAAC;gBAChD,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,KAAK;gBACb,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,cAAc;gBACtB,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,MAAM,EAAE,KAAK;gBACb,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACnD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,KAAK,EAAE,kCAAkC;aAC1C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,cAAc,EAAE;gBACvE,MAAM,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE;aACvD,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAAE,CAAC;gBAC1D,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,qBAAqB,CAAC;gBAChD,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,KAAK;gBACb,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAE5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,+CAA+C,EAC/C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,cAAc,GAAG,IAAI,uCAA6B,CAAC,cAAc,EAAE;gBACvE,MAAM,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE;aACzC,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAAE,CAAC;gBAC1D,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,qBAAqB,CAAC;gBAChD,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,KAAK;gBACb,MAAM,EAAE,GAAG;gBACX,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,MAAM,cAAc,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAE5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,+CAA+C,EAC/C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,IAAI,QAAkC,CAAC;QAEvC,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,GAAG,IAAI,kCAAwB,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC;iBAClD,CAAC;gBACF,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;aACrD,CAAC,EACF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,yCAAyC;aACtF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE1F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,UAAU,EAAE,CAAC;oBACb,MAAM,EAAE,CAAC;iBACV;aACF,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACnD,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAC1B,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aACzB,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,EACtC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEzD,MAAM,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG,IAAI,kCAAwB,CAAC;gBAClD,MAAM,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE;aACvD,CAAC,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YACxE,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;aAC7C,CAAC;YACF,IAAI;iBACD,MAAM,CAAC,sBAAc,CAAC;iBACtB,iBAAiB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3F,MAAM,cAAc,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,CAAC,oBAAoB,CACtD,yCAAyC,EACzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -177,6 +177,64 @@ describe('WatsonXProvider', () => {
177
177
  authenticator: expect.any(ibm_cloud_sdk_core_1.IamAuthenticator),
178
178
  });
179
179
  });
180
+ it('should use IAM Authentication when WATSONX_AI_AUTH_TYPE is set to iam', async () => {
181
+ const dualAuthConfig = { ...config, apiBearerToken: 'test-bearer-token' };
182
+ const mockedWatsonXAIClient = {
183
+ generateText: jest.fn(),
184
+ };
185
+ jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
186
+ const provider = new watsonx_1.WatsonXProvider(modelName, {
187
+ config: dualAuthConfig,
188
+ env: { WATSONX_AI_AUTH_TYPE: 'iam' },
189
+ });
190
+ await provider.getClient();
191
+ expect(logger_1.default.info).toHaveBeenCalledWith('Using IAM Authentication based on WATSONX_AI_AUTH_TYPE.');
192
+ expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
193
+ version: '2023-05-29',
194
+ serviceUrl: 'https://us-south.ml.cloud.ibm.com',
195
+ authenticator: expect.any(ibm_cloud_sdk_core_1.IamAuthenticator),
196
+ });
197
+ });
198
+ it('should use Bearer Token Authentication when WATSONX_AI_AUTH_TYPE is set to bearertoken', async () => {
199
+ const dualAuthConfig = {
200
+ ...config,
201
+ apiKey: 'test-api-key',
202
+ apiBearerToken: 'test-bearer-token',
203
+ };
204
+ const mockedWatsonXAIClient = {
205
+ generateText: jest.fn(),
206
+ };
207
+ jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
208
+ const provider = new watsonx_1.WatsonXProvider(modelName, {
209
+ config: dualAuthConfig,
210
+ env: { WATSONX_AI_AUTH_TYPE: 'bearertoken' },
211
+ });
212
+ await provider.getClient();
213
+ expect(logger_1.default.info).toHaveBeenCalledWith('Using Bearer Token Authentication based on WATSONX_AI_AUTH_TYPE.');
214
+ expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
215
+ version: '2023-05-29',
216
+ serviceUrl: 'https://us-south.ml.cloud.ibm.com',
217
+ authenticator: expect.any(ibm_cloud_sdk_core_1.BearerTokenAuthenticator),
218
+ });
219
+ });
220
+ it('should fallback to default behavior when WATSONX_AI_AUTH_TYPE is invalid', async () => {
221
+ const dualAuthConfig = { ...config, apiBearerToken: 'test-bearer-token' };
222
+ const mockedWatsonXAIClient = {
223
+ generateText: jest.fn(),
224
+ };
225
+ jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
226
+ const provider = new watsonx_1.WatsonXProvider(modelName, {
227
+ config: dualAuthConfig,
228
+ env: { WATSONX_AI_AUTH_TYPE: 'invalid' },
229
+ });
230
+ await provider.getClient();
231
+ expect(logger_1.default.info).toHaveBeenCalledWith('Using IAM Authentication.');
232
+ expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
233
+ version: '2023-05-29',
234
+ serviceUrl: 'https://us-south.ml.cloud.ibm.com',
235
+ authenticator: expect.any(ibm_cloud_sdk_core_1.IamAuthenticator),
236
+ });
237
+ });
180
238
  });
181
239
  describe('callApi', () => {
182
240
  it('should call generateText with correct parameters and return the correct response', async () => {
@@ -1 +1 @@
1
- {"version":3,"file":"watsonx.test.js","sourceRoot":"","sources":["../../../test/providers/watsonx.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,sDAAkD;AAClD,2DAAgF;AAChF,2CAA2D;AAC3D,+DAAiD;AACjD,8DAAsC;AACtC,yDAAkF;AAElF,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,SAAS,EAAE;QACT,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KACvB;CACF,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;CACpC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;CAChB,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;IACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,SAAS,GAAG,YAAY,CAAC;IAC/B,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,cAAc;QACtB,SAAS,EAAE,iBAAiB;QAC5B,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,EAAE;KACjB,CAAC;IACF,MAAM,MAAM,GAAG,aAAa,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEhD,uCAAuC;YACvC,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE;QAClB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,SAAS,GAAG,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE1C,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;YAEH,MAAM,CAAC,qCAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,SAAgB,CAAC,CAAC;YAE3E,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,SAAS,EAAE;aACpE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAChD,uKAAuK,CACxK,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,iBAAiB,GAAG;gBACxB,GAAG,MAAM;gBACT,MAAM,EAAE,SAAS;gBACjB,cAAc,EAAE,mBAAmB;aACpC,CAAC;YACF,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAC/E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oCAAoC,CAAC,CAAC;YAC/E,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,6CAAwB,CAAC;aACpD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8FAA8F,EAAE,KAAK,IAAI,EAAE;YAC5G,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC5E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;YAChG,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBACxC,MAAM,EAAE;wBACN,QAAQ,EAAE,gBAAgB;wBAC1B,aAAa,EAAE,OAAO;wBACtB,UAAU,EAAE,sBAAsB;wBAClC,OAAO,EAAE;4BACP;gCACE,cAAc,EAAE,4BAA4B;gCAC5C,qBAAqB,EAAE,EAAE;gCACzB,iBAAiB,EAAE,CAAC;gCACpB,WAAW,EAAE,YAAY;6BAC1B;yBACF;qBACF;iBACF,CAAC;aACH,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBAC9D,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,UAAU,EAAE;oBACV,cAAc,EAAE,MAAM,CAAC,YAAY;iBACpC;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,4BAA4B;gBACpC,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACpC,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAC9D,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,iBAAiB;gBACzB,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC;YAEF,MAAM,QAAQ,GAAG,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAChF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBAChE,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,SAAS,EAAE,EAAE,cAAc,CAAC,CAAC;YAC/E,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;aAClE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YACjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,kCAAkC;gBACzC,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YACH,MAAM,CAAC,gBAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,2CAA2C,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"watsonx.test.js","sourceRoot":"","sources":["../../../test/providers/watsonx.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,sDAAkD;AAClD,2DAAgF;AAChF,2CAA2D;AAC3D,+DAAiD;AACjD,8DAAsC;AACtC,yDAAkF;AAElF,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,SAAS,EAAE;QACT,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KACvB;CACF,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;CACpC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;CAChB,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;IACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,SAAS,GAAG,YAAY,CAAC;IAC/B,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,cAAc;QACtB,SAAS,EAAE,iBAAiB;QAC5B,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,EAAE;KACjB,CAAC;IACF,MAAM,MAAM,GAAG,aAAa,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEhD,uCAAuC;YACvC,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE;QAClB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,SAAS,GAAG,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE1C,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;YAEH,MAAM,CAAC,qCAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,SAAgB,CAAC,CAAC;YAE3E,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,SAAS,EAAE;aACpE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAChD,uKAAuK,CACxK,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,iBAAiB,GAAG;gBACxB,GAAG,MAAM;gBACT,MAAM,EAAE,SAAS;gBACjB,cAAc,EAAE,mBAAmB;aACpC,CAAC;YACF,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAC/E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oCAAoC,CAAC,CAAC;YAC/E,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,6CAAwB,CAAC;aACpD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8FAA8F,EAAE,KAAK,IAAI,EAAE;YAC5G,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC5E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,cAAc;gBACtB,GAAG,EAAE,EAAE,oBAAoB,EAAE,KAAK,EAAE;aACrC,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,yDAAyD,CAC1D,CAAC;YACF,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;YACtG,MAAM,cAAc,GAAG;gBACrB,GAAG,MAAM;gBACT,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE,mBAAmB;aACpC,CAAC;YACF,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,cAAc;gBACtB,GAAG,EAAE,EAAE,oBAAoB,EAAE,aAAa,EAAE;aAC7C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,kEAAkE,CACnE,CAAC;YACF,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,6CAAwB,CAAC;aACpD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,cAAc;gBACtB,GAAG,EAAE,EAAE,oBAAoB,EAAE,SAAS,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;YAChG,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBACxC,MAAM,EAAE;wBACN,QAAQ,EAAE,gBAAgB;wBAC1B,aAAa,EAAE,OAAO;wBACtB,UAAU,EAAE,sBAAsB;wBAClC,OAAO,EAAE;4BACP;gCACE,cAAc,EAAE,4BAA4B;gCAC5C,qBAAqB,EAAE,EAAE;gCACzB,iBAAiB,EAAE,CAAC;gCACpB,WAAW,EAAE,YAAY;6BAC1B;yBACF;qBACF;iBACF,CAAC;aACH,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBAC9D,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,UAAU,EAAE;oBACV,cAAc,EAAE,MAAM,CAAC,YAAY;iBACpC;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,4BAA4B;gBACpC,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACpC,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAC9D,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,iBAAiB;gBACzB,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC;YAEF,MAAM,QAAQ,GAAG,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAChF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBAChE,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,SAAS,EAAE,EAAE,cAAc,CAAC,CAAC;YAC/E,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;aAClE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YACjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,kCAAkC;gBACzC,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YACH,MAAM,CAAC,gBAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,2CAA2C,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -1,6 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  const globals_1 = require("@jest/globals");
4
+ const uuid_1 = require("uuid");
4
5
  const iterativeTree_1 = require("../../../src/redteam/providers/iterativeTree");
5
6
  const prompts_1 = require("../../../src/redteam/providers/prompts");
6
7
  const shared_1 = require("../../../src/redteam/providers/shared");
@@ -171,104 +172,6 @@ describe('RedteamIterativeProvider', () => {
171
172
  await expect((0, iterativeTree_1.checkIfOnTopic)(mockRedteamProvider, 'On-topic system prompt', 'Target prompt')).rejects.toThrow('Invariant failed: Expected onTopic to be a boolean');
172
173
  });
173
174
  });
174
- describe('calculateBranches', () => {
175
- it('should calculate branches correctly for various scores and depths', () => {
176
- // Test base case
177
- expect((0, iterativeTree_1.calculateBranches)(5, 0)).toBe(3);
178
- // Test high scores
179
- expect((0, iterativeTree_1.calculateBranches)(8, 0)).toBe(5);
180
- expect((0, iterativeTree_1.calculateBranches)(9, 0)).toBe(5);
181
- expect((0, iterativeTree_1.calculateBranches)(6, 0)).toBe(4);
182
- expect((0, iterativeTree_1.calculateBranches)(7, 0)).toBe(4);
183
- // Test low scores
184
- expect((0, iterativeTree_1.calculateBranches)(3, 0)).toBe(2);
185
- expect((0, iterativeTree_1.calculateBranches)(2, 0)).toBe(2);
186
- // Test depth impact
187
- expect((0, iterativeTree_1.calculateBranches)(5, 2)).toBe(2);
188
- expect((0, iterativeTree_1.calculateBranches)(5, 4)).toBe(1);
189
- expect((0, iterativeTree_1.calculateBranches)(8, 2)).toBe(4);
190
- expect((0, iterativeTree_1.calculateBranches)(8, 4)).toBe(3);
191
- // Test minimum branches
192
- expect((0, iterativeTree_1.calculateBranches)(1, 6)).toBe(1);
193
- // Test maximum branches
194
- expect((0, iterativeTree_1.calculateBranches)(10, 0)).toBe(5);
195
- });
196
- it('should never return less than MIN_BRANCHES', () => {
197
- expect((0, iterativeTree_1.calculateBranches)(1, 10)).toBe(1);
198
- });
199
- it('should never return more than MAX_BRANCHES', () => {
200
- expect((0, iterativeTree_1.calculateBranches)(10, 0)).toBe(5);
201
- });
202
- it('should decrease branches as depth increases', () => {
203
- const score = 7;
204
- const branchesAtDepth0 = (0, iterativeTree_1.calculateBranches)(score, 0);
205
- const branchesAtDepth2 = (0, iterativeTree_1.calculateBranches)(score, 2);
206
- const branchesAtDepth4 = (0, iterativeTree_1.calculateBranches)(score, 4);
207
- expect(branchesAtDepth2).toBeLessThan(branchesAtDepth0);
208
- expect(branchesAtDepth4).toBeLessThan(branchesAtDepth2);
209
- });
210
- });
211
- describe('calculateSimilarity', () => {
212
- it('should calculate similarity correctly', () => {
213
- expect((0, iterativeTree_1.calculateSimilarity)('hello world', 'hello there')).toBeCloseTo(0.33, 2);
214
- expect((0, iterativeTree_1.calculateSimilarity)('test prompt', 'test prompt')).toBe(1);
215
- expect((0, iterativeTree_1.calculateSimilarity)('completely different', 'totally unrelated')).toBe(0);
216
- });
217
- it('should be case-insensitive', () => {
218
- expect((0, iterativeTree_1.calculateSimilarity)('Hello World', 'hello world')).toBe(1);
219
- });
220
- it('should handle empty strings', () => {
221
- expect((0, iterativeTree_1.calculateSimilarity)('', '')).toBe(1);
222
- expect((0, iterativeTree_1.calculateSimilarity)('test', '')).toBe(0);
223
- });
224
- it('should handle different word orders', () => {
225
- expect((0, iterativeTree_1.calculateSimilarity)('world hello', 'hello world')).toBeGreaterThan(0.8);
226
- });
227
- });
228
- describe('selectDiverseBestNodes', () => {
229
- it('should select diverse best nodes', () => {
230
- const nodes = [
231
- { prompt: 'test 1', score: 8, children: [], depth: 0 },
232
- { prompt: 'test 2', score: 7, children: [], depth: 0 },
233
- { prompt: 'very different', score: 6, children: [], depth: 0 },
234
- { prompt: 'test 3', score: 5, children: [], depth: 0 },
235
- ];
236
- const result = (0, iterativeTree_1.selectDiverseBestNodes)(nodes, 3);
237
- expect(result).toHaveLength(3);
238
- expect(result).toEqual([
239
- expect.objectContaining({ score: 8 }),
240
- expect.objectContaining({ score: 7 }),
241
- expect.objectContaining({ prompt: 'very different' }),
242
- ]);
243
- });
244
- it('should handle cases where there are fewer nodes than requested', () => {
245
- const nodes = [
246
- { prompt: 'test 1', score: 8, children: [], depth: 0 },
247
- { prompt: 'test 2', score: 7, children: [], depth: 0 },
248
- ];
249
- const result = (0, iterativeTree_1.selectDiverseBestNodes)(nodes, 3);
250
- expect(result).toHaveLength(2);
251
- expect(result).toEqual([
252
- expect.objectContaining({ score: 8 }),
253
- expect.objectContaining({ score: 7 }),
254
- ]);
255
- });
256
- it('should select best remaining nodes if not enough diverse nodes', () => {
257
- const nodes = [
258
- { prompt: 'test 1', score: 8, children: [], depth: 0 },
259
- { prompt: 'test 2', score: 7, children: [], depth: 0 },
260
- { prompt: 'test 3', score: 6, children: [], depth: 0 },
261
- { prompt: 'test 4', score: 5, children: [], depth: 0 },
262
- ];
263
- const result = (0, iterativeTree_1.selectDiverseBestNodes)(nodes, 3);
264
- expect(result).toHaveLength(3);
265
- expect(result).toEqual([
266
- expect.objectContaining({ score: 8 }),
267
- expect.objectContaining({ score: 7 }),
268
- expect.objectContaining({ score: 6 }),
269
- ]);
270
- });
271
- });
272
175
  describe('updateRedteamHistory', () => {
273
176
  it('should create a new history entry with correct content', () => {
274
177
  const targetResponse = 'Test response';
@@ -331,6 +234,7 @@ describe('RedteamIterativeProvider', () => {
331
234
  let mockTargetProvider;
332
235
  beforeEach(() => {
333
236
  mockTargetProvider = {
237
+ id: globals_1.jest.fn().mockReturnValue('mock-provider'),
334
238
  callApi: globals_1.jest.fn(),
335
239
  };
336
240
  });
@@ -365,4 +269,331 @@ describe('RedteamIterativeProvider', () => {
365
269
  });
366
270
  });
367
271
  });
272
+ describe('TreeNode', () => {
273
+ describe('createTreeNode', () => {
274
+ it('should create a node with unique UUID', () => {
275
+ const node1 = (0, iterativeTree_1.createTreeNode)('prompt1', 5, 0);
276
+ const node2 = (0, iterativeTree_1.createTreeNode)('prompt2', 5, 0);
277
+ expect(node1.id).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
278
+ expect(node2.id).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
279
+ expect(node1.id).not.toBe(node2.id);
280
+ });
281
+ it('should use provided UUID if given', () => {
282
+ const customId = (0, uuid_1.v4)();
283
+ const node = (0, iterativeTree_1.createTreeNode)('prompt', 5, 0, customId);
284
+ expect(node.id).toBe(customId);
285
+ });
286
+ });
287
+ });
288
+ describe('Tree Structure', () => {
289
+ it('should track parent-child relationships in treeOutputs', async () => {
290
+ const parentId = (0, uuid_1.v4)();
291
+ const childId = (0, uuid_1.v4)();
292
+ const parentNode = (0, iterativeTree_1.createTreeNode)('parent', 5, 0, parentId);
293
+ const childNode = (0, iterativeTree_1.createTreeNode)('child', 7, 1, childId);
294
+ const treeOutputs = [];
295
+ treeOutputs.push({
296
+ depth: 0,
297
+ id: parentNode.id,
298
+ isOnTopic: true,
299
+ output: 'parent output',
300
+ prompt: 'parent prompt',
301
+ score: 5,
302
+ wasSelected: true,
303
+ });
304
+ treeOutputs.push({
305
+ depth: 1,
306
+ id: childNode.id,
307
+ improvement: 'test improvement',
308
+ isOnTopic: true,
309
+ output: 'child output',
310
+ parentId: parentNode.id,
311
+ prompt: 'child prompt',
312
+ score: 7,
313
+ wasSelected: false,
314
+ });
315
+ const childOutput = treeOutputs.find((o) => o.id === childNode.id);
316
+ expect(childOutput?.parentId).toBe(parentNode.id);
317
+ expect(childOutput?.depth).toBe(1);
318
+ expect(childOutput?.improvement).toBe('test improvement');
319
+ });
320
+ describe('selectNodes', () => {
321
+ let mockRedteamProvider;
322
+ beforeEach(() => {
323
+ mockRedteamProvider = {
324
+ callApi: globals_1.jest.fn().mockResolvedValue({
325
+ output: JSON.stringify({ onTopic: true }),
326
+ }),
327
+ id: globals_1.jest.fn().mockReturnValue('mock-provider'),
328
+ };
329
+ });
330
+ it('should mark selected nodes in treeOutputs', async () => {
331
+ const nodes = [
332
+ (0, iterativeTree_1.createTreeNode)('node1', 3, 0),
333
+ (0, iterativeTree_1.createTreeNode)('node2', 8, 0),
334
+ (0, iterativeTree_1.createTreeNode)('node3', 5, 0),
335
+ ];
336
+ const treeOutputs = nodes.map((node) => ({
337
+ depth: node.depth,
338
+ id: node.id,
339
+ isOnTopic: true,
340
+ output: 'test output',
341
+ prompt: node.prompt,
342
+ score: node.score,
343
+ wasSelected: false,
344
+ }));
345
+ const selectedNodes = await (0, iterativeTree_1.selectNodes)(nodes, mockRedteamProvider, 'test prompt', 'test goal');
346
+ selectedNodes.forEach((node) => {
347
+ const output = treeOutputs.find((o) => o.id === node.id);
348
+ if (output) {
349
+ output.wasSelected = true;
350
+ }
351
+ });
352
+ const selectedOutputs = treeOutputs.filter((o) => o.wasSelected);
353
+ expect(selectedOutputs.length).toBeLessThanOrEqual(iterativeTree_1.MAX_WIDTH);
354
+ const allSortedByScore = [...treeOutputs].sort((a, b) => b.score - a.score);
355
+ const expectedLength = Math.min(iterativeTree_1.MAX_WIDTH, allSortedByScore.length);
356
+ expect(selectedOutputs).toHaveLength(expectedLength);
357
+ const expectedScores = allSortedByScore.slice(0, expectedLength).map((n) => n.score);
358
+ const actualScores = selectedOutputs.map((n) => n.score).sort((a, b) => b - a);
359
+ expect(actualScores).toEqual(expectedScores);
360
+ });
361
+ });
362
+ describe('Tree Reconstruction', () => {
363
+ it('should be able to reconstruct tree from treeOutputs', () => {
364
+ const treeOutputs = [
365
+ {
366
+ depth: 0,
367
+ id: 'root',
368
+ isOnTopic: true,
369
+ output: 'root output',
370
+ prompt: 'root prompt',
371
+ score: 5,
372
+ wasSelected: true,
373
+ },
374
+ {
375
+ depth: 1,
376
+ id: 'child1',
377
+ improvement: 'improvement1',
378
+ isOnTopic: true,
379
+ output: 'child1 output',
380
+ parentId: 'root',
381
+ prompt: 'child1 prompt',
382
+ score: 7,
383
+ wasSelected: true,
384
+ },
385
+ {
386
+ depth: 1,
387
+ id: 'child2',
388
+ improvement: 'improvement2',
389
+ isOnTopic: true,
390
+ output: 'child2 output',
391
+ parentId: 'root',
392
+ prompt: 'child2 prompt',
393
+ score: 6,
394
+ wasSelected: false,
395
+ },
396
+ ];
397
+ function reconstructTree(outputs) {
398
+ const nodes = new Map();
399
+ outputs.forEach((output) => {
400
+ nodes.set(output.id, { children: [], output });
401
+ });
402
+ outputs.forEach((output) => {
403
+ if (output.parentId && nodes.has(output.parentId)) {
404
+ nodes.get(output.parentId)?.children.push(output.id);
405
+ }
406
+ });
407
+ return nodes;
408
+ }
409
+ const tree = reconstructTree(treeOutputs);
410
+ expect(tree.get('root')?.children).toHaveLength(2);
411
+ expect(tree.get('child1')?.output.wasSelected).toBe(true);
412
+ expect(tree.get('child2')?.output.wasSelected).toBe(false);
413
+ expect(tree.get('child1')?.output.improvement).toBe('improvement1');
414
+ });
415
+ });
416
+ });
417
+ describe('Tree Structure and Metadata', () => {
418
+ let mockRedteamProvider;
419
+ let mockTargetProvider;
420
+ beforeEach(() => {
421
+ mockRedteamProvider = {
422
+ id: globals_1.jest.fn().mockReturnValue('mock-provider'),
423
+ callApi: globals_1.jest.fn().mockResolvedValue({
424
+ output: JSON.stringify({ onTopic: true }),
425
+ }),
426
+ };
427
+ mockTargetProvider = {
428
+ id: globals_1.jest.fn().mockReturnValue('mock-provider'),
429
+ callApi: globals_1.jest.fn().mockResolvedValue({
430
+ output: 'test response',
431
+ }),
432
+ };
433
+ });
434
+ it('should track parent-child relationships in metadata', async () => {
435
+ const parentPrompt = 'parent prompt';
436
+ const childPrompt = 'child prompt';
437
+ const improvement = 'test improvement';
438
+ mockRedteamProvider.callApi
439
+ .mockResolvedValueOnce({ output: JSON.stringify({ onTopic: true }) })
440
+ .mockResolvedValueOnce({ output: JSON.stringify({ improvement, prompt: childPrompt }) });
441
+ mockTargetProvider.callApi.mockResolvedValue({ output: 'test response' });
442
+ const treeOutputs = [];
443
+ const parentNode = (0, iterativeTree_1.createTreeNode)(parentPrompt, 5, 0);
444
+ const childNode = (0, iterativeTree_1.createTreeNode)(childPrompt, 7, 1);
445
+ treeOutputs.push({
446
+ depth: 0,
447
+ id: parentNode.id,
448
+ isOnTopic: true,
449
+ output: 'parent output',
450
+ prompt: parentPrompt,
451
+ score: 5,
452
+ wasSelected: true,
453
+ });
454
+ treeOutputs.push({
455
+ depth: 1,
456
+ id: childNode.id,
457
+ improvement,
458
+ isOnTopic: true,
459
+ output: 'child output',
460
+ parentId: parentNode.id,
461
+ prompt: childPrompt,
462
+ score: 7,
463
+ wasSelected: false,
464
+ });
465
+ const childOutput = treeOutputs.find((o) => o.id === childNode.id);
466
+ expect(childOutput?.parentId).toBe(parentNode.id);
467
+ expect(childOutput?.depth).toBe(1);
468
+ expect(childOutput?.improvement).toBe(improvement);
469
+ });
470
+ it('should track tree structure across multiple depths', () => {
471
+ const rootId = (0, uuid_1.v4)();
472
+ const child1Id = (0, uuid_1.v4)();
473
+ const child2Id = (0, uuid_1.v4)();
474
+ const grandchild1Id = (0, uuid_1.v4)();
475
+ const treeOutputs = [
476
+ {
477
+ depth: 0,
478
+ id: rootId,
479
+ isOnTopic: true,
480
+ output: 'root output',
481
+ prompt: 'root prompt',
482
+ score: 5,
483
+ wasSelected: true,
484
+ },
485
+ {
486
+ depth: 1,
487
+ id: child1Id,
488
+ improvement: 'improvement1',
489
+ isOnTopic: true,
490
+ output: 'child1 output',
491
+ parentId: rootId,
492
+ prompt: 'child1 prompt',
493
+ score: 7,
494
+ wasSelected: true,
495
+ },
496
+ {
497
+ depth: 1,
498
+ id: child2Id,
499
+ improvement: 'improvement2',
500
+ isOnTopic: true,
501
+ output: 'child2 output',
502
+ parentId: rootId,
503
+ prompt: 'child2 prompt',
504
+ score: 6,
505
+ wasSelected: false,
506
+ },
507
+ {
508
+ depth: 2,
509
+ id: grandchild1Id,
510
+ improvement: 'improvement3',
511
+ isOnTopic: true,
512
+ output: 'grandchild1 output',
513
+ parentId: child1Id,
514
+ prompt: 'grandchild1 prompt',
515
+ score: 8,
516
+ wasSelected: true,
517
+ },
518
+ ];
519
+ function reconstructTree(outputs) {
520
+ const nodes = new Map();
521
+ outputs.forEach((output) => {
522
+ nodes.set(output.id, { children: [], output });
523
+ });
524
+ outputs.forEach((output) => {
525
+ if (output.parentId && nodes.has(output.parentId)) {
526
+ nodes.get(output.parentId)?.children.push(output.id);
527
+ }
528
+ });
529
+ return nodes;
530
+ }
531
+ const tree = reconstructTree(treeOutputs);
532
+ expect(tree.get(rootId)?.children).toHaveLength(2);
533
+ expect(tree.get(child1Id)?.children).toHaveLength(1);
534
+ expect(tree.get(child2Id)?.children).toHaveLength(0);
535
+ expect(tree.get(grandchild1Id)?.output.parentId).toBe(child1Id);
536
+ expect(tree.get(rootId)?.output.wasSelected).toBe(true);
537
+ expect(tree.get(child1Id)?.output.wasSelected).toBe(true);
538
+ expect(tree.get(child2Id)?.output.wasSelected).toBe(false);
539
+ expect(tree.get(grandchild1Id)?.output.wasSelected).toBe(true);
540
+ expect(tree.get(child1Id)?.output.improvement).toBe('improvement1');
541
+ expect(tree.get(child2Id)?.output.improvement).toBe('improvement2');
542
+ expect(tree.get(grandchild1Id)?.output.improvement).toBe('improvement3');
543
+ expect(tree.get(rootId)?.output.depth).toBe(0);
544
+ expect(tree.get(child1Id)?.output.depth).toBe(1);
545
+ expect(tree.get(child2Id)?.output.depth).toBe(1);
546
+ expect(tree.get(grandchild1Id)?.output.depth).toBe(2);
547
+ });
548
+ it('should validate metadata format', () => {
549
+ const metadata = {
550
+ attempts: 10,
551
+ highestScore: 8,
552
+ redteamFinalPrompt: 'final prompt',
553
+ stoppingReason: 'TARGET_SCORE',
554
+ treeOutputs: JSON.stringify([
555
+ {
556
+ depth: 0,
557
+ id: 'root',
558
+ isOnTopic: true,
559
+ output: 'root output',
560
+ prompt: 'root prompt',
561
+ score: 5,
562
+ wasSelected: true,
563
+ graderPassed: true,
564
+ },
565
+ {
566
+ depth: 1,
567
+ id: 'child',
568
+ improvement: 'improvement',
569
+ isOnTopic: true,
570
+ output: 'child output',
571
+ parentId: 'root',
572
+ prompt: 'child prompt',
573
+ score: 8,
574
+ wasSelected: true,
575
+ graderPassed: false,
576
+ },
577
+ ]),
578
+ };
579
+ expect(metadata).toHaveProperty('highestScore');
580
+ expect(metadata).toHaveProperty('redteamFinalPrompt');
581
+ expect(metadata).toHaveProperty('stoppingReason');
582
+ expect(metadata).toHaveProperty('attempts');
583
+ expect(metadata).toHaveProperty('treeOutputs');
584
+ const treeOutputs = JSON.parse(metadata.treeOutputs);
585
+ expect(Array.isArray(treeOutputs)).toBe(true);
586
+ expect(treeOutputs[0]).toHaveProperty('id');
587
+ expect(treeOutputs[0]).toHaveProperty('prompt');
588
+ expect(treeOutputs[0]).toHaveProperty('output');
589
+ expect(treeOutputs[0]).toHaveProperty('score');
590
+ expect(treeOutputs[0]).toHaveProperty('isOnTopic');
591
+ expect(treeOutputs[0]).toHaveProperty('depth');
592
+ expect(treeOutputs[0]).toHaveProperty('wasSelected');
593
+ expect(treeOutputs[0]).toHaveProperty('graderPassed');
594
+ expect(treeOutputs[1].parentId).toBe('root');
595
+ expect(treeOutputs[1].improvement).toBe('improvement');
596
+ expect(treeOutputs[1].graderPassed).toBe(false);
597
+ });
598
+ });
368
599
  //# sourceMappingURL=iterativeTree.test.js.map