ai-speedometer 2.1.7 → 2.2.0

package/dist/ai-speedometer

@@ -1821,16 +1821,17 @@ async function benchmarkSingleModelRest(model, logger) {
           continue;
         try {
           if (model.providerType === "anthropic") {
-            if (trimmedLine.startsWith("data: ")) {
-              const jsonStr = trimmedLine.slice(6);
+            const anthropicDataPrefix = trimmedLine.startsWith("data: ") ? 6 : trimmedLine.startsWith("data:") ? 5 : -1;
+            if (anthropicDataPrefix !== -1) {
+              const jsonStr = trimmedLine.slice(anthropicDataPrefix);
               if (jsonStr === "[DONE]")
                 continue;
               const chunk = JSON.parse(jsonStr);
               const chunkTyped = chunk;
-              if (chunkTyped.type === "content_block_delta" && chunkTyped.delta?.text) {
+              if (chunkTyped.type === "content_block_delta" && (chunkTyped.delta?.text || chunkTyped.delta?.thinking)) {
                 if (!firstParsedTokenTime)
                   firstParsedTokenTime = Date.now();
-                streamedText += chunkTyped.delta.text;
+                streamedText += chunkTyped.delta?.text || chunkTyped.delta?.thinking || "";
               } else if (chunkTyped.type === "message_start" && chunkTyped.message?.usage) {
                 inputTokens = chunkTyped.message.usage.input_tokens || 0;
               } else if (chunkTyped.type === "message_delta") {
@@ -1843,10 +1844,10 @@ async function benchmarkSingleModelRest(model, logger) {
               continue;
             } else {
               const chunk = JSON.parse(trimmedLine);
-              if (chunk.type === "content_block_delta" && chunk.delta?.text) {
+              if (chunk.type === "content_block_delta" && (chunk.delta?.text || chunk.delta?.thinking)) {
                 if (!firstParsedTokenTime)
                   firstParsedTokenTime = Date.now();
-                streamedText += chunk.delta.text;
+                streamedText += chunk.delta?.text || chunk.delta?.thinking || "";
               } else if (chunk.type === "message_start" && chunk.message?.usage) {
                 inputTokens = chunk.message.usage.input_tokens || 0;
               } else if (chunk.type === "message_delta") {
@@ -1868,24 +1869,42 @@ async function benchmarkSingleModelRest(model, logger) {
             if (chunk.usageMetadata?.candidatesTokenCount)
               outputTokens = chunk.usageMetadata.candidatesTokenCount;
           } else {
-            if (trimmedLine.startsWith("data: ")) {
-              const jsonStr = trimmedLine.slice(6);
-              if (jsonStr === "[DONE]")
-                continue;
-              const chunk = JSON.parse(jsonStr);
-              if (chunk.choices?.[0]?.delta?.content) {
-                if (!firstParsedTokenTime)
-                  firstParsedTokenTime = Date.now();
-                streamedText += chunk.choices[0].delta.content;
-              } else if (chunk.choices?.[0]?.delta?.reasoning) {
-                if (!firstParsedTokenTime)
-                  firstParsedTokenTime = Date.now();
-                streamedText += chunk.choices[0].delta.reasoning;
-              }
-              if (chunk.usage?.prompt_tokens)
-                inputTokens = chunk.usage.prompt_tokens;
-              if (chunk.usage?.completion_tokens)
-                outputTokens = chunk.usage.completion_tokens;
+            const dataPrefix = trimmedLine.startsWith("data: ") ? 6 : trimmedLine.startsWith("data:") ? 5 : -1;
+            if (dataPrefix === -1)
+              continue;
+            const jsonStr = trimmedLine.slice(dataPrefix);
+            if (jsonStr === "[DONE]")
+              continue;
+            const chunk = JSON.parse(jsonStr);
+            if (chunk.choices?.[0]?.delta?.content) {
+              if (!firstParsedTokenTime)
+                firstParsedTokenTime = Date.now();
+              streamedText += chunk.choices[0].delta.content;
+            } else if (chunk.choices?.[0]?.delta?.reasoning) {
+              if (!firstParsedTokenTime)
+                firstParsedTokenTime = Date.now();
+              streamedText += chunk.choices[0].delta.reasoning;
+            } else if (chunk.choices?.[0]?.delta?.reasoning_content) {
+              if (!firstParsedTokenTime)
+                firstParsedTokenTime = Date.now();
+              streamedText += chunk.choices[0].delta.reasoning_content;
+            } else if (chunk.type === "content_block_delta" && chunk.delta?.text) {
+              if (!firstParsedTokenTime)
+                firstParsedTokenTime = Date.now();
+              streamedText += chunk.delta.text;
+            } else if (chunk.type === "content_block_delta" && chunk.delta?.thinking) {
+              if (!firstParsedTokenTime)
+                firstParsedTokenTime = Date.now();
+              streamedText += chunk.delta.thinking;
+            }
+            if (chunk.usage?.prompt_tokens)
+              inputTokens = chunk.usage.prompt_tokens;
+            if (chunk.usage?.completion_tokens)
+              outputTokens = chunk.usage.completion_tokens;
+            if (chunk.type === "message_start" && chunk.message?.usage?.input_tokens)
+              inputTokens = chunk.message.usage.input_tokens;
+            if (chunk.type === "message_delta" && chunk.usage?.output_tokens) {
+              outputTokens = chunk.usage.output_tokens;
             }
           }
         } catch {
@@ -2269,7 +2288,7 @@ var package_default;
 var init_package = __esm(() => {
   package_default = {
     name: "ai-speedometer",
-    version: "2.1.7",
+    version: "2.2.0",
     description: "A comprehensive CLI tool for benchmarking AI models across multiple providers with parallel execution and professional metrics",
     bin: {
       "ai-speedometer": "dist/ai-speedometer",
@@ -3059,6 +3078,7 @@ var init_ResultsTable = () => {};
 // src/tui/screens/BenchmarkScreen.tsx
 import { useState as useState4, useEffect as useEffect4, useRef as useRef3, useMemo } from "react";
 import { useKeyboard as useKeyboard5 } from "@opentui/react";
+import { engine, Timeline } from "@opentui/core";
 import { jsxDEV as jsxDEV10 } from "@opentui/react/jsx-dev-runtime";
 function rankBadge(rank) {
   if (rank === 1)
@@ -3080,9 +3100,23 @@ function BenchmarkScreen() {
   const navigate = useNavigate();
   const [modelStates, setModelStates] = useState4([]);
   const [spinnerFrame, setSpinnerFrame] = useState4(0);
+  const [shimmerPos, setShimmerPos] = useState4(0);
   const [allDone, setAllDone] = useState4(false);
   const spinnerRef = useRef3(null);
   const startedRef = useRef3(false);
+  useEffect4(() => {
+    const tl = new Timeline({ loop: true, duration: 1400 });
+    const target = { pos: 0 };
+    tl.add(target, { pos: 1, duration: 1400, ease: "inOutSine", onUpdate: (anim) => {
+      setShimmerPos(anim.targets[0].pos);
+    } }, 0);
+    tl.play();
+    engine.register(tl);
+    return () => {
+      tl.pause();
+      engine.unregister(tl);
+    };
+  }, []);
   useEffect4(() => {
     if (startedRef.current)
       return;
@@ -3151,6 +3185,27 @@ function BenchmarkScreen() {
       const total = modelStates.length || 1;
       const filled = Math.round((done.length + errors.length) / total * BAR_W);
       const empty = BAR_W - filled;
+      const shimmerCenter = shimmerPos * (BAR_W - 1);
+      const shimmerHalf = 1.5;
+      const barChars = Array.from({ length: BAR_W }, (_, i) => {
+        const isFilled = i < filled;
+        const dist = Math.abs(i - shimmerCenter);
+        const inWindow = dist <= shimmerHalf;
+        const intensity = inWindow ? 1 - dist / shimmerHalf : 0;
+        if (isFilled) {
+          if (intensity > 0.6)
+            return { ch: "\u2588", fg: "#c8eeff" };
+          if (intensity > 0)
+            return { ch: "\u2593", fg: "#a0d8f0" };
+          return { ch: "\u2588", fg: "#4a9abb" };
+        } else {
+          if (intensity > 0.6)
+            return { ch: "\u2591", fg: "#2a3a52" };
+          if (intensity > 0)
+            return { ch: "\u2591", fg: "#222d40" };
+          return { ch: "\u2591", fg: "#1a2030" };
+        }
+      });
       rows.push(/* @__PURE__ */ jsxDEV10("box", {
         height: 1,
         flexDirection: "row",
@@ -3169,14 +3224,10 @@ function BenchmarkScreen() {
               "  "
             ]
           }, undefined, true, undefined, this),
-          /* @__PURE__ */ jsxDEV10("text", {
-            fg: "#7dcfff",
-            children: "\u2588".repeat(filled)
-          }, undefined, false, undefined, this),
-          /* @__PURE__ */ jsxDEV10("text", {
-            fg: "#292e42",
-            children: "\u2591".repeat(empty)
-          }, undefined, false, undefined, this),
+          barChars.map((b, i) => /* @__PURE__ */ jsxDEV10("text", {
+            fg: b.fg,
+            children: b.ch
+          }, i, false, undefined, this)),
           /* @__PURE__ */ jsxDEV10("text", {
             fg: "#ff9e64",
             children: [
@@ -3512,7 +3563,7 @@ function BenchmarkScreen() {
       }, "results-empty", false, undefined, this));
     }
     return rows;
-  }, [modelStates, allDone, tpsRanked, ttftRanked, doneResults, pendingCount, maxTps, maxTtftForBar]);
+  }, [modelStates, allDone, shimmerPos, tpsRanked, ttftRanked, doneResults, pendingCount, maxTps, maxTtftForBar]);
   useKeyboard5((key) => {
     if (!allDone)
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-speedometer",
-  "version": "2.1.7",
+  "version": "2.2.0",
   "description": "A comprehensive CLI tool for benchmarking AI models across multiple providers with parallel execution and professional metrics",
   "bin": {
     "ai-speedometer": "dist/ai-speedometer",