* Initialize tokenizer for the current model * Should be called when model changes or on first stream
(model: string, metadataModelOverride?: string)
| 19 | * Should be called when model changes or on first stream |
| 20 | */ |
| 21 | async setModel(model: string, metadataModelOverride?: string): Promise<void> { |
| 22 | const tokenizerModelKey = metadataModelOverride ?? model; |
| 23 | if (this.tokenizer && this.tokenizerModelKey === tokenizerModelKey) { |
| 24 | return; |
| 25 | } |
| 26 | |
| 27 | this.tokenizer = await getTokenizerForModel(model, metadataModelOverride); |
| 28 | this.tokenizerModelKey = tokenizerModelKey; |
| 29 | } |
| 30 | |
| 31 | /** |
| 32 | * Count tokens in a text string synchronously |
no test coverage detected