( modelsToWarm: string[] = Array.from(DEFAULT_WARM_MODELS) )
| 178 | } |
| 179 | |
| 180 | export function loadTokenizerModules( |
| 181 | modelsToWarm: string[] = Array.from(DEFAULT_WARM_MODELS) |
| 182 | ): Promise<Array<PromiseSettledResult<string>>> { |
| 183 | if (shouldUseApproxTokenizer()) { |
| 184 | const fulfilled: Array<PromiseFulfilledResult<string>> = modelsToWarm.map(() => ({ |
| 185 | status: "fulfilled", |
| 186 | value: APPROX_ENCODING, |
| 187 | })); |
| 188 | return Promise.resolve(fulfilled); |
| 189 | } |
| 190 | |
| 191 | return Promise.allSettled( |
| 192 | modelsToWarm.map((modelString) => { |
| 193 | const modelName = normalizeModelKey(modelString); |
| 194 | // Skip unknown models during warmup |
| 195 | if (!modelName) { |
| 196 | return Promise.reject(new Error(`Unknown model: ${modelString}`)); |
| 197 | } |
| 198 | return resolveEncoding(modelName); |
| 199 | }) |
| 200 | ); |
| 201 | } |
| 202 | |
| 203 | export async function getTokenizerForModel( |
| 204 | modelString: string, |
no test coverage detected