返回提交历史
Modified
g4f/gui/client/static/js/chat.v1.js
+6
-6
XFEstudio/gpt4free
Fix track usage
ad59df30
代码差异
1 个文件
+6
-6
@@ -941,17 +941,12 @@ const ask_gpt = async (message_id, message_index = -1, regenerate = false, provi
941
941
usage = usage_storage[message_id];
942
942
delete usage_storage[message_id];
943
943
}
944
usage = {
945
model: message_provider?.model,
946
provider: message_provider?.name,
947
...usage
948
}
949
944
// Calculate usage if we don't have it jet
950
945
if (countTokensEnabled && document.getElementById("track_usage").checked && !usage.prompt_tokens && window.GPTTokenizer_cl100k_base) {
951
946
const prompt_token_model = model?.startsWith("gpt-3") ? "gpt-3.5-turbo" : "gpt-4"
952
947
const prompt_tokens = GPTTokenizer_cl100k_base?.encodeChat(messages, prompt_token_model).length;
953
948
const completion_tokens = count_tokens(message_provider?.model, message_storage[message_id])
954
+ reasoning_storage[message_id] ? count_tokens(message_provider?.model, reasoning_storage[message_id].text) : 0;
949
+ (reasoning_storage[message_id] ? count_tokens(message_provider?.model, reasoning_storage[message_id].text) : 0);
955
950
usage = {
956
951
...usage,
957
952
prompt_tokens: prompt_tokens,
@@ -986,6 +981,11 @@ const ask_gpt = async (message_id, message_index = -1, regenerate = false, provi
986
981
delete message_storage[message_id];
987
982
// Send usage to the server
988
983
if (document.getElementById("track_usage").checked) {
984
usage = {
985
model: message_provider?.model,
986
provider: message_provider?.name,
987
...usage
988
};
989
989
const user = localStorage.getItem("user");
990
990
if (user) {
991
991
usage = {user: user, ...usage};