重构绘图相关

This commit is contained in:
sijinhui
2024-02-22 00:21:36 +08:00
parent 7c3c19e965
commit 747bed9826
11 changed files with 116 additions and 648 deletions

View File

@@ -31,13 +31,18 @@ async function handle(
const regex_message = /(?<="content":")(.*?)(?="}[,\]])/g;
const matchAllMessage = request_data.logEntry.match(regex_message);
// console.log(matchAllMessage, "=====");
if (matchAllMessage.length > 0) {
if (matchAllMessage && matchAllMessage.length > 0) {
request_data.logToken =
getTokenLength(matchAllMessage.join(" ")) +
matchAllMessage.length * 3;
} else {
request_data.logToken = 0;
}
delete request_data["logEntry"];
}
if (request_data?.model == "midjourney") {
request_data.logToken = 1000;
}
} catch (e) {
console.log("[LOG]", "logToken", e);
request_data.logToken = 0;