@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,120 +0,0 @@
1
- import { OptionType, SharedOptions } from "../types.js";
2
- import { textOptionsFallback } from "../options.js";
3
- export function getOpenAiOptions(model, _option) {
4
- const visionOptions = isVisionModel(model) ? [
5
- {
6
- name: "image_detail", type: OptionType.enum, enum: { "Low": "low", "High": "high", "Auto": "auto" },
7
- default: "auto", description: "Controls how the model processes an input image."
8
- },
9
- ] : [];
10
- if (model.includes("o1") || model.includes("o3")) {
11
- //Is thinking text model
12
- let max_tokens_limit = 4096;
13
- if (model.includes("o1")) {
14
- if (model.includes("preview")) {
15
- max_tokens_limit = 32768;
16
- }
17
- else if (model.includes("mini")) {
18
- max_tokens_limit = 65536;
19
- }
20
- else {
21
- max_tokens_limit = 100000;
22
- }
23
- }
24
- else if (model.includes("o3")) {
25
- max_tokens_limit = 100000;
26
- }
27
- const commonOptions = [
28
- {
29
- name: SharedOptions.max_tokens, type: OptionType.numeric, min: 1, max: max_tokens_limit,
30
- integer: true, description: "The maximum number of tokens to generate",
31
- },
32
- {
33
- name: SharedOptions.stop_sequence, type: OptionType.string_list, value: [],
34
- description: "The stop sequence of the generated image",
35
- },
36
- ];
37
- const reasoningOptions = model.includes("o3") || isO1Full(model) ? [
38
- {
39
- name: "reasoning_effort", type: OptionType.enum, enum: { "Low": "low", "Medium": "medium", "High": "high" },
40
- default: "medium", description: "How much effort the model should put into reasoning, lower values result in faster responses and less tokens used."
41
- },
42
- ] : [];
43
- return {
44
- _option_id: "openai-thinking",
45
- options: [
46
- ...commonOptions,
47
- ...reasoningOptions,
48
- ...visionOptions,
49
- ],
50
- };
51
- }
52
- else {
53
- let max_tokens_limit = 4096;
54
- if (model.includes("gpt-4o")) {
55
- max_tokens_limit = 16384;
56
- if (model.includes("gpt-4o-2024-05-13") || model.includes("realtime")) {
57
- max_tokens_limit = 4096;
58
- }
59
- }
60
- else if (model.includes("gpt-4")) {
61
- if (model.includes("turbo")) {
62
- max_tokens_limit = 4096;
63
- }
64
- else {
65
- max_tokens_limit = 8192;
66
- }
67
- }
68
- else if (model.includes("gpt-3-5")) {
69
- max_tokens_limit = 4096;
70
- }
71
- //Is non-thinking text model
72
- const commonOptions = [
73
- {
74
- name: SharedOptions.max_tokens, type: OptionType.numeric, min: 1, max: max_tokens_limit,
75
- integer: true, step: 200, description: "The maximum number of tokens to generate",
76
- },
77
- {
78
- name: "temperature", type: OptionType.numeric, min: 0.0, max: 2.0, default: 0.7,
79
- integer: false, step: 0.1, description: "A higher temperature biases toward less likely tokens, making the model more creative"
80
- },
81
- {
82
- name: "top_p", type: OptionType.numeric, min: 0, max: 1,
83
- integer: false, step: 0.1, description: "Limits token sampling to the cumulative probability of the top p tokens"
84
- },
85
- {
86
- name: "presence_penalty", type: OptionType.numeric, min: -2.0, max: 2.0,
87
- integer: false, step: 0.1, description: "Penalise tokens if they appear at least once in the text"
88
- },
89
- {
90
- name: "frequency_penalty", type: OptionType.numeric, min: -2.0, max: 2.0,
91
- integer: false, step: 0.1, description: "Penalise tokens based on their frequency in the text"
92
- },
93
- {
94
- name: SharedOptions.stop_sequence, type: OptionType.string_list, value: [],
95
- description: "The generation will halt if one of the stop sequences is output",
96
- }
97
- ];
98
- return {
99
- _option_id: "openai-text",
100
- options: [
101
- ...commonOptions,
102
- ...visionOptions,
103
- ],
104
- };
105
- }
106
- return textOptionsFallback;
107
- }
108
- function isO1Full(model) {
109
- if (model.includes("o1")) {
110
- if (model.includes("mini") || model.includes("preview")) {
111
- return false;
112
- }
113
- return true;
114
- }
115
- return false;
116
- }
117
- function isVisionModel(model) {
118
- return model.includes("gpt-4o") || isO1Full(model) || model.includes("gpt-4-turbo");
119
- }
120
- //# sourceMappingURL=openai.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/options/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuD,UAAU,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC7G,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAwBpD,MAAM,UAAU,gBAAgB,CAAC,KAAa,EAAE,OAAsB;IAClE,MAAM,aAAa,GAA0B,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5D;YACI,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,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,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,UAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,gBAAgB;gBACvF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,0CAA0C;aACzE;YACD;gBACI,IAAI,EAAE,aAAa,CAAC,aAAa,EAAE,IAAI,EAAE,UAAU,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,UAAU,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,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,aAAa,CAAC,aAAa,EAAE,IAAI,EAAE,UAAU,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,mBAAmB,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,253 +0,0 @@
1
- import { OptionType, SharedOptions } from "../types.js";
2
- import { textOptionsFallback } from "../options.js";
3
- export var ImagenTaskType;
4
- (function (ImagenTaskType) {
5
- ImagenTaskType["TEXT_IMAGE"] = "TEXT_IMAGE";
6
- ImagenTaskType["EDIT_MODE_INPAINT_REMOVAL"] = "EDIT_MODE_INPAINT_REMOVAL";
7
- ImagenTaskType["EDIT_MODE_INPAINT_INSERTION"] = "EDIT_MODE_INPAINT_INSERTION";
8
- ImagenTaskType["EDIT_MODE_BGSWAP"] = "EDIT_MODE_BGSWAP";
9
- ImagenTaskType["EDIT_MODE_OUTPAINT"] = "EDIT_MODE_OUTPAINT";
10
- ImagenTaskType["CUSTOMIZATION_SUBJECT"] = "CUSTOMIZATION_SUBJECT";
11
- ImagenTaskType["CUSTOMIZATION_STYLE"] = "CUSTOMIZATION_STYLE";
12
- ImagenTaskType["CUSTOMIZATION_CONTROLLED"] = "CUSTOMIZATION_CONTROLLED";
13
- ImagenTaskType["CUSTOMIZATION_INSTRUCT"] = "CUSTOMIZATION_INSTRUCT";
14
- })(ImagenTaskType || (ImagenTaskType = {}));
15
- export var ImagenMaskMode;
16
- (function (ImagenMaskMode) {
17
- ImagenMaskMode["MASK_MODE_USER_PROVIDED"] = "MASK_MODE_USER_PROVIDED";
18
- ImagenMaskMode["MASK_MODE_BACKGROUND"] = "MASK_MODE_BACKGROUND";
19
- ImagenMaskMode["MASK_MODE_FOREGROUND"] = "MASK_MODE_FOREGROUND";
20
- ImagenMaskMode["MASK_MODE_SEMANTIC"] = "MASK_MODE_SEMANTIC";
21
- })(ImagenMaskMode || (ImagenMaskMode = {}));
22
- export function getVertexAiOptions(model, option) {
23
- if (model.includes("imagen-3.0")) {
24
- const commonOptions = [
25
- {
26
- name: SharedOptions.number_of_images, type: OptionType.numeric, min: 1, max: 4, default: 1,
27
- integer: true, description: "Number of Images to generate",
28
- },
29
- {
30
- name: SharedOptions.seed, type: OptionType.numeric, min: 0, max: 4294967295, default: 12,
31
- integer: true, description: "The seed of the generated image"
32
- },
33
- {
34
- name: "person_generation", type: 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" },
35
- default: "allow_adult", description: "The safety setting for allowing the generation of people in the image"
36
- },
37
- {
38
- name: "safety_setting", type: 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" },
39
- default: "block_medium_and_above", description: "The overall safety setting"
40
- },
41
- ];
42
- const outputOptions = [
43
- {
44
- name: "image_file_type", type: OptionType.enum, enum: { "JPEG": "image/jpeg", "PNG": "image/png" },
45
- default: "image/png", description: "The file type of the generated image",
46
- refresh: true,
47
- },
48
- ];
49
- const jpegQuality = {
50
- name: "jpeg_compression_quality", type: OptionType.numeric, min: 0, max: 100, default: 75,
51
- integer: true, description: "The compression quality of the JPEG image",
52
- };
53
- if (option?.image_file_type === "image/jpeg") {
54
- outputOptions.push(jpegQuality);
55
- }
56
- if (model.includes("generate")) {
57
- //Generate models
58
- const modeOptions = [
59
- {
60
- name: "aspect_ratio", type: OptionType.enum, enum: { "1:1": "1:1", "4:3": "4:3", "3:4": "3:4", "16:9": "16:9", "9:16": "9:16" },
61
- default: "1:1", description: "The aspect ratio of the generated image"
62
- },
63
- {
64
- name: "add_watermark", type: OptionType.boolean, default: false, description: "Add an invisible watermark to the generated image, useful for detection of AI images"
65
- },
66
- ];
67
- const enhanceOptions = !model.includes("generate-001") ? [
68
- {
69
- name: "enhance_prompt", type: OptionType.boolean, default: true, description: "VertexAI automatically rewrites the prompt to better reflect the prompt's intent."
70
- },
71
- ] : [];
72
- return {
73
- _option_id: "vertexai-imagen",
74
- options: [
75
- ...commonOptions,
76
- ...modeOptions,
77
- ...outputOptions,
78
- ...enhanceOptions,
79
- ]
80
- };
81
- }
82
- if (model.includes("capability")) {
83
- //Edit models
84
- let guidanceScaleDefault = 75;
85
- if (option?.edit_mode === ImagenTaskType.EDIT_MODE_INPAINT_INSERTION) {
86
- guidanceScaleDefault = 60;
87
- }
88
- const modeOptions = [
89
- {
90
- name: "edit_mode", type: OptionType.enum,
91
- enum: {
92
- "EDIT_MODE_INPAINT_REMOVAL": "EDIT_MODE_INPAINT_REMOVAL",
93
- "EDIT_MODE_INPAINT_INSERTION": "EDIT_MODE_INPAINT_INSERTION",
94
- "EDIT_MODE_BGSWAP": "EDIT_MODE_BGSWAP",
95
- "EDIT_MODE_OUTPAINT": "EDIT_MODE_OUTPAINT",
96
- "CUSTOMIZATION_SUBJECT": "CUSTOMIZATION_SUBJECT",
97
- "CUSTOMIZATION_STYLE": "CUSTOMIZATION_STYLE",
98
- "CUSTOMIZATION_CONTROLLED": "CUSTOMIZATION_CONTROLLED",
99
- "CUSTOMIZATION_INSTRUCT": "CUSTOMIZATION_INSTRUCT",
100
- },
101
- description: "The editing mode. CUSTOMIZATION options use few-shot learning to generate images based on a few examples."
102
- },
103
- {
104
- name: "guidance_scale", type: OptionType.numeric, min: 0, max: 500, default: guidanceScaleDefault,
105
- integer: true, description: "How closely the generation follows the prompt"
106
- }
107
- ];
108
- const maskOptions = (option?.edit_mode?.includes("EDIT")) ? [
109
- {
110
- name: "mask_mode", type: OptionType.enum,
111
- enum: {
112
- "MASK_MODE_USER_PROVIDED": "MASK_MODE_USER_PROVIDED",
113
- "MASK_MODE_BACKGROUND": "MASK_MODE_BACKGROUND",
114
- "MASK_MODE_FOREGROUND": "MASK_MODE_FOREGROUND",
115
- "MASK_MODE_SEMANTIC": "MASK_MODE_SEMANTIC",
116
- },
117
- default: "MASK_MODE_USER_PROVIDED",
118
- description: "How should the mask for the generation be provided"
119
- },
120
- {
121
- name: "mask_dilation", type: OptionType.numeric, min: 0, max: 1,
122
- integer: true, description: "The mask dilation, grows the mask by a percetage of image width to compensate for imprecise masks."
123
- },
124
- ] : [];
125
- const maskClassOptions = (option?.mask_mode === ImagenMaskMode.MASK_MODE_SEMANTIC) ? [
126
- {
127
- name: "mask_class", type: OptionType.string_list, default: [],
128
- 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"
129
- }
130
- ] : [];
131
- const editOptions = option?.edit_mode?.includes("EDIT") ? [
132
- {
133
- name: "edit_steps", type: OptionType.numeric, default: 75,
134
- integer: true, description: "The number of steps for the base image generation, more steps means more time and better quality"
135
- },
136
- ] : [];
137
- const customizationOptions = option?.edit_mode === ImagenTaskType.CUSTOMIZATION_CONTROLLED
138
- || option?.edit_mode === ImagenTaskType.CUSTOMIZATION_SUBJECT ? [
139
- {
140
- name: "controlType", type: OptionType.enum, enum: { "Face Mesh": "CONTROL_TYPE_FACE_MESH", "Canny": "CONTROL_TYPE_CANNY", "Scribble": "CONTROL_TYPE_SCRIBBLE" },
141
- default: "CONTROL_TYPE_CANNY", description: "Method used to generate the control image"
142
- },
143
- {
144
- name: "controlImageComputation", type: OptionType.boolean, default: true, description: "Should the control image be computed from the input image, or is it provided"
145
- }
146
- ] : [];
147
- return {
148
- _option_id: "vertexai-imagen",
149
- options: [
150
- ...modeOptions,
151
- ...commonOptions,
152
- ...maskOptions,
153
- ...maskClassOptions,
154
- ...editOptions,
155
- ...customizationOptions,
156
- ...outputOptions,
157
- ]
158
- };
159
- }
160
- }
161
- else if (model.includes("gemini")) {
162
- const max_tokens_limit = getGeminiMaxTokensLimit(model);
163
- const excludeOptions = ["max_tokens", "presence_penalty"];
164
- let commonOptions = textOptionsFallback.options.filter((option) => !excludeOptions.includes(option.name));
165
- if (model.includes("1.5")) {
166
- commonOptions = commonOptions.filter((option) => option.name !== "frequency_penalty");
167
- }
168
- const max_tokens = [{
169
- name: SharedOptions.max_tokens, type: OptionType.numeric, min: 1, max: max_tokens_limit,
170
- integer: true, step: 200, description: "The maximum number of tokens to generate"
171
- }];
172
- return {
173
- _option_id: "vertexai-gemini",
174
- options: [
175
- ...max_tokens,
176
- ...commonOptions,
177
- ]
178
- };
179
- }
180
- else if (model.includes("claude")) {
181
- const max_tokens_limit = getClaudeMaxTokensLimit(model, option);
182
- const excludeOptions = ["max_tokens", "presence_penalty", "frequency_penalty"];
183
- let commonOptions = textOptionsFallback.options.filter((option) => !excludeOptions.includes(option.name));
184
- const max_tokens = [{
185
- name: SharedOptions.max_tokens, type: OptionType.numeric, min: 1, max: max_tokens_limit,
186
- integer: true, step: 200, description: "The maximum number of tokens to generate"
187
- }];
188
- if (model.includes("3-7")) {
189
- const claudeModeOptions = [
190
- {
191
- name: "thinking_mode",
192
- type: OptionType.boolean,
193
- default: false,
194
- description: "If true, use the extended reasoning mode"
195
- },
196
- ];
197
- const claudeThinkingOptions = option?.thinking_mode ? [
198
- {
199
- name: "thinking_budget_tokens",
200
- type: OptionType.numeric,
201
- min: 1024,
202
- default: 4000,
203
- integer: true,
204
- step: 100,
205
- description: "The target number of tokens to use for reasoning, not a hard limit."
206
- },
207
- ] : [];
208
- return {
209
- _option_id: "vertexai-claude",
210
- options: [
211
- ...max_tokens,
212
- ...commonOptions,
213
- ...claudeModeOptions,
214
- ...claudeThinkingOptions,
215
- ]
216
- };
217
- }
218
- return {
219
- _option_id: "vertexai-claude",
220
- options: [
221
- ...max_tokens,
222
- ...commonOptions,
223
- ]
224
- };
225
- }
226
- return textOptionsFallback;
227
- }
228
- function getGeminiMaxTokensLimit(model) {
229
- if (model.includes("thinking")) {
230
- return 65536;
231
- }
232
- if (model.includes("ultra") || model.includes("vision")) {
233
- return 2048;
234
- }
235
- return 8192;
236
- }
237
- function getClaudeMaxTokensLimit(model, option) {
238
- if (model.includes("3-7")) {
239
- if (option && option?.thinking_mode) {
240
- return 128000;
241
- }
242
- else {
243
- return 8192;
244
- }
245
- }
246
- else if (model.includes("3-5")) {
247
- return 8192;
248
- }
249
- else {
250
- return 4096;
251
- }
252
- }
253
- //# sourceMappingURL=vertexai.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"vertexai.js","sourceRoot":"","sources":["../../../src/options/vertexai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyC,UAAU,EAAE,aAAa,EAAgB,MAAM,aAAa,CAAC;AAC7G,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAKpD,MAAM,CAAN,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,KAAd,cAAc,QAUzB;AAED,MAAM,CAAN,IAAY,cAKX;AALD,WAAY,cAAc;IACtB,qEAAmD,CAAA;IACnD,+DAA6C,CAAA;IAC7C,+DAA6C,CAAA;IAC7C,2DAAyC,CAAA;AAC7C,CAAC,EALW,cAAc,KAAd,cAAc,QAKzB;AAyCD,MAAM,UAAU,kBAAkB,CAAC,KAAa,EAAE,MAAqB;IACnE,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QAC/B,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,aAAa,CAAC,gBAAgB,EAAE,IAAI,EAAE,UAAU,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,aAAa,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,UAAU,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,mBAAmB,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,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,UAAU,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,mBAAmB,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,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,UAAU,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,UAAU,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,UAAU,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,mBAAmB,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,16 +0,0 @@
1
- import { getBedrockOptions, textOptionsFallback, getGroqOptions, getOpenAiOptions, getVertexAiOptions } from "@llumiverse/common";
2
- export function getOptions(model, provider, options) {
3
- switch (provider?.toLowerCase()) {
4
- case "bedrock":
5
- return getBedrockOptions(model ?? "", options);
6
- case "vertexai":
7
- return getVertexAiOptions(model ?? "", options);
8
- case "openai":
9
- return getOpenAiOptions(model ?? "", options);
10
- case "groq":
11
- return getGroqOptions(model ?? "", options);
12
- default:
13
- return textOptionsFallback;
14
- }
15
- }
16
- //# sourceMappingURL=options.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"options.js","sourceRoot":"","sources":["../../src/options.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,iBAAiB,EACjB,mBAAmB,EACnB,cAAc,EACd,gBAAgB,EAChB,kBAAkB,EAGrB,MAAM,oBAAoB,CAAC;AAE5B,MAAM,UAAU,UAAU,CAAC,KAAa,EAAE,QAAiB,EAAE,OAAsB;IAC/E,QAAQ,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC;QAC9B,KAAK,SAAS;YACV,OAAO,iBAAiB,CAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QACnD,KAAK,UAAU;YACX,OAAO,kBAAkB,CAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QACpD,KAAK,QAAQ;YACT,OAAO,gBAAgB,CAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAClD,KAAK,MAAM;YACP,OAAO,cAAc,CAAC,KAAK,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAChD;YACI,OAAO,mBAAmB,CAAC;IACnC,CAAC;AACL,CAAC"}
package/lib/esm/types.js DELETED
@@ -1,77 +0,0 @@
1
- //Common names to share between different models
2
- export var SharedOptions;
3
- (function (SharedOptions) {
4
- //Text
5
- SharedOptions["max_tokens"] = "max_tokens";
6
- SharedOptions["temperature"] = "temperature";
7
- SharedOptions["top_p"] = "top_p";
8
- SharedOptions["top_k"] = "top_k";
9
- SharedOptions["presence_penalty"] = "presence_penalty";
10
- SharedOptions["frequency_penalty"] = "frequency_penalty";
11
- SharedOptions["stop_sequence"] = "stop_sequence";
12
- //Image
13
- SharedOptions["seed"] = "seed";
14
- SharedOptions["number_of_images"] = "number_of_images";
15
- })(SharedOptions || (SharedOptions = {}));
16
- export var OptionType;
17
- (function (OptionType) {
18
- OptionType["numeric"] = "numeric";
19
- OptionType["enum"] = "enum";
20
- OptionType["boolean"] = "boolean";
21
- OptionType["string_list"] = "string_list";
22
- })(OptionType || (OptionType = {}));
23
- // ============== Prompts ===============
24
- export var PromptRole;
25
- (function (PromptRole) {
26
- PromptRole["safety"] = "safety";
27
- PromptRole["system"] = "system";
28
- PromptRole["user"] = "user";
29
- PromptRole["assistant"] = "assistant";
30
- PromptRole["negative"] = "negative";
31
- PromptRole["mask"] = "mask";
32
- /**
33
- * Used to send the response of a tool
34
- */
35
- PromptRole["tool"] = "tool";
36
- })(PromptRole || (PromptRole = {}));
37
- export var Modalities;
38
- (function (Modalities) {
39
- Modalities["text"] = "text";
40
- Modalities["image"] = "image";
41
- })(Modalities || (Modalities = {}));
42
- export var AIModelStatus;
43
- (function (AIModelStatus) {
44
- AIModelStatus["Available"] = "available";
45
- AIModelStatus["Pending"] = "pending";
46
- AIModelStatus["Stopped"] = "stopped";
47
- AIModelStatus["Unavailable"] = "unavailable";
48
- AIModelStatus["Unknown"] = "unknown";
49
- })(AIModelStatus || (AIModelStatus = {}));
50
- export var ModelType;
51
- (function (ModelType) {
52
- ModelType["Classifier"] = "classifier";
53
- ModelType["Regressor"] = "regressor";
54
- ModelType["Clustering"] = "clustering";
55
- ModelType["AnomalyDetection"] = "anomaly-detection";
56
- ModelType["TimeSeries"] = "time-series";
57
- ModelType["Text"] = "text";
58
- ModelType["Image"] = "image";
59
- ModelType["Audio"] = "audio";
60
- ModelType["Video"] = "video";
61
- ModelType["Embedding"] = "embedding";
62
- ModelType["Chat"] = "chat";
63
- ModelType["Code"] = "code";
64
- ModelType["NLP"] = "nlp";
65
- ModelType["MultiModal"] = "multi-modal";
66
- ModelType["Test"] = "test";
67
- ModelType["Other"] = "other";
68
- ModelType["Unknown"] = "unknown";
69
- })(ModelType || (ModelType = {}));
70
- export var TrainingJobStatus;
71
- (function (TrainingJobStatus) {
72
- TrainingJobStatus["running"] = "running";
73
- TrainingJobStatus["succeeded"] = "succeeded";
74
- TrainingJobStatus["failed"] = "failed";
75
- TrainingJobStatus["cancelled"] = "cancelled";
76
- })(TrainingJobStatus || (TrainingJobStatus = {}));
77
- //# sourceMappingURL=types.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"AAmMA,gDAAgD;AAChD,MAAM,CAAN,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,KAAb,aAAa,QAaxB;AAED,MAAM,CAAN,IAAY,UAKX;AALD,WAAY,UAAU;IAClB,iCAAmB,CAAA;IACnB,2BAAa,CAAA;IACb,iCAAmB,CAAA;IACnB,yCAA2B,CAAA;AAC/B,CAAC,EALW,UAAU,KAAV,UAAU,QAKrB;AAqDD,yCAAyC;AACzC,MAAM,CAAN,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,KAAV,UAAU,QAWrB;AAkBD,MAAM,CAAN,IAAY,UAGX;AAHD,WAAY,UAAU;IAClB,2BAAa,CAAA;IACb,6BAAe,CAAA;AACnB,CAAC,EAHW,UAAU,KAAV,UAAU,QAGrB;AAsBD,MAAM,CAAN,IAAY,aAMX;AAND,WAAY,aAAa;IACrB,wCAAuB,CAAA;IACvB,oCAAmB,CAAA;IACnB,oCAAmB,CAAA;IACnB,4CAA2B,CAAA;IAC3B,oCAAmB,CAAA;AACvB,CAAC,EANW,aAAa,KAAb,aAAa,QAMxB;AAiBD,MAAM,CAAN,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,KAAT,SAAS,QAkBpB;AA2BD,MAAM,CAAN,IAAY,iBAKX;AALD,WAAY,iBAAiB;IACzB,wCAAmB,CAAA;IACnB,4CAAuB,CAAA;IACvB,sCAAiB,CAAA;IACjB,4CAAuB,CAAA;AAC3B,CAAC,EALW,iBAAiB,KAAjB,iBAAiB,QAK5B"}