feat: use js-tiktoken/lite to reduce module size

pull/610/head
LiYang 2023-09-04 14:23:48 +08:00
rodzic 73fb30bc11
commit f5c876a1fa
1 zmienionych plików z 6 dodań i 5 usunięć

Wyświetl plik

@ -1,8 +1,9 @@
import { getEncoding } from 'js-tiktoken'
// TODO: make this configurable
const tokenizer = getEncoding('cl100k_base')
import { Tiktoken } from 'js-tiktoken/lite'
import cl100k_base from 'js-tiktoken/ranks/cl100k_base'
export function encode(input: string): Uint32Array {
return new Uint32Array(tokenizer.encode(input))
// TODO: make this configurable
const encoding = new Tiktoken(cl100k_base)
return new Uint32Array(encoding.encode(input))
}