You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
gcgj-dify-1.7.0/api/tests/integration_tests/model_runtime/test_tiktoken.py

12 lines
320 B
Python

import os
import tiktoken
from core.model_runtime.model_providers.__base.tokenizers.gpt2_tokenzier import GPT2Tokenizer
def test_tiktoken():
os.environ["TIKTOKEN_CACHE_DIR"] = "/tmp/.tiktoken_cache"
GPT2Tokenizer.get_num_tokens("Hello, world!")
assert tiktoken.registry.ENCODING_CONSTRUCTORS is not None