@yishiguji/tokenarena 0.8.4 → 0.8.6

package/dist/index.js

@@ -2120,9 +2120,14 @@ import { homedir as homedir12 } from "os";
 import { join as join13 } from "path";
 var TOOL_ID8 = "kimi-code";
 var TOOL_NAME8 = "Kimi Code";
-var DEFAULT_SESSIONS_DIR3 = join13(homedir12(), ".kimi", "sessions");
-var DEFAULT_CONFIG_PATH = join13(homedir12(), ".kimi", "kimi.json");
-var USER_EVENT_TYPES = /* @__PURE__ */ new Set(["UserMessage", "user_message", "Input"]);
+var DEFAULT_SESSIONS_DIR3 = join13(homedir12(), ".kimi-code", "sessions");
+var DEFAULT_CONFIG_PATH = join13(homedir12(), ".kimi-code", "workspaces.json");
+var USER_EVENT_TYPES = /* @__PURE__ */ new Set([
+  "UserMessage",
+  "user_message",
+  "Input",
+  "turn.prompt"
+]);
 var ASSISTANT_EVENT_TYPES = /* @__PURE__ */ new Set([
   "AssistantMessage",
   "assistant_message",
@@ -2158,9 +2163,23 @@ function findWireFiles(baseDir) {
           withFileTypes: true
         })) {
           if (!session.isDirectory()) continue;
-          const wireFile = join13(workDirPath, session.name, "wire.jsonl");
-          if (existsSync13(wireFile)) {
-            results.push({ filePath: wireFile, workDirHash: workDir.name });
+          const newWireFile = join13(
+            workDirPath,
+            session.name,
+            "agents",
+            "main",
+            "wire.jsonl"
+          );
+          if (existsSync13(newWireFile)) {
+            results.push({ filePath: newWireFile, workDirHash: workDir.name });
+            continue;
+          }
+          const legacyWireFile = join13(workDirPath, session.name, "wire.jsonl");
+          if (existsSync13(legacyWireFile)) {
+            results.push({
+              filePath: legacyWireFile,
+              workDirHash: workDir.name
+            });
           }
         }
       } catch {
@@ -2199,9 +2218,15 @@ function loadProjectMap(configPath) {
     const config = JSON.parse(content);
     const workspaces = config.workspaces || config.projects || {};
     for (const [hash, info] of Object.entries(workspaces)) {
-      const pathValue = typeof info === "string" ? info : info.path || info.dir || void 0;
+      let pathValue;
+      if (typeof info === "string") {
+        pathValue = info;
+      } else {
+        pathValue = info.root || info.path || info.dir || void 0;
+      }
       if (!pathValue) continue;
-      projectMap.set(hash, getPathLeaf5(pathValue));
+      const name = typeof info === "object" && info.name ? info.name : getPathLeaf5(pathValue);
+      projectMap.set(hash, name);
     }
   } catch {
   }
@@ -2240,6 +2265,42 @@ var KimiCodeParser = class {
         } catch {
           continue;
         }
+        if (obj.type === "usage.record") {
+          const timestampValue2 = obj.time;
+          const timestamp2 = timestampValue2 ? new Date(timestampValue2) : null;
+          if (!timestamp2 || Number.isNaN(timestamp2.getTime())) continue;
+          const usage = obj.usage;
+          if (!usage) continue;
+          const inputTokens2 = toSafeNumber7(
+            usage.input_other ?? usage.inputOther
+          );
+          const outputTokens2 = toSafeNumber7(usage.output);
+          const cachedTokens2 = toSafeNumber7(
+            usage.input_cache_read ?? usage.inputCacheRead
+          );
+          if (inputTokens2 === 0 && outputTokens2 === 0 && cachedTokens2 === 0) {
+            continue;
+          }
+          sessionEvents.push({
+            sessionId,
+            source: TOOL_ID8,
+            project,
+            timestamp: timestamp2,
+            role: "assistant"
+          });
+          entries.push({
+            sessionId,
+            source: TOOL_ID8,
+            model: obj.model || currentModel,
+            project,
+            timestamp: timestamp2,
+            inputTokens: inputTokens2,
+            outputTokens: outputTokens2,
+            reasoningTokens: 0,
+            cachedTokens: cachedTokens2
+          });
+          continue;
+        }
         const payload = obj.payload;
         if (!payload) continue;
         if (payload.model) {