2023-01-10 14:54:50 +08:00
|
|
|
# 计费模块
|
|
|
|
|
# 已弃用 https://github.com/RockChinQ/QChatGPT/issues/81
|
|
|
|
|
|
2023-01-02 20:42:08 +08:00
|
|
|
import logging
|
|
|
|
|
|
2022-12-28 00:05:25 +08:00
|
|
|
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
|
2023-01-02 20:42:08 +08:00
|
|
|
logging.debug("text length: %d" % length)
|
2022-12-28 00:05:25 +08:00
|
|
|
|
|
|
|
|
return pricing["base"][model] * length / 1000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def image_price(size):
|
2023-01-02 20:42:08 +08:00
|
|
|
logging.debug("image size: %s" % size)
|
2022-12-28 00:05:25 +08:00
|
|
|
return pricing["image"][size]
|
