@llumiverse/common 1.3.0 → 1.4.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/capability/anthropic.d.ts +3 -0
- package/lib/capability/anthropic.d.ts.map +1 -0
- package/lib/capability/anthropic.js +92 -0
- package/lib/capability/anthropic.js.map +1 -0
- package/lib/{types/capability → capability}/azure_foundry.d.ts +1 -1
- package/lib/capability/azure_foundry.d.ts.map +1 -0
- package/lib/capability/azure_foundry.js +394 -0
- package/lib/capability/azure_foundry.js.map +1 -0
- package/lib/{types/capability → capability}/bedrock.d.ts +1 -1
- package/lib/capability/bedrock.d.ts.map +1 -0
- package/lib/capability/bedrock.js +300 -0
- package/lib/capability/bedrock.js.map +1 -0
- package/lib/{types/capability → capability}/openai.d.ts +1 -1
- package/lib/capability/openai.d.ts.map +1 -0
- package/lib/capability/openai.js +179 -0
- package/lib/capability/openai.js.map +1 -0
- package/lib/{types/capability → capability}/vertexai.d.ts +1 -1
- package/lib/capability/vertexai.d.ts.map +1 -0
- package/lib/capability/vertexai.js +137 -0
- package/lib/capability/vertexai.js.map +1 -0
- package/lib/{types/capability.d.ts → capability.d.ts} +1 -1
- package/lib/capability.d.ts.map +1 -0
- package/lib/{esm/capability.js → capability.js} +17 -17
- package/lib/capability.js.map +1 -0
- package/lib/index.d.ts +14 -0
- package/lib/index.d.ts.map +1 -0
- package/lib/index.js +14 -0
- package/lib/index.js.map +1 -0
- package/lib/{types/options → options}/anthropic.d.ts +2 -2
- package/lib/options/anthropic.d.ts.map +1 -0
- package/lib/{esm/options → options}/anthropic.js +14 -11
- package/lib/options/anthropic.js.map +1 -0
- package/lib/{types/options → options}/azure_foundry.d.ts +14 -14
- package/lib/options/azure_foundry.d.ts.map +1 -0
- package/lib/{esm/options → options}/azure_foundry.js +126 -126
- package/lib/options/azure_foundry.js.map +1 -0
- package/lib/{types/options → options}/bedrock.d.ts +16 -16
- package/lib/{types/options → options}/bedrock.d.ts.map +1 -1
- package/lib/options/bedrock.js +489 -0
- package/lib/options/bedrock.js.map +1 -0
- package/lib/options/context-windows.d.ts.map +1 -0
- package/lib/options/context-windows.js.map +1 -0
- package/lib/options/embedding.d.ts +38 -0
- package/lib/options/embedding.d.ts.map +1 -0
- package/lib/options/embedding.js +77 -0
- package/lib/options/embedding.js.map +1 -0
- package/lib/{types/options → options}/fallback.d.ts +2 -2
- package/lib/options/fallback.d.ts.map +1 -0
- package/lib/options/fallback.js +65 -0
- package/lib/options/fallback.js.map +1 -0
- package/lib/{types/options → options}/groq.d.ts +2 -2
- package/lib/options/groq.d.ts.map +1 -0
- package/lib/options/groq.js +54 -0
- package/lib/options/groq.js.map +1 -0
- package/lib/options/openai.d.ts +40 -0
- package/lib/options/openai.d.ts.map +1 -0
- package/lib/options/openai.js +284 -0
- package/lib/options/openai.js.map +1 -0
- package/lib/{types/options → options}/shared-parsing.d.ts +1 -1
- package/lib/options/shared-parsing.d.ts.map +1 -0
- package/lib/{esm/options → options}/shared-parsing.js +14 -17
- package/lib/options/shared-parsing.js.map +1 -0
- package/lib/options/version-parsing.d.ts.map +1 -0
- package/lib/{esm/options → options}/version-parsing.js +15 -15
- package/lib/options/version-parsing.js.map +1 -0
- package/lib/{types/options → options}/vertexai.d.ts +15 -15
- package/lib/options/vertexai.d.ts.map +1 -0
- package/lib/options/vertexai.js +606 -0
- package/lib/options/vertexai.js.map +1 -0
- package/lib/{types/options.d.ts → options.d.ts} +1 -1
- package/lib/options.d.ts.map +1 -0
- package/lib/{esm/options.js → options.js} +8 -8
- package/lib/options.js.map +1 -0
- package/lib/{types/types.d.ts → types.d.ts} +143 -35
- package/lib/types.d.ts.map +1 -0
- package/lib/{esm/types.js → types.js} +21 -23
- package/lib/types.js.map +1 -0
- package/package.json +11 -34
- package/src/LlumiverseError.test.ts +18 -72
- package/src/capability/anthropic.ts +61 -11
- package/src/capability/azure_foundry.ts +303 -65
- package/src/capability/bedrock.ts +201 -56
- package/src/capability/openai.ts +135 -41
- package/src/capability/vertexai.ts +105 -35
- package/src/capability.ts +18 -18
- package/src/index.ts +13 -12
- package/src/options/anthropic.ts +16 -13
- package/src/options/azure_foundry.ts +157 -143
- package/src/options/bedrock.ts +247 -185
- package/src/options/context-windows.ts +2 -2
- package/src/options/embedding.ts +92 -0
- package/src/options/fallback.ts +50 -17
- package/src/options/groq.ts +48 -21
- package/src/options/openai.ts +228 -183
- package/src/options/shared-parsing.ts +16 -21
- package/src/options/version-parsing.ts +17 -17
- package/src/options/vertexai.ts +382 -288
- package/src/options.ts +8 -8
- package/src/types.ts +260 -155
- package/lib/cjs/capability/anthropic.js +0 -45
- package/lib/cjs/capability/anthropic.js.map +0 -1
- package/lib/cjs/capability/azure_foundry.js +0 -160
- package/lib/cjs/capability/azure_foundry.js.map +0 -1
- package/lib/cjs/capability/bedrock.js +0 -158
- package/lib/cjs/capability/bedrock.js.map +0 -1
- package/lib/cjs/capability/openai.js +0 -98
- package/lib/cjs/capability/openai.js.map +0 -1
- package/lib/cjs/capability/vertexai.js +0 -80
- package/lib/cjs/capability/vertexai.js.map +0 -1
- package/lib/cjs/capability.js +0 -115
- package/lib/cjs/capability.js.map +0 -1
- package/lib/cjs/index.js +0 -29
- package/lib/cjs/index.js.map +0 -1
- package/lib/cjs/options/anthropic.js +0 -34
- package/lib/cjs/options/anthropic.js.map +0 -1
- package/lib/cjs/options/azure_foundry.js +0 -423
- package/lib/cjs/options/azure_foundry.js.map +0 -1
- package/lib/cjs/options/bedrock.js +0 -427
- package/lib/cjs/options/bedrock.js.map +0 -1
- package/lib/cjs/options/context-windows.js +0 -138
- package/lib/cjs/options/context-windows.js.map +0 -1
- package/lib/cjs/options/fallback.js +0 -35
- package/lib/cjs/options/fallback.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 -246
- package/lib/cjs/options/openai.js.map +0 -1
- package/lib/cjs/options/shared-parsing.js +0 -144
- package/lib/cjs/options/shared-parsing.js.map +0 -1
- package/lib/cjs/options/version-parsing.js +0 -326
- package/lib/cjs/options/version-parsing.js.map +0 -1
- package/lib/cjs/options/vertexai.js +0 -525
- package/lib/cjs/options/vertexai.js.map +0 -1
- package/lib/cjs/options.js +0 -33
- package/lib/cjs/options.js.map +0 -1
- package/lib/cjs/package.json +0 -3
- package/lib/cjs/types.js +0 -307
- package/lib/cjs/types.js.map +0 -1
- package/lib/esm/capability/anthropic.js +0 -42
- package/lib/esm/capability/anthropic.js.map +0 -1
- package/lib/esm/capability/azure_foundry.js +0 -157
- package/lib/esm/capability/azure_foundry.js.map +0 -1
- package/lib/esm/capability/bedrock.js +0 -155
- package/lib/esm/capability/bedrock.js.map +0 -1
- package/lib/esm/capability/openai.js +0 -95
- package/lib/esm/capability/openai.js.map +0 -1
- package/lib/esm/capability/vertexai.js +0 -77
- package/lib/esm/capability/vertexai.js.map +0 -1
- package/lib/esm/capability.js.map +0 -1
- package/lib/esm/index.js +0 -13
- package/lib/esm/index.js.map +0 -1
- package/lib/esm/options/anthropic.js.map +0 -1
- package/lib/esm/options/azure_foundry.js.map +0 -1
- package/lib/esm/options/bedrock.js +0 -423
- package/lib/esm/options/bedrock.js.map +0 -1
- package/lib/esm/options/context-windows.js.map +0 -1
- package/lib/esm/options/fallback.js +0 -32
- package/lib/esm/options/fallback.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 -243
- package/lib/esm/options/openai.js.map +0 -1
- package/lib/esm/options/shared-parsing.js.map +0 -1
- package/lib/esm/options/version-parsing.js.map +0 -1
- package/lib/esm/options/vertexai.js +0 -519
- package/lib/esm/options/vertexai.js.map +0 -1
- package/lib/esm/options.js.map +0 -1
- package/lib/esm/types.js.map +0 -1
- package/lib/types/capability/anthropic.d.ts +0 -3
- package/lib/types/capability/anthropic.d.ts.map +0 -1
- package/lib/types/capability/azure_foundry.d.ts.map +0 -1
- package/lib/types/capability/bedrock.d.ts.map +0 -1
- package/lib/types/capability/openai.d.ts.map +0 -1
- package/lib/types/capability/vertexai.d.ts.map +0 -1
- package/lib/types/capability.d.ts.map +0 -1
- package/lib/types/index.d.ts +0 -13
- package/lib/types/index.d.ts.map +0 -1
- package/lib/types/options/anthropic.d.ts.map +0 -1
- package/lib/types/options/azure_foundry.d.ts.map +0 -1
- package/lib/types/options/context-windows.d.ts.map +0 -1
- package/lib/types/options/fallback.d.ts.map +0 -1
- package/lib/types/options/groq.d.ts.map +0 -1
- package/lib/types/options/openai.d.ts +0 -40
- package/lib/types/options/openai.d.ts.map +0 -1
- package/lib/types/options/shared-parsing.d.ts.map +0 -1
- package/lib/types/options/version-parsing.d.ts.map +0 -1
- package/lib/types/options/vertexai.d.ts.map +0 -1
- package/lib/types/options.d.ts.map +0 -1
- package/lib/types/types.d.ts.map +0 -1
- /package/lib/{types/options → options}/context-windows.d.ts +0 -0
- /package/lib/{esm/options → options}/context-windows.js +0 -0
- /package/lib/{types/options → options}/version-parsing.d.ts +0 -0
|
@@ -0,0 +1,137 @@
|
|
|
1
|
+
// Record of Vertex AI model capabilities keyed by model ID (last path segment, lowercased)
|
|
2
|
+
const RECORD_MODEL_CAPABILITIES = {
|
|
3
|
+
'gemini-2.0-flash-lite-001': {
|
|
4
|
+
input: { text: true, image: true, video: true, audio: true, embed: false },
|
|
5
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
6
|
+
tool_support: false,
|
|
7
|
+
},
|
|
8
|
+
};
|
|
9
|
+
// Populate RECORD_FAMILY_CAPABILITIES as a const record (lowest common denominator for each family)
|
|
10
|
+
const RECORD_FAMILY_CAPABILITIES = {
|
|
11
|
+
gemini: {
|
|
12
|
+
input: { text: true, image: true, video: true, audio: true, embed: false },
|
|
13
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
14
|
+
tool_support: true,
|
|
15
|
+
},
|
|
16
|
+
'gemini-2.5-flash-image': {
|
|
17
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
18
|
+
output: { text: true, image: true, video: false, audio: false, embed: false },
|
|
19
|
+
tool_support: false,
|
|
20
|
+
},
|
|
21
|
+
'gemini-3.0-pro-image': {
|
|
22
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
23
|
+
output: { text: true, image: true, video: false, audio: false, embed: false },
|
|
24
|
+
tool_support: false,
|
|
25
|
+
},
|
|
26
|
+
'gemini-3.0-flash-image': {
|
|
27
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
28
|
+
output: { text: true, image: true, video: false, audio: false, embed: false },
|
|
29
|
+
tool_support: false,
|
|
30
|
+
},
|
|
31
|
+
imagen: {
|
|
32
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
33
|
+
output: { text: false, image: true, video: false, audio: false, embed: false },
|
|
34
|
+
tool_support: false,
|
|
35
|
+
},
|
|
36
|
+
'imagen-3.0': {
|
|
37
|
+
input: { text: true, image: false, video: false, audio: false, embed: false },
|
|
38
|
+
output: { text: false, image: true, video: false, audio: false, embed: false },
|
|
39
|
+
tool_support: false,
|
|
40
|
+
},
|
|
41
|
+
'imagen-3.0-capability': {
|
|
42
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
43
|
+
output: { text: false, image: true, video: false, audio: false, embed: false },
|
|
44
|
+
tool_support: false,
|
|
45
|
+
},
|
|
46
|
+
'imagen-4.0': {
|
|
47
|
+
input: { text: true, image: false, video: false, audio: false, embed: false },
|
|
48
|
+
output: { text: false, image: true, video: false, audio: false, embed: false },
|
|
49
|
+
tool_support: false,
|
|
50
|
+
},
|
|
51
|
+
'imagen-4.0-capability': {
|
|
52
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
53
|
+
output: { text: false, image: true, video: false, audio: false, embed: false },
|
|
54
|
+
tool_support: false,
|
|
55
|
+
},
|
|
56
|
+
claude: {
|
|
57
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
58
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
59
|
+
tool_support: true,
|
|
60
|
+
},
|
|
61
|
+
llama: {
|
|
62
|
+
input: { text: true, image: false, video: false, audio: false, embed: false },
|
|
63
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
64
|
+
tool_support: true,
|
|
65
|
+
},
|
|
66
|
+
'llama-2': {
|
|
67
|
+
input: { text: true, image: false, video: false, audio: false, embed: false },
|
|
68
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
69
|
+
tool_support: false,
|
|
70
|
+
},
|
|
71
|
+
'llama-3.2': {
|
|
72
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
73
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
74
|
+
tool_support: true,
|
|
75
|
+
},
|
|
76
|
+
'llama-4': {
|
|
77
|
+
input: { text: true, image: true, video: false, audio: false, embed: false },
|
|
78
|
+
output: { text: true, image: false, video: false, audio: false, embed: false },
|
|
79
|
+
tool_support: true,
|
|
80
|
+
},
|
|
81
|
+
};
|
|
82
|
+
// Fallback pattern lists for inferring modalities and tool support
|
|
83
|
+
const IMAGE_INPUT_MODELS = ['image'];
|
|
84
|
+
const VIDEO_INPUT_MODELS = ['video'];
|
|
85
|
+
const AUDIO_INPUT_MODELS = ['audio'];
|
|
86
|
+
const TEXT_INPUT_MODELS = ['text'];
|
|
87
|
+
const IMAGE_OUTPUT_MODELS = ['image'];
|
|
88
|
+
const VIDEO_OUTPUT_MODELS = ['video'];
|
|
89
|
+
const AUDIO_OUTPUT_MODELS = ['audio'];
|
|
90
|
+
const TEXT_OUTPUT_MODELS = ['text'];
|
|
91
|
+
const EMBEDDING_OUTPUT_MODELS = ['embed'];
|
|
92
|
+
const TOOL_SUPPORT_MODELS = ['tool', 'sonnet', 'opus', 'gemini', 'claude'];
|
|
93
|
+
function modelMatches(modelName, patterns) {
|
|
94
|
+
return patterns.some((pattern) => modelName.includes(pattern));
|
|
95
|
+
}
|
|
96
|
+
function normalizeVertexAIModelName(modelName) {
|
|
97
|
+
const segments = modelName.toLowerCase().split('/');
|
|
98
|
+
return segments[segments.length - 1];
|
|
99
|
+
}
|
|
100
|
+
/**
|
|
101
|
+
* Get the full ModelCapabilities for a Vertex AI model.
|
|
102
|
+
* Checks RECORD_MODEL_CAPABILITIES first, then falls back to pattern-based inference.
|
|
103
|
+
*/
|
|
104
|
+
export function getModelCapabilitiesVertexAI(model) {
|
|
105
|
+
const normalized = normalizeVertexAIModelName(model);
|
|
106
|
+
const record = RECORD_MODEL_CAPABILITIES[normalized];
|
|
107
|
+
if (record)
|
|
108
|
+
return record;
|
|
109
|
+
let bestFamilyKey;
|
|
110
|
+
let bestFamilyLength = 0;
|
|
111
|
+
for (const key of Object.keys(RECORD_FAMILY_CAPABILITIES)) {
|
|
112
|
+
if (normalized.startsWith(key) && key.length > bestFamilyLength) {
|
|
113
|
+
bestFamilyKey = key;
|
|
114
|
+
bestFamilyLength = key.length;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
if (bestFamilyKey) {
|
|
118
|
+
return RECORD_FAMILY_CAPABILITIES[bestFamilyKey];
|
|
119
|
+
}
|
|
120
|
+
const input = {
|
|
121
|
+
text: modelMatches(normalized, TEXT_INPUT_MODELS) || undefined,
|
|
122
|
+
image: modelMatches(normalized, IMAGE_INPUT_MODELS) || undefined,
|
|
123
|
+
video: modelMatches(normalized, VIDEO_INPUT_MODELS) || undefined,
|
|
124
|
+
audio: modelMatches(normalized, AUDIO_INPUT_MODELS) || undefined,
|
|
125
|
+
embed: false,
|
|
126
|
+
};
|
|
127
|
+
const output = {
|
|
128
|
+
text: modelMatches(normalized, TEXT_OUTPUT_MODELS) || undefined,
|
|
129
|
+
image: modelMatches(normalized, IMAGE_OUTPUT_MODELS) || undefined,
|
|
130
|
+
video: modelMatches(normalized, VIDEO_OUTPUT_MODELS) || undefined,
|
|
131
|
+
audio: modelMatches(normalized, AUDIO_OUTPUT_MODELS) || undefined,
|
|
132
|
+
embed: modelMatches(normalized, EMBEDDING_OUTPUT_MODELS) || undefined,
|
|
133
|
+
};
|
|
134
|
+
const tool_support = modelMatches(normalized, TOOL_SUPPORT_MODELS) || undefined;
|
|
135
|
+
return { input, output, tool_support };
|
|
136
|
+
}
|
|
137
|
+
//# sourceMappingURL=vertexai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vertexai.js","sourceRoot":"","sources":["../../src/capability/vertexai.ts"],"names":[],"mappings":"AAEA,2FAA2F;AAC3F,MAAM,yBAAyB,GAG3B;IACA,2BAA2B,EAAE;QACzB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;QAC1E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;CACJ,CAAC;AAEF,oGAAoG;AACpG,MAAM,0BAA0B,GAG5B;IACA,MAAM,EAAE;QACJ,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;QAC1E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,IAAI;KACrB;IACD,wBAAwB,EAAE;QACtB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,YAAY,EAAE,KAAK;KACtB;IACD,sBAAsB,EAAE;QACpB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,YAAY,EAAE,KAAK;KACtB;IACD,wBAAwB,EAAE;QACtB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,YAAY,EAAE,KAAK;KACtB;IACD,MAAM,EAAE;QACJ,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,YAAY,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,uBAAuB,EAAE;QACrB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,YAAY,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,uBAAuB,EAAE;QACrB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,MAAM,EAAE;QACJ,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,IAAI;KACrB;IACD,KAAK,EAAE;QACH,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,IAAI;KACrB;IACD,SAAS,EAAE;QACP,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,KAAK;KACtB;IACD,WAAW,EAAE;QACT,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,IAAI;KACrB;IACD,SAAS,EAAE;QACP,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC9E,YAAY,EAAE,IAAI;KACrB;CACJ,CAAC;AAEF,mEAAmE;AACnE,MAAM,kBAAkB,GAAG,CAAC,OAAO,CAAC,CAAC;AACrC,MAAM,kBAAkB,GAAG,CAAC,OAAO,CAAC,CAAC;AACrC,MAAM,kBAAkB,GAAG,CAAC,OAAO,CAAC,CAAC;AACrC,MAAM,iBAAiB,GAAG,CAAC,MAAM,CAAC,CAAC;AACnC,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,CAAC;AACtC,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,CAAC;AACtC,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,CAAC;AACtC,MAAM,kBAAkB,GAAG,CAAC,MAAM,CAAC,CAAC;AACpC,MAAM,uBAAuB,GAAG,CAAC,OAAO,CAAC,CAAC;AAC1C,MAAM,mBAAmB,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;AAE3E,SAAS,YAAY,CAAC,SAAiB,EAAE,QAAkB;IACvD,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,0BAA0B,CAAC,SAAiB;IACjD,MAAM,QAAQ,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACpD,OAAO,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACzC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,4BAA4B,CAAC,KAAa;IAKtD,MAAM,UAAU,GAAG,0BAA0B,CAAC,KAAK,CAAC,CAAC;IACrD,MAAM,MAAM,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;IACrD,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAC1B,IAAI,aAAiC,CAAC;IACtC,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,EAAE,CAAC;QACxD,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YAC9D,aAAa,GAAG,GAAG,CAAC;YACpB,gBAAgB,GAAG,GAAG,CAAC,MAAM,CAAC;QAClC,CAAC;IACL,CAAC;IACD,IAAI,aAAa,EAAE,CAAC;QAChB,OAAO,0BAA0B,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC;IACD,MAAM,KAAK,GAAoB;QAC3B,IAAI,EAAE,YAAY,CAAC,UAAU,EAAE,iBAAiB,CAAC,IAAI,SAAS;QAC9D,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,kBAAkB,CAAC,IAAI,SAAS;QAChE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,kBAAkB,CAAC,IAAI,SAAS;QAChE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,kBAAkB,CAAC,IAAI,SAAS;QAChE,KAAK,EAAE,KAAK;KACf,CAAC;IACF,MAAM,MAAM,GAAoB;QAC5B,IAAI,EAAE,YAAY,CAAC,UAAU,EAAE,kBAAkB,CAAC,IAAI,SAAS;QAC/D,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,mBAAmB,CAAC,IAAI,SAAS;QACjE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,mBAAmB,CAAC,IAAI,SAAS;QACjE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,mBAAmB,CAAC,IAAI,SAAS;QACjE,KAAK,EAAE,YAAY,CAAC,UAAU,EAAE,uBAAuB,CAAC,IAAI,SAAS;KACxE,CAAC;IACF,MAAM,YAAY,GAAG,YAAY,CAAC,UAAU,EAAE,mBAAmB,CAAC,IAAI,SAAS,CAAC;IAChF,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;AAC3C,CAAC"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { type ModelCapabilities, type ModelModalities, Providers } from
|
|
1
|
+
import { type ModelCapabilities, type ModelModalities, Providers } from './types.js';
|
|
2
2
|
export declare function getModelCapabilities(model: string, provider?: string | Providers): ModelCapabilities;
|
|
3
3
|
export declare function supportsToolUse(model: string, provider?: string | Providers, streaming?: boolean): boolean;
|
|
4
4
|
export declare function modelModalitiesToArray(modalities: ModelModalities): string[];
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"capability.d.ts","sourceRoot":"","sources":["../src/capability.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,KAAK,iBAAiB,EAAE,KAAK,eAAe,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAErF,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,iBAAiB,CAiBpG;AA6ED,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,EAAE,SAAS,GAAE,OAAe,GAAG,OAAO,CAGjH;AAED,wBAAgB,sBAAsB,CAAC,UAAU,EAAE,eAAe,GAAG,MAAM,EAAE,CAI5E"}
|
|
@@ -1,15 +1,15 @@
|
|
|
1
|
-
import { getModelCapabilitiesAnthropic } from
|
|
2
|
-
import { getModelCapabilitiesAzureFoundry } from
|
|
3
|
-
import { getModelCapabilitiesBedrock } from
|
|
4
|
-
import { getModelCapabilitiesOpenAI } from
|
|
5
|
-
import { getModelCapabilitiesVertexAI } from
|
|
6
|
-
import { Providers } from
|
|
1
|
+
import { getModelCapabilitiesAnthropic } from './capability/anthropic.js';
|
|
2
|
+
import { getModelCapabilitiesAzureFoundry } from './capability/azure_foundry.js';
|
|
3
|
+
import { getModelCapabilitiesBedrock } from './capability/bedrock.js';
|
|
4
|
+
import { getModelCapabilitiesOpenAI } from './capability/openai.js';
|
|
5
|
+
import { getModelCapabilitiesVertexAI } from './capability/vertexai.js';
|
|
6
|
+
import { Providers } from './types.js';
|
|
7
7
|
export function getModelCapabilities(model, provider) {
|
|
8
8
|
//Check for locations/<location>/ prefix and remove it
|
|
9
|
-
if (model.startsWith(
|
|
10
|
-
const parts = model.split(
|
|
9
|
+
if (model.startsWith('locations/')) {
|
|
10
|
+
const parts = model.split('/');
|
|
11
11
|
if (parts.length >= 3) {
|
|
12
|
-
model = parts.slice(2).join(
|
|
12
|
+
model = parts.slice(2).join('/');
|
|
13
13
|
}
|
|
14
14
|
}
|
|
15
15
|
const capabilities = _getModelCapabilities(model, provider);
|
|
@@ -45,32 +45,32 @@ function _getModelCapabilities(model, provider) {
|
|
|
45
45
|
case Providers.xai:
|
|
46
46
|
// xAI (Grok) models support tool use and are text-based
|
|
47
47
|
return {
|
|
48
|
-
input: { text: true, image: model.includes(
|
|
48
|
+
input: { text: true, image: model.includes('vision') },
|
|
49
49
|
output: { text: true },
|
|
50
50
|
tool_support: true,
|
|
51
51
|
tool_support_streaming: false, // Conservative - may work but not tested
|
|
52
52
|
};
|
|
53
53
|
default:
|
|
54
54
|
// Guess the provider based on the model name
|
|
55
|
-
if (model.startsWith(
|
|
55
|
+
if (model.startsWith('gpt')) {
|
|
56
56
|
return getModelCapabilitiesOpenAI(model);
|
|
57
57
|
}
|
|
58
|
-
else if (model.startsWith(
|
|
58
|
+
else if (model.startsWith('claude')) {
|
|
59
59
|
return getModelCapabilitiesAnthropic(model);
|
|
60
60
|
}
|
|
61
|
-
else if (model.startsWith(
|
|
61
|
+
else if (model.startsWith('grok')) {
|
|
62
62
|
// xAI Grok models
|
|
63
63
|
return {
|
|
64
|
-
input: { text: true, image: model.includes(
|
|
64
|
+
input: { text: true, image: model.includes('vision') },
|
|
65
65
|
output: { text: true },
|
|
66
66
|
tool_support: true,
|
|
67
67
|
tool_support_streaming: false,
|
|
68
68
|
};
|
|
69
69
|
}
|
|
70
|
-
else if (model.startsWith(
|
|
70
|
+
else if (model.startsWith('publishers/')) {
|
|
71
71
|
return getModelCapabilitiesVertexAI(model);
|
|
72
72
|
}
|
|
73
|
-
else if (model.startsWith(
|
|
73
|
+
else if (model.startsWith('arn:aws')) {
|
|
74
74
|
return getModelCapabilitiesBedrock(model);
|
|
75
75
|
}
|
|
76
76
|
// Fallback to a generic model with no capabilities
|
|
@@ -90,7 +90,7 @@ function getModelCapabilitiesOpenAICompatible(model) {
|
|
|
90
90
|
return caps;
|
|
91
91
|
}
|
|
92
92
|
const normalized = model.toLowerCase();
|
|
93
|
-
const isNonToolModel = NO_TOOL_SUPPORT_PATTERNS.some(p => normalized.includes(p));
|
|
93
|
+
const isNonToolModel = NO_TOOL_SUPPORT_PATTERNS.some((p) => normalized.includes(p));
|
|
94
94
|
return {
|
|
95
95
|
input: { text: true },
|
|
96
96
|
output: { text: true },
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"capability.js","sourceRoot":"","sources":["../src/capability.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EAAE,gCAAgC,EAAE,MAAM,+BAA+B,CAAC;AACjF,OAAO,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AACtE,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AACxE,OAAO,EAAgD,SAAS,EAAE,MAAM,YAAY,CAAC;AAErF,MAAM,UAAU,oBAAoB,CAAC,KAAa,EAAE,QAA6B;IAC7E,sDAAsD;IACtD,IAAI,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/B,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACpB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,CAAC;IACL,CAAC;IACD,MAAM,YAAY,GAAG,qBAAqB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC5D,gFAAgF;IAChF,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,8EAA8E;IAC9E,sEAAsE;IACtE,OAAO,YAAY,CAAC;AACxB,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAa,EAAE,QAA6B;IACvE,QAAQ,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC;QAC9B,KAAK,SAAS,CAAC,SAAS;YACpB,OAAO,6BAA6B,CAAC,KAAK,CAAC,CAAC;QAChD,KAAK,SAAS,CAAC,QAAQ;YACnB,OAAO,4BAA4B,CAAC,KAAK,CAAC,CAAC;QAC/C,KAAK,SAAS,CAAC,MAAM;YACjB,OAAO,0BAA0B,CAAC,KAAK,CAAC,CAAC;QAC7C,KAAK,SAAS,CAAC,iBAAiB;YAC5B,OAAO,oCAAoC,CAAC,KAAK,CAAC,CAAC;QACvD,KAAK,SAAS,CAAC,OAAO;YAClB,OAAO,2BAA2B,CAAC,KAAK,CAAC,CAAC;QAC9C,KAAK,SAAS,CAAC,aAAa;YACxB,yCAAyC;YACzC,OAAO,gCAAgC,CAAC,KAAK,CAAC,CAAC;QACnD,KAAK,SAAS,CAAC,IAAI,CAAC;QACpB,KAAK,SAAS,CAAC,UAAU,CAAC;QAC1B,KAAK,SAAS,CAAC,SAAS;YACpB,mEAAmE;YACnE,OAAO,oCAAoC,CAAC,KAAK,CAAC,CAAC;QACvD,KAAK,SAAS,CAAC,GAAG;YACd,wDAAwD;YACxD,OAAO;gBACH,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;gBACtD,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;gBACtB,YAAY,EAAE,IAAI;gBAClB,sBAAsB,EAAE,KAAK,EAAE,yCAAyC;aAC3E,CAAC;QACN;YACI,6CAA6C;YAC7C,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,0BAA0B,CAAC,KAAK,CAAC,CAAC;YAC7C,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACpC,OAAO,6BAA6B,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClC,kBAAkB;gBAClB,OAAO;oBACH,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;oBACtD,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;oBACtB,YAAY,EAAE,IAAI;oBAClB,sBAAsB,EAAE,KAAK;iBAChC,CAAC;YACN,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,4BAA4B,CAAC,KAAK,CAAC,CAAC;YAC/C,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,OAAO,2BAA2B,CAAC,KAAK,CAAC,CAAC;YAC9C,CAAC;YACD,mDAAmD;YACnD,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAA8B,CAAC;IACrE,CAAC;AACL,CAAC;AAED,mFAAmF;AACnF,MAAM,wBAAwB,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;AAEtG;;;;GAIG;AACH,SAAS,oCAAoC,CAAC,KAAa;IACvD,MAAM,IAAI,GAAG,0BAA0B,CAAC,KAAK,CAAC,CAAC;IAC/C,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IACvC,MAAM,cAAc,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACpF,OAAO;QACH,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;QACrB,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;QACtB,YAAY,EAAE,CAAC,cAAc;QAC7B,sBAAsB,EAAE,CAAC,cAAc;KAC1C,CAAC;AACN,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAa,EAAE,QAA6B,EAAE,YAAqB,KAAK;IACpG,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,MAAM,UAAU,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"}
|
package/lib/index.d.ts
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export * from './capability.js';
|
|
2
|
+
export * from './options/anthropic.js';
|
|
3
|
+
export * from './options/bedrock.js';
|
|
4
|
+
export * from './options/context-windows.js';
|
|
5
|
+
export * from './options/embedding.js';
|
|
6
|
+
export * from './options/fallback.js';
|
|
7
|
+
export * from './options/groq.js';
|
|
8
|
+
export * from './options/openai.js';
|
|
9
|
+
export * from './options/shared-parsing.js';
|
|
10
|
+
export * from './options/version-parsing.js';
|
|
11
|
+
export * from './options/vertexai.js';
|
|
12
|
+
export * from './options.js';
|
|
13
|
+
export * from './types.js';
|
|
14
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,uBAAuB,CAAC;AACtC,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,CAAC"}
|
package/lib/index.js
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export * from './capability.js';
|
|
2
|
+
export * from './options/anthropic.js';
|
|
3
|
+
export * from './options/bedrock.js';
|
|
4
|
+
export * from './options/context-windows.js';
|
|
5
|
+
export * from './options/embedding.js';
|
|
6
|
+
export * from './options/fallback.js';
|
|
7
|
+
export * from './options/groq.js';
|
|
8
|
+
export * from './options/openai.js';
|
|
9
|
+
export * from './options/shared-parsing.js';
|
|
10
|
+
export * from './options/version-parsing.js';
|
|
11
|
+
export * from './options/vertexai.js';
|
|
12
|
+
export * from './options.js';
|
|
13
|
+
export * from './types.js';
|
|
14
|
+
//# sourceMappingURL=index.js.map
|
package/lib/index.js.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,uBAAuB,CAAC;AACtC,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { type ModelOptions, type ModelOptionsInfo } from
|
|
1
|
+
import { type ModelOptions, type ModelOptionsInfo } from '../types.js';
|
|
2
2
|
export interface AnthropicClaudeOptions {
|
|
3
|
-
_option_id:
|
|
3
|
+
_option_id: 'anthropic-claude';
|
|
4
4
|
max_tokens?: number;
|
|
5
5
|
temperature?: number;
|
|
6
6
|
top_p?: number;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/options/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAE,KAAK,gBAAgB,EAA6B,MAAM,aAAa,CAAC;AAYlG,MAAM,WAAW,sBAAsB;IACnC,UAAU,EAAE,kBAAkB,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;IACrD,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;CAC3B;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,GAAG,gBAAgB,CAgC1F"}
|
|
@@ -1,23 +1,26 @@
|
|
|
1
|
-
import { OptionType, SharedOptions } from
|
|
2
|
-
import { textOptionsFallback } from
|
|
3
|
-
import { buildClaudeCacheOptions, buildClaudeCacheTtlOptions, buildClaudeEffortOptions, buildClaudeIncludeThoughtsOption, buildClaudeThinkingBudgetOption, getClaudeMaxTokensLimit, } from
|
|
4
|
-
import { hasSamplingParameterRestriction } from
|
|
1
|
+
import { OptionType, SharedOptions } from '../types.js';
|
|
2
|
+
import { textOptionsFallback } from './fallback.js';
|
|
3
|
+
import { buildClaudeCacheOptions, buildClaudeCacheTtlOptions, buildClaudeEffortOptions, buildClaudeIncludeThoughtsOption, buildClaudeThinkingBudgetOption, getClaudeMaxTokensLimit, } from './shared-parsing.js';
|
|
4
|
+
import { hasSamplingParameterRestriction } from './version-parsing.js';
|
|
5
5
|
export function getAnthropicOptions(model, option) {
|
|
6
6
|
const max_tokens_limit = getClaudeMaxTokensLimit(model);
|
|
7
|
-
const excludeOptions = [
|
|
7
|
+
const excludeOptions = ['max_tokens', 'presence_penalty', 'frequency_penalty'];
|
|
8
8
|
let commonOptions = textOptionsFallback.options.filter((o) => !excludeOptions.includes(o.name));
|
|
9
9
|
const hasSamplingRestriction = hasSamplingParameterRestriction(model);
|
|
10
10
|
if (hasSamplingRestriction) {
|
|
11
|
-
commonOptions = commonOptions.filter((o) => o.name !== SharedOptions.temperature &&
|
|
12
|
-
o.name !== SharedOptions.top_p &&
|
|
13
|
-
o.name !== "top_k");
|
|
11
|
+
commonOptions = commonOptions.filter((o) => o.name !== SharedOptions.temperature && o.name !== SharedOptions.top_p && o.name !== 'top_k');
|
|
14
12
|
}
|
|
15
13
|
return {
|
|
16
|
-
_option_id:
|
|
14
|
+
_option_id: 'anthropic-claude',
|
|
17
15
|
options: [
|
|
18
16
|
{
|
|
19
|
-
name: SharedOptions.max_tokens,
|
|
20
|
-
|
|
17
|
+
name: SharedOptions.max_tokens,
|
|
18
|
+
type: OptionType.numeric,
|
|
19
|
+
min: 1,
|
|
20
|
+
max: max_tokens_limit,
|
|
21
|
+
integer: true,
|
|
22
|
+
step: 200,
|
|
23
|
+
description: 'The maximum number of tokens to generate',
|
|
21
24
|
},
|
|
22
25
|
...commonOptions,
|
|
23
26
|
...buildClaudeEffortOptions(model),
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/options/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4C,UAAU,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAClG,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACH,uBAAuB,EACvB,0BAA0B,EAC1B,wBAAwB,EACxB,gCAAgC,EAChC,+BAA+B,EAC/B,uBAAuB,GAC1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,+BAA+B,EAAE,MAAM,sBAAsB,CAAC;AAgBvE,MAAM,UAAU,mBAAmB,CAAC,KAAa,EAAE,MAAqB;IACpE,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;IACxD,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,kBAAkB,EAAE,mBAAmB,CAAC,CAAC;IAC/E,IAAI,aAAa,GAAG,mBAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAEhG,MAAM,sBAAsB,GAAG,+BAA+B,CAAC,KAAK,CAAC,CAAC;IACtE,IAAI,sBAAsB,EAAE,CAAC;QACzB,aAAa,GAAG,aAAa,CAAC,MAAM,CAChC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,WAAW,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,CACtG,CAAC;IACN,CAAC;IAED,OAAO;QACH,UAAU,EAAE,kBAAkB;QAC9B,OAAO,EAAE;YACL;gBACI,IAAI,EAAE,aAAa,CAAC,UAAU;gBAC9B,IAAI,EAAE,UAAU,CAAC,OAAO;gBACxB,GAAG,EAAE,CAAC;gBACN,GAAG,EAAE,gBAAgB;gBACrB,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,GAAG;gBACT,WAAW,EAAE,0CAA0C;aAC1D;YACD,GAAG,aAAa;YAChB,GAAG,wBAAwB,CAAC,KAAK,CAAC;YAClC,GAAG,+BAA+B,CAAC,KAAK,CAAC;YACzC,GAAG,gCAAgC,CAAC,KAAK,CAAC;YAC1C,GAAG,uBAAuB,EAAE;YAC5B,GAAG,0BAA0B,CAAE,MAA4C,EAAE,aAAa,CAAC;SAC9F;KACJ,CAAC;AACN,CAAC"}
|
|
@@ -1,34 +1,34 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { type ModelOptions, type ModelOptionsInfo } from '../types.js';
|
|
2
2
|
export type AzureFoundryOptions = AzureFoundryOpenAIOptions | AzureFoundryDeepSeekOptions | AzureFoundryThinkingOptions | AzureFoundryTextOptions | AzureFoundryImageOptions;
|
|
3
3
|
export interface AzureFoundryOpenAIOptions {
|
|
4
|
-
_option_id:
|
|
4
|
+
_option_id: 'azure-foundry-openai';
|
|
5
5
|
max_tokens?: number;
|
|
6
6
|
temperature?: number;
|
|
7
7
|
top_p?: number;
|
|
8
8
|
presence_penalty?: number;
|
|
9
9
|
frequency_penalty?: number;
|
|
10
10
|
stop_sequence?: string[];
|
|
11
|
-
image_detail?:
|
|
12
|
-
reasoning_effort?:
|
|
11
|
+
image_detail?: 'low' | 'high' | 'auto';
|
|
12
|
+
reasoning_effort?: 'low' | 'medium' | 'high';
|
|
13
13
|
}
|
|
14
14
|
export interface AzureFoundryDeepSeekOptions {
|
|
15
|
-
_option_id:
|
|
15
|
+
_option_id: 'azure-foundry-deepseek';
|
|
16
16
|
max_tokens?: number;
|
|
17
17
|
temperature?: number;
|
|
18
18
|
top_p?: number;
|
|
19
19
|
stop_sequence?: string[];
|
|
20
20
|
}
|
|
21
21
|
export interface AzureFoundryThinkingOptions {
|
|
22
|
-
_option_id:
|
|
22
|
+
_option_id: 'azure-foundry-thinking';
|
|
23
23
|
max_tokens?: number;
|
|
24
24
|
temperature?: number;
|
|
25
25
|
top_p?: number;
|
|
26
26
|
stop_sequence?: string[];
|
|
27
|
-
reasoning_effort?:
|
|
28
|
-
image_detail?:
|
|
27
|
+
reasoning_effort?: 'low' | 'medium' | 'high';
|
|
28
|
+
image_detail?: 'low' | 'high' | 'auto';
|
|
29
29
|
}
|
|
30
30
|
export interface AzureFoundryTextOptions {
|
|
31
|
-
_option_id:
|
|
31
|
+
_option_id: 'azure-foundry-text';
|
|
32
32
|
max_tokens?: number;
|
|
33
33
|
temperature?: number;
|
|
34
34
|
top_p?: number;
|
|
@@ -39,13 +39,13 @@ export interface AzureFoundryTextOptions {
|
|
|
39
39
|
seed?: number;
|
|
40
40
|
}
|
|
41
41
|
export interface AzureFoundryImageOptions {
|
|
42
|
-
_option_id:
|
|
42
|
+
_option_id: 'azure-foundry-image';
|
|
43
43
|
width?: number;
|
|
44
44
|
height?: number;
|
|
45
|
-
quality?:
|
|
46
|
-
style?:
|
|
47
|
-
response_format?:
|
|
48
|
-
size?:
|
|
45
|
+
quality?: 'standard' | 'hd';
|
|
46
|
+
style?: 'vivid' | 'natural';
|
|
47
|
+
response_format?: 'url' | 'b64_json';
|
|
48
|
+
size?: '256x256' | '512x512' | '1024x1024' | '1792x1024' | '1024x1792';
|
|
49
49
|
}
|
|
50
50
|
export declare function getMaxTokensLimitAzureFoundry(model: string): number | undefined;
|
|
51
51
|
export declare function getAzureFoundryOptions(model: string, _option?: ModelOptions): ModelOptionsInfo;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"azure_foundry.d.ts","sourceRoot":"","sources":["../../src/options/azure_foundry.ts"],"names":[],"mappings":"AAAA,OAAO,EAEH,KAAK,YAAY,EACjB,KAAK,gBAAgB,EAGxB,MAAM,aAAa,CAAC;AAqBrB,MAAM,MAAM,mBAAmB,GACzB,yBAAyB,GACzB,2BAA2B,GAC3B,2BAA2B,GAC3B,uBAAuB,GACvB,wBAAwB,CAAC;AAE/B,MAAM,WAAW,yBAAyB;IACtC,UAAU,EAAE,sBAAsB,CAAC;IACnC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;IACvC,gBAAgB,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;CAChD;AAED,MAAM,WAAW,2BAA2B;IACxC,UAAU,EAAE,wBAAwB,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,2BAA2B;IACxC,UAAU,EAAE,wBAAwB,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;CAC1C;AAED,MAAM,WAAW,uBAAuB;IACpC,UAAU,EAAE,oBAAoB,CAAC;IACjC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,wBAAwB;IACrC,UAAU,EAAE,qBAAqB,CAAC;IAClC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;IAC5B,KAAK,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC5B,eAAe,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC;IACrC,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,WAAW,GAAG,WAAW,CAAC;CAC1E;AAED,wBAAgB,6BAA6B,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAkG/E;AAED,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,gBAAgB,CAmS9F"}
|