mirror of
https://github.com/Mintplex-Labs/anything-llm.git
synced 2024-11-14 02:20:12 +01:00
10 lines
316 B
Python
10 lines
316 B
Python
|
import tiktoken
|
||
|
|
encoder = tiktoken.encoding_for_model("text-embedding-ada-002")
|
||
|
|
|
||
|
|
def tokenize(fullText):
|
||
|
|
return encoder.encode(fullText)
|
||
|
|
|
||
|
|
def ada_v2_cost(tokenCount):
|
||
|
|
rate_per = 0.0004 / 1_000 # $0.0004 / 1K tokens
|
||
|
|
total = tokenCount * rate_per
|
||
|
|
return '${:,.2f}'.format(total) if total >= 0.01 else '< $0.01'
|