blun-king-cli 2.6.0 → 2.6.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/bin/blun.js +12 -2
- package/package.json +1 -1
package/bin/blun.js
CHANGED
|
@@ -495,9 +495,17 @@ async function sendChat(message) {
|
|
|
495
495
|
chatHistory.push({ role: "user", content: message });
|
|
496
496
|
chatHistory.push({ role: "assistant", content: resp.data.answer });
|
|
497
497
|
|
|
498
|
+
// Token tracking
|
|
499
|
+
var inTok = Math.ceil(message.length / 3.5);
|
|
500
|
+
var outTok = Math.ceil((resp.data.answer || "").length / 3.5);
|
|
501
|
+
sessionCost.requests++;
|
|
502
|
+
sessionCost.inputTokensEst += inTok;
|
|
503
|
+
sessionCost.outputTokensEst += outTok;
|
|
504
|
+
|
|
498
505
|
var q = resp.data.quality || {};
|
|
506
|
+
var tokInfo = C.dim + (inTok + outTok) + " tok" + C.reset;
|
|
499
507
|
var meta = resp.data.task_type + "/" + resp.data.role + " " + BOX.dot + " score: " + (q.score || "?") +
|
|
500
|
-
(q.retried ? " " + BOX.dot + " retried" : "");
|
|
508
|
+
(q.retried ? " " + BOX.dot + " retried" : "") + " " + BOX.dot + " " + tokInfo;
|
|
501
509
|
printAnswer(resp.data.answer, meta);
|
|
502
510
|
|
|
503
511
|
// Auto Memory Dream Mode
|
|
@@ -2442,7 +2450,9 @@ async function main() {
|
|
|
2442
2450
|
var permColor = isDangerous ? C.red + C.bold : permMode === "deny" ? C.red : C.yellow;
|
|
2443
2451
|
var modelName = typeof config.model === "string" ? config.model : (config.model && config.model.name ? config.model.name : "default");
|
|
2444
2452
|
var wdShort = config.workdir ? path.basename(config.workdir) : "~";
|
|
2445
|
-
|
|
2453
|
+
var totalTok = sessionCost.inputTokensEst + sessionCost.outputTokensEst;
|
|
2454
|
+
var tokStr = totalTok > 1000 ? (totalTok / 1000).toFixed(1) + "k" : String(totalTok);
|
|
2455
|
+
lines.push(permColor + " " + permIcon + " " + permLabel + C.reset + C.dim + " \u2502 " + C.reset + C.cyan + modelName + C.reset + C.dim + " \u2502 " + C.reset + C.dim + wdShort + C.reset + C.dim + " \u2502 " + C.reset + C.yellow + tokStr + " tok" + C.reset);
|
|
2446
2456
|
|
|
2447
2457
|
// Menu
|
|
2448
2458
|
if (menuVisible && menuItems.length > 0) {
|