@llumiverse/core 0.21.0 → 0.22.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 (80) hide show
  1. package/lib/cjs/CompletionStream.js +44 -25
  2. package/lib/cjs/CompletionStream.js.map +1 -1
  3. package/lib/cjs/Driver.js +4 -1
  4. package/lib/cjs/Driver.js.map +1 -1
  5. package/lib/esm/CompletionStream.js +44 -25
  6. package/lib/esm/CompletionStream.js.map +1 -1
  7. package/lib/esm/Driver.js +4 -1
  8. package/lib/esm/Driver.js.map +1 -1
  9. package/lib/types/CompletionStream.d.ts.map +1 -1
  10. package/lib/types/Driver.d.ts.map +1 -1
  11. package/package.json +8 -8
  12. package/src/CompletionStream.ts +42 -27
  13. package/src/Driver.ts +4 -1
  14. package/lib/cjs/capability/bedrock.js +0 -186
  15. package/lib/cjs/capability/bedrock.js.map +0 -1
  16. package/lib/cjs/capability/openai.js +0 -122
  17. package/lib/cjs/capability/openai.js.map +0 -1
  18. package/lib/cjs/capability/vertexai.js +0 -90
  19. package/lib/cjs/capability/vertexai.js.map +0 -1
  20. package/lib/cjs/capability.js +0 -52
  21. package/lib/cjs/capability.js.map +0 -1
  22. package/lib/cjs/formatters/openai.js +0 -113
  23. package/lib/cjs/formatters/openai.js.map +0 -1
  24. package/lib/cjs/options/bedrock.js +0 -343
  25. package/lib/cjs/options/bedrock.js.map +0 -1
  26. package/lib/cjs/options/groq.js +0 -37
  27. package/lib/cjs/options/groq.js.map +0 -1
  28. package/lib/cjs/options/openai.js +0 -123
  29. package/lib/cjs/options/openai.js.map +0 -1
  30. package/lib/cjs/options/vertexai.js +0 -257
  31. package/lib/cjs/options/vertexai.js.map +0 -1
  32. package/lib/cjs/options.js +0 -19
  33. package/lib/cjs/options.js.map +0 -1
  34. package/lib/cjs/types.js +0 -80
  35. package/lib/cjs/types.js.map +0 -1
  36. package/lib/esm/capability/bedrock.js +0 -183
  37. package/lib/esm/capability/bedrock.js.map +0 -1
  38. package/lib/esm/capability/openai.js +0 -119
  39. package/lib/esm/capability/openai.js.map +0 -1
  40. package/lib/esm/capability/vertexai.js +0 -87
  41. package/lib/esm/capability/vertexai.js.map +0 -1
  42. package/lib/esm/capability.js +0 -47
  43. package/lib/esm/capability.js.map +0 -1
  44. package/lib/esm/formatters/openai.js +0 -109
  45. package/lib/esm/formatters/openai.js.map +0 -1
  46. package/lib/esm/options/bedrock.js +0 -340
  47. package/lib/esm/options/bedrock.js.map +0 -1
  48. package/lib/esm/options/groq.js +0 -34
  49. package/lib/esm/options/groq.js.map +0 -1
  50. package/lib/esm/options/openai.js +0 -120
  51. package/lib/esm/options/openai.js.map +0 -1
  52. package/lib/esm/options/vertexai.js +0 -253
  53. package/lib/esm/options/vertexai.js.map +0 -1
  54. package/lib/esm/options.js +0 -16
  55. package/lib/esm/options.js.map +0 -1
  56. package/lib/esm/types.js +0 -77
  57. package/lib/esm/types.js.map +0 -1
  58. package/lib/tsconfig.tsbuildinfo +0 -1
  59. package/lib/types/capability/bedrock.d.ts +0 -6
  60. package/lib/types/capability/bedrock.d.ts.map +0 -1
  61. package/lib/types/capability/openai.d.ts +0 -10
  62. package/lib/types/capability/openai.d.ts.map +0 -1
  63. package/lib/types/capability/vertexai.d.ts +0 -10
  64. package/lib/types/capability/vertexai.d.ts.map +0 -1
  65. package/lib/types/capability.d.ts +0 -4
  66. package/lib/types/capability.d.ts.map +0 -1
  67. package/lib/types/formatters/openai.d.ts +0 -41
  68. package/lib/types/formatters/openai.d.ts.map +0 -1
  69. package/lib/types/options/bedrock.d.ts +0 -31
  70. package/lib/types/options/bedrock.d.ts.map +0 -1
  71. package/lib/types/options/groq.d.ts +0 -11
  72. package/lib/types/options/groq.d.ts.map +0 -1
  73. package/lib/types/options/openai.d.ts +0 -20
  74. package/lib/types/options/openai.d.ts.map +0 -1
  75. package/lib/types/options/vertexai.d.ts +0 -51
  76. package/lib/types/options/vertexai.d.ts.map +0 -1
  77. package/lib/types/options.d.ts +0 -2
  78. package/lib/types/options.d.ts.map +0 -1
  79. package/lib/types/types.d.ts +0 -322
  80. package/lib/types/types.d.ts.map +0 -1
@@ -1,123 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.getOpenAiOptions = getOpenAiOptions;
4
- const types_js_1 = require("../types.js");
5
- const options_js_1 = require("../options.js");
6
- function getOpenAiOptions(model, _option) {
7
- const visionOptions = isVisionModel(model) ? [
8
- {
9
- name: "image_detail", type: types_js_1.OptionType.enum, enum: { "Low": "low", "High": "high", "Auto": "auto" },
10
- default: "auto", description: "Controls how the model processes an input image."
11
- },
12
- ] : [];
13
- if (model.includes("o1") || model.includes("o3")) {
14
- //Is thinking text model
15
- let max_tokens_limit = 4096;
16
- if (model.includes("o1")) {
17
- if (model.includes("preview")) {
18
- max_tokens_limit = 32768;
19
- }
20
- else if (model.includes("mini")) {
21
- max_tokens_limit = 65536;
22
- }
23
- else {
24
- max_tokens_limit = 100000;
25
- }
26
- }
27
- else if (model.includes("o3")) {
28
- max_tokens_limit = 100000;
29
- }
30
- const commonOptions = [
31
- {
32
- name: types_js_1.SharedOptions.max_tokens, type: types_js_1.OptionType.numeric, min: 1, max: max_tokens_limit,
33
- integer: true, description: "The maximum number of tokens to generate",
34
- },
35
- {
36
- name: types_js_1.SharedOptions.stop_sequence, type: types_js_1.OptionType.string_list, value: [],
37
- description: "The stop sequence of the generated image",
38
- },
39
- ];
40
- const reasoningOptions = model.includes("o3") || isO1Full(model) ? [
41
- {
42
- name: "reasoning_effort", type: types_js_1.OptionType.enum, enum: { "Low": "low", "Medium": "medium", "High": "high" },
43
- default: "medium", description: "How much effort the model should put into reasoning, lower values result in faster responses and less tokens used."
44
- },
45
- ] : [];
46
- return {
47
- _option_id: "openai-thinking",
48
- options: [
49
- ...commonOptions,
50
- ...reasoningOptions,
51
- ...visionOptions,
52
- ],
53
- };
54
- }
55
- else {
56
- let max_tokens_limit = 4096;
57
- if (model.includes("gpt-4o")) {
58
- max_tokens_limit = 16384;
59
- if (model.includes("gpt-4o-2024-05-13") || model.includes("realtime")) {
60
- max_tokens_limit = 4096;
61
- }
62
- }
63
- else if (model.includes("gpt-4")) {
64
- if (model.includes("turbo")) {
65
- max_tokens_limit = 4096;
66
- }
67
- else {
68
- max_tokens_limit = 8192;
69
- }
70
- }
71
- else if (model.includes("gpt-3-5")) {
72
- max_tokens_limit = 4096;
73
- }
74
- //Is non-thinking text model
75
- const commonOptions = [
76
- {
77
- name: types_js_1.SharedOptions.max_tokens, type: types_js_1.OptionType.numeric, min: 1, max: max_tokens_limit,
78
- integer: true, step: 200, description: "The maximum number of tokens to generate",
79
- },
80
- {
81
- name: "temperature", type: types_js_1.OptionType.numeric, min: 0.0, max: 2.0, default: 0.7,
82
- integer: false, step: 0.1, description: "A higher temperature biases toward less likely tokens, making the model more creative"
83
- },
84
- {
85
- name: "top_p", type: types_js_1.OptionType.numeric, min: 0, max: 1,
86
- integer: false, step: 0.1, description: "Limits token sampling to the cumulative probability of the top p tokens"
87
- },
88
- {
89
- name: "presence_penalty", type: types_js_1.OptionType.numeric, min: -2.0, max: 2.0,
90
- integer: false, step: 0.1, description: "Penalise tokens if they appear at least once in the text"
91
- },
92
- {
93
- name: "frequency_penalty", type: types_js_1.OptionType.numeric, min: -2.0, max: 2.0,
94
- integer: false, step: 0.1, description: "Penalise tokens based on their frequency in the text"
95
- },
96
- {
97
- name: types_js_1.SharedOptions.stop_sequence, type: types_js_1.OptionType.string_list, value: [],
98
- description: "The generation will halt if one of the stop sequences is output",
99
- }
100
- ];
101
- return {
102
- _option_id: "openai-text",
103
- options: [
104
- ...commonOptions,
105
- ...visionOptions,
106
- ],
107
- };
108
- }
109
- return options_js_1.textOptionsFallback;
110
- }
111
- function isO1Full(model) {
112
- if (model.includes("o1")) {
113
- if (model.includes("mini") || model.includes("preview")) {
114
- return false;
115
- }
116
- return true;
117
- }
118
- return false;
119
- }
120
- function isVisionModel(model) {
121
- return model.includes("gpt-4o") || isO1Full(model) || model.includes("gpt-4-turbo");
122
- }
123
- //# sourceMappingURL=openai.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/options/openai.ts"],"names":[],"mappings":";;AAyBA,4CA4GC;AArID,0CAA6G;AAC7G,8CAAoD;AAwBpD,SAAgB,gBAAgB,CAAC,KAAa,EAAE,OAAsB;IAClE,MAAM,aAAa,GAA0B,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5D;YACI,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;YACnG,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,kDAAkD;SACnF;KACR,CAAC,CAAC,CAAC,EAAE,CAAC;IAEP,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/C,wBAAwB;QACxB,IAAI,gBAAgB,GAAG,IAAI,CAAC;QAC5B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5B,gBAAgB,GAAG,KAAK,CAAC;YAC7B,CAAC;iBACI,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9B,gBAAgB,GAAG,KAAK,CAAC;YAC7B,CAAC;iBACI,CAAC;gBACF,gBAAgB,GAAG,MAAM,CAAC;YAC9B,CAAC;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,gBAAgB,GAAG,MAAM,CAAC;QAC9B,CAAC;QAED,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,wBAAa,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,gBAAgB;gBACvF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,0CAA0C;aACzE;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;gBAC1E,WAAW,EAAE,0CAA0C;aAC1D;SACJ,CAAC;QAEF,MAAM,gBAAgB,GAA0B,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACtF;gBACI,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE;gBAC3G,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,oHAAoH;aACvJ;SACJ,CAAC,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO;YACH,UAAU,EAAE,iBAAiB;YAC7B,OAAO,EAAE;gBACL,GAAG,aAAa;gBAChB,GAAG,gBAAgB;gBACnB,GAAG,aAAa;aACnB;SACJ,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,IAAI,gBAAgB,GAAG,IAAI,CAAC;QAC5B,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,gBAAgB,GAAG,KAAK,CAAC;YACzB,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpE,gBAAgB,GAAG,IAAI,CAAC;YAC5B,CAAC;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1B,gBAAgB,GAAG,IAAI,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACJ,gBAAgB,GAAG,IAAI,CAAC;YAC5B,CAAC;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACjC,gBAAgB,GAAG,IAAI,CAAC;QAC5B,CAAC;QAED,4BAA4B;QAC5B,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,wBAAa,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,gBAAgB;gBACvF,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,0CAA0C;aACpF;YACD;gBACI,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG;gBAC/E,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,uFAAuF;aAClI;YACD;gBACI,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC;gBACvD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,yEAAyE;aACpH;YACD;gBACI,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG;gBACvE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,0DAA0D;aACrG;YACD;gBACI,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG;gBACxE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,sDAAsD;aACjG;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;gBAC1E,WAAW,EAAE,iEAAiE;aACjF;SACJ,CAAA;QAED,OAAO;YACH,UAAU,EAAE,aAAa;YACzB,OAAO,EAAE;gBACL,GAAG,aAAa;gBAChB,GAAG,aAAa;aACnB;SACJ,CAAA;IACL,CAAC;IACD,OAAO,gCAAmB,CAAC;AAC/B,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa;IAC3B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACvB,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACtD,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAChC,OAAO,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;AACxF,CAAC"}
@@ -1,257 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.ImagenMaskMode = exports.ImagenTaskType = void 0;
4
- exports.getVertexAiOptions = getVertexAiOptions;
5
- const types_js_1 = require("../types.js");
6
- const options_js_1 = require("../options.js");
7
- var ImagenTaskType;
8
- (function (ImagenTaskType) {
9
- ImagenTaskType["TEXT_IMAGE"] = "TEXT_IMAGE";
10
- ImagenTaskType["EDIT_MODE_INPAINT_REMOVAL"] = "EDIT_MODE_INPAINT_REMOVAL";
11
- ImagenTaskType["EDIT_MODE_INPAINT_INSERTION"] = "EDIT_MODE_INPAINT_INSERTION";
12
- ImagenTaskType["EDIT_MODE_BGSWAP"] = "EDIT_MODE_BGSWAP";
13
- ImagenTaskType["EDIT_MODE_OUTPAINT"] = "EDIT_MODE_OUTPAINT";
14
- ImagenTaskType["CUSTOMIZATION_SUBJECT"] = "CUSTOMIZATION_SUBJECT";
15
- ImagenTaskType["CUSTOMIZATION_STYLE"] = "CUSTOMIZATION_STYLE";
16
- ImagenTaskType["CUSTOMIZATION_CONTROLLED"] = "CUSTOMIZATION_CONTROLLED";
17
- ImagenTaskType["CUSTOMIZATION_INSTRUCT"] = "CUSTOMIZATION_INSTRUCT";
18
- })(ImagenTaskType || (exports.ImagenTaskType = ImagenTaskType = {}));
19
- var ImagenMaskMode;
20
- (function (ImagenMaskMode) {
21
- ImagenMaskMode["MASK_MODE_USER_PROVIDED"] = "MASK_MODE_USER_PROVIDED";
22
- ImagenMaskMode["MASK_MODE_BACKGROUND"] = "MASK_MODE_BACKGROUND";
23
- ImagenMaskMode["MASK_MODE_FOREGROUND"] = "MASK_MODE_FOREGROUND";
24
- ImagenMaskMode["MASK_MODE_SEMANTIC"] = "MASK_MODE_SEMANTIC";
25
- })(ImagenMaskMode || (exports.ImagenMaskMode = ImagenMaskMode = {}));
26
- function getVertexAiOptions(model, option) {
27
- if (model.includes("imagen-3.0")) {
28
- const commonOptions = [
29
- {
30
- name: types_js_1.SharedOptions.number_of_images, type: types_js_1.OptionType.numeric, min: 1, max: 4, default: 1,
31
- integer: true, description: "Number of Images to generate",
32
- },
33
- {
34
- name: types_js_1.SharedOptions.seed, type: types_js_1.OptionType.numeric, min: 0, max: 4294967295, default: 12,
35
- integer: true, description: "The seed of the generated image"
36
- },
37
- {
38
- name: "person_generation", type: types_js_1.OptionType.enum, enum: { "Disallow the inclusion of people or faces in images": "dont_allow", "Allow generation of adults only": "allow_adult", "Allow generation of people of all ages": "allow_all" },
39
- default: "allow_adult", description: "The safety setting for allowing the generation of people in the image"
40
- },
41
- {
42
- name: "safety_setting", type: types_js_1.OptionType.enum, enum: { "Block very few problematic prompts and responses": "block_none", "Block only few problematic prompts and responses": "block_only_high", "Block some problematic prompts and responses": "block_medium_and_above", "Strictest filtering": "block_low_and_above" },
43
- default: "block_medium_and_above", description: "The overall safety setting"
44
- },
45
- ];
46
- const outputOptions = [
47
- {
48
- name: "image_file_type", type: types_js_1.OptionType.enum, enum: { "JPEG": "image/jpeg", "PNG": "image/png" },
49
- default: "image/png", description: "The file type of the generated image",
50
- refresh: true,
51
- },
52
- ];
53
- const jpegQuality = {
54
- name: "jpeg_compression_quality", type: types_js_1.OptionType.numeric, min: 0, max: 100, default: 75,
55
- integer: true, description: "The compression quality of the JPEG image",
56
- };
57
- if (option?.image_file_type === "image/jpeg") {
58
- outputOptions.push(jpegQuality);
59
- }
60
- if (model.includes("generate")) {
61
- //Generate models
62
- const modeOptions = [
63
- {
64
- name: "aspect_ratio", type: types_js_1.OptionType.enum, enum: { "1:1": "1:1", "4:3": "4:3", "3:4": "3:4", "16:9": "16:9", "9:16": "9:16" },
65
- default: "1:1", description: "The aspect ratio of the generated image"
66
- },
67
- {
68
- name: "add_watermark", type: types_js_1.OptionType.boolean, default: false, description: "Add an invisible watermark to the generated image, useful for detection of AI images"
69
- },
70
- ];
71
- const enhanceOptions = !model.includes("generate-001") ? [
72
- {
73
- name: "enhance_prompt", type: types_js_1.OptionType.boolean, default: true, description: "VertexAI automatically rewrites the prompt to better reflect the prompt's intent."
74
- },
75
- ] : [];
76
- return {
77
- _option_id: "vertexai-imagen",
78
- options: [
79
- ...commonOptions,
80
- ...modeOptions,
81
- ...outputOptions,
82
- ...enhanceOptions,
83
- ]
84
- };
85
- }
86
- if (model.includes("capability")) {
87
- //Edit models
88
- let guidanceScaleDefault = 75;
89
- if (option?.edit_mode === ImagenTaskType.EDIT_MODE_INPAINT_INSERTION) {
90
- guidanceScaleDefault = 60;
91
- }
92
- const modeOptions = [
93
- {
94
- name: "edit_mode", type: types_js_1.OptionType.enum,
95
- enum: {
96
- "EDIT_MODE_INPAINT_REMOVAL": "EDIT_MODE_INPAINT_REMOVAL",
97
- "EDIT_MODE_INPAINT_INSERTION": "EDIT_MODE_INPAINT_INSERTION",
98
- "EDIT_MODE_BGSWAP": "EDIT_MODE_BGSWAP",
99
- "EDIT_MODE_OUTPAINT": "EDIT_MODE_OUTPAINT",
100
- "CUSTOMIZATION_SUBJECT": "CUSTOMIZATION_SUBJECT",
101
- "CUSTOMIZATION_STYLE": "CUSTOMIZATION_STYLE",
102
- "CUSTOMIZATION_CONTROLLED": "CUSTOMIZATION_CONTROLLED",
103
- "CUSTOMIZATION_INSTRUCT": "CUSTOMIZATION_INSTRUCT",
104
- },
105
- description: "The editing mode. CUSTOMIZATION options use few-shot learning to generate images based on a few examples."
106
- },
107
- {
108
- name: "guidance_scale", type: types_js_1.OptionType.numeric, min: 0, max: 500, default: guidanceScaleDefault,
109
- integer: true, description: "How closely the generation follows the prompt"
110
- }
111
- ];
112
- const maskOptions = (option?.edit_mode?.includes("EDIT")) ? [
113
- {
114
- name: "mask_mode", type: types_js_1.OptionType.enum,
115
- enum: {
116
- "MASK_MODE_USER_PROVIDED": "MASK_MODE_USER_PROVIDED",
117
- "MASK_MODE_BACKGROUND": "MASK_MODE_BACKGROUND",
118
- "MASK_MODE_FOREGROUND": "MASK_MODE_FOREGROUND",
119
- "MASK_MODE_SEMANTIC": "MASK_MODE_SEMANTIC",
120
- },
121
- default: "MASK_MODE_USER_PROVIDED",
122
- description: "How should the mask for the generation be provided"
123
- },
124
- {
125
- name: "mask_dilation", type: types_js_1.OptionType.numeric, min: 0, max: 1,
126
- integer: true, description: "The mask dilation, grows the mask by a percetage of image width to compensate for imprecise masks."
127
- },
128
- ] : [];
129
- const maskClassOptions = (option?.mask_mode === ImagenMaskMode.MASK_MODE_SEMANTIC) ? [
130
- {
131
- name: "mask_class", type: types_js_1.OptionType.string_list, default: [],
132
- description: "Input Class IDs. Create a mask based on image class, based on https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/imagen-api-customization#segment-ids"
133
- }
134
- ] : [];
135
- const editOptions = option?.edit_mode?.includes("EDIT") ? [
136
- {
137
- name: "edit_steps", type: types_js_1.OptionType.numeric, default: 75,
138
- integer: true, description: "The number of steps for the base image generation, more steps means more time and better quality"
139
- },
140
- ] : [];
141
- const customizationOptions = option?.edit_mode === ImagenTaskType.CUSTOMIZATION_CONTROLLED
142
- || option?.edit_mode === ImagenTaskType.CUSTOMIZATION_SUBJECT ? [
143
- {
144
- name: "controlType", type: types_js_1.OptionType.enum, enum: { "Face Mesh": "CONTROL_TYPE_FACE_MESH", "Canny": "CONTROL_TYPE_CANNY", "Scribble": "CONTROL_TYPE_SCRIBBLE" },
145
- default: "CONTROL_TYPE_CANNY", description: "Method used to generate the control image"
146
- },
147
- {
148
- name: "controlImageComputation", type: types_js_1.OptionType.boolean, default: true, description: "Should the control image be computed from the input image, or is it provided"
149
- }
150
- ] : [];
151
- return {
152
- _option_id: "vertexai-imagen",
153
- options: [
154
- ...modeOptions,
155
- ...commonOptions,
156
- ...maskOptions,
157
- ...maskClassOptions,
158
- ...editOptions,
159
- ...customizationOptions,
160
- ...outputOptions,
161
- ]
162
- };
163
- }
164
- }
165
- else if (model.includes("gemini")) {
166
- const max_tokens_limit = getGeminiMaxTokensLimit(model);
167
- const excludeOptions = ["max_tokens", "presence_penalty"];
168
- let commonOptions = options_js_1.textOptionsFallback.options.filter((option) => !excludeOptions.includes(option.name));
169
- if (model.includes("1.5")) {
170
- commonOptions = commonOptions.filter((option) => option.name !== "frequency_penalty");
171
- }
172
- const max_tokens = [{
173
- name: types_js_1.SharedOptions.max_tokens, type: types_js_1.OptionType.numeric, min: 1, max: max_tokens_limit,
174
- integer: true, step: 200, description: "The maximum number of tokens to generate"
175
- }];
176
- return {
177
- _option_id: "vertexai-gemini",
178
- options: [
179
- ...max_tokens,
180
- ...commonOptions,
181
- ]
182
- };
183
- }
184
- else if (model.includes("claude")) {
185
- const max_tokens_limit = getClaudeMaxTokensLimit(model, option);
186
- const excludeOptions = ["max_tokens", "presence_penalty", "frequency_penalty"];
187
- let commonOptions = options_js_1.textOptionsFallback.options.filter((option) => !excludeOptions.includes(option.name));
188
- const max_tokens = [{
189
- name: types_js_1.SharedOptions.max_tokens, type: types_js_1.OptionType.numeric, min: 1, max: max_tokens_limit,
190
- integer: true, step: 200, description: "The maximum number of tokens to generate"
191
- }];
192
- if (model.includes("3-7")) {
193
- const claudeModeOptions = [
194
- {
195
- name: "thinking_mode",
196
- type: types_js_1.OptionType.boolean,
197
- default: false,
198
- description: "If true, use the extended reasoning mode"
199
- },
200
- ];
201
- const claudeThinkingOptions = option?.thinking_mode ? [
202
- {
203
- name: "thinking_budget_tokens",
204
- type: types_js_1.OptionType.numeric,
205
- min: 1024,
206
- default: 4000,
207
- integer: true,
208
- step: 100,
209
- description: "The target number of tokens to use for reasoning, not a hard limit."
210
- },
211
- ] : [];
212
- return {
213
- _option_id: "vertexai-claude",
214
- options: [
215
- ...max_tokens,
216
- ...commonOptions,
217
- ...claudeModeOptions,
218
- ...claudeThinkingOptions,
219
- ]
220
- };
221
- }
222
- return {
223
- _option_id: "vertexai-claude",
224
- options: [
225
- ...max_tokens,
226
- ...commonOptions,
227
- ]
228
- };
229
- }
230
- return options_js_1.textOptionsFallback;
231
- }
232
- function getGeminiMaxTokensLimit(model) {
233
- if (model.includes("thinking")) {
234
- return 65536;
235
- }
236
- if (model.includes("ultra") || model.includes("vision")) {
237
- return 2048;
238
- }
239
- return 8192;
240
- }
241
- function getClaudeMaxTokensLimit(model, option) {
242
- if (model.includes("3-7")) {
243
- if (option && option?.thinking_mode) {
244
- return 128000;
245
- }
246
- else {
247
- return 8192;
248
- }
249
- }
250
- else if (model.includes("3-5")) {
251
- return 8192;
252
- }
253
- else {
254
- return 4096;
255
- }
256
- }
257
- //# sourceMappingURL=vertexai.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"vertexai.js","sourceRoot":"","sources":["../../../src/options/vertexai.ts"],"names":[],"mappings":";;;AAgEA,gDA8NC;AA9RD,0CAA6G;AAC7G,8CAAoD;AAKpD,IAAY,cAUX;AAVD,WAAY,cAAc;IACtB,2CAAyB,CAAA;IACzB,yEAAuD,CAAA;IACvD,6EAA2D,CAAA;IAC3D,uDAAqC,CAAA;IACrC,2DAAyC,CAAA;IACzC,iEAA+C,CAAA;IAC/C,6DAA2C,CAAA;IAC3C,uEAAqD,CAAA;IACrD,mEAAiD,CAAA;AACrD,CAAC,EAVW,cAAc,8BAAd,cAAc,QAUzB;AAED,IAAY,cAKX;AALD,WAAY,cAAc;IACtB,qEAAmD,CAAA;IACnD,+DAA6C,CAAA;IAC7C,+DAA6C,CAAA;IAC7C,2DAAyC,CAAA;AAC7C,CAAC,EALW,cAAc,8BAAd,cAAc,QAKzB;AAyCD,SAAgB,kBAAkB,CAAC,KAAa,EAAE,MAAqB;IACnE,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QAC/B,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,wBAAa,CAAC,gBAAgB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC;gBAC1F,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,8BAA8B;aAC7D;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,IAAI,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE;gBACxF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,iCAAiC;aAChE;YACD;gBACI,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,qDAAqD,EAAE,YAAY,EAAE,iCAAiC,EAAE,aAAa,EAAE,wCAAwC,EAAE,WAAW,EAAE;gBACxO,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,uEAAuE;aAC/G;YACD;gBACI,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,kDAAkD,EAAE,YAAY,EAAE,kDAAkD,EAAE,iBAAiB,EAAE,8CAA8C,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,qBAAqB,EAAE;gBACxT,OAAO,EAAE,wBAAwB,EAAE,WAAW,EAAE,4BAA4B;aAC/E;SACJ,CAAC;QAGF,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE;gBAClG,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,sCAAsC;gBACzE,OAAO,EAAE,IAAI;aAChB;SACJ,CAAA;QAED,MAAM,WAAW,GAAwB;YACrC,IAAI,EAAE,0BAA0B,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;YACzF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,2CAA2C;SAC1E,CAAA;QAED,IAAK,MAAwB,EAAE,eAAe,KAAK,YAAY,EAAE,CAAC;YAC9D,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7B,iBAAiB;YACjB,MAAM,WAAW,GACX;gBACE;oBACI,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;oBAC/H,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,yCAAyC;iBACzE;gBACD;oBACI,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,sFAAsF;iBACvK;aAEJ,CAAC;YAEN,MAAM,cAAc,GAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;gBAC5E;oBACI,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,mFAAmF;iBACpK;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,OAAO;gBACH,UAAU,EAAE,iBAAiB;gBAC7B,OAAO,EAAE;oBACL,GAAG,aAAa;oBAChB,GAAG,WAAW;oBACd,GAAG,aAAa;oBAChB,GAAG,cAAc;iBACpB;aACJ,CAAC;QACN,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAC/B,aAAa;YACb,IAAI,oBAAoB,GAAG,EAAE,CAAC;YAC9B,IAAK,MAAwB,EAAE,SAAS,KAAK,cAAc,CAAC,2BAA2B,EAAE,CAAC;gBACtF,oBAAoB,GAAG,EAAE,CAAC;YAC9B,CAAC;YAED,MAAM,WAAW,GAA0B;gBACvC;oBACI,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI;oBACxC,IAAI,EAAE;wBACF,2BAA2B,EAAE,2BAA2B;wBACxD,6BAA6B,EAAE,6BAA6B;wBAC5D,kBAAkB,EAAE,kBAAkB;wBACtC,oBAAoB,EAAE,oBAAoB;wBAC1C,uBAAuB,EAAE,uBAAuB;wBAChD,qBAAqB,EAAE,qBAAqB;wBAC5C,0BAA0B,EAAE,0BAA0B;wBACtD,wBAAwB,EAAE,wBAAwB;qBACrD;oBACD,WAAW,EAAE,2GAA2G;iBAC3H;gBAED;oBACI,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,oBAAoB;oBACjG,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,+CAA+C;iBAC9E;aACJ,CAAC;YAEF,MAAM,WAAW,GAA0B,CAAE,MAAwB,EAAE,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClG;oBACI,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI;oBACxC,IAAI,EAAE;wBACF,yBAAyB,EAAE,yBAAyB;wBACpD,sBAAsB,EAAE,sBAAsB;wBAC9C,sBAAsB,EAAE,sBAAsB;wBAC9C,oBAAoB,EAAE,oBAAoB;qBAC7C;oBACD,OAAO,EAAE,yBAAyB;oBAClC,WAAW,EAAE,oDAAoD;iBACpE;gBACD;oBACI,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC;oBAC/D,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,oGAAoG;iBACnI;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,MAAM,gBAAgB,GAA0B,CAAE,MAAwB,EAAE,SAAS,KAAK,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;gBAC3H;oBACI,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,qBAAU,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE;oBAC7D,WAAW,EAAE,0KAA0K;iBAC1L;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,MAAM,WAAW,GAA2B,MAAwB,EAAE,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChG;oBACI,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;oBACzD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kGAAkG;iBACjI;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,MAAM,oBAAoB,GAA2B,MAAwB,EAAE,SAAS,KAAK,cAAc,CAAC,wBAAwB;mBAChI,MAAwB,EAAE,SAAS,KAAK,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBAC/E;oBACI,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,WAAW,EAAE,wBAAwB,EAAE,OAAO,EAAE,oBAAoB,EAAE,UAAU,EAAE,uBAAuB,EAAE;oBAC/J,OAAO,EAAE,oBAAoB,EAAE,WAAW,EAAE,2CAA2C;iBAC1F;gBACD;oBACI,IAAI,EAAE,yBAAyB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,8EAA8E;iBACxK;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,OAAO;gBACH,UAAU,EAAE,iBAAiB;gBAC7B,OAAO,EAAE;oBACL,GAAG,WAAW;oBACd,GAAG,aAAa;oBAChB,GAAG,WAAW;oBACd,GAAG,gBAAgB;oBACnB,GAAG,WAAW;oBACd,GAAG,oBAAoB;oBACvB,GAAG,aAAa;iBACnB;aACJ,CAAC;QACN,CAAC;IACL,CAAC;SACI,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;QACxD,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;QAC1D,IAAI,aAAa,GAAG,gCAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1G,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC;QAC1F,CAAC;QACD,MAAM,UAAU,GAA0B,CAAC;gBACvC,IAAI,EAAE,wBAAa,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,gBAAgB;gBACvF,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,0CAA0C;aACpF,CAAC,CAAC;QACH,OAAO;YACH,UAAU,EAAE,iBAAiB;YAC7B,OAAO,EAAE;gBACL,GAAG,UAAU;gBACb,GAAG,aAAa;aACnB;SACJ,CAAC;IACN,CAAC;SACI,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,KAAK,EAAE,MAA+B,CAAC,CAAC;QACzF,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,kBAAkB,EAAE,mBAAmB,CAAC,CAAC;QAC/E,IAAI,aAAa,GAAG,gCAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1G,MAAM,UAAU,GAA0B,CAAC;gBACvC,IAAI,EAAE,wBAAa,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,gBAAgB;gBACvF,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,0CAA0C;aACpF,CAAC,CAAC;QAEH,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,iBAAiB,GAA0B;gBAC7C;oBACI,IAAI,EAAE,eAAe;oBACrB,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,OAAO,EAAE,KAAK;oBACd,WAAW,EAAE,0CAA0C;iBAC1D;aACJ,CAAC;YACF,MAAM,qBAAqB,GAA2B,MAAgC,EAAE,aAAa,CAAC,CAAC,CAAC;gBACpG;oBACI,IAAI,EAAE,wBAAwB;oBAC9B,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,GAAG,EAAE,IAAI;oBACT,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,IAAI;oBACb,IAAI,EAAE,GAAG;oBACT,WAAW,EAAE,qEAAqE;iBACrF;aACJ,CAAC,CAAC,CAAC,EAAE,CAAC;YAEP,OAAO;gBACH,UAAU,EAAE,iBAAiB;gBAC7B,OAAO,EAAE;oBACL,GAAG,UAAU;oBACb,GAAG,aAAa;oBAChB,GAAG,iBAAiB;oBACpB,GAAG,qBAAqB;iBAC3B;aACJ,CAAC;QACN,CAAC;QACD,OAAO;YACH,UAAU,EAAE,iBAAiB;YAC7B,OAAO,EAAE;gBACL,GAAG,UAAU;gBACb,GAAG,aAAa;aACnB;SACJ,CAAC;IACN,CAAC;IACD,OAAO,gCAAmB,CAAC;AAC/B,CAAC;AACD,SAAS,uBAAuB,CAAC,KAAa;IAC1C,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtD,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AACD,SAAS,uBAAuB,CAAC,KAAa,EAAE,MAA8B;IAC1E,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,IAAI,MAAM,IAAI,MAAM,EAAE,aAAa,EAAE,CAAC;YAClC,OAAO,MAAM,CAAC;QAClB,CAAC;aAAM,CAAC;YACJ,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;SACI,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;SACI,CAAC;QACF,OAAO,IAAI,CAAC;IAChB,CAAC;AACL,CAAC"}
@@ -1,19 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.getOptions = getOptions;
4
- const common_1 = require("@llumiverse/common");
5
- function getOptions(model, provider, options) {
6
- switch (provider?.toLowerCase()) {
7
- case "bedrock":
8
- return (0, common_1.getBedrockOptions)(model ?? "", options);
9
- case "vertexai":
10
- return (0, common_1.getVertexAiOptions)(model ?? "", options);
11
- case "openai":
12
- return (0, common_1.getOpenAiOptions)(model ?? "", options);
13
- case "groq":
14
- return (0, common_1.getGroqOptions)(model ?? "", options);
15
- default:
16
- return common_1.textOptionsFallback;
17
- }
18
- }
19
- //# sourceMappingURL=options.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"options.js","sourceRoot":"","sources":["../../src/options.ts"],"names":[],"mappings":";;AAUA,gCAaC;AAvBD,+CAQ4B;AAE5B,SAAgB,UAAU,CAAC,KAAa,EAAE,QAAiB,EAAE,OAAsB;IAC/E,QAAQ,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC;QAC9B,KAAK,SAAS;YACV,OAAO,IAAA,0BAAiB,EAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QACnD,KAAK,UAAU;YACX,OAAO,IAAA,2BAAkB,EAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QACpD,KAAK,QAAQ;YACT,OAAO,IAAA,yBAAgB,EAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAClD,KAAK,MAAM;YACP,OAAO,IAAA,uBAAc,EAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAChD;YACI,OAAO,4BAAmB,CAAC;IACnC,CAAC;AACL,CAAC"}
package/lib/cjs/types.js DELETED
@@ -1,80 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.TrainingJobStatus = exports.ModelType = exports.AIModelStatus = exports.Modalities = exports.PromptRole = exports.OptionType = exports.SharedOptions = void 0;
4
- //Common names to share between different models
5
- var SharedOptions;
6
- (function (SharedOptions) {
7
- //Text
8
- SharedOptions["max_tokens"] = "max_tokens";
9
- SharedOptions["temperature"] = "temperature";
10
- SharedOptions["top_p"] = "top_p";
11
- SharedOptions["top_k"] = "top_k";
12
- SharedOptions["presence_penalty"] = "presence_penalty";
13
- SharedOptions["frequency_penalty"] = "frequency_penalty";
14
- SharedOptions["stop_sequence"] = "stop_sequence";
15
- //Image
16
- SharedOptions["seed"] = "seed";
17
- SharedOptions["number_of_images"] = "number_of_images";
18
- })(SharedOptions || (exports.SharedOptions = SharedOptions = {}));
19
- var OptionType;
20
- (function (OptionType) {
21
- OptionType["numeric"] = "numeric";
22
- OptionType["enum"] = "enum";
23
- OptionType["boolean"] = "boolean";
24
- OptionType["string_list"] = "string_list";
25
- })(OptionType || (exports.OptionType = OptionType = {}));
26
- // ============== Prompts ===============
27
- var PromptRole;
28
- (function (PromptRole) {
29
- PromptRole["safety"] = "safety";
30
- PromptRole["system"] = "system";
31
- PromptRole["user"] = "user";
32
- PromptRole["assistant"] = "assistant";
33
- PromptRole["negative"] = "negative";
34
- PromptRole["mask"] = "mask";
35
- /**
36
- * Used to send the response of a tool
37
- */
38
- PromptRole["tool"] = "tool";
39
- })(PromptRole || (exports.PromptRole = PromptRole = {}));
40
- var Modalities;
41
- (function (Modalities) {
42
- Modalities["text"] = "text";
43
- Modalities["image"] = "image";
44
- })(Modalities || (exports.Modalities = Modalities = {}));
45
- var AIModelStatus;
46
- (function (AIModelStatus) {
47
- AIModelStatus["Available"] = "available";
48
- AIModelStatus["Pending"] = "pending";
49
- AIModelStatus["Stopped"] = "stopped";
50
- AIModelStatus["Unavailable"] = "unavailable";
51
- AIModelStatus["Unknown"] = "unknown";
52
- })(AIModelStatus || (exports.AIModelStatus = AIModelStatus = {}));
53
- var ModelType;
54
- (function (ModelType) {
55
- ModelType["Classifier"] = "classifier";
56
- ModelType["Regressor"] = "regressor";
57
- ModelType["Clustering"] = "clustering";
58
- ModelType["AnomalyDetection"] = "anomaly-detection";
59
- ModelType["TimeSeries"] = "time-series";
60
- ModelType["Text"] = "text";
61
- ModelType["Image"] = "image";
62
- ModelType["Audio"] = "audio";
63
- ModelType["Video"] = "video";
64
- ModelType["Embedding"] = "embedding";
65
- ModelType["Chat"] = "chat";
66
- ModelType["Code"] = "code";
67
- ModelType["NLP"] = "nlp";
68
- ModelType["MultiModal"] = "multi-modal";
69
- ModelType["Test"] = "test";
70
- ModelType["Other"] = "other";
71
- ModelType["Unknown"] = "unknown";
72
- })(ModelType || (exports.ModelType = ModelType = {}));
73
- var TrainingJobStatus;
74
- (function (TrainingJobStatus) {
75
- TrainingJobStatus["running"] = "running";
76
- TrainingJobStatus["succeeded"] = "succeeded";
77
- TrainingJobStatus["failed"] = "failed";
78
- TrainingJobStatus["cancelled"] = "cancelled";
79
- })(TrainingJobStatus || (exports.TrainingJobStatus = TrainingJobStatus = {}));
80
- //# sourceMappingURL=types.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":";;;AAmMA,gDAAgD;AAChD,IAAY,aAaX;AAbD,WAAY,aAAa;IACrB,MAAM;IACN,0CAAyB,CAAA;IACzB,4CAA2B,CAAA;IAC3B,gCAAe,CAAA;IACf,gCAAe,CAAA;IACf,sDAAqC,CAAA;IACrC,wDAAuC,CAAA;IACvC,gDAA+B,CAAA;IAE/B,OAAO;IACP,8BAAa,CAAA;IACb,sDAAqC,CAAA;AACzC,CAAC,EAbW,aAAa,6BAAb,aAAa,QAaxB;AAED,IAAY,UAKX;AALD,WAAY,UAAU;IAClB,iCAAmB,CAAA;IACnB,2BAAa,CAAA;IACb,iCAAmB,CAAA;IACnB,yCAA2B,CAAA;AAC/B,CAAC,EALW,UAAU,0BAAV,UAAU,QAKrB;AAqDD,yCAAyC;AACzC,IAAY,UAWX;AAXD,WAAY,UAAU;IAClB,+BAAiB,CAAA;IACjB,+BAAiB,CAAA;IACjB,2BAAa,CAAA;IACb,qCAAuB,CAAA;IACvB,mCAAqB,CAAA;IACrB,2BAAa,CAAA;IACb;;OAEG;IACH,2BAAa,CAAA;AACjB,CAAC,EAXW,UAAU,0BAAV,UAAU,QAWrB;AAkBD,IAAY,UAGX;AAHD,WAAY,UAAU;IAClB,2BAAa,CAAA;IACb,6BAAe,CAAA;AACnB,CAAC,EAHW,UAAU,0BAAV,UAAU,QAGrB;AAsBD,IAAY,aAMX;AAND,WAAY,aAAa;IACrB,wCAAuB,CAAA;IACvB,oCAAmB,CAAA;IACnB,oCAAmB,CAAA;IACnB,4CAA2B,CAAA;IAC3B,oCAAmB,CAAA;AACvB,CAAC,EANW,aAAa,6BAAb,aAAa,QAMxB;AAiBD,IAAY,SAkBX;AAlBD,WAAY,SAAS;IACjB,sCAAyB,CAAA;IACzB,oCAAuB,CAAA;IACvB,sCAAyB,CAAA;IACzB,mDAAsC,CAAA;IACtC,uCAA0B,CAAA;IAC1B,0BAAa,CAAA;IACb,4BAAe,CAAA;IACf,4BAAe,CAAA;IACf,4BAAe,CAAA;IACf,oCAAuB,CAAA;IACvB,0BAAa,CAAA;IACb,0BAAa,CAAA;IACb,wBAAW,CAAA;IACX,uCAA0B,CAAA;IAC1B,0BAAa,CAAA;IACb,4BAAe,CAAA;IACf,gCAAmB,CAAA;AACvB,CAAC,EAlBW,SAAS,yBAAT,SAAS,QAkBpB;AA2BD,IAAY,iBAKX;AALD,WAAY,iBAAiB;IACzB,wCAAmB,CAAA;IACnB,4CAAuB,CAAA;IACvB,sCAAiB,CAAA;IACjB,4CAAuB,CAAA;AAC3B,CAAC,EALW,iBAAiB,iCAAjB,iBAAiB,QAK5B"}