From 9309cc8655a49f834cf9795272aa835f278fe315 Mon Sep 17 00:00:00 2001
From: musistudio <m@musiiot.top>
Date: Fri, 2 Jan 2026 21:49:25 +0800
Subject: [PATCH] fix token-speed plugin

---
 packages/cli/src/utils/statusline.ts | 26 ++++----------------------
 1 file changed, 4 insertions(+), 22 deletions(-)

diff --git a/packages/cli/src/utils/statusline.ts b/packages/cli/src/utils/statusline.ts
index a410925..cc2c434 100644
--- a/packages/cli/src/utils/statusline.ts
+++ b/packages/cli/src/utils/statusline.ts
@@ -441,31 +441,13 @@ async function getTokenSpeedStats(sessionId: string): Promise<{
             return null;
         }
 
-        // List all files in temp directory
-        const files = await fs.readdir(tempDir);
-
-        // Find files matching pattern: session-{sessionId}-{timestamp}.json
-        const pattern = new RegExp(`^session-${sessionId}-(\\d+)\\.json$`);
-        const matchingFiles = files
-            .map(file => {
-                const match = file.match(pattern);
-                if (!match) return null;
-                return {
-                    file,
-                    timestamp: parseInt(match[1])
-                };
-            })
-            .filter(Boolean) as Array<{ file: string; timestamp: number }>;
-
-        if (matchingFiles.length === 0) {
+        const statsFilePath = path.join(tempDir, `session-${sessionId}.json`);
+        try {
+            await fs.access(statsFilePath);
+        } catch {
             return null;
         }
 
-        // Sort by timestamp descending and get the most recent file
-        matchingFiles.sort((a, b) => b.timestamp - a.timestamp);
-        const latestFile = matchingFiles[0];
-        const statsFilePath = path.join(tempDir, latestFile.file);
-
         // Read stats file
         const content = await fs.readFile(statsFilePath, 'utf-8');
         const data = JSON.parse(content);