@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.
- package/lib/cjs/CompletionStream.js +44 -25
- package/lib/cjs/CompletionStream.js.map +1 -1
- package/lib/cjs/Driver.js +4 -1
- package/lib/cjs/Driver.js.map +1 -1
- package/lib/esm/CompletionStream.js +44 -25
- package/lib/esm/CompletionStream.js.map +1 -1
- package/lib/esm/Driver.js +4 -1
- package/lib/esm/Driver.js.map +1 -1
- package/lib/types/CompletionStream.d.ts.map +1 -1
- package/lib/types/Driver.d.ts.map +1 -1
- package/package.json +8 -8
- package/src/CompletionStream.ts +42 -27
- package/src/Driver.ts +4 -1
- package/lib/cjs/capability/bedrock.js +0 -186
- package/lib/cjs/capability/bedrock.js.map +0 -1
- package/lib/cjs/capability/openai.js +0 -122
- package/lib/cjs/capability/openai.js.map +0 -1
- package/lib/cjs/capability/vertexai.js +0 -90
- package/lib/cjs/capability/vertexai.js.map +0 -1
- package/lib/cjs/capability.js +0 -52
- package/lib/cjs/capability.js.map +0 -1
- package/lib/cjs/formatters/openai.js +0 -113
- package/lib/cjs/formatters/openai.js.map +0 -1
- package/lib/cjs/options/bedrock.js +0 -343
- package/lib/cjs/options/bedrock.js.map +0 -1
- package/lib/cjs/options/groq.js +0 -37
- package/lib/cjs/options/groq.js.map +0 -1
- package/lib/cjs/options/openai.js +0 -123
- package/lib/cjs/options/openai.js.map +0 -1
- package/lib/cjs/options/vertexai.js +0 -257
- package/lib/cjs/options/vertexai.js.map +0 -1
- package/lib/cjs/options.js +0 -19
- package/lib/cjs/options.js.map +0 -1
- package/lib/cjs/types.js +0 -80
- package/lib/cjs/types.js.map +0 -1
- package/lib/esm/capability/bedrock.js +0 -183
- package/lib/esm/capability/bedrock.js.map +0 -1
- package/lib/esm/capability/openai.js +0 -119
- package/lib/esm/capability/openai.js.map +0 -1
- package/lib/esm/capability/vertexai.js +0 -87
- package/lib/esm/capability/vertexai.js.map +0 -1
- package/lib/esm/capability.js +0 -47
- package/lib/esm/capability.js.map +0 -1
- package/lib/esm/formatters/openai.js +0 -109
- package/lib/esm/formatters/openai.js.map +0 -1
- package/lib/esm/options/bedrock.js +0 -340
- package/lib/esm/options/bedrock.js.map +0 -1
- package/lib/esm/options/groq.js +0 -34
- package/lib/esm/options/groq.js.map +0 -1
- package/lib/esm/options/openai.js +0 -120
- package/lib/esm/options/openai.js.map +0 -1
- package/lib/esm/options/vertexai.js +0 -253
- package/lib/esm/options/vertexai.js.map +0 -1
- package/lib/esm/options.js +0 -16
- package/lib/esm/options.js.map +0 -1
- package/lib/esm/types.js +0 -77
- package/lib/esm/types.js.map +0 -1
- package/lib/tsconfig.tsbuildinfo +0 -1
- package/lib/types/capability/bedrock.d.ts +0 -6
- package/lib/types/capability/bedrock.d.ts.map +0 -1
- package/lib/types/capability/openai.d.ts +0 -10
- package/lib/types/capability/openai.d.ts.map +0 -1
- package/lib/types/capability/vertexai.d.ts +0 -10
- package/lib/types/capability/vertexai.d.ts.map +0 -1
- package/lib/types/capability.d.ts +0 -4
- package/lib/types/capability.d.ts.map +0 -1
- package/lib/types/formatters/openai.d.ts +0 -41
- package/lib/types/formatters/openai.d.ts.map +0 -1
- package/lib/types/options/bedrock.d.ts +0 -31
- package/lib/types/options/bedrock.d.ts.map +0 -1
- package/lib/types/options/groq.d.ts +0 -11
- package/lib/types/options/groq.d.ts.map +0 -1
- package/lib/types/options/openai.d.ts +0 -20
- package/lib/types/options/openai.d.ts.map +0 -1
- package/lib/types/options/vertexai.d.ts +0 -51
- package/lib/types/options/vertexai.d.ts.map +0 -1
- package/lib/types/options.d.ts +0 -2
- package/lib/types/options.d.ts.map +0 -1
- package/lib/types/types.d.ts +0 -322
- package/lib/types/types.d.ts.map +0 -1
package/lib/cjs/capability.js
DELETED
|
@@ -1,52 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.getModelCapabilities = getModelCapabilities;
|
|
4
|
-
exports.supportsToolUse = supportsToolUse;
|
|
5
|
-
exports.modelModalitiesToArray = modelModalitiesToArray;
|
|
6
|
-
const bedrock_js_1 = require("./capability/bedrock.js");
|
|
7
|
-
const openai_js_1 = require("./capability/openai.js");
|
|
8
|
-
const vertexai_js_1 = require("./capability/vertexai.js");
|
|
9
|
-
function getModelCapabilities(model, provider) {
|
|
10
|
-
const capabilities = _getModelCapabilities(model, provider);
|
|
11
|
-
// Globally disable audio and video for all models, as we don't support them yet
|
|
12
|
-
// We also do not support tool use while streaming
|
|
13
|
-
// TODO: Remove this when we add support.
|
|
14
|
-
capabilities.input.audio = false;
|
|
15
|
-
capabilities.output.audio = false;
|
|
16
|
-
capabilities.output.video = false;
|
|
17
|
-
capabilities.tool_support_streaming = false;
|
|
18
|
-
return capabilities;
|
|
19
|
-
}
|
|
20
|
-
function _getModelCapabilities(model, provider) {
|
|
21
|
-
switch (provider?.toLowerCase()) {
|
|
22
|
-
case "vertexai":
|
|
23
|
-
return (0, vertexai_js_1.getModelCapabilitiesVertexAI)(model);
|
|
24
|
-
case "openai":
|
|
25
|
-
return (0, openai_js_1.getModelCapabilitiesOpenAI)(model);
|
|
26
|
-
case "bedrock":
|
|
27
|
-
return (0, bedrock_js_1.getModelCapabilitiesBedrock)(model);
|
|
28
|
-
default:
|
|
29
|
-
// Guess the provider based on the model name
|
|
30
|
-
if (model.startsWith("gpt")) {
|
|
31
|
-
return (0, openai_js_1.getModelCapabilitiesOpenAI)(model);
|
|
32
|
-
}
|
|
33
|
-
else if (model.startsWith("publishers/")) {
|
|
34
|
-
return (0, vertexai_js_1.getModelCapabilitiesVertexAI)(model);
|
|
35
|
-
}
|
|
36
|
-
else if (model.startsWith("arn:aws")) {
|
|
37
|
-
return (0, bedrock_js_1.getModelCapabilitiesBedrock)(model);
|
|
38
|
-
}
|
|
39
|
-
// Fallback to a generic model with no capabilities
|
|
40
|
-
return { input: {}, output: {} };
|
|
41
|
-
}
|
|
42
|
-
}
|
|
43
|
-
function supportsToolUse(model, provider, streaming = false) {
|
|
44
|
-
const capabilities = getModelCapabilities(model, provider);
|
|
45
|
-
return streaming ? !!capabilities.tool_support_streaming : !!capabilities.tool_support;
|
|
46
|
-
}
|
|
47
|
-
function modelModalitiesToArray(modalities) {
|
|
48
|
-
return Object.entries(modalities)
|
|
49
|
-
.filter(([_, isSupported]) => isSupported)
|
|
50
|
-
.map(([modality]) => modality);
|
|
51
|
-
}
|
|
52
|
-
//# sourceMappingURL=capability.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"capability.js","sourceRoot":"","sources":["../../src/capability.ts"],"names":[],"mappings":";;AAKA,oDAUC;AAwBD,0CAGC;AAED,wDAIC;AAhDD,wDAAsE;AACtE,sDAAoE;AACpE,0DAAwE;AAGxE,SAAgB,oBAAoB,CAAC,KAAa,EAAE,QAAiB;IACjE,MAAM,YAAY,GAAG,qBAAqB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC5D,gFAAgF;IAChF,kDAAkD;IAClD,yCAAyC;IACzC,YAAY,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC;IACjC,YAAY,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IAClC,YAAY,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IAClC,YAAY,CAAC,sBAAsB,GAAG,KAAK,CAAC;IAC5C,OAAO,YAAY,CAAC;AACxB,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAa,EAAE,QAAiB;IAC3D,QAAQ,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC;QAC9B,KAAK,UAAU;YACX,OAAO,IAAA,0CAA4B,EAAC,KAAK,CAAC,CAAC;QAC/C,KAAK,QAAQ;YACT,OAAO,IAAA,sCAA0B,EAAC,KAAK,CAAC,CAAC;QAC7C,KAAK,SAAS;YACV,OAAO,IAAA,wCAA2B,EAAC,KAAK,CAAC,CAAC;QAC9C;YACI,6CAA6C;YAC7C,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,IAAA,sCAA0B,EAAC,KAAK,CAAC,CAAC;YAC7C,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,IAAA,0CAA4B,EAAC,KAAK,CAAC,CAAC;YAC/C,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,OAAO,IAAA,wCAA2B,EAAC,KAAK,CAAC,CAAC;YAC9C,CAAC;YACD,mDAAmD;YACnD,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAA8B,CAAC;IACrE,CAAC;AACL,CAAC;AAED,SAAgB,eAAe,CAAC,KAAa,EAAE,QAAiB,EAAE,YAAqB,KAAK;IACxF,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC3D,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,YAAY,CAAC;AAC3F,CAAC;AAED,SAAgB,sBAAsB,CAAC,UAA2B;IAC9D,OAAO,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;SAC5B,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC;SACzC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC;AACvC,CAAC"}
|
|
@@ -1,113 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.formatOpenAILikeTextPrompt = formatOpenAILikeTextPrompt;
|
|
4
|
-
exports.formatOpenAILikeMultimodalPrompt = formatOpenAILikeMultimodalPrompt;
|
|
5
|
-
const common_1 = require("@llumiverse/common");
|
|
6
|
-
const stream_js_1 = require("../stream.js");
|
|
7
|
-
/**
|
|
8
|
-
* OpenAI text only prompts
|
|
9
|
-
* @param segments
|
|
10
|
-
* @returns
|
|
11
|
-
*/
|
|
12
|
-
function formatOpenAILikeTextPrompt(segments) {
|
|
13
|
-
const system = [];
|
|
14
|
-
const safety = [];
|
|
15
|
-
const user = [];
|
|
16
|
-
for (const msg of segments) {
|
|
17
|
-
if (msg.role === common_1.PromptRole.system) {
|
|
18
|
-
system.push({ content: msg.content, role: "system" });
|
|
19
|
-
}
|
|
20
|
-
else if (msg.role === common_1.PromptRole.safety) {
|
|
21
|
-
safety.push({ content: "IMPORTANT: " + msg.content, role: "system" });
|
|
22
|
-
}
|
|
23
|
-
else if (msg.role !== common_1.PromptRole.negative && msg.role !== common_1.PromptRole.mask && msg.role !== common_1.PromptRole.tool) {
|
|
24
|
-
user.push({
|
|
25
|
-
content: msg.content,
|
|
26
|
-
role: msg.role || 'user',
|
|
27
|
-
});
|
|
28
|
-
}
|
|
29
|
-
}
|
|
30
|
-
// put system messages first and safety last
|
|
31
|
-
return system.concat(user).concat(safety);
|
|
32
|
-
}
|
|
33
|
-
async function formatOpenAILikeMultimodalPrompt(segments, opts) {
|
|
34
|
-
const system = [];
|
|
35
|
-
const safety = [];
|
|
36
|
-
const others = [];
|
|
37
|
-
for (const msg of segments) {
|
|
38
|
-
const parts = [];
|
|
39
|
-
//generate the parts based on PromptSegment
|
|
40
|
-
if (msg.files) {
|
|
41
|
-
for (const file of msg.files) {
|
|
42
|
-
const stream = await file.getStream();
|
|
43
|
-
const data = await (0, stream_js_1.readStreamAsBase64)(stream);
|
|
44
|
-
parts.push({
|
|
45
|
-
type: "image_url",
|
|
46
|
-
image_url: {
|
|
47
|
-
url: `data:${file.mime_type || "image/jpeg"};base64,${data}`,
|
|
48
|
-
//detail: "auto" //This is modified just before execution to "low" | "high" | "auto"
|
|
49
|
-
},
|
|
50
|
-
});
|
|
51
|
-
}
|
|
52
|
-
}
|
|
53
|
-
if (msg.content) {
|
|
54
|
-
parts.push({
|
|
55
|
-
text: msg.content,
|
|
56
|
-
type: "text"
|
|
57
|
-
});
|
|
58
|
-
}
|
|
59
|
-
if (msg.role === common_1.PromptRole.system) {
|
|
60
|
-
system.push({
|
|
61
|
-
role: "system",
|
|
62
|
-
content: parts
|
|
63
|
-
});
|
|
64
|
-
if (opts.useToolForFormatting && opts.schema) {
|
|
65
|
-
system.forEach(s => {
|
|
66
|
-
s.content.forEach(c => {
|
|
67
|
-
if (c.type === "text")
|
|
68
|
-
c.text = "TOOL: " + c.text;
|
|
69
|
-
});
|
|
70
|
-
});
|
|
71
|
-
}
|
|
72
|
-
}
|
|
73
|
-
else if (msg.role === common_1.PromptRole.safety) {
|
|
74
|
-
const safetyMsg = {
|
|
75
|
-
role: "system",
|
|
76
|
-
content: parts
|
|
77
|
-
};
|
|
78
|
-
safetyMsg.content.forEach(c => {
|
|
79
|
-
if (c.type === "text")
|
|
80
|
-
c.text = "DO NOT IGNORE - IMPORTANT: " + c.text;
|
|
81
|
-
});
|
|
82
|
-
system.push(safetyMsg);
|
|
83
|
-
}
|
|
84
|
-
else if (msg.role === common_1.PromptRole.tool) {
|
|
85
|
-
if (!msg.tool_use_id) {
|
|
86
|
-
throw new Error("Tool use id is required for tool messages");
|
|
87
|
-
}
|
|
88
|
-
others.push({
|
|
89
|
-
role: "tool",
|
|
90
|
-
tool_call_id: msg.tool_use_id,
|
|
91
|
-
content: msg.content
|
|
92
|
-
});
|
|
93
|
-
}
|
|
94
|
-
else if (msg.role !== common_1.PromptRole.negative && msg.role !== common_1.PromptRole.mask) {
|
|
95
|
-
others.push({
|
|
96
|
-
role: msg.role ?? 'user',
|
|
97
|
-
content: parts
|
|
98
|
-
});
|
|
99
|
-
}
|
|
100
|
-
}
|
|
101
|
-
if (opts.result_schema && !opts.useToolForFormatting) {
|
|
102
|
-
system.push({
|
|
103
|
-
role: "system",
|
|
104
|
-
content: [{
|
|
105
|
-
type: "text",
|
|
106
|
-
text: "IMPORTANT: only answer using JSON, and respecting the schema included below, between the <response_schema> tags. " + `<response_schema>${JSON.stringify(opts.result_schema)}</response_schema>`
|
|
107
|
-
}]
|
|
108
|
-
});
|
|
109
|
-
}
|
|
110
|
-
// put system messages first and safety last
|
|
111
|
-
return [].concat(system).concat(others).concat(safety);
|
|
112
|
-
}
|
|
113
|
-
//# sourceMappingURL=openai.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/formatters/openai.ts"],"names":[],"mappings":";;AAwCA,gEAoBC;AAGD,4EAyFC;AAxJD,+CAA+D;AAC/D,4CAAkD;AAkClD;;;;GAIG;AACH,SAAgB,0BAA0B,CAAC,QAAyB;IAChE,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,MAAM,IAAI,GAAwB,EAAE,CAAC;IAErC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,MAAM,EAAE,CAAC;YACjC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,aAAa,GAAG,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,QAAQ,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,IAAI,EAAE,CAAC;YAC1G,IAAI,CAAC,IAAI,CAAC;gBACN,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,MAAM;aAC3B,CAAC,CAAA;QACN,CAAC;IACL,CAAC;IAED,4CAA4C;IAC5C,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AAC9C,CAAC;AAGM,KAAK,UAAU,gCAAgC,CAAC,QAAyB,EAAE,IAAkD;IAChI,MAAM,MAAM,GAAoB,EAAE,CAAC;IACnC,MAAM,MAAM,GAAoB,EAAE,CAAC;IACnC,MAAM,MAAM,GAAyB,EAAE,CAAC;IAExC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAEzB,MAAM,KAAK,GAAuD,EAAE,CAAC;QAErE,2CAA2C;QAC3C,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACZ,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,MAAM,IAAA,8BAAkB,EAAC,MAAM,CAAC,CAAC;gBAC9C,KAAK,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,WAAW;oBACjB,SAAS,EAAE;wBACP,GAAG,EAAE,QAAQ,IAAI,CAAC,SAAS,IAAI,YAAY,WAAW,IAAI,EAAE;wBAC5D,qFAAqF;qBACxF;iBACJ,CAAC,CAAA;YACN,CAAC;QACL,CAAC;QAED,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YACd,KAAK,CAAC,IAAI,CAAC;gBACP,IAAI,EAAE,GAAG,CAAC,OAAO;gBACjB,IAAI,EAAE,MAAM;aACf,CAAC,CAAA;QACN,CAAC;QAGD,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,MAAM,EAAE,CAAC;YACjC,MAAM,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,KAAK;aACjB,CAAC,CAAA;YAGF,IAAI,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;oBACf,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;wBAClB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;4BAAE,CAAC,CAAC,IAAI,GAAG,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC;oBACtD,CAAC,CAAC,CAAA;gBACN,CAAC,CAAC,CAAA;YACN,CAAC;QAEL,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,SAAS,GAAkB;gBAC7B,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,KAAK;aACjB,CAAA;YAED,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;gBAC1B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;oBAAE,CAAC,CAAC,IAAI,GAAG,6BAA6B,GAAG,CAAC,CAAC,IAAI,CAAC;YAC3E,CAAC,CAAC,CAAA;YAEF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC1B,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,IAAI,EAAE,CAAC;YACtC,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAA;YAChE,CAAC;YACD,MAAM,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,GAAG,CAAC,WAAW;gBAC7B,OAAO,EAAE,GAAG,CAAC,OAAO;aACvB,CAAC,CAAA;QACN,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,QAAQ,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAU,CAAC,IAAI,EAAE,CAAC;YAC1E,MAAM,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,MAAM;gBACxB,OAAO,EAAE,KAAK;aACjB,CAAC,CAAA;QACN,CAAC;IAEL,CAAC;IAED,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACnD,MAAM,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,CAAC;oBACN,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,mHAAmH,GAAG,oBAAoB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,oBAAoB;iBACzM,CAAC;SACL,CAAC,CAAA;IACN,CAAC;IAED,4CAA4C;IAC5C,OAAQ,EAA2B,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AAErF,CAAC"}
|
|
@@ -1,343 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.getBedrockOptions = getBedrockOptions;
|
|
4
|
-
const types_js_1 = require("../types.js");
|
|
5
|
-
const options_js_1 = require("../options.js");
|
|
6
|
-
function getMaxTokensLimit(model, option) {
|
|
7
|
-
// Claude models
|
|
8
|
-
if (model.includes("claude")) {
|
|
9
|
-
if (model.includes("3-7")) {
|
|
10
|
-
if (option && option?.thinking_mode) {
|
|
11
|
-
return 128000;
|
|
12
|
-
}
|
|
13
|
-
else {
|
|
14
|
-
return 8192;
|
|
15
|
-
}
|
|
16
|
-
}
|
|
17
|
-
else if (model.includes("3-5")) {
|
|
18
|
-
if (model.includes("claude-3-5-sonnet")) {
|
|
19
|
-
return 4096;
|
|
20
|
-
}
|
|
21
|
-
return 8192;
|
|
22
|
-
}
|
|
23
|
-
else {
|
|
24
|
-
return 4096;
|
|
25
|
-
}
|
|
26
|
-
}
|
|
27
|
-
// Amazon models
|
|
28
|
-
else if (model.includes("amazon")) {
|
|
29
|
-
if (model.includes("titan")) {
|
|
30
|
-
if (model.includes("lite")) {
|
|
31
|
-
return 4096;
|
|
32
|
-
}
|
|
33
|
-
else if (model.includes("express")) {
|
|
34
|
-
return 8192;
|
|
35
|
-
}
|
|
36
|
-
else if (model.includes("premier")) {
|
|
37
|
-
return 3072;
|
|
38
|
-
}
|
|
39
|
-
}
|
|
40
|
-
else if (model.includes("nova")) {
|
|
41
|
-
return 5000;
|
|
42
|
-
}
|
|
43
|
-
}
|
|
44
|
-
// Mistral models
|
|
45
|
-
else if (model.includes("mistral")) {
|
|
46
|
-
if (model.includes("8x7b")) {
|
|
47
|
-
return 4096;
|
|
48
|
-
}
|
|
49
|
-
return 8192;
|
|
50
|
-
}
|
|
51
|
-
// AI21 models
|
|
52
|
-
else if (model.includes("ai21")) {
|
|
53
|
-
if (model.includes("j2")) {
|
|
54
|
-
if (model.includes("large") || model.includes("mid") || model.includes("ultra")) {
|
|
55
|
-
return 8191;
|
|
56
|
-
}
|
|
57
|
-
return 2048;
|
|
58
|
-
}
|
|
59
|
-
if (model.includes("jamba")) {
|
|
60
|
-
return 4096;
|
|
61
|
-
}
|
|
62
|
-
}
|
|
63
|
-
// Cohere models
|
|
64
|
-
else if (model.includes("cohere.command")) {
|
|
65
|
-
if (model.includes("command-r")) {
|
|
66
|
-
return 128000;
|
|
67
|
-
}
|
|
68
|
-
return 4096;
|
|
69
|
-
}
|
|
70
|
-
// Meta models
|
|
71
|
-
else if (model.includes("llama")) {
|
|
72
|
-
if (model.includes("3-70b") || model.includes("3-8b")) {
|
|
73
|
-
return 2048;
|
|
74
|
-
}
|
|
75
|
-
return 8192;
|
|
76
|
-
}
|
|
77
|
-
// Default fallback
|
|
78
|
-
return undefined;
|
|
79
|
-
}
|
|
80
|
-
function getBedrockOptions(model, option) {
|
|
81
|
-
if (model.includes("canvas")) {
|
|
82
|
-
const tasktypeList = {
|
|
83
|
-
name: "taskType",
|
|
84
|
-
type: types_js_1.OptionType.enum,
|
|
85
|
-
enum: {
|
|
86
|
-
"Text-To-Image": "TEXT_IMAGE",
|
|
87
|
-
"Text-To-Image-with-Image-Conditioning": "TEXT_IMAGE_WITH_IMAGE_CONDITIONING",
|
|
88
|
-
"Color-Guided-Generation": "COLOR_GUIDED_GENERATION",
|
|
89
|
-
"Image-Variation": "IMAGE_VARIATION",
|
|
90
|
-
"Inpainting": "INPAINTING",
|
|
91
|
-
"Outpainting": "OUTPAINTING",
|
|
92
|
-
"Background-Removal": "BACKGROUND_REMOVAL",
|
|
93
|
-
},
|
|
94
|
-
default: "TEXT_IMAGE",
|
|
95
|
-
description: "The type of task to perform",
|
|
96
|
-
refresh: true,
|
|
97
|
-
};
|
|
98
|
-
let otherOptions = [
|
|
99
|
-
{ name: "width", type: types_js_1.OptionType.numeric, min: 320, max: 4096, default: 512, step: 16, integer: true, description: "The width of the generated image" },
|
|
100
|
-
{ name: "height", type: types_js_1.OptionType.numeric, min: 320, max: 4096, default: 512, step: 16, integer: true, description: "The height of the generated image" },
|
|
101
|
-
{
|
|
102
|
-
name: "quality",
|
|
103
|
-
type: types_js_1.OptionType.enum,
|
|
104
|
-
enum: { "standard": "standard", "premium": "premium" },
|
|
105
|
-
default: "standard",
|
|
106
|
-
description: "The quality of the generated image"
|
|
107
|
-
},
|
|
108
|
-
{ name: "cfgScale", type: types_js_1.OptionType.numeric, min: 1.1, max: 10.0, default: 6.5, step: 0.1, integer: false, description: "The scale of the generated image" },
|
|
109
|
-
{ name: "seed", type: types_js_1.OptionType.numeric, min: 0, max: 858993459, default: 12, integer: true, description: "The seed of the generated image" },
|
|
110
|
-
{ name: "numberOfImages", type: types_js_1.OptionType.numeric, min: 1, max: 5, default: 1, integer: true, description: "The number of images to generate" },
|
|
111
|
-
];
|
|
112
|
-
let dependentOptions = [];
|
|
113
|
-
switch (option?.taskType ?? "TEXT_IMAGE") {
|
|
114
|
-
case "TEXT_IMAGE_WITH_IMAGE_CONDITIONING":
|
|
115
|
-
dependentOptions.push({
|
|
116
|
-
name: "controlMode", type: types_js_1.OptionType.enum, enum: { "CANNY_EDGE": "CANNY_EDGE", "SEGMENTATION": "SEGMENTATION" },
|
|
117
|
-
default: "CANNY_EDGE", description: "The control mode of the generated image"
|
|
118
|
-
}, { name: "controlStrength", type: types_js_1.OptionType.numeric, min: 0, max: 1, default: 0.7, description: "The control strength of the generated image" });
|
|
119
|
-
break;
|
|
120
|
-
case "COLOR_GUIDED_GENERATION":
|
|
121
|
-
dependentOptions.push({ name: "colors", type: types_js_1.OptionType.string_list, value: [], description: "Hexadecimal color values to guide generation" });
|
|
122
|
-
break;
|
|
123
|
-
case "IMAGE_VARIATION":
|
|
124
|
-
dependentOptions.push({ name: "similarityStrength", type: types_js_1.OptionType.numeric, min: 0.2, max: 1, default: 0.7, description: "The similarity strength of the generated image" });
|
|
125
|
-
break;
|
|
126
|
-
case "INPAINTING":
|
|
127
|
-
//No changes
|
|
128
|
-
break;
|
|
129
|
-
case "OUTPAINTING":
|
|
130
|
-
dependentOptions.push({
|
|
131
|
-
name: "outPaintingMode", type: types_js_1.OptionType.enum, enum: { "DEFAULT": "DEFAULT", "PRECISE": "PRECISE" },
|
|
132
|
-
default: "default", description: "The outpainting mode of the generated image"
|
|
133
|
-
});
|
|
134
|
-
break;
|
|
135
|
-
case "BACKGROUND_REMOVAL":
|
|
136
|
-
dependentOptions = [];
|
|
137
|
-
otherOptions = [];
|
|
138
|
-
break;
|
|
139
|
-
}
|
|
140
|
-
return {
|
|
141
|
-
_option_id: "bedrock-nova-canvas",
|
|
142
|
-
options: [
|
|
143
|
-
tasktypeList,
|
|
144
|
-
...otherOptions,
|
|
145
|
-
...dependentOptions,
|
|
146
|
-
]
|
|
147
|
-
};
|
|
148
|
-
}
|
|
149
|
-
else {
|
|
150
|
-
const max_tokens_limit = getMaxTokensLimit(model, option);
|
|
151
|
-
//Not canvas, i.e normal AWS bedrock converse
|
|
152
|
-
const baseConverseOptions = [
|
|
153
|
-
{
|
|
154
|
-
name: "max_tokens",
|
|
155
|
-
type: types_js_1.OptionType.numeric,
|
|
156
|
-
min: 1,
|
|
157
|
-
max: max_tokens_limit,
|
|
158
|
-
integer: true,
|
|
159
|
-
step: 200,
|
|
160
|
-
description: "The maximum number of tokens to generate",
|
|
161
|
-
},
|
|
162
|
-
{
|
|
163
|
-
name: "temperature",
|
|
164
|
-
type: types_js_1.OptionType.numeric,
|
|
165
|
-
min: 0.0,
|
|
166
|
-
default: 0.7,
|
|
167
|
-
step: 0.1,
|
|
168
|
-
description: "A higher temperature biases toward less likely tokens, making the model more creative"
|
|
169
|
-
},
|
|
170
|
-
{
|
|
171
|
-
name: "top_p",
|
|
172
|
-
type: types_js_1.OptionType.numeric,
|
|
173
|
-
min: 0,
|
|
174
|
-
max: 1,
|
|
175
|
-
step: 0.1,
|
|
176
|
-
description: "Limits token sampling to the cumulative probability of the top p tokens"
|
|
177
|
-
},
|
|
178
|
-
{
|
|
179
|
-
name: "stop_sequence",
|
|
180
|
-
type: types_js_1.OptionType.string_list,
|
|
181
|
-
value: [],
|
|
182
|
-
description: "The generation will halt if one of the stop sequences is output"
|
|
183
|
-
}
|
|
184
|
-
];
|
|
185
|
-
if (model.includes("claude")) {
|
|
186
|
-
const claudeConverseOptions = [
|
|
187
|
-
{
|
|
188
|
-
name: "top_k",
|
|
189
|
-
type: types_js_1.OptionType.numeric,
|
|
190
|
-
min: 1,
|
|
191
|
-
integer: true,
|
|
192
|
-
step: 1,
|
|
193
|
-
description: "Limits token sampling to the top k tokens"
|
|
194
|
-
},
|
|
195
|
-
];
|
|
196
|
-
if (model.includes("3-7")) {
|
|
197
|
-
const claudeModeOptions = [
|
|
198
|
-
{
|
|
199
|
-
name: "thinking_mode",
|
|
200
|
-
type: types_js_1.OptionType.boolean,
|
|
201
|
-
default: false,
|
|
202
|
-
description: "If true, use the extended reasoning mode"
|
|
203
|
-
},
|
|
204
|
-
];
|
|
205
|
-
const claudeThinkingOptions = option?.thinking_mode ? [
|
|
206
|
-
{
|
|
207
|
-
name: "thinking_budget_tokens",
|
|
208
|
-
type: types_js_1.OptionType.numeric,
|
|
209
|
-
min: 1024,
|
|
210
|
-
default: 4000,
|
|
211
|
-
integer: true,
|
|
212
|
-
step: 100,
|
|
213
|
-
description: "The target number of tokens to use for reasoning, not a hard limit."
|
|
214
|
-
},
|
|
215
|
-
] : [];
|
|
216
|
-
return {
|
|
217
|
-
_option_id: "bedrock-claude",
|
|
218
|
-
options: [
|
|
219
|
-
...baseConverseOptions,
|
|
220
|
-
...claudeConverseOptions,
|
|
221
|
-
...claudeModeOptions,
|
|
222
|
-
...claudeThinkingOptions
|
|
223
|
-
]
|
|
224
|
-
};
|
|
225
|
-
}
|
|
226
|
-
return {
|
|
227
|
-
_option_id: "bedrock-claude",
|
|
228
|
-
options: [...baseConverseOptions, ...claudeConverseOptions]
|
|
229
|
-
};
|
|
230
|
-
}
|
|
231
|
-
else if (model.includes("amazon")) {
|
|
232
|
-
//Titan models also exists but does not support any additional options
|
|
233
|
-
if (model.includes("nova")) {
|
|
234
|
-
const novaConverseOptions = [
|
|
235
|
-
{
|
|
236
|
-
name: "top_k",
|
|
237
|
-
type: types_js_1.OptionType.numeric,
|
|
238
|
-
min: 1,
|
|
239
|
-
integer: true,
|
|
240
|
-
step: 1,
|
|
241
|
-
description: "Limits token sampling to the top k tokens"
|
|
242
|
-
},
|
|
243
|
-
];
|
|
244
|
-
return {
|
|
245
|
-
_option_id: "bedrock-nova",
|
|
246
|
-
options: [...baseConverseOptions, ...novaConverseOptions]
|
|
247
|
-
};
|
|
248
|
-
}
|
|
249
|
-
}
|
|
250
|
-
else if (model.includes("mistral")) {
|
|
251
|
-
//7b and 8x7b instruct
|
|
252
|
-
if (model.includes("7b")) {
|
|
253
|
-
const mistralConverseOptions = [
|
|
254
|
-
{
|
|
255
|
-
name: "top_k",
|
|
256
|
-
type: types_js_1.OptionType.numeric,
|
|
257
|
-
min: 1,
|
|
258
|
-
integer: true,
|
|
259
|
-
step: 1,
|
|
260
|
-
description: "Limits token sampling to the top k tokens"
|
|
261
|
-
},
|
|
262
|
-
];
|
|
263
|
-
return {
|
|
264
|
-
_option_id: "bedrock-mistral",
|
|
265
|
-
options: [...baseConverseOptions, ...mistralConverseOptions]
|
|
266
|
-
};
|
|
267
|
-
}
|
|
268
|
-
//Other models such as Mistral Small, Large and Large 2
|
|
269
|
-
//Support no additional options
|
|
270
|
-
}
|
|
271
|
-
else if (model.includes("ai21")) {
|
|
272
|
-
const ai21ConverseOptions = [
|
|
273
|
-
{
|
|
274
|
-
name: "presence_penalty",
|
|
275
|
-
type: types_js_1.OptionType.numeric,
|
|
276
|
-
min: -2,
|
|
277
|
-
max: 2,
|
|
278
|
-
default: 0,
|
|
279
|
-
step: 0.1,
|
|
280
|
-
description: "A higher presence penalty encourages the model to talk about new topics"
|
|
281
|
-
},
|
|
282
|
-
{
|
|
283
|
-
name: "frequency_penalty",
|
|
284
|
-
type: types_js_1.OptionType.numeric,
|
|
285
|
-
min: -2,
|
|
286
|
-
max: 2,
|
|
287
|
-
default: 0,
|
|
288
|
-
step: 0.1,
|
|
289
|
-
description: "A higher frequency penalty encourages the model to use less common words"
|
|
290
|
-
},
|
|
291
|
-
];
|
|
292
|
-
return {
|
|
293
|
-
_option_id: "bedrock-ai21",
|
|
294
|
-
options: [...baseConverseOptions, ...ai21ConverseOptions]
|
|
295
|
-
};
|
|
296
|
-
}
|
|
297
|
-
else if (model.includes("cohere.command")) {
|
|
298
|
-
const cohereCommandOptions = [
|
|
299
|
-
{
|
|
300
|
-
name: "top_k",
|
|
301
|
-
type: types_js_1.OptionType.numeric,
|
|
302
|
-
min: 1,
|
|
303
|
-
integer: true,
|
|
304
|
-
step: 1,
|
|
305
|
-
description: "Limits token sampling to the top k tokens"
|
|
306
|
-
},
|
|
307
|
-
];
|
|
308
|
-
if (model.includes("command-r")) {
|
|
309
|
-
const cohereCommandROptions = [
|
|
310
|
-
{
|
|
311
|
-
name: "frequency_penalty",
|
|
312
|
-
type: types_js_1.OptionType.numeric,
|
|
313
|
-
min: -2,
|
|
314
|
-
max: 2,
|
|
315
|
-
default: 0,
|
|
316
|
-
step: 0.1,
|
|
317
|
-
description: "A higher frequency penalty encourages the model to use less common words"
|
|
318
|
-
},
|
|
319
|
-
{
|
|
320
|
-
name: "presence_penalty",
|
|
321
|
-
type: types_js_1.OptionType.numeric,
|
|
322
|
-
min: -2,
|
|
323
|
-
max: 2,
|
|
324
|
-
default: 0,
|
|
325
|
-
step: 0.1,
|
|
326
|
-
description: "A higher presence penalty encourages the model to talk about new topics"
|
|
327
|
-
},
|
|
328
|
-
];
|
|
329
|
-
return {
|
|
330
|
-
_option_id: "bedrock-cohere-command",
|
|
331
|
-
options: [...baseConverseOptions, ...cohereCommandOptions, ...cohereCommandROptions]
|
|
332
|
-
};
|
|
333
|
-
}
|
|
334
|
-
}
|
|
335
|
-
//Fallback to converse standard.
|
|
336
|
-
return {
|
|
337
|
-
_option_id: "bedrock-converse",
|
|
338
|
-
options: baseConverseOptions
|
|
339
|
-
};
|
|
340
|
-
}
|
|
341
|
-
return options_js_1.textOptionsFallback;
|
|
342
|
-
}
|
|
343
|
-
//# sourceMappingURL=bedrock.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../../src/options/bedrock.ts"],"names":[],"mappings":";;AA+GA,8CAoRC;AAnYD,0CAA8F;AAC9F,8CAAoD;AAoCpD,SAAS,iBAAiB,CAAC,KAAa,EAAE,MAAqB;IAC3D,gBAAgB;IAChB,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3B,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,IAAI,MAAM,IAAK,MAA+B,EAAE,aAAa,EAAE,CAAC;gBAC5D,OAAO,MAAM,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACJ,OAAO,IAAI,CAAC;YAChB,CAAC;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,OAAO,IAAI,CAAC;QAChB,CAAC;aACI,CAAC;YACF,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IACD,gBAAgB;SACX,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1B,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC;YAChB,CAAC;iBAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,OAAO,IAAI,CAAC;YAChB,CAAC;iBAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,OAAO,IAAI,CAAC;YAChB,CAAC;QAEL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IACD,iBAAiB;SACZ,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,cAAc;SACT,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9E,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IACD,gBAAgB;SACX,IAAI,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACxC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAC9B,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,cAAc;SACT,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACpD,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,mBAAmB;IACnB,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,iBAAiB,CAAC,KAAa,EAAE,MAAqB;IAClE,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAwB;YACtC,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,qBAAU,CAAC,IAAI;YACrB,IAAI,EAAE;gBACF,eAAe,EAAE,YAAY;gBAC7B,uCAAuC,EAAE,oCAAoC;gBAC7E,yBAAyB,EAAE,yBAAyB;gBACpD,iBAAiB,EAAE,iBAAiB;gBACpC,YAAY,EAAE,YAAY;gBAC1B,aAAa,EAAE,aAAa;gBAC5B,oBAAoB,EAAE,oBAAoB;aAC7C;YACD,OAAO,EAAE,YAAY;YACrB,WAAW,EAAE,6BAA6B;YAC1C,OAAO,EAAE,IAAI;SAChB,CAAC;QAEF,IAAI,YAAY,GAA0B;YACtC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kCAAkC,EAAE;YACxJ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,mCAAmC,EAAE;YAC1J;gBACI,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,qBAAU,CAAC,IAAI;gBACrB,IAAI,EAAE,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE;gBACtD,OAAO,EAAE,UAAU;gBACnB,WAAW,EAAE,oCAAoC;aACpD;YACD,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,kCAAkC,EAAE;YAC7J,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,iCAAiC,EAAE;YAC9I,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kCAAkC,EAAE;SACnJ,CAAC;QAEF,IAAI,gBAAgB,GAA0B,EAAE,CAAC;QAEjD,QAAS,MAA4B,EAAE,QAAQ,IAAI,YAAY,EAAE,CAAC;YAC9D,KAAK,oCAAoC;gBACrC,gBAAgB,CAAC,IAAI,CACjB;oBACI,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE;oBAChH,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,yCAAyC;iBAChF,EACD,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,6CAA6C,EAAE,CAClJ,CAAC;gBACF,MAAM;YACV,KAAK,yBAAyB;gBAC1B,gBAAgB,CAAC,IAAI,CACjB,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,qBAAU,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE,EAAE,WAAW,EAAE,8CAA8C,EAAE,CAC3H,CAAA;gBACD,MAAM;YACV,KAAK,iBAAiB;gBAClB,gBAAgB,CAAC,IAAI,CACjB,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,gDAAgD,EAAE,CAC1J,CAAA;gBACD,MAAM;YACV,KAAK,YAAY;gBACb,YAAY;gBACZ,MAAM;YACV,KAAK,aAAa;gBACd,gBAAgB,CAAC,IAAI,CACjB;oBACI,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;oBACpG,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,6CAA6C;iBACjF,CACJ,CAAA;gBACD,MAAM;YACV,KAAK,oBAAoB;gBACrB,gBAAgB,GAAG,EAAE,CAAC;gBACtB,YAAY,GAAG,EAAE,CAAC;gBAClB,MAAM;QACd,CAAC;QAED,OAAO;YACH,UAAU,EAAE,qBAAqB;YACjC,OAAO,EAAE;gBACL,YAAY;gBACZ,GAAG,YAAY;gBACf,GAAG,gBAAgB;aACtB;SACJ,CAAC;IACN,CAAC;SAAM,CAAC;QACJ,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1D,6CAA6C;QAC7C,MAAM,mBAAmB,GAA0B;YAC/C;gBACI,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,qBAAU,CAAC,OAAO;gBACxB,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,gBAAgB;gBACrB,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,GAAG;gBACT,WAAW,EAAE,0CAA0C;aAC1D;YACD;gBACI,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,qBAAU,CAAC,OAAO;gBACxB,GAAG,EAAE,GAAG;gBACR,OAAO,EAAE,GAAG;gBACZ,IAAI,EAAE,GAAG;gBACT,WAAW,EAAE,uFAAuF;aACvG;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,qBAAU,CAAC,OAAO;gBACxB,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,CAAC;gBACN,IAAI,EAAE,GAAG;gBACT,WAAW,EAAE,yEAAyE;aACzF;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,qBAAU,CAAC,WAAW;gBAC5B,KAAK,EAAE,EAAE;gBACT,WAAW,EAAE,iEAAiE;aACjF;SAAC,CAAC;QAEP,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,MAAM,qBAAqB,GAA0B;gBACjD;oBACI,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,GAAG,EAAE,CAAC;oBACN,OAAO,EAAE,IAAI;oBACb,IAAI,EAAE,CAAC;oBACP,WAAW,EAAE,2CAA2C;iBAC3D;aACJ,CAAC;YACF,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxB,MAAM,iBAAiB,GAA0B;oBAC7C;wBACI,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,OAAO,EAAE,KAAK;wBACd,WAAW,EAAE,0CAA0C;qBAC1D;iBACJ,CAAC;gBACF,MAAM,qBAAqB,GAA2B,MAA+B,EAAE,aAAa,CAAC,CAAC,CAAC;oBACnG;wBACI,IAAI,EAAE,wBAAwB;wBAC9B,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,GAAG,EAAE,IAAI;wBACT,OAAO,EAAE,IAAI;wBACb,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,GAAG;wBACT,WAAW,EAAE,qEAAqE;qBACrF;iBACJ,CAAC,CAAC,CAAC,EAAE,CAAC;gBAEP,OAAO;oBACH,UAAU,EAAE,gBAAgB;oBAC5B,OAAO,EAAE;wBACL,GAAG,mBAAmB;wBACtB,GAAG,qBAAqB;wBACxB,GAAG,iBAAiB;wBACpB,GAAG,qBAAqB;qBAAC;iBAChC,CAAA;YACL,CAAC;YACD,OAAO;gBACH,UAAU,EAAE,gBAAgB;gBAC5B,OAAO,EAAE,CAAC,GAAG,mBAAmB,EAAE,GAAG,qBAAqB,CAAC;aAC9D,CAAA;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,sEAAsE;YACtE,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,MAAM,mBAAmB,GAA0B;oBAC/C;wBACI,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,GAAG,EAAE,CAAC;wBACN,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,CAAC;wBACP,WAAW,EAAE,2CAA2C;qBAC3D;iBACJ,CAAC;gBACF,OAAO;oBACH,UAAU,EAAE,cAAc;oBAC1B,OAAO,EAAE,CAAC,GAAG,mBAAmB,EAAE,GAAG,mBAAmB,CAAC;iBAC5D,CAAA;YACL,CAAC;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACjC,sBAAsB;YACtB,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvB,MAAM,sBAAsB,GAA0B;oBAClD;wBACI,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,GAAG,EAAE,CAAC;wBACN,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,CAAC;wBACP,WAAW,EAAE,2CAA2C;qBAC3D;iBACJ,CAAC;gBACF,OAAO;oBACH,UAAU,EAAE,iBAAiB;oBAC7B,OAAO,EAAE,CAAC,GAAG,mBAAmB,EAAE,GAAG,sBAAsB,CAAC;iBAC/D,CAAA;YACL,CAAC;YACD,uDAAuD;YACvD,+BAA+B;QACnC,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9B,MAAM,mBAAmB,GAA0B;gBAC/C;oBACI,IAAI,EAAE,kBAAkB;oBACxB,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,GAAG,EAAE,CAAC,CAAC;oBACP,GAAG,EAAE,CAAC;oBACN,OAAO,EAAE,CAAC;oBACV,IAAI,EAAE,GAAG;oBACT,WAAW,EAAE,yEAAyE;iBACzF;gBACD;oBACI,IAAI,EAAE,mBAAmB;oBACzB,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,GAAG,EAAE,CAAC,CAAC;oBACP,GAAG,EAAE,CAAC;oBACN,OAAO,EAAE,CAAC;oBACV,IAAI,EAAE,GAAG;oBACT,WAAW,EAAE,0EAA0E;iBAC1F;aACJ,CAAC;YAEF,OAAO;gBACH,UAAU,EAAE,cAAc;gBAC1B,OAAO,EAAE,CAAC,GAAG,mBAAmB,EAAE,GAAG,mBAAmB,CAAC;aAC5D,CAAA;QACL,CAAC;aACI,IAAI,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACxC,MAAM,oBAAoB,GAA0B;gBAChD;oBACI,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,qBAAU,CAAC,OAAO;oBACxB,GAAG,EAAE,CAAC;oBACN,OAAO,EAAE,IAAI;oBACb,IAAI,EAAE,CAAC;oBACP,WAAW,EAAE,2CAA2C;iBAC3D;aACJ,CAAC;YACF,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC9B,MAAM,qBAAqB,GAA0B;oBACjD;wBACI,IAAI,EAAE,mBAAmB;wBACzB,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,GAAG,EAAE,CAAC,CAAC;wBACP,GAAG,EAAE,CAAC;wBACN,OAAO,EAAE,CAAC;wBACV,IAAI,EAAE,GAAG;wBACT,WAAW,EAAE,0EAA0E;qBAC1F;oBACD;wBACI,IAAI,EAAE,kBAAkB;wBACxB,IAAI,EAAE,qBAAU,CAAC,OAAO;wBACxB,GAAG,EAAE,CAAC,CAAC;wBACP,GAAG,EAAE,CAAC;wBACN,OAAO,EAAE,CAAC;wBACV,IAAI,EAAE,GAAG;wBACT,WAAW,EAAE,yEAAyE;qBACzF;iBACJ,CAAC;gBACF,OAAO;oBACH,UAAU,EAAE,wBAAwB;oBACpC,OAAO,EAAE,CAAC,GAAG,mBAAmB,EAAE,GAAG,oBAAoB,EAAE,GAAG,qBAAqB,CAAC;iBACvF,CAAA;YACL,CAAC;QACL,CAAC;QAED,gCAAgC;QAChC,OAAO;YACH,UAAU,EAAE,kBAAkB;YAC9B,OAAO,EAAE,mBAAmB;SAC/B,CAAC;IACN,CAAC;IACD,OAAO,gCAAmB,CAAC;AAC/B,CAAC"}
|
package/lib/cjs/options/groq.js
DELETED
|
@@ -1,37 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.getGroqOptions = getGroqOptions;
|
|
4
|
-
const types_js_1 = require("../types.js");
|
|
5
|
-
const options_js_1 = require("../options.js");
|
|
6
|
-
function getGroqOptions(model, _option) {
|
|
7
|
-
if (model.includes("deepseek") && model.includes("r1")) {
|
|
8
|
-
const commonOptions = [
|
|
9
|
-
{
|
|
10
|
-
name: types_js_1.SharedOptions.max_tokens, type: types_js_1.OptionType.numeric, min: 1, max: 131072,
|
|
11
|
-
integer: true, description: "The maximum number of tokens to generate",
|
|
12
|
-
},
|
|
13
|
-
{
|
|
14
|
-
name: types_js_1.SharedOptions.temperature, type: types_js_1.OptionType.numeric, min: 0.0, default: 0.7, max: 2.0,
|
|
15
|
-
integer: false, step: 0.1, description: "A higher temperature biases toward less likely tokens, making the model more creative. A lower temperature than other models is recommended for deepseek R1, 0.3-0.7 approximately.",
|
|
16
|
-
},
|
|
17
|
-
{
|
|
18
|
-
name: types_js_1.SharedOptions.top_p, type: types_js_1.OptionType.numeric, min: 0, max: 1,
|
|
19
|
-
integer: false, step: 0.1, description: "Limits token sampling to the cumulative probability of the top p tokens",
|
|
20
|
-
},
|
|
21
|
-
{
|
|
22
|
-
name: types_js_1.SharedOptions.stop_sequence, type: types_js_1.OptionType.string_list, value: [],
|
|
23
|
-
description: "The generation will halt if one of the stop sequences is output",
|
|
24
|
-
},
|
|
25
|
-
{
|
|
26
|
-
name: "reasoning_format", type: types_js_1.OptionType.enum, enum: { "Parsed": "parsed", "Raw": "raw", "Hidden": "hidden" },
|
|
27
|
-
default: "parsed", description: "Controls how the reasoning is returned.",
|
|
28
|
-
},
|
|
29
|
-
];
|
|
30
|
-
return {
|
|
31
|
-
_option_id: "groq-deepseek-thinking",
|
|
32
|
-
options: commonOptions,
|
|
33
|
-
};
|
|
34
|
-
}
|
|
35
|
-
return options_js_1.textOptionsFallback;
|
|
36
|
-
}
|
|
37
|
-
//# sourceMappingURL=groq.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"groq.js","sourceRoot":"","sources":["../../../src/options/groq.ts"],"names":[],"mappings":";;AAeA,wCA+BC;AA9CD,0CAA6G;AAC7G,8CAAoD;AAcpD,SAAgB,cAAc,CAAC,KAAa,EAAE,OAAsB;IAChE,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,aAAa,GAA0B;YACzC;gBACI,IAAI,EAAE,wBAAa,CAAC,UAAU,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,MAAM;gBAC7E,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,0CAA0C;aACzE;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,WAAW,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG;gBAC3F,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,qLAAqL;aAChO;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,KAAK,EAAE,IAAI,EAAE,qBAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC;gBACnE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,WAAW,EAAE,yEAAyE;aACpH;YACD;gBACI,IAAI,EAAE,wBAAa,CAAC,aAAa,EAAE,IAAI,EAAE,qBAAU,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;gBAC1E,WAAW,EAAE,iEAAiE;aACjF;YACD;gBACI,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,qBAAU,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE;gBAC/G,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,yCAAyC;aAC5E;SACJ,CAAC;QAEF,OAAO;YACH,UAAU,EAAE,wBAAwB;YACpC,OAAO,EAAE,aAAa;SACzB,CAAC;IACN,CAAC;IACD,OAAO,gCAAmB,CAAC;AAC/B,CAAC"}
|