promptfoo 0.89.3 → 0.90.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (188) hide show
  1. package/dist/package.json +16 -12
  2. package/dist/src/app/assets/index-B0eIt9gO.js +781 -0
  3. package/dist/src/app/assets/{index-Dl-FhnW1.css → index-DdvQdOby.css} +1 -1
  4. package/dist/src/app/assets/{index.es-mBX9PPNc.js → index.es-6LwcIYWb.js} +6 -6
  5. package/dist/src/app/assets/{sync-DKaPkQuY.js → sync-DvHcfyRC.js} +1 -1
  6. package/dist/src/app/index.html +2 -2
  7. package/dist/src/assertions.d.ts.map +1 -1
  8. package/dist/src/assertions.js +2 -1
  9. package/dist/src/assertions.js.map +1 -1
  10. package/dist/src/cache.js +2 -2
  11. package/dist/src/cache.js.map +1 -1
  12. package/dist/src/commands/auth.d.ts +3 -0
  13. package/dist/src/commands/auth.d.ts.map +1 -0
  14. package/dist/src/commands/auth.js +86 -0
  15. package/dist/src/commands/auth.js.map +1 -0
  16. package/dist/src/commands/config.js +1 -1
  17. package/dist/src/commands/config.js.map +1 -1
  18. package/dist/src/commands/eval.d.ts.map +1 -1
  19. package/dist/src/commands/eval.js +27 -13
  20. package/dist/src/commands/eval.js.map +1 -1
  21. package/dist/src/commands/generate/dataset.d.ts.map +1 -1
  22. package/dist/src/commands/generate/dataset.js +2 -2
  23. package/dist/src/commands/generate/dataset.js.map +1 -1
  24. package/dist/src/commands/share.d.ts +2 -0
  25. package/dist/src/commands/share.d.ts.map +1 -1
  26. package/dist/src/commands/share.js +20 -8
  27. package/dist/src/commands/share.js.map +1 -1
  28. package/dist/src/commands/view.js +2 -2
  29. package/dist/src/commands/view.js.map +1 -1
  30. package/dist/src/configTypes.d.ts +5 -1
  31. package/dist/src/configTypes.d.ts.map +1 -1
  32. package/dist/src/constants.d.ts +1 -1
  33. package/dist/src/constants.d.ts.map +1 -1
  34. package/dist/src/constants.js +2 -1
  35. package/dist/src/constants.js.map +1 -1
  36. package/dist/src/csv.d.ts.map +1 -1
  37. package/dist/src/csv.js +3 -2
  38. package/dist/src/csv.js.map +1 -1
  39. package/dist/src/database/index.js +3 -3
  40. package/dist/src/database/index.js.map +1 -1
  41. package/dist/src/envars.d.ts +19 -0
  42. package/dist/src/envars.d.ts.map +1 -1
  43. package/dist/src/envars.js.map +1 -1
  44. package/dist/src/globalConfig/accounts.d.ts.map +1 -0
  45. package/dist/src/{accounts.js → globalConfig/accounts.js} +2 -2
  46. package/dist/src/globalConfig/accounts.js.map +1 -0
  47. package/dist/src/globalConfig/cloud.d.ts +17 -0
  48. package/dist/src/globalConfig/cloud.d.ts.map +1 -0
  49. package/dist/src/globalConfig/cloud.js +82 -0
  50. package/dist/src/globalConfig/cloud.js.map +1 -0
  51. package/dist/src/{globalConfig.d.ts → globalConfig/globalConfig.d.ts} +1 -3
  52. package/dist/src/globalConfig/globalConfig.d.ts.map +1 -0
  53. package/dist/src/{globalConfig.js → globalConfig/globalConfig.js} +13 -36
  54. package/dist/src/globalConfig/globalConfig.js.map +1 -0
  55. package/dist/src/main.d.ts +1 -5
  56. package/dist/src/main.d.ts.map +1 -1
  57. package/dist/src/main.js +15 -31
  58. package/dist/src/main.js.map +1 -1
  59. package/dist/src/onboarding.d.ts.map +1 -1
  60. package/dist/src/onboarding.js +9 -8
  61. package/dist/src/onboarding.js.map +1 -1
  62. package/dist/src/prompts/index.d.ts.map +1 -1
  63. package/dist/src/prompts/index.js +4 -0
  64. package/dist/src/prompts/index.js.map +1 -1
  65. package/dist/src/providers/bedrock.d.ts +15 -0
  66. package/dist/src/providers/bedrock.d.ts.map +1 -1
  67. package/dist/src/providers/bedrock.js +49 -20
  68. package/dist/src/providers/bedrock.js.map +1 -1
  69. package/dist/src/providers/browser.d.ts +28 -0
  70. package/dist/src/providers/browser.d.ts.map +1 -0
  71. package/dist/src/providers/browser.js +151 -0
  72. package/dist/src/providers/browser.js.map +1 -0
  73. package/dist/src/providers/http.d.ts +9 -5
  74. package/dist/src/providers/http.d.ts.map +1 -1
  75. package/dist/src/providers/http.js +81 -12
  76. package/dist/src/providers/http.js.map +1 -1
  77. package/dist/src/providers/openai.d.ts.map +1 -1
  78. package/dist/src/providers/openai.js +7 -0
  79. package/dist/src/providers/openai.js.map +1 -1
  80. package/dist/src/providers/vertexUtil.d.ts +1 -1
  81. package/dist/src/providers/websocket.d.ts +2 -2
  82. package/dist/src/providers/websocket.d.ts.map +1 -1
  83. package/dist/src/providers/websocket.js +3 -3
  84. package/dist/src/providers/websocket.js.map +1 -1
  85. package/dist/src/providers.d.ts.map +1 -1
  86. package/dist/src/providers.js +21 -4
  87. package/dist/src/providers.js.map +1 -1
  88. package/dist/src/redteam/commands/generate.d.ts.map +1 -1
  89. package/dist/src/redteam/commands/generate.js +7 -7
  90. package/dist/src/redteam/commands/generate.js.map +1 -1
  91. package/dist/src/redteam/commands/init.d.ts.map +1 -1
  92. package/dist/src/redteam/commands/init.js +15 -24
  93. package/dist/src/redteam/commands/init.js.map +1 -1
  94. package/dist/src/redteam/plugins/base.d.ts +7 -1
  95. package/dist/src/redteam/plugins/base.d.ts.map +1 -1
  96. package/dist/src/redteam/plugins/base.js +9 -1
  97. package/dist/src/redteam/plugins/base.js.map +1 -1
  98. package/dist/src/redteam/plugins/indirectPromptInjection.d.ts +7 -1
  99. package/dist/src/redteam/plugins/indirectPromptInjection.d.ts.map +1 -1
  100. package/dist/src/redteam/plugins/indirectPromptInjection.js +37 -0
  101. package/dist/src/redteam/plugins/indirectPromptInjection.js.map +1 -1
  102. package/dist/src/server/server.d.ts.map +1 -1
  103. package/dist/src/server/server.js +12 -0
  104. package/dist/src/server/server.js.map +1 -1
  105. package/dist/src/share.d.ts +1 -1
  106. package/dist/src/share.d.ts.map +1 -1
  107. package/dist/src/share.js +38 -13
  108. package/dist/src/share.js.map +1 -1
  109. package/dist/src/telemetry.d.ts +2 -1
  110. package/dist/src/telemetry.d.ts.map +1 -1
  111. package/dist/src/telemetry.js +48 -33
  112. package/dist/src/telemetry.js.map +1 -1
  113. package/dist/src/types/index.d.ts +6 -0
  114. package/dist/src/types/index.d.ts.map +1 -1
  115. package/dist/src/types/index.js.map +1 -1
  116. package/dist/src/util/config/default.d.ts +14 -0
  117. package/dist/src/util/config/default.d.ts.map +1 -0
  118. package/dist/src/util/config/default.js +36 -0
  119. package/dist/src/util/config/default.js.map +1 -0
  120. package/dist/src/{config.d.ts → util/config/load.d.ts} +2 -2
  121. package/dist/src/util/config/load.d.ts.map +1 -0
  122. package/dist/src/{config.js → util/config/load.js} +18 -13
  123. package/dist/src/util/config/load.js.map +1 -0
  124. package/dist/src/util/{config.d.ts → config/manage.d.ts} +2 -2
  125. package/dist/src/util/config/manage.d.ts.map +1 -0
  126. package/dist/src/util/{config.js → config/manage.js} +3 -3
  127. package/dist/src/util/config/manage.js.map +1 -0
  128. package/dist/src/util/index.js +8 -8
  129. package/dist/src/util/index.js.map +1 -1
  130. package/dist/test/account.test.js +5 -4
  131. package/dist/test/account.test.js.map +1 -1
  132. package/dist/test/csv.test.js +21 -0
  133. package/dist/test/csv.test.js.map +1 -1
  134. package/dist/test/globalConfig.test.js +5 -26
  135. package/dist/test/globalConfig.test.js.map +1 -1
  136. package/dist/test/prompts.test.js +17 -2
  137. package/dist/test/prompts.test.js.map +1 -1
  138. package/dist/test/providers.bedrock.test.js +46 -0
  139. package/dist/test/providers.bedrock.test.js.map +1 -1
  140. package/dist/test/providers.http.test.js +122 -8
  141. package/dist/test/providers.http.test.js.map +1 -1
  142. package/dist/test/providers.pythonCompletion.test.js +21 -0
  143. package/dist/test/providers.pythonCompletion.test.js.map +1 -1
  144. package/dist/test/providers.test.js +25 -0
  145. package/dist/test/providers.test.js.map +1 -1
  146. package/dist/test/redteam/commands/generate.test.js +7 -7
  147. package/dist/test/redteam/commands/generate.test.js.map +1 -1
  148. package/dist/test/redteam/extraction/entities.test.js +4 -1
  149. package/dist/test/redteam/extraction/entities.test.js.map +1 -1
  150. package/dist/test/redteam/extraction/purpose.test.js +4 -1
  151. package/dist/test/redteam/extraction/purpose.test.js.map +1 -1
  152. package/dist/test/redteam/extraction/util.test.js +3 -0
  153. package/dist/test/redteam/extraction/util.test.js.map +1 -1
  154. package/dist/test/telemetry.test.js +49 -7
  155. package/dist/test/telemetry.test.js.map +1 -1
  156. package/dist/test/util.config.default.test.d.ts +2 -0
  157. package/dist/test/util.config.default.test.d.ts.map +1 -0
  158. package/dist/test/util.config.default.test.js +62 -0
  159. package/dist/test/util.config.default.test.js.map +1 -0
  160. package/dist/test/util.config.load.test.ts.d.ts +2 -0
  161. package/dist/test/util.config.load.test.ts.d.ts.map +1 -0
  162. package/dist/test/{config.test.js → util.config.load.test.ts.js} +32 -27
  163. package/dist/test/util.config.load.test.ts.js.map +1 -0
  164. package/dist/test/util.config.main.test.d.ts +2 -0
  165. package/dist/test/util.config.main.test.d.ts.map +1 -0
  166. package/dist/test/{util.config.test.js → util.config.main.test.js} +16 -16
  167. package/dist/test/util.config.main.test.js.map +1 -0
  168. package/package.json +16 -12
  169. package/dist/src/accounts.d.ts.map +0 -1
  170. package/dist/src/accounts.js.map +0 -1
  171. package/dist/src/app/assets/index-CW_4LMda.js +0 -781
  172. package/dist/src/config.d.ts.map +0 -1
  173. package/dist/src/config.js.map +0 -1
  174. package/dist/src/globalConfig.d.ts.map +0 -1
  175. package/dist/src/globalConfig.js.map +0 -1
  176. package/dist/src/util/config.d.ts.map +0 -1
  177. package/dist/src/util/config.js.map +0 -1
  178. package/dist/test/config.test.d.ts +0 -2
  179. package/dist/test/config.test.d.ts.map +0 -1
  180. package/dist/test/config.test.js.map +0 -1
  181. package/dist/test/main.test.d.ts +0 -2
  182. package/dist/test/main.test.d.ts.map +0 -1
  183. package/dist/test/main.test.js +0 -49
  184. package/dist/test/main.test.js.map +0 -1
  185. package/dist/test/util.config.test.d.ts +0 -2
  186. package/dist/test/util.config.test.d.ts.map +0 -1
  187. package/dist/test/util.config.test.js.map +0 -1
  188. /package/dist/src/{accounts.d.ts → globalConfig/accounts.d.ts} +0 -0
@@ -26,11 +26,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
26
26
  return (mod && mod.__esModule) ? mod : { "default": mod };
27
27
  };
28
28
  Object.defineProperty(exports, "__esModule", { value: true });
29
- exports.resetGlobalConfig = resetGlobalConfig;
30
29
  exports.readGlobalConfig = readGlobalConfig;
31
30
  exports.writeGlobalConfig = writeGlobalConfig;
32
31
  exports.writeGlobalConfigPartial = writeGlobalConfigPartial;
33
- exports.maybeRecordFirstRun = maybeRecordFirstRun;
34
32
  /**
35
33
  * Functions for manipulating the global configuration file, which lives at
36
34
  * ~/.promptfoo/promptfoo.yaml by default.
@@ -38,30 +36,24 @@ exports.maybeRecordFirstRun = maybeRecordFirstRun;
38
36
  const fs = __importStar(require("fs"));
39
37
  const js_yaml_1 = __importDefault(require("js-yaml"));
40
38
  const path = __importStar(require("path"));
41
- const config_1 = require("./util/config");
42
- let globalConfigCache = null;
43
- function resetGlobalConfig() {
44
- globalConfigCache = null;
45
- }
39
+ const manage_1 = require("../util/config/manage");
46
40
  function readGlobalConfig() {
47
- if (!globalConfigCache) {
48
- const configDir = (0, config_1.getConfigDirectoryPath)();
49
- const configFilePath = path.join(configDir, 'promptfoo.yaml');
50
- if (fs.existsSync(configFilePath)) {
51
- globalConfigCache = js_yaml_1.default.load(fs.readFileSync(configFilePath, 'utf-8'));
52
- }
53
- else {
54
- if (!fs.existsSync(configDir)) {
55
- fs.mkdirSync(configDir, { recursive: true });
56
- }
57
- globalConfigCache = { hasRun: false };
58
- fs.writeFileSync(configFilePath, js_yaml_1.default.dump(globalConfigCache));
41
+ const configDir = (0, manage_1.getConfigDirectoryPath)();
42
+ const configFilePath = path.join(configDir, 'promptfoo.yaml');
43
+ let globalConfig = {};
44
+ if (fs.existsSync(configFilePath)) {
45
+ globalConfig = js_yaml_1.default.load(fs.readFileSync(configFilePath, 'utf-8')) || {};
46
+ }
47
+ else {
48
+ if (!fs.existsSync(configDir)) {
49
+ fs.mkdirSync(configDir, { recursive: true });
59
50
  }
51
+ fs.writeFileSync(configFilePath, js_yaml_1.default.dump(globalConfig));
60
52
  }
61
- return globalConfigCache;
53
+ return globalConfig;
62
54
  }
63
55
  function writeGlobalConfig(config) {
64
- fs.writeFileSync(path.join((0, config_1.getConfigDirectoryPath)(true), 'promptfoo.yaml') /* createIfNotExists */, js_yaml_1.default.dump(config));
56
+ fs.writeFileSync(path.join((0, manage_1.getConfigDirectoryPath)(true), 'promptfoo.yaml') /* createIfNotExists */, js_yaml_1.default.dump(config));
65
57
  }
66
58
  /**
67
59
  * Merges the top-level keys into existing config.
@@ -81,19 +73,4 @@ function writeGlobalConfigPartial(partialConfig) {
81
73
  }
82
74
  writeGlobalConfig(updatedConfig);
83
75
  }
84
- function maybeRecordFirstRun() {
85
- // Return true if first run
86
- try {
87
- const config = readGlobalConfig();
88
- if (!config.hasRun) {
89
- config.hasRun = true;
90
- writeGlobalConfig(config);
91
- return true;
92
- }
93
- return false;
94
- }
95
- catch {
96
- return false;
97
- }
98
- }
99
76
  //# sourceMappingURL=globalConfig.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"globalConfig.js","sourceRoot":"","sources":["../../../src/globalConfig/globalConfig.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAUA,4CAcC;AAED,8CAKC;AAMD,4DAcC;AAnDD;;;GAGG;AACH,uCAAyB;AACzB,sDAA2B;AAC3B,2CAA6B;AAE7B,kDAA+D;AAE/D,SAAgB,gBAAgB;IAC9B,MAAM,SAAS,GAAG,IAAA,+BAAsB,GAAE,CAAC;IAC3C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAC9D,IAAI,YAAY,GAAiB,EAAE,CAAC;IACpC,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QAClC,YAAY,GAAI,iBAAI,CAAC,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAkB,IAAI,EAAE,CAAC;IAC7F,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;QACD,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,iBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAgB,iBAAiB,CAAC,MAAoB;IACpD,EAAE,CAAC,aAAa,CACd,IAAI,CAAC,IAAI,CAAC,IAAA,+BAAsB,EAAC,IAAI,CAAC,EAAE,gBAAgB,CAAC,CAAC,uBAAuB,EACjF,iBAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAClB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAgB,wBAAwB,CAAC,aAAoC;IAC3E,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;IACzC,MAAM,aAAa,GAAG,EAAE,GAAG,aAAa,EAAE,CAAC;IAE3C,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,aAAa,CAAC,GAAyB,CAAC,CAAC;QACvD,IAAI,KAAK,EAAE,CAAC;YACV,aAAa,CAAC,GAAyB,CAAC,GAAG,KAAY,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,OAAO,aAAa,CAAC,GAAyB,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,aAAa,CAAC,CAAC;AACnC,CAAC"}
@@ -1,7 +1,3 @@
1
1
  #!/usr/bin/env node
2
- import { type UnifiedConfig } from './types';
3
- export declare function loadDefaultConfig(): Promise<{
4
- defaultConfig: Partial<UnifiedConfig>;
5
- defaultConfigPath: string | undefined;
6
- }>;
2
+ export {};
7
3
  //# sourceMappingURL=main.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":";AAuBA,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,SAAS,CAAC;AAG7C,wBAAsB,iBAAiB,IAAI,OAAO,CAAC;IACjD,aAAa,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IACtC,iBAAiB,EAAE,MAAM,GAAG,SAAS,CAAC;CACvC,CAAC,CAkBD"}
1
+ {"version":3,"file":"main.d.ts","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":""}
package/dist/src/main.js CHANGED
@@ -1,13 +1,9 @@
1
1
  #!/usr/bin/env node
2
2
  "use strict";
3
- var __importDefault = (this && this.__importDefault) || function (mod) {
4
- return (mod && mod.__esModule) ? mod : { "default": mod };
5
- };
6
3
  Object.defineProperty(exports, "__esModule", { value: true });
7
- exports.loadDefaultConfig = loadDefaultConfig;
8
4
  const commander_1 = require("commander");
9
- const path_1 = __importDefault(require("path"));
10
5
  const checkNodeVersion_1 = require("./checkNodeVersion");
6
+ const auth_1 = require("./commands/auth");
11
7
  const cache_1 = require("./commands/cache");
12
8
  const config_1 = require("./commands/config");
13
9
  const delete_1 = require("./commands/delete");
@@ -22,54 +18,42 @@ const share_1 = require("./commands/share");
22
18
  const show_1 = require("./commands/show");
23
19
  const version_1 = require("./commands/version");
24
20
  const view_1 = require("./commands/view");
25
- const config_2 = require("./config");
26
21
  const migrate_1 = require("./migrate");
27
22
  const generate_1 = require("./redteam/commands/generate");
28
23
  const init_2 = require("./redteam/commands/init");
29
24
  const plugins_1 = require("./redteam/commands/plugins");
30
25
  const updates_1 = require("./updates");
31
- async function loadDefaultConfig() {
32
- const pwd = process.cwd();
33
- let defaultConfig = {};
34
- let defaultConfigPath;
35
- // NOTE: sorted by frequency of use
36
- const extensions = ['yaml', 'yml', 'json', 'cjs', 'cts', 'js', 'mjs', 'mts', 'ts'];
37
- for (const ext of extensions) {
38
- const configPath = path_1.default.join(pwd, `promptfooconfig.${ext}`);
39
- const maybeConfig = await (0, config_2.maybeReadConfig)(configPath);
40
- if (maybeConfig) {
41
- defaultConfig = maybeConfig;
42
- defaultConfigPath = configPath;
43
- break;
44
- }
45
- }
46
- return { defaultConfig, defaultConfigPath };
47
- }
26
+ const default_1 = require("./util/config/default");
48
27
  async function main() {
49
28
  await (0, updates_1.checkForUpdates)();
50
29
  await (0, migrate_1.runDbMigrations)();
51
- const { defaultConfig, defaultConfigPath } = await loadDefaultConfig();
30
+ const { defaultConfig, defaultConfigPath } = await (0, default_1.loadDefaultConfig)(['promptfooconfig']);
52
31
  const program = new commander_1.Command();
32
+ // Main commands
33
+ (0, eval_1.evalCommand)(program, defaultConfig, defaultConfigPath);
34
+ (0, init_1.initCommand)(program);
35
+ (0, view_1.viewCommand)(program);
36
+ const redteamBaseCommand = program.command('redteam').description('Red team LLM applications');
37
+ (0, share_1.shareCommand)(program);
38
+ // Alphabetical order
39
+ (0, auth_1.authCommand)(program);
53
40
  (0, cache_1.cacheCommand)(program);
54
41
  (0, config_1.configCommand)(program);
55
42
  (0, delete_1.deleteCommand)(program);
56
- (0, eval_1.evalCommand)(program, defaultConfig, defaultConfigPath);
57
43
  (0, export_1.exportCommand)(program);
58
44
  (0, feedback_1.feedbackCommand)(program);
45
+ const generateCommand = program.command('generate').description('Generate synthetic data');
59
46
  (0, import_1.importCommand)(program);
60
- (0, init_1.initCommand)(program);
61
47
  (0, list_1.listCommand)(program);
62
- (0, share_1.shareCommand)(program);
63
48
  (0, show_1.showCommand)(program);
64
49
  (0, version_1.versionCommand)(program);
65
- (0, view_1.viewCommand)(program);
66
- const generateCommand = program.command('generate').description('Generate synthetic data');
67
50
  (0, dataset_1.generateDatasetCommand)(generateCommand, defaultConfig, defaultConfigPath);
68
51
  (0, generate_1.generateRedteamCommand)(generateCommand, 'redteam', defaultConfig, defaultConfigPath);
69
- const redteamBaseCommand = program.command('redteam').description('Red team LLM applications');
52
+ const { defaultConfig: redteamConfig, defaultConfigPath: redteamConfigPath } = await (0, default_1.loadDefaultConfig)(['redteam', 'promptfooconfig']);
70
53
  (0, init_2.initCommand)(redteamBaseCommand);
71
- (0, plugins_1.pluginsCommand)(redteamBaseCommand);
54
+ (0, eval_1.evalCommand)(redteamBaseCommand, redteamConfig, redteamConfigPath);
72
55
  (0, generate_1.generateRedteamCommand)(redteamBaseCommand, 'generate', defaultConfig, defaultConfigPath);
56
+ (0, plugins_1.pluginsCommand)(redteamBaseCommand);
73
57
  if (!process.argv.slice(2).length) {
74
58
  program.outputHelp();
75
59
  process.exit(0);
@@ -1 +1 @@
1
- {"version":3,"file":"main.js","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":";;;;;;AA0BA,8CAqBC;AA9CD,yCAAoC;AACpC,gDAAwB;AACxB,yDAAsD;AACtD,4CAAgD;AAChD,8CAAkD;AAClD,8CAAkD;AAClD,0CAA8C;AAC9C,8CAAkD;AAClD,kDAAsD;AACtD,yDAAqE;AACrE,8CAAkD;AAClD,0CAA8C;AAC9C,0CAA8C;AAC9C,4CAAgD;AAChD,0CAA8C;AAC9C,gDAAoD;AACpD,0CAA8C;AAC9C,qCAA2C;AAC3C,uCAA4C;AAC5C,0DAAqE;AACrE,kDAA4E;AAC5E,wDAAqF;AAErF,uCAA4C;AAErC,KAAK,UAAU,iBAAiB;IAIrC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,IAAI,aAAa,GAA2B,EAAE,CAAC;IAC/C,IAAI,iBAAqC,CAAC;IAE1C,mCAAmC;IACnC,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IACnF,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,GAAG,EAAE,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,IAAA,wBAAe,EAAC,UAAU,CAAC,CAAC;QACtD,IAAI,WAAW,EAAE,CAAC;YAChB,aAAa,GAAG,WAAW,CAAC;YAC5B,iBAAiB,GAAG,UAAU,CAAC;YAC/B,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,IAAA,yBAAe,GAAE,CAAC;IACxB,MAAM,IAAA,yBAAe,GAAE,CAAC;IAExB,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAEvE,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;IAE9B,IAAA,oBAAY,EAAC,OAAO,CAAC,CAAC;IACtB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,kBAAW,EAAC,OAAO,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IACvD,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,0BAAe,EAAC,OAAO,CAAC,CAAC;IACzB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,oBAAY,EAAC,OAAO,CAAC,CAAC;IACtB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,wBAAc,EAAC,OAAO,CAAC,CAAC;IACxB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IAErB,MAAM,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,WAAW,CAAC,yBAAyB,CAAC,CAAC;IAC3F,IAAA,gCAAsB,EAAC,eAAe,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAC1E,IAAA,iCAAsB,EAAC,eAAe,EAAE,SAAS,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAErF,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,2BAA2B,CAAC,CAAC;IAC/F,IAAA,kBAAkB,EAAC,kBAAkB,CAAC,CAAC;IACvC,IAAA,wBAAqB,EAAC,kBAAkB,CAAC,CAAC;IAC1C,IAAA,iCAAsB,EAAC,kBAAkB,EAAE,UAAU,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAEzF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAClC,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC5B,IAAA,mCAAgB,GAAE,CAAC;IACnB,IAAI,EAAE,CAAC;AACT,CAAC"}
1
+ {"version":3,"file":"main.js","sourceRoot":"","sources":["../../src/main.ts"],"names":[],"mappings":";;;AACA,yCAAoC;AACpC,yDAAsD;AACtD,0CAA8C;AAC9C,4CAAgD;AAChD,8CAAkD;AAClD,8CAAkD;AAClD,0CAA8C;AAC9C,8CAAkD;AAClD,kDAAsD;AACtD,yDAAqE;AACrE,8CAAkD;AAClD,0CAA8C;AAC9C,0CAA8C;AAC9C,4CAAgD;AAChD,0CAA8C;AAC9C,gDAAoD;AACpD,0CAA8C;AAC9C,uCAA4C;AAC5C,0DAAqE;AACrE,kDAA4E;AAC5E,wDAAqF;AACrF,uCAA4C;AAC5C,mDAA0D;AAE1D,KAAK,UAAU,IAAI;IACjB,MAAM,IAAA,yBAAe,GAAE,CAAC;IACxB,MAAM,IAAA,yBAAe,GAAE,CAAC;IAExB,MAAM,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAAG,MAAM,IAAA,2BAAiB,EAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAE1F,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;IAE9B,gBAAgB;IAChB,IAAA,kBAAW,EAAC,OAAO,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IACvD,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,2BAA2B,CAAC,CAAC;IAC/F,IAAA,oBAAY,EAAC,OAAO,CAAC,CAAC;IAEtB,qBAAqB;IACrB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,oBAAY,EAAC,OAAO,CAAC,CAAC;IACtB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,0BAAe,EAAC,OAAO,CAAC,CAAC;IACzB,MAAM,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,WAAW,CAAC,yBAAyB,CAAC,CAAC;IAC3F,IAAA,sBAAa,EAAC,OAAO,CAAC,CAAC;IACvB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;IACrB,IAAA,wBAAc,EAAC,OAAO,CAAC,CAAC;IAExB,IAAA,gCAAsB,EAAC,eAAe,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAC1E,IAAA,iCAAsB,EAAC,eAAe,EAAE,SAAS,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAErF,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,GAC1E,MAAM,IAAA,2BAAiB,EAAC,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAE1D,IAAA,kBAAkB,EAAC,kBAAkB,CAAC,CAAC;IACvC,IAAA,kBAAW,EAAC,kBAAkB,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IAClE,IAAA,iCAAsB,EAAC,kBAAkB,EAAE,UAAU,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;IACzF,IAAA,wBAAqB,EAAC,kBAAkB,CAAC,CAAC;IAE1C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAClC,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC5B,IAAA,mCAAgB,GAAE,CAAC;IACnB,IAAI,EAAE,CAAC;AACT,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"onboarding.d.ts","sourceRoot":"","sources":["../../src/onboarding.ts"],"names":[],"mappings":"AAeA,eAAO,MAAM,eAAe,4hFAmF3B,CAAC;AAEF,eAAO,MAAM,eAAe,62CAgC3B,CAAC;AAEF,eAAO,MAAM,mBAAmB,0/BAsC/B,CAAC;AAEF,eAAO,MAAM,aAAa,gTAQzB,CAAC;AAEF,eAAO,MAAM,UAAU,upBAgBtB,CAAC;AAEF,eAAO,MAAM,cAAc,2pBAoB1B,CAAC;AAEF,eAAO,MAAM,cAAc,0QAU1B,CAAC;AAUF;;;GAGG;AACH,wBAAgB,4BAA4B,CAAC,eAAe,EAAE,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,GAAG,MAAM,EAAE,CAgB3F;AAED,wBAAsB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,GAAE,OAAc;;;;;GAsS3F;AAED,wBAAsB,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,GAAE,OAAc;;;;;GAsB5F"}
1
+ {"version":3,"file":"onboarding.d.ts","sourceRoot":"","sources":["../../src/onboarding.ts"],"names":[],"mappings":"AAeA,eAAO,MAAM,eAAe,4hFAmF3B,CAAC;AAEF,eAAO,MAAM,eAAe,62CAgC3B,CAAC;AAEF,eAAO,MAAM,mBAAmB,0/BAsC/B,CAAC;AAEF,eAAO,MAAM,aAAa,gTAQzB,CAAC;AAEF,eAAO,MAAM,UAAU,upBAgBtB,CAAC;AAEF,eAAO,MAAM,cAAc,2pBAoB1B,CAAC;AAEF,eAAO,MAAM,cAAc,0QAU1B,CAAC;AAUF;;;GAGG;AACH,wBAAgB,4BAA4B,CAAC,eAAe,EAAE,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,GAAG,MAAM,EAAE,CAgB3F;AAED,wBAAsB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,GAAE,OAAc;;;;;GA4S3F;AAED,wBAAsB,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,GAAE,OAAc;;;;;GAsB5F"}
@@ -10,7 +10,7 @@ exports.initializeProject = initializeProject;
10
10
  const checkbox_1 = __importDefault(require("@inquirer/checkbox"));
11
11
  const confirm_1 = __importDefault(require("@inquirer/confirm"));
12
12
  const core_1 = require("@inquirer/core");
13
- const rawlist_1 = __importDefault(require("@inquirer/rawlist"));
13
+ const select_1 = __importDefault(require("@inquirer/select"));
14
14
  const chalk_1 = __importDefault(require("chalk"));
15
15
  const dedent_1 = __importDefault(require("dedent"));
16
16
  const fs_1 = __importDefault(require("fs"));
@@ -288,7 +288,7 @@ async function createDummyFiles(directory, interactive = true) {
288
288
  if (interactive) {
289
289
  recordOnboardingStep('start');
290
290
  // Choose use case
291
- action = await (0, rawlist_1.default)({
291
+ action = await (0, select_1.default)({
292
292
  message: 'What would you like to do?',
293
293
  choices: [
294
294
  { name: 'Not sure yet', value: 'compare' },
@@ -312,7 +312,7 @@ async function createDummyFiles(directory, interactive = true) {
312
312
  }
313
313
  language = 'not_sure';
314
314
  if (action === 'rag' || action === 'agent') {
315
- language = await (0, rawlist_1.default)({
315
+ language = await (0, select_1.default)({
316
316
  message: 'What programming language are you developing the app in?',
317
317
  choices: [
318
318
  { name: 'Not sure yet', value: 'not_sure' },
@@ -374,7 +374,7 @@ async function createDummyFiles(directory, interactive = true) {
374
374
  },
375
375
  {
376
376
  name: 'Local Python script',
377
- value: ['python:provider.py'],
377
+ value: ['file://provider.py'],
378
378
  },
379
379
  {
380
380
  name: 'Local Javascript script',
@@ -410,9 +410,8 @@ async function createDummyFiles(directory, interactive = true) {
410
410
  * for openai as a value choice
411
411
  */
412
412
  const providerChoices = (await (0, checkbox_1.default)({
413
- message: 'Which model providers would you like to use? (press space to select, enter to complete selection)',
413
+ message: 'Which model providers would you like to use?',
414
414
  choices,
415
- required: true,
416
415
  loop: false,
417
416
  pageSize: process.stdout.rows - 6,
418
417
  })).flat();
@@ -428,7 +427,7 @@ async function createDummyFiles(directory, interactive = true) {
428
427
  else {
429
428
  providers.push(...providerChoices);
430
429
  }
431
- if (providerChoices.some((choice) => typeof choice === 'string' && choice.startsWith('file://'))) {
430
+ if (providerChoices.some((choice) => typeof choice === 'string' && choice.startsWith('file://') && choice.endsWith('.js'))) {
432
431
  fs_1.default.writeFileSync(path_1.default.join(process.cwd(), directory, 'provider.js'), exports.JAVASCRIPT_PROVIDER);
433
432
  logger_1.default.info('⌛ Wrote provider.js');
434
433
  }
@@ -436,7 +435,9 @@ async function createDummyFiles(directory, interactive = true) {
436
435
  fs_1.default.writeFileSync(path_1.default.join(process.cwd(), directory, 'provider.sh'), exports.BASH_PROVIDER);
437
436
  logger_1.default.info('⌛ Wrote provider.sh');
438
437
  }
439
- if (providerChoices.some((choice) => typeof choice === 'string' && choice.startsWith('python:'))) {
438
+ if (providerChoices.some((choice) => typeof choice === 'string' &&
439
+ (choice.startsWith('python:') ||
440
+ (choice.startsWith('file://') && choice.endsWith('.py'))))) {
440
441
  fs_1.default.writeFileSync(path_1.default.join(process.cwd(), directory, 'provider.py'), exports.PYTHON_PROVIDER);
441
442
  logger_1.default.info('⌛ Wrote provider.py');
442
443
  }
@@ -1 +1 @@
1
- {"version":3,"file":"onboarding.js","sourceRoot":"","sources":["../../src/onboarding.ts"],"names":[],"mappings":";;;;;;AAwPA,oEAgBC;AAED,4CAsSC;AAED,8CAsBC;AAxkBD,kEAA0C;AAC1C,gEAAwC;AACxC,yCAAiD;AACjD,gEAAwC;AACxC,kDAA0B;AAC1B,oDAA4B;AAC5B,4CAAoB;AACpB,gDAAwB;AACxB,qCAAwC;AACxC,sDAA8B;AAC9B,kDAAsD;AACtD,4DAAyD;AAEzD,gDAAqD;AAExC,QAAA,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmF9B,CAAC;AAEW,QAAA,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgC9B,CAAC;AAEW,QAAA,mBAAmB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsClC,CAAC;AAEW,QAAA,aAAa,GAAG;;;;;;;;CAQ5B,CAAC;AAEW,QAAA,UAAU,GAAG;;;;;;;;;;;;;;;;CAgBzB,CAAC;AAEW,QAAA,cAAc,GAAG;;;;;;;;;;;;;;;;;;;;CAoB7B,CAAC;AAEW,QAAA,cAAc,GAAG;;;;;;;;;;CAU7B,CAAC;AAEF,SAAS,oBAAoB,CAAC,IAAY,EAAE,aAAyC,EAAE;IACrF,mBAAS,CAAC,aAAa,CAAC,QAAQ,EAAE;QAChC,IAAI,EAAE,iBAAiB;QACvB,IAAI;QACJ,GAAG,UAAU;KACd,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAgB,4BAA4B,CAAC,eAAoC;IAC/E,MAAM,GAAG,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEpF,MAAM,GAAG,GAAuC;QAC9C,MAAM,EAAE,gBAAgB;QACxB,SAAS,EAAE,mBAAmB;KAC/B,CAAC;IAEF,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;SACvB,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAA,qBAAY,EAAC,GAAG,CAAC,CAAC;SACxF,GAAG,CACF,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,IAAA,gBAAM,EAAA;MACzB,eAAK,CAAC,IAAI,CAAC,YAAY,GAAG,mCAAmC,CAAC;wDACZ,GAAG;GACxD,CACE,CAAC;AACN,CAAC;AAEM,KAAK,UAAU,gBAAgB,CAAC,SAAwB,EAAE,cAAuB,IAAI;IAC1F,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,IAAI,SAAS,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3C,YAAE,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAED,SAAS,GAAG,SAAS,IAAI,GAAG,CAAC;IAE7B,oDAAoD;IACpD,MAAM,YAAY,GAAG,CAAC,sBAAsB,EAAE,WAAW,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACjD,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CACzD,CAAC;IAEF,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAO,EAAC;YAC9B,OAAO,EAAE,GAAG,QAAQ,iBAAiB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,SAAS,EAAE,8BAA8B,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG;YAC/L,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAC1E,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,GAAG,CACP,gBAAgB,UAAU,wCAAwC,UAAU,2CAA2C,CACxH,CACF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,SAAS,GAAwB,EAAE,CAAC;IAC1C,IAAI,MAAc,CAAC;IACnB,IAAI,QAAgB,CAAC;IAErB,IAAI,WAAW,EAAE,CAAC;QAChB,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAE9B,kBAAkB;QAClB,MAAM,GAAG,MAAM,IAAA,iBAAO,EAAC;YACrB,OAAO,EAAE,4BAA4B;YACrC,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE;gBAC1C,EAAE,IAAI,EAAE,sCAAsC,EAAE,KAAK,EAAE,SAAS,EAAE;gBAClE,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,KAAK,EAAE;gBACjD,EAAE,IAAI,EAAE,4CAA4C,EAAE,KAAK,EAAE,OAAO,EAAE;gBACtE,EAAE,IAAI,EAAE,2BAA2B,EAAE,KAAK,EAAE,SAAS,EAAE;aACxD;SACF,CAAC,CAAC;QAEH,oBAAoB,CAAC,iBAAiB,EAAE;YACtC,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;QAEH,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,IAAA,kBAAW,EAAC,SAAS,IAAI,GAAG,CAAC,CAAC;YACpC,OAAO;gBACL,UAAU,EAAE,CAAC;gBACb,gBAAgB,EAAE,EAAE;gBACpB,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,gBAAgB;aAC3B,CAAC;QACJ,CAAC;QAED,QAAQ,GAAG,UAAU,CAAC;QACtB,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC3C,QAAQ,GAAG,MAAM,IAAA,iBAAO,EAAC;gBACvB,OAAO,EAAE,0DAA0D;gBACnE,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,EAAE;oBAC3C,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE;iBAC5C;aACF,CAAC,CAAC;YAEH,oBAAoB,CAAC,iBAAiB,EAAE;gBACtC,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,OAAO,GAAmD;YAC9D,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,CAAC,oBAAoB,EAAE,eAAe,CAAC,EAAE;YAC7E;gBACE,IAAI,EAAE,4CAA4C;gBAClD,KAAK,EACH,MAAM,KAAK,OAAO;oBAChB,CAAC,CAAC;wBACE;4BACE,EAAE,EAAE,eAAe;4BACnB,MAAM,EAAE;gCACN,KAAK,EAAE;oCACL;wCACE,IAAI,EAAE,UAAU;wCAChB,QAAQ,EAAE;4CACR,IAAI,EAAE,qBAAqB;4CAC3B,WAAW,EAAE,6CAA6C;4CAC1D,UAAU,EAAE;gDACV,IAAI,EAAE,QAAQ;gDACd,UAAU,EAAE;oDACV,QAAQ,EAAE;wDACR,IAAI,EAAE,QAAQ;wDACd,WAAW,EAAE,4CAA4C;qDAC1D;iDACF;gDACD,QAAQ,EAAE,CAAC,UAAU,CAAC;6CACvB;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;oBACH,CAAC,CAAC,CAAC,oBAAoB,EAAE,eAAe,CAAC;aAC9C;YACD;gBACE,IAAI,EAAE,6CAA6C;gBACnD,KAAK,EAAE;oBACL,+CAA+C;oBAC/C,2CAA2C;iBAC5C;aACF;YACD;gBACE,IAAI,EAAE,sCAAsC;gBAC5C,KAAK,EAAE;oBACL,iEAAiE;oBACjE,8DAA8D;oBAC9D,gDAAgD;iBACjD;aACF;YACD;gBACE,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,CAAC,oBAAoB,CAAC;aAC9B;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,KAAK,EAAE,CAAC,oBAAoB,CAAC;aAC9B;YACD;gBACE,IAAI,EAAE,kBAAkB;gBACxB,KAAK,EAAE,CAAC,kBAAkB,CAAC;aAC5B;YACD;gBACE,IAAI,EAAE,eAAe;gBACrB,KAAK,EAAE,CAAC,kCAAkC,CAAC;aAC5C;YACD;gBACE,IAAI,EAAE,yCAAyC;gBAC/C,KAAK,EAAE;oBACL,gDAAgD;oBAChD,+CAA+C;iBAChD;aACF;YACD;gBACE,IAAI,EAAE,qCAAqC;gBAC3C,KAAK,EAAE,CAAC,kBAAkB,EAAE,uBAAuB,CAAC;aACrD;YACD,EAAE,IAAI,EAAE,wCAAwC,EAAE,KAAK,EAAE,CAAC,mBAAmB,CAAC,EAAE;YAChF;gBACE,IAAI,EAAE,gCAAgC;gBACtC,KAAK,EAAE,CAAC,oBAAoB,EAAE,2BAA2B,CAAC;aAC3D;SACF,CAAC;QAEF;;;WAGG;QACH,MAAM,eAAe,GAAwB,CAC3C,MAAM,IAAA,kBAAQ,EAAC;YACb,OAAO,EACL,mGAAmG;YACrG,OAAO;YACP,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,KAAK;YACX,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC;SAClC,CAAC,CACH,CAAC,IAAI,EAAE,CAAC;QAET,oBAAoB,CAAC,kBAAkB,EAAE;YACvC,KAAK,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAC7D;SACF,CAAC,CAAC;QAEH,sFAAsF;QACtF,4BAA4B,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CACpE,gBAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CACzB,CAAC;QAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,SAAS,CAAC,IAAI,CACZ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CACjF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;YACrC,CAAC;YAED,IACE,eAAe,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAC5F,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,2BAAmB,CAAC,CAAC;gBAC1F,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;YACD,IACE,eAAe,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,EAC1F,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,qBAAa,CAAC,CAAC;gBACpF,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;YACD,IACE,eAAe,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAC5F,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,uBAAe,CAAC,CAAC;gBACtF,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC9C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;aAAM,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CACV,6FAA6F,CAC9F,CAAC;QACJ,CAAC;aAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC3C,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;gBAC9B,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,YAAY,CAAC,EAAE,sBAAc,CAAC,CAAC;gBACpF,gBAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,YAAY,CAAC,EAAE,kBAAU,CAAC,CAAC;gBAChF,gBAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,oBAAoB,CAAC,UAAU,CAAC,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,SAAS,CAAC;QACnB,QAAQ,GAAG,UAAU,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAC9C,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QAC7D,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACrC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,CAAC,uBAAe,EAAE;QACpD,OAAO;QACP,SAAS;QACT,IAAI,EAAE,MAAM;QACZ,QAAQ;KACT,CAAC,CAAC;IAEH,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,sBAAsB,CAAC,EAAE,MAAM,CAAC,CAAC;IACtF,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,WAAW,CAAC,EAAE,sBAAc,CAAC,CAAC;IAEnF,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;IAC1E,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;QACtB,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,KAAK,CACT,uCAAuC,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAClF,CACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,gBAAM,CAAC,IAAI,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;QAC9D,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,KAAK,CACT,SAAS,eAAK,CAAC,IAAI,CAAC,MAAM,SAAS,EAAE,CAAC,iBAAiB,eAAK,CAAC,IAAI,CAC/D,UAAU,CACX,oBAAoB,CACtB,CACF,CAAC;IACJ,CAAC;IAED,OAAO;QACL,UAAU,EAAE,OAAO,CAAC,MAAM;QAC1B,gBAAgB,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC7F,MAAM;QACN,QAAQ;KACT,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,iBAAiB,CAAC,SAAwB,EAAE,cAAuB,IAAI;IAC3F,IAAI,CAAC;QACH,OAAO,MAAM,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IACxD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,sBAAe,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAC1E,gBAAM,CAAC,IAAI,CACT,IAAI;gBACF,eAAK,CAAC,IAAI,CAAC,mEAAmE,CAAC;gBAC/E,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CACzB,CAAC;YACF,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC;gBACxC,eAAK,CAAC,KAAK,CAAC,oCAAoC,CAAC,CACpD,CAAC;YACF,MAAM,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACzC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;AACH,CAAC"}
1
+ {"version":3,"file":"onboarding.js","sourceRoot":"","sources":["../../src/onboarding.ts"],"names":[],"mappings":";;;;;;AAwPA,oEAgBC;AAED,4CA4SC;AAED,8CAsBC;AA9kBD,kEAA0C;AAC1C,gEAAwC;AACxC,yCAAiD;AACjD,8DAAsC;AACtC,kDAA0B;AAC1B,oDAA4B;AAC5B,4CAAoB;AACpB,gDAAwB;AACxB,qCAAwC;AACxC,sDAA8B;AAC9B,kDAAsD;AACtD,4DAAyD;AAEzD,gDAAqD;AAExC,QAAA,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmF9B,CAAC;AAEW,QAAA,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgC9B,CAAC;AAEW,QAAA,mBAAmB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsClC,CAAC;AAEW,QAAA,aAAa,GAAG;;;;;;;;CAQ5B,CAAC;AAEW,QAAA,UAAU,GAAG;;;;;;;;;;;;;;;;CAgBzB,CAAC;AAEW,QAAA,cAAc,GAAG;;;;;;;;;;;;;;;;;;;;CAoB7B,CAAC;AAEW,QAAA,cAAc,GAAG;;;;;;;;;;CAU7B,CAAC;AAEF,SAAS,oBAAoB,CAAC,IAAY,EAAE,aAAyC,EAAE;IACrF,mBAAS,CAAC,aAAa,CAAC,QAAQ,EAAE;QAChC,IAAI,EAAE,iBAAiB;QACvB,IAAI;QACJ,GAAG,UAAU;KACd,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAgB,4BAA4B,CAAC,eAAoC;IAC/E,MAAM,GAAG,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEpF,MAAM,GAAG,GAAuC;QAC9C,MAAM,EAAE,gBAAgB;QACxB,SAAS,EAAE,mBAAmB;KAC/B,CAAC;IAEF,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;SACvB,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAA,qBAAY,EAAC,GAAG,CAAC,CAAC;SACxF,GAAG,CACF,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,IAAA,gBAAM,EAAA;MACzB,eAAK,CAAC,IAAI,CAAC,YAAY,GAAG,mCAAmC,CAAC;wDACZ,GAAG;GACxD,CACE,CAAC;AACN,CAAC;AAEM,KAAK,UAAU,gBAAgB,CAAC,SAAwB,EAAE,cAAuB,IAAI;IAC1F,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,IAAI,SAAS,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3C,YAAE,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAED,SAAS,GAAG,SAAS,IAAI,GAAG,CAAC;IAE7B,oDAAoD;IACpD,MAAM,YAAY,GAAG,CAAC,sBAAsB,EAAE,WAAW,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACjD,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CACzD,CAAC;IAEF,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAO,EAAC;YAC9B,OAAO,EAAE,GAAG,QAAQ,iBAAiB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,SAAS,EAAE,8BAA8B,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG;YAC/L,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAC1E,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,GAAG,CACP,gBAAgB,UAAU,wCAAwC,UAAU,2CAA2C,CACxH,CACF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,SAAS,GAAwB,EAAE,CAAC;IAC1C,IAAI,MAAc,CAAC;IACnB,IAAI,QAAgB,CAAC;IAErB,IAAI,WAAW,EAAE,CAAC;QAChB,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAE9B,kBAAkB;QAClB,MAAM,GAAG,MAAM,IAAA,gBAAM,EAAC;YACpB,OAAO,EAAE,4BAA4B;YACrC,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE;gBAC1C,EAAE,IAAI,EAAE,sCAAsC,EAAE,KAAK,EAAE,SAAS,EAAE;gBAClE,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,KAAK,EAAE;gBACjD,EAAE,IAAI,EAAE,4CAA4C,EAAE,KAAK,EAAE,OAAO,EAAE;gBACtE,EAAE,IAAI,EAAE,2BAA2B,EAAE,KAAK,EAAE,SAAS,EAAE;aACxD;SACF,CAAC,CAAC;QAEH,oBAAoB,CAAC,iBAAiB,EAAE;YACtC,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;QAEH,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,IAAA,kBAAW,EAAC,SAAS,IAAI,GAAG,CAAC,CAAC;YACpC,OAAO;gBACL,UAAU,EAAE,CAAC;gBACb,gBAAgB,EAAE,EAAE;gBACpB,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,gBAAgB;aAC3B,CAAC;QACJ,CAAC;QAED,QAAQ,GAAG,UAAU,CAAC;QACtB,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC3C,QAAQ,GAAG,MAAM,IAAA,gBAAM,EAAC;gBACtB,OAAO,EAAE,0DAA0D;gBACnE,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,EAAE;oBAC3C,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE;iBAC5C;aACF,CAAC,CAAC;YAEH,oBAAoB,CAAC,iBAAiB,EAAE;gBACtC,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,OAAO,GAAmD;YAC9D,EAAE,IAAI,EAAE,mBAAmB,EAAE,KAAK,EAAE,CAAC,oBAAoB,EAAE,eAAe,CAAC,EAAE;YAC7E;gBACE,IAAI,EAAE,4CAA4C;gBAClD,KAAK,EACH,MAAM,KAAK,OAAO;oBAChB,CAAC,CAAC;wBACE;4BACE,EAAE,EAAE,eAAe;4BACnB,MAAM,EAAE;gCACN,KAAK,EAAE;oCACL;wCACE,IAAI,EAAE,UAAU;wCAChB,QAAQ,EAAE;4CACR,IAAI,EAAE,qBAAqB;4CAC3B,WAAW,EAAE,6CAA6C;4CAC1D,UAAU,EAAE;gDACV,IAAI,EAAE,QAAQ;gDACd,UAAU,EAAE;oDACV,QAAQ,EAAE;wDACR,IAAI,EAAE,QAAQ;wDACd,WAAW,EAAE,4CAA4C;qDAC1D;iDACF;gDACD,QAAQ,EAAE,CAAC,UAAU,CAAC;6CACvB;yCACF;qCACF;iCACF;6BACF;yBACF;qBACF;oBACH,CAAC,CAAC,CAAC,oBAAoB,EAAE,eAAe,CAAC;aAC9C;YACD;gBACE,IAAI,EAAE,6CAA6C;gBACnD,KAAK,EAAE;oBACL,+CAA+C;oBAC/C,2CAA2C;iBAC5C;aACF;YACD;gBACE,IAAI,EAAE,sCAAsC;gBAC5C,KAAK,EAAE;oBACL,iEAAiE;oBACjE,8DAA8D;oBAC9D,gDAAgD;iBACjD;aACF;YACD;gBACE,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,CAAC,oBAAoB,CAAC;aAC9B;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,KAAK,EAAE,CAAC,oBAAoB,CAAC;aAC9B;YACD;gBACE,IAAI,EAAE,kBAAkB;gBACxB,KAAK,EAAE,CAAC,kBAAkB,CAAC;aAC5B;YACD;gBACE,IAAI,EAAE,eAAe;gBACrB,KAAK,EAAE,CAAC,kCAAkC,CAAC;aAC5C;YACD;gBACE,IAAI,EAAE,yCAAyC;gBAC/C,KAAK,EAAE;oBACL,gDAAgD;oBAChD,+CAA+C;iBAChD;aACF;YACD;gBACE,IAAI,EAAE,qCAAqC;gBAC3C,KAAK,EAAE,CAAC,kBAAkB,EAAE,uBAAuB,CAAC;aACrD;YACD,EAAE,IAAI,EAAE,wCAAwC,EAAE,KAAK,EAAE,CAAC,mBAAmB,CAAC,EAAE;YAChF;gBACE,IAAI,EAAE,gCAAgC;gBACtC,KAAK,EAAE,CAAC,oBAAoB,EAAE,2BAA2B,CAAC;aAC3D;SACF,CAAC;QAEF;;;WAGG;QACH,MAAM,eAAe,GAAwB,CAC3C,MAAM,IAAA,kBAAQ,EAAC;YACb,OAAO,EAAE,8CAA8C;YACvD,OAAO;YACP,IAAI,EAAE,KAAK;YACX,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC;SAClC,CAAC,CACH,CAAC,IAAI,EAAE,CAAC;QAET,oBAAoB,CAAC,kBAAkB,EAAE;YACvC,KAAK,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpC,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAC7D;SACF,CAAC,CAAC;QAEH,sFAAsF;QACtF,4BAA4B,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CACpE,gBAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CACzB,CAAC;QAEF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,SAAS,CAAC,IAAI,CACZ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CACjF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;YACrC,CAAC;YAED,IACE,eAAe,CAAC,IAAI,CAClB,CAAC,MAAM,EAAE,EAAE,CACT,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CACvF,EACD,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,2BAAmB,CAAC,CAAC;gBAC1F,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;YACD,IACE,eAAe,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,EAC1F,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,qBAAa,CAAC,CAAC;gBACpF,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;YACD,IACE,eAAe,CAAC,IAAI,CAClB,CAAC,MAAM,EAAE,EAAE,CACT,OAAO,MAAM,KAAK,QAAQ;gBAC1B,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC;oBAC3B,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAC9D,EACD,CAAC;gBACD,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,EAAE,uBAAe,CAAC,CAAC;gBACtF,gBAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC9C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;aAAM,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CACV,6FAA6F,CAC9F,CAAC;QACJ,CAAC;aAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC3C,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;gBAC9B,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,YAAY,CAAC,EAAE,sBAAc,CAAC,CAAC;gBACpF,gBAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,YAAY,CAAC,EAAE,kBAAU,CAAC,CAAC;gBAChF,gBAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,oBAAoB,CAAC,UAAU,CAAC,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,SAAS,CAAC;QACnB,QAAQ,GAAG,UAAU,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAC9C,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QAC7D,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACrC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;IACrC,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,CAAC,uBAAe,EAAE;QACpD,OAAO;QACP,SAAS;QACT,IAAI,EAAE,MAAM;QACZ,QAAQ;KACT,CAAC,CAAC;IAEH,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,sBAAsB,CAAC,EAAE,MAAM,CAAC,CAAC;IACtF,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,WAAW,CAAC,EAAE,sBAAc,CAAC,CAAC;IAEnF,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;IAC1E,IAAI,SAAS,KAAK,GAAG,EAAE,CAAC;QACtB,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,KAAK,CACT,uCAAuC,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAClF,CACF,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,gBAAM,CAAC,IAAI,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;QAC9D,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,KAAK,CACT,SAAS,eAAK,CAAC,IAAI,CAAC,MAAM,SAAS,EAAE,CAAC,iBAAiB,eAAK,CAAC,IAAI,CAC/D,UAAU,CACX,oBAAoB,CACtB,CACF,CAAC;IACJ,CAAC;IAED,OAAO;QACL,UAAU,EAAE,OAAO,CAAC,MAAM;QAC1B,gBAAgB,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC7F,MAAM;QACN,QAAQ;KACT,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,iBAAiB,CAAC,SAAwB,EAAE,cAAuB,IAAI;IAC3F,IAAI,CAAC;QACH,OAAO,MAAM,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IACxD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,sBAAe,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAC1E,gBAAM,CAAC,IAAI,CACT,IAAI;gBACF,eAAK,CAAC,IAAI,CAAC,mEAAmE,CAAC;gBAC/E,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CACzB,CAAC;YACF,gBAAM,CAAC,IAAI,CACT,eAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC;gBACxC,eAAK,CAAC,KAAK,CAAC,oCAAoC,CAAC,CACpD,CAAC;YACF,MAAM,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACzC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;AACH,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/prompts/index.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,aAAa,EACb,MAAM,EAEN,SAAS,EAEV,MAAM,UAAU,CAAC;AAYlB,cAAc,WAAW,CAAC;AAE1B;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,EAC9B,aAAa,EAAE,MAAM,EAAE,GACtB,SAAS,CAAC,mBAAmB,CAAC,CA4ChC;AAED;;;;;;GAMG;AACH,wBAAsB,aAAa,CACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,EACvB,QAAQ,GAAE,MAAW,EACrB,iBAAiB,GAAE,MAAU,GAC5B,OAAO,CAAC,MAAM,EAAE,CAAC,CAsEnB;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,CAC/B,iBAAiB,EAAE,MAAM,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACjF,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,MAAM,EAAE,CAAC,CAYnB"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/prompts/index.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,aAAa,EACb,MAAM,EAEN,SAAS,EAEV,MAAM,UAAU,CAAC;AAYlB,cAAc,WAAW,CAAC;AAE1B;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,EAC9B,aAAa,EAAE,MAAM,EAAE,GACtB,SAAS,CAAC,mBAAmB,CAAC,CA4ChC;AAED;;;;;;GAMG;AACH,wBAAsB,aAAa,CACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,EACvB,QAAQ,GAAE,MAAW,EACrB,iBAAiB,GAAE,MAAU,GAC5B,OAAO,CAAC,MAAM,EAAE,CAAC,CA2EnB;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,CAC/B,iBAAiB,EAAE,MAAM,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACjF,QAAQ,GAAE,MAAW,GACpB,OAAO,CAAC,MAAM,EAAE,CAAC,CAYnB"}
@@ -86,6 +86,10 @@ function readProviderPromptMap(config, parsedPrompts) {
86
86
  */
87
87
  async function processPrompt(prompt, basePath = '', maxRecursionDepth = 1) {
88
88
  (0, tiny_invariant_1.default)(typeof prompt.raw === 'string', `prompt.raw must be a string, but got ${JSON.stringify(prompt.raw)}`);
89
+ // Handling when the prompt is a raw function (e.g. javascript function)
90
+ if (prompt.function) {
91
+ return [prompt];
92
+ }
89
93
  if (!(0, utils_1.maybeFilePath)(prompt.raw)) {
90
94
  return (0, string_1.processString)(prompt);
91
95
  }
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/prompts/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AA6BA,sDA+CC;AASD,sCA0EC;AAQD,kCAeC;AAtLD,+BAAgC;AAChC,oEAAuC;AACvC,uDAA+B;AAQ/B,kCAA4D;AAC5D,wDAAwD;AACxD,4CAAoD;AACpD,8CAAsD;AACtD,oDAA4D;AAC5D,gDAAwD;AACxD,gDAAoD;AACpD,4CAAmD;AACnD,4CAAoD;AACpD,mCAAwD;AAExD,4CAA0B;AAE1B;;;;;GAKG;AACH,SAAgB,qBAAqB,CACnC,MAA8B,EAC9B,aAAuB;IAEvB,MAAM,GAAG,GAA6B,EAAE,CAAC;IAEzC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACtB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,UAAU,GAAG,EAAE,CAAC;IACtB,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;QACnC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IAC5C,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACxC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACjC,wDAAwD;YACxD,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAChB,MAAM,WAAW,GAAG,QAA2B,CAAC;gBAChD,IAAA,wBAAS,EACP,WAAW,CAAC,EAAE,EACd,4EAA4E,CAC7E,CAAC;gBACF,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,OAAO,IAAI,UAAU,CAAC;gBACxD,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;oBACtB,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,WAAW,CAAC,OAAO,IAAI,UAAU,CAAC;gBAC7D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,QAA8B,CAAC;gBACnD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/C,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;gBAC/C,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,IAAI,UAAU,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC,OAAO,IAAI,UAAU,CAAC;YAC1D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,aAAa,CACjC,MAAuB,EACvB,WAAmB,EAAE,EACrB,oBAA4B,CAAC;IAE7B,IAAA,wBAAS,EACP,OAAO,MAAM,CAAC,GAAG,KAAK,QAAQ,EAC9B,wCAAwC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CACrE,CAAC;IAEF,IAAI,CAAC,IAAA,qBAAa,EAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,IAAA,sBAAa,EAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,EACJ,SAAS,EACT,YAAY,EACZ,aAAa,EACb,QAAQ,GACT,GAKG,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;IAE1C,IAAI,aAAa,IAAI,iBAAiB,GAAG,CAAC,EAAE,CAAC;QAC3C,MAAM,WAAW,GAAG,IAAA,eAAQ,EAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE;YACzD,oBAAoB,EAAE,IAAI;SAC3B,CAAC,CAAC;QACH,gBAAM,CAAC,KAAK,CACV,mBAAmB,MAAM,CAAC,GAAG,OAAO,QAAQ,gBAAgB,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAC1F,CAAC;QACF,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,eAAe,IAAI,WAAW,EAAE,CAAC;YAC1C,MAAM,gBAAgB,GAAG,MAAM,aAAa,CAC1C,EAAE,GAAG,EAAE,eAAe,EAAE,EACxB,QAAQ,EACR,iBAAiB,GAAG,CAAC,CACtB,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,sEAAsE;YACtE,gBAAM,CAAC,KAAK,CACV,8BAA8B,MAAM,CAAC,GAAG,yCAAyC,CAClF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,IAAA,sBAAa,EAAC,MAAM,CAAC,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,IAAA,wBAAgB,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,SAAS,IAAI,IAAA,uBAAgB,EAAC,SAAS,CAAC,EAAE,CAAC;QAC7C,OAAO,IAAA,0BAAa,EAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACxB,OAAO,IAAA,8BAAmB,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACxB,OAAO,IAAA,0BAAiB,EAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACzB,OAAO,IAAA,qBAAc,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,SAAS,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACvD,OAAO,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,WAAW,CAC/B,iBAAiF,EACjF,WAAmB,EAAE;IAErB,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAsB,IAAA,sBAAc,EAAC,iBAAiB,CAAC,CAAC;IAC5E,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/prompts/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AA6BA,sDA+CC;AASD,sCA+EC;AAQD,kCAeC;AA3LD,+BAAgC;AAChC,oEAAuC;AACvC,uDAA+B;AAQ/B,kCAA4D;AAC5D,wDAAwD;AACxD,4CAAoD;AACpD,8CAAsD;AACtD,oDAA4D;AAC5D,gDAAwD;AACxD,gDAAoD;AACpD,4CAAmD;AACnD,4CAAoD;AACpD,mCAAwD;AAExD,4CAA0B;AAE1B;;;;;GAKG;AACH,SAAgB,qBAAqB,CACnC,MAA8B,EAC9B,aAAuB;IAEvB,MAAM,GAAG,GAA6B,EAAE,CAAC;IAEzC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACtB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,UAAU,GAAG,EAAE,CAAC;IACtB,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;QACnC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IAC5C,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACxC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACjC,wDAAwD;YACxD,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAChB,MAAM,WAAW,GAAG,QAA2B,CAAC;gBAChD,IAAA,wBAAS,EACP,WAAW,CAAC,EAAE,EACd,4EAA4E,CAC7E,CAAC;gBACF,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,OAAO,IAAI,UAAU,CAAC;gBACxD,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;oBACtB,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,WAAW,CAAC,OAAO,IAAI,UAAU,CAAC;gBAC7D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,QAA8B,CAAC;gBACnD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/C,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;gBAC/C,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,IAAI,UAAU,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC,OAAO,IAAI,UAAU,CAAC;YAC1D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,aAAa,CACjC,MAAuB,EACvB,WAAmB,EAAE,EACrB,oBAA4B,CAAC;IAE7B,IAAA,wBAAS,EACP,OAAO,MAAM,CAAC,GAAG,KAAK,QAAQ,EAC9B,wCAAwC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CACrE,CAAC;IAEF,wEAAwE;IACxE,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,OAAO,CAAC,MAAgB,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,CAAC,IAAA,qBAAa,EAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,IAAA,sBAAa,EAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,EACJ,SAAS,EACT,YAAY,EACZ,aAAa,EACb,QAAQ,GACT,GAKG,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;IAE1C,IAAI,aAAa,IAAI,iBAAiB,GAAG,CAAC,EAAE,CAAC;QAC3C,MAAM,WAAW,GAAG,IAAA,eAAQ,EAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE;YACzD,oBAAoB,EAAE,IAAI;SAC3B,CAAC,CAAC;QACH,gBAAM,CAAC,KAAK,CACV,mBAAmB,MAAM,CAAC,GAAG,OAAO,QAAQ,gBAAgB,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAC1F,CAAC;QACF,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,eAAe,IAAI,WAAW,EAAE,CAAC;YAC1C,MAAM,gBAAgB,GAAG,MAAM,aAAa,CAC1C,EAAE,GAAG,EAAE,eAAe,EAAE,EACxB,QAAQ,EACR,iBAAiB,GAAG,CAAC,CACtB,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,sEAAsE;YACtE,gBAAM,CAAC,KAAK,CACV,8BAA8B,MAAM,CAAC,GAAG,yCAAyC,CAClF,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,IAAA,sBAAa,EAAC,MAAM,CAAC,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,IAAA,wBAAgB,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,SAAS,IAAI,IAAA,uBAAgB,EAAC,SAAS,CAAC,EAAE,CAAC;QAC7C,OAAO,IAAA,0BAAa,EAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACxB,OAAO,IAAA,8BAAmB,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;QACxB,OAAO,IAAA,0BAAiB,EAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACzB,OAAO,IAAA,qBAAc,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,SAAS,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACvD,OAAO,IAAA,sBAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,WAAW,CAC/B,iBAAiF,EACjF,WAAmB,EAAE;IAErB,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IAC1E,MAAM,cAAc,GAAsB,IAAA,sBAAc,EAAC,iBAAiB,CAAC,CAAC;IAC5E,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}
@@ -143,6 +143,17 @@ interface BedrockMistralGenerationOptions extends BedrockOptions {
143
143
  top_p?: number;
144
144
  top_k?: number;
145
145
  }
146
+ export interface BedrockAI21GenerationOptions extends BedrockOptions {
147
+ frequency_penalty?: number;
148
+ max_tokens?: number;
149
+ presence_penalty?: number;
150
+ response_format?: {
151
+ type: 'json_object' | 'text';
152
+ };
153
+ stop?: string[];
154
+ temperature?: number;
155
+ top_p?: number;
156
+ }
146
157
  export declare function parseValue(value: string | number, defaultValue: any): string | number;
147
158
  export declare function addConfigParam(params: any, key: string, configValue: any, envValue?: string | number | undefined, defaultValue?: any): void;
148
159
  export declare enum LlamaVersion {
@@ -164,6 +175,10 @@ export declare const getLlamaModelHandler: (version: LlamaVersion) => {
164
175
  output: (responseJson: any) => any;
165
176
  };
166
177
  export declare const BEDROCK_MODEL: {
178
+ AI21: {
179
+ params: (config: BedrockAI21GenerationOptions, prompt: string) => any;
180
+ output: (responseJson: any) => any;
181
+ };
167
182
  CLAUDE_COMPLETION: {
168
183
  params: (config: BedrockClaudeLegacyCompletionOptions, prompt: string, stop: string[]) => any;
169
184
  output: (responseJson: any) => any;
@@ -1 +1 @@
1
- {"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAMtE,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,YAAY,EACZ,gBAAgB,EAChB,yBAAyB,EAC1B,MAAM,UAAU,CAAC;AAKlB,UAAU,cAAc;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,qBAAqB;IAC7B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,4BAA6B,SAAQ,cAAc;IAC3D,oBAAoB,CAAC,EAAE,qBAAqB,CAAC;CAC9C;AAED,UAAU,oCAAqC,SAAQ,cAAc;IACnE,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,sCAAuC,SAAQ,cAAc;IAC5E,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,GAAG,CAAC;KACnB,EAAE,CAAC;IACJ,WAAW,CAAC,EAAE;QACZ,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;QAC9B,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;CACH;AAED,UAAU,6BAA8B,SAAQ,cAAc;IAC5D,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,UAAU,qCAAsC,SAAQ,cAAc;IACpE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,UAAU,sCAAuC,SAAQ,cAAc;IACrE,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,SAAS,CAAC,EAAE,KAAK,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,qBAAqB,EAAE,MAAM,CAC3B,MAAM,EACN;YACE,WAAW,EAAE,MAAM,CAAC;YACpB,IAAI,EAAE,MAAM,CAAC;YACb,QAAQ,EAAE,OAAO,CAAC;SACnB,CACF,CAAC;KACH,CAAC,CAAC;IACH,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;SACpC,CAAC;QACF,OAAO,EAAE,KAAK,CAAC;YACb,IAAI,EAAE,MAAM,CAAC;SACd,CAAC,CAAC;KACJ,CAAC,CAAC;IACH,cAAc,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,UAAU,2BAA2B;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE;QACZ,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC;KACjB,EAAE,CAAC;IACJ,SAAS,CAAC,EAAE;QACV,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;KACjB,EAAE,CAAC;IACJ,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,qBAAqB,EAAE;YACrB,CAAC,aAAa,EAAE,MAAM,GAAG;gBACvB,WAAW,EAAE,MAAM,CAAC;gBACpB,IAAI,EAAE,MAAM,CAAC;gBACb,QAAQ,EAAE,OAAO,CAAC;aACnB,CAAC;SACH,CAAC;KACH,EAAE,CAAC;IACJ,YAAY,CAAC,EAAE;QACb,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,EAAE;gBACV,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,CAAC;aACjC,CAAC;SACH,CAAC;QACF,OAAO,EAAE;YACP,IAAI,EAAE,MAAM,CAAC;SACd,EAAE,CAAC;KACL,EAAE,CAAC;IACJ,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,UAAU,+BAAgC,SAAQ,cAAc;IAC9D,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAOD,wBAAgB,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,YAAY,EAAE,GAAG,mBAQnE;AAED,wBAAgB,cAAc,CAC5B,MAAM,EAAE,GAAG,EACX,GAAG,EAAE,MAAM,EACX,WAAW,EAAE,GAAG,EAChB,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,EACtC,YAAY,CAAC,EAAE,GAAG,QAMnB;AAED,oBAAY,YAAY;IACtB,EAAE,IAAI;IACN,EAAE,IAAI;CACP;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAGD,eAAO,MAAM,sBAAsB,aAAc,YAAY,EAAE,KAAG,MAyCjE,CAAC;AAgBF,eAAO,MAAM,oBAAoB,YAAa,YAAY;qBAMrC,6BAA6B,UAAU,MAAM;gBAcpC,MAAM;sBAAgB,MAAM;gBAAU,MAAM;sBAAgB,MAAM;;2BAqBrE,GAAG;CAE7B,CAAC;AAEF,eAAO,MAAM,aAAa;;yBAEL,oCAAoC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BAqB9D,GAAG;;;yBAGT,sCAAsC,UAAU,MAAM;+BA6BhD,GAAG;;;yBAKT,4BAA4B,UAAU,MAAM,QAAQ,MAAM,EAAE;;;;+BAgCtD,GAAG;;;yBAnIT,6BAA6B,UAAU,MAAM;oBAcpC,MAAM;0BAAgB,MAAM;oBAAU,MAAM;0BAAgB,MAAM;;+BAqBrE,GAAG;;;yBAnCT,6BAA6B,UAAU,MAAM;oBAcpC,MAAM;0BAAgB,MAAM;oBAAU,MAAM;0BAAgB,MAAM;;+BAqBrE,GAAG;;;yBAqGT,qCAAqC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BAc/D,GAAG;;;yBAGT,sCAAsC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BA+BhE,GAAG;;;yBAGT,+BAA+B,UAAU,MAAM,QAAQ,MAAM,EAAE;+BAoBzD,GAAG;;CAE7B,CAAC;AAuDF,8BAAsB,yBAAyB;IAC7C,SAAS,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;IACnB,OAAO,CAAC,EAAE,cAAc,CAAC;IACzB,MAAM,EAAE,cAAc,CAAC;gBAGrB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAS5E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,kBAAkB;IAiCxB,SAAS,IAAI,MAAM;CAQpB;AAED,qBAAa,4BAA6B,SAAQ,yBAA0B,YAAW,WAAW;IAChG,MAAM,CAAC,6BAA6B,WAAmC;IAEjE,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAwEzD;AAED,qBAAa,2BACX,SAAQ,yBACR,YAAW,oBAAoB;IAEzB,OAAO,IAAI,OAAO,CAAC,yBAAyB,CAAC;IAI7C,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAiDzE"}
1
+ {"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/providers/bedrock.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAMtE,OAAO,KAAK,EACV,WAAW,EACX,oBAAoB,EACpB,YAAY,EACZ,gBAAgB,EAChB,yBAAyB,EAC1B,MAAM,UAAU,CAAC;AAKlB,UAAU,cAAc;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,qBAAqB;IAC7B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,4BAA6B,SAAQ,cAAc;IAC3D,oBAAoB,CAAC,EAAE,qBAAqB,CAAC;CAC9C;AAED,UAAU,oCAAqC,SAAQ,cAAc;IACnE,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,sCAAuC,SAAQ,cAAc;IAC5E,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,GAAG,CAAC;KACnB,EAAE,CAAC;IACJ,WAAW,CAAC,EAAE;QACZ,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;QAC9B,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;CACH;AAED,UAAU,6BAA8B,SAAQ,cAAc;IAC5D,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,UAAU,qCAAsC,SAAQ,cAAc;IACpE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,UAAU,sCAAuC,SAAQ,cAAc;IACrE,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,SAAS,CAAC,EAAE,KAAK,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,qBAAqB,EAAE,MAAM,CAC3B,MAAM,EACN;YACE,WAAW,EAAE,MAAM,CAAC;YACpB,IAAI,EAAE,MAAM,CAAC;YACb,QAAQ,EAAE,OAAO,CAAC;SACnB,CACF,CAAC;KACH,CAAC,CAAC;IACH,YAAY,CAAC,EAAE,KAAK,CAAC;QACnB,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;SACpC,CAAC;QACF,OAAO,EAAE,KAAK,CAAC;YACb,IAAI,EAAE,MAAM,CAAC;SACd,CAAC,CAAC;KACJ,CAAC,CAAC;IACH,cAAc,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,UAAU,2BAA2B;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE;QACZ,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,EAAE,MAAM,CAAC;KACjB,EAAE,CAAC;IACJ,SAAS,CAAC,EAAE;QACV,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;KACjB,EAAE,CAAC;IACJ,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,qBAAqB,EAAE;YACrB,CAAC,aAAa,EAAE,MAAM,GAAG;gBACvB,WAAW,EAAE,MAAM,CAAC;gBACpB,IAAI,EAAE,MAAM,CAAC;gBACb,QAAQ,EAAE,OAAO,CAAC;aACnB,CAAC;SACH,CAAC;KACH,EAAE,CAAC;IACJ,YAAY,CAAC,EAAE;QACb,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,EAAE;gBACV,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,CAAC;aACjC,CAAC;SACH,CAAC;QACF,OAAO,EAAE;YACP,IAAI,EAAE,MAAM,CAAC;SACd,EAAE,CAAC;KACL,EAAE,CAAC;IACJ,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,UAAU,+BAAgC,SAAQ,cAAc;IAC9D,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,4BAA6B,SAAQ,cAAc;IAClE,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE;QAAE,IAAI,EAAE,aAAa,GAAG,MAAM,CAAA;KAAE,CAAC;IACnD,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAOD,wBAAgB,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,YAAY,EAAE,GAAG,mBAQnE;AAED,wBAAgB,cAAc,CAC5B,MAAM,EAAE,GAAG,EACX,GAAG,EAAE,MAAM,EACX,WAAW,EAAE,GAAG,EAChB,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,EACtC,YAAY,CAAC,EAAE,GAAG,QAMnB;AAED,oBAAY,YAAY;IACtB,EAAE,IAAI;IACN,EAAE,IAAI;CACP;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAGD,eAAO,MAAM,sBAAsB,aAAc,YAAY,EAAE,KAAG,MAyCjE,CAAC;AAgBF,eAAO,MAAM,oBAAoB,YAAa,YAAY;qBAMrC,6BAA6B,UAAU,MAAM;gBAcpC,MAAM;sBAAgB,MAAM;gBAAU,MAAM;sBAAgB,MAAM;;2BAqBrE,GAAG;CAE7B,CAAC;AAEF,eAAO,MAAM,aAAa;;yBAEL,4BAA4B,UAAU,MAAM;+BAmCtC,GAAG;;;yBAQT,oCAAoC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BAqB9D,GAAG;;;yBAGT,sCAAsC,UAAU,MAAM;+BA6BhD,GAAG;;;yBAKT,4BAA4B,UAAU,MAAM,QAAQ,MAAM,EAAE;;;;+BAgCtD,GAAG;;;yBA9KT,6BAA6B,UAAU,MAAM;oBAcpC,MAAM;0BAAgB,MAAM;oBAAU,MAAM;0BAAgB,MAAM;;+BAqBrE,GAAG;;;yBAnCT,6BAA6B,UAAU,MAAM;oBAcpC,MAAM;0BAAgB,MAAM;oBAAU,MAAM;0BAAgB,MAAM;;+BAqBrE,GAAG;;;yBAgJT,qCAAqC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BA0B/D,GAAG;;;yBAGT,sCAAsC,UAAU,MAAM,QAAQ,MAAM,EAAE;+BA+BhE,GAAG;;;yBAGT,+BAA+B,UAAU,MAAM,QAAQ,MAAM,EAAE;+BAoBzD,GAAG;;CAE7B,CAAC;AA+DF,8BAAsB,yBAAyB;IAC7C,SAAS,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;IACnB,OAAO,CAAC,EAAE,cAAc,CAAC;IACzB,MAAM,EAAE,cAAc,CAAC;gBAGrB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAS5E,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIZ,kBAAkB;IAiCxB,SAAS,IAAI,MAAM;CAQpB;AAED,qBAAa,4BAA6B,SAAQ,yBAA0B,YAAW,WAAW;IAChG,MAAM,CAAC,6BAA6B,WAAmC;IAEjE,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAwEzD;AAED,qBAAa,2BACX,SAAQ,yBACR,YAAW,oBAAoB;IAEzB,OAAO,IAAI,OAAO,CAAC,yBAAyB,CAAC;IAI7C,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAiDzE"}
@@ -130,7 +130,7 @@ const getLlamaModelHandler = (version) => {
130
130
  prompt: finalPrompt,
131
131
  };
132
132
  addConfigParam(params, 'temperature', config?.temperature, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TEMPERATURE'), 0.01);
133
- addConfigParam(params, 'top_p', config?.top_p, process.env.AWS_BEDROCK_TOP_P, 1);
133
+ addConfigParam(params, 'top_p', config?.top_p, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TOP_P'), 1);
134
134
  addConfigParam(params, 'max_gen_len', config?.max_gen_len, (0, envars_1.getEnvInt)('AWS_BEDROCK_MAX_GEN_LEN'), 1024);
135
135
  return params;
136
136
  },
@@ -139,14 +139,35 @@ const getLlamaModelHandler = (version) => {
139
139
  };
140
140
  exports.getLlamaModelHandler = getLlamaModelHandler;
141
141
  exports.BEDROCK_MODEL = {
142
+ AI21: {
143
+ params: (config, prompt) => {
144
+ const messages = (0, shared_1.parseChatPrompt)(prompt, [{ role: 'user', content: prompt }]);
145
+ const params = {
146
+ messages,
147
+ };
148
+ addConfigParam(params, 'max_tokens', config?.max_tokens, (0, envars_1.getEnvInt)('AWS_BEDROCK_MAX_TOKENS'), 1024);
149
+ addConfigParam(params, 'temperature', config?.temperature, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TEMPERATURE'), 0);
150
+ addConfigParam(params, 'top_p', config?.top_p, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TOP_P'), 1.0);
151
+ addConfigParam(params, 'stop', config?.stop, (0, envars_1.getEnvString)('AWS_BEDROCK_STOP'));
152
+ addConfigParam(params, 'frequency_penalty', config?.frequency_penalty, (0, envars_1.getEnvFloat)('AWS_BEDROCK_FREQUENCY_PENALTY'));
153
+ addConfigParam(params, 'presence_penalty', config?.presence_penalty, (0, envars_1.getEnvFloat)('AWS_BEDROCK_PRESENCE_PENALTY'));
154
+ return params;
155
+ },
156
+ output: (responseJson) => {
157
+ if (responseJson.error) {
158
+ throw new Error(`AI21 API error: ${responseJson.error}`);
159
+ }
160
+ return responseJson.choices[0].message.content;
161
+ },
162
+ },
142
163
  CLAUDE_COMPLETION: {
143
164
  params: (config, prompt, stop) => {
144
165
  const params = {
145
166
  prompt: `${sdk_1.default.HUMAN_PROMPT} ${prompt} ${sdk_1.default.AI_PROMPT}`,
146
167
  stop_sequences: stop,
147
168
  };
148
- addConfigParam(params, 'max_tokens_to_sample', config?.max_tokens_to_sample, process.env.AWS_BEDROCK_MAX_TOKENS, 1024);
149
- addConfigParam(params, 'temperature', config?.temperature, process.env.AWS_BEDROCK_TEMPERATURE, 0);
169
+ addConfigParam(params, 'max_tokens_to_sample', config?.max_tokens_to_sample, (0, envars_1.getEnvInt)('AWS_BEDROCK_MAX_TOKENS'), 1024);
170
+ addConfigParam(params, 'temperature', config?.temperature, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TEMPERATURE'), 0);
150
171
  return params;
151
172
  },
152
173
  output: (responseJson) => responseJson?.completion,
@@ -155,7 +176,7 @@ exports.BEDROCK_MODEL = {
155
176
  params: (config, prompt) => {
156
177
  const { system, extractedMessages } = (0, anthropic_1.parseMessages)(prompt);
157
178
  const params = { messages: extractedMessages };
158
- addConfigParam(params, 'max_tokens', config?.max_tokens, process.env.AWS_BEDROCK_MAX_TOKENS, 1024);
179
+ addConfigParam(params, 'max_tokens', config?.max_tokens, (0, envars_1.getEnvInt)('AWS_BEDROCK_MAX_TOKENS'), 1024);
159
180
  addConfigParam(params, 'temperature', config?.temperature, undefined, 0);
160
181
  addConfigParam(params, 'anthropic_version', config?.anthropic_version, undefined, 'bedrock-2023-05-31');
161
182
  addConfigParam(params, 'tools', (0, util_1.maybeLoadFromExternalFile)(config?.tools), undefined, undefined);
@@ -170,9 +191,9 @@ exports.BEDROCK_MODEL = {
170
191
  TITAN_TEXT: {
171
192
  params: (config, prompt, stop) => {
172
193
  const textGenerationConfig = {};
173
- addConfigParam(textGenerationConfig, 'maxTokenCount', config?.textGenerationConfig?.maxTokenCount, process.env.AWS_BEDROCK_MAX_TOKENS, 1024);
174
- addConfigParam(textGenerationConfig, 'temperature', config?.textGenerationConfig?.temperature, process.env.AWS_BEDROCK_TEMPERATURE, 0);
175
- addConfigParam(textGenerationConfig, 'topP', config?.textGenerationConfig?.topP, process.env.AWS_BEDROCK_TOP_P, 1);
194
+ addConfigParam(textGenerationConfig, 'maxTokenCount', config?.textGenerationConfig?.maxTokenCount, (0, envars_1.getEnvInt)('AWS_BEDROCK_MAX_TOKENS'), 1024);
195
+ addConfigParam(textGenerationConfig, 'temperature', config?.textGenerationConfig?.temperature, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TEMPERATURE'), 0);
196
+ addConfigParam(textGenerationConfig, 'topP', config?.textGenerationConfig?.topP, (0, envars_1.getEnvFloat)('AWS_BEDROCK_TOP_P'), 1);
176
197
  addConfigParam(textGenerationConfig, 'stopSequences', config?.textGenerationConfig?.stopSequences, undefined, stop);
177
198
  return { inputText: prompt, textGenerationConfig };
178
199
  },
@@ -183,10 +204,10 @@ exports.BEDROCK_MODEL = {
183
204
  COHERE_COMMAND: {
184
205
  params: (config, prompt, stop) => {
185
206
  const params = { prompt };
186
- addConfigParam(params, 'temperature', config?.temperature, process.env.COHERE_TEMPERATURE, 0);
187
- addConfigParam(params, 'p', config?.p, process.env.COHERE_P, 1);
188
- addConfigParam(params, 'k', config?.k, process.env.COHERE_K, 0);
189
- addConfigParam(params, 'max_tokens', config?.max_tokens, process.env.COHERE_MAX_TOKENS, 1024);
207
+ addConfigParam(params, 'temperature', config?.temperature, (0, envars_1.getEnvFloat)('COHERE_TEMPERATURE'), 0);
208
+ addConfigParam(params, 'p', config?.p, (0, envars_1.getEnvFloat)('COHERE_P'), 1);
209
+ addConfigParam(params, 'k', config?.k, (0, envars_1.getEnvInt)('COHERE_K'), 0);
210
+ addConfigParam(params, 'max_tokens', config?.max_tokens, (0, envars_1.getEnvInt)('COHERE_MAX_TOKENS'), 1024);
190
211
  addConfigParam(params, 'return_likelihoods', config?.return_likelihoods, undefined, 'NONE');
191
212
  addConfigParam(params, 'stream', config?.stream, undefined, false);
192
213
  addConfigParam(params, 'num_generations', config?.num_generations, undefined, 1);
@@ -234,16 +255,18 @@ exports.BEDROCK_MODEL = {
234
255
  MISTRAL: {
235
256
  params: (config, prompt, stop) => {
236
257
  const params = { prompt, stop };
237
- addConfigParam(params, 'max_tokens', config?.max_tokens, process.env.MISTRAL_MAX_TOKENS, 1024);
238
- addConfigParam(params, 'temperature', config?.temperature, process.env.MISTRAL_TEMPERATURE, 0);
239
- addConfigParam(params, 'top_p', config?.top_p, process.env.MISTRAL_TOP_P, 1);
240
- addConfigParam(params, 'top_k', config?.top_k, process.env.MISTRAL_TOP_K, 0);
258
+ addConfigParam(params, 'max_tokens', config?.max_tokens, (0, envars_1.getEnvInt)('MISTRAL_MAX_TOKENS'), 1024);
259
+ addConfigParam(params, 'temperature', config?.temperature, (0, envars_1.getEnvFloat)('MISTRAL_TEMPERATURE'), 0);
260
+ addConfigParam(params, 'top_p', config?.top_p, (0, envars_1.getEnvFloat)('MISTRAL_TOP_P'), 1);
261
+ addConfigParam(params, 'top_k', config?.top_k, (0, envars_1.getEnvFloat)('MISTRAL_TOP_K'), 0);
241
262
  return params;
242
263
  },
243
264
  output: (responseJson) => responseJson?.outputs[0]?.text,
244
265
  },
245
266
  };
246
267
  const AWS_BEDROCK_MODELS = {
268
+ 'ai21.jamba-1-5-large-v1:0': exports.BEDROCK_MODEL.AI21,
269
+ 'ai21.jamba-1-5-mini-v1:0': exports.BEDROCK_MODEL.AI21,
247
270
  'amazon.titan-text-express-v1': exports.BEDROCK_MODEL.TITAN_TEXT,
248
271
  'amazon.titan-text-lite-v1': exports.BEDROCK_MODEL.TITAN_TEXT,
249
272
  'amazon.titan-text-premier-v1:0': exports.BEDROCK_MODEL.TITAN_TEXT,
@@ -261,6 +284,9 @@ const AWS_BEDROCK_MODELS = {
261
284
  'cohere.command-text-v14': exports.BEDROCK_MODEL.COHERE_COMMAND,
262
285
  'meta.llama2-13b-chat-v1': exports.BEDROCK_MODEL.LLAMA2,
263
286
  'meta.llama2-70b-chat-v1': exports.BEDROCK_MODEL.LLAMA2,
287
+ 'meta.llama3-1-405b-instruct-v1:0': exports.BEDROCK_MODEL.LLAMA3,
288
+ 'meta.llama3-1-70b-instruct-v1:0': exports.BEDROCK_MODEL.LLAMA3,
289
+ 'meta.llama3-1-8b-instruct-v1:0': exports.BEDROCK_MODEL.LLAMA3,
264
290
  'meta.llama3-70b-instruct-v1:0': exports.BEDROCK_MODEL.LLAMA3,
265
291
  'meta.llama3-8b-instruct-v1:0': exports.BEDROCK_MODEL.LLAMA3,
266
292
  'mistral.mistral-7b-instruct-v0:2': exports.BEDROCK_MODEL.MISTRAL,
@@ -274,6 +300,9 @@ function getHandlerForModel(modelName) {
274
300
  if (ret) {
275
301
  return ret;
276
302
  }
303
+ if (modelName.startsWith('ai21.')) {
304
+ return exports.BEDROCK_MODEL.AI21;
305
+ }
277
306
  if (modelName.startsWith('anthropic.claude')) {
278
307
  return exports.BEDROCK_MODEL.CLAUDE_MESSAGES;
279
308
  }
@@ -341,7 +370,7 @@ class AwsBedrockGenericProvider {
341
370
  getRegion() {
342
371
  return (this.config?.region ||
343
372
  this.env?.AWS_BEDROCK_REGION ||
344
- process.env.AWS_BEDROCK_REGION ||
373
+ (0, envars_1.getEnvString)('AWS_BEDROCK_REGION') ||
345
374
  'us-west-2');
346
375
  }
347
376
  }
@@ -350,7 +379,7 @@ class AwsBedrockCompletionProvider extends AwsBedrockGenericProvider {
350
379
  async callApi(prompt) {
351
380
  let stop;
352
381
  try {
353
- stop = process.env.AWS_BEDROCK_STOP ? JSON.parse(process.env.AWS_BEDROCK_STOP) : [];
382
+ stop = (0, envars_1.getEnvString)('AWS_BEDROCK_STOP') ? JSON.parse((0, envars_1.getEnvString)('AWS_BEDROCK_STOP')) : [];
354
383
  }
355
384
  catch (err) {
356
385
  throw new Error(`BEDROCK_STOP is not a valid JSON string: ${err}`);
@@ -404,9 +433,9 @@ class AwsBedrockCompletionProvider extends AwsBedrockGenericProvider {
404
433
  return {
405
434
  output: model.output(output),
406
435
  tokenUsage: {
407
- total: output.prompt_token_count + output.generation_token_count,
408
- prompt: output.prompt_token_count,
409
- completion: output.generation_token_count,
436
+ total: output.total_tokens ?? output.prompt_token_count + output.generation_token_count,
437
+ prompt: output.prompt_tokens ?? output.prompt_token_count,
438
+ completion: output.completion_tokens ?? output.generation_token_count,
410
439
  },
411
440
  };
412
441
  }