mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
29 lines
639 B
Python
29 lines
639 B
Python
# 计费模块
|
|
# 已弃用 https://github.com/RockChinQ/QChatGPT/issues/81
|
|
|
|
import logging
|
|
|
|
pricing = {
|
|
"base": { # 文字模型单位是1000字符
|
|
"text-davinci-003": 0.02,
|
|
},
|
|
"image": {
|
|
"256x256": 0.016,
|
|
"512x512": 0.018,
|
|
"1024x1024": 0.02,
|
|
}
|
|
}
|
|
|
|
|
|
def language_base_price(model, text):
|
|
salt_rate = 0.93
|
|
length = ((len(text.encode('utf-8')) - len(text)) / 2 + len(text)) * salt_rate
|
|
logging.debug("text length: %d" % length)
|
|
|
|
return pricing["base"][model] * length / 1000
|
|
|
|
|
|
def image_price(size):
|
|
logging.debug("image size: %s" % size)
|
|
return pricing["image"][size]
|