@unikode/cli 1.0.20 → 1.0.21

package/dist/index.js

@@ -3078,9 +3078,11 @@ function SessionShell({
   onSubmit,
   inputDisabled = false,
   loading = false,
-  interruptible = false
+  interruptible = false,
+  tokenCount
 }) {
   const { mode } = usePromptConfig();
+  const { colors } = useTheme();
   return /* @__PURE__ */ jsxDEV21("box", {
     flexDirection: "column",
     flexGrow: 1,
@@ -3099,6 +3101,37 @@ function SessionShell({
           children
         }, undefined, false, undefined, this)
       }, undefined, false, undefined, this),
+      tokenCount !== undefined && /* @__PURE__ */ jsxDEV21("box", {
+        flexShrink: 0,
+        flexDirection: "row",
+        justifyContent: "flex-end",
+        alignItems: "center",
+        width: "100%",
+        paddingX: 1,
+        children: /* @__PURE__ */ jsxDEV21("box", {
+          flexDirection: "row",
+          alignItems: "center",
+          gap: 1,
+          paddingX: 1,
+          backgroundColor: colors.surface,
+          children: [
+            /* @__PURE__ */ jsxDEV21("text", {
+              fg: mode === Mode.PLAN ? colors.planMode : colors.primary,
+              children: "\u26A1"
+            }, undefined, false, undefined, this),
+            /* @__PURE__ */ jsxDEV21("text", {
+              attributes: TextAttributes8.DIM,
+              fg: colors.info,
+              children: "Tokens:"
+            }, undefined, false, undefined, this),
+            /* @__PURE__ */ jsxDEV21("text", {
+              fg: mode === Mode.PLAN ? colors.planMode : colors.primary,
+              attributes: TextAttributes8.BOLD,
+              children: tokenCount.toLocaleString()
+            }, undefined, false, undefined, this)
+          ]
+        }, undefined, true, undefined, this)
+      }, undefined, false, undefined, this),
       /* @__PURE__ */ jsxDEV21("box", {
         flexShrink: 0,
         children: /* @__PURE__ */ jsxDEV21(InputBar, {
@@ -4406,7 +4439,24 @@ function SessionChat({ session, initialPrompt }) {
       model: initialPrompt.model
     });
   }, [initialPrompt, submit]);
+  const totalTokenCount = useMemo6(() => {
+    let initialUsageSum = 0;
+    let currentUsageSum = 0;
+    for (let i = 0;i < messages.length; i++) {
+      const msg = messages[i];
+      const metadata = msg?.metadata;
+      const tokens = metadata?.usage?.totalTokens ?? 0;
+      currentUsageSum += tokens;
+      if (i < initialMessages.length) {
+        initialUsageSum += tokens;
+      }
+    }
+    const baseTokens = Math.max(session.tokenCount ?? 0, initialUsageSum);
+    const newTokens = currentUsageSum - initialUsageSum;
+    return baseTokens + newTokens;
+  }, [messages, initialMessages.length, session.tokenCount]);
   return /* @__PURE__ */ jsxDEV28(SessionShell, {
+    tokenCount: totalTokenCount,
     onSubmit: (text) => {
       deactivateQuestionPrompts();
       submit({ userText: text, mode, model });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@unikode/cli",
-  "version": "1.0.20",
+  "version": "1.0.21",
   "type": "module",
   "main": "./dist/index.js",
   "bin": {