@firstpick/pi-extension-stats 0.1.5 → 0.1.6

package/index.ts

@@ -2,6 +2,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 import { buildSessionContext, formatSkillsForPrompt } from "@earendil-works/pi-coding-agent";
 import type { BuildSystemPromptOptions, ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { estimatePromptInjectionTokens, estimateTokensFromCharCount, formatTokens } from "@firstpick/pi-utils";
 
 type DayUsage = {
   input: number;
@@ -44,24 +45,6 @@ const DEFAULT_DAYS = 14;
 const MAX_BAR_WIDTH = 24;
 const COST_BAR_WIDTH = 10;
 
-function formatTokens(count: number): string {
-  if (count < 1000) return count.toString();
-  if (count < 10000) return `${(count / 1000).toFixed(1)}k`;
-  if (count < 1000000) return `${Math.round(count / 1000)}k`;
-  if (count < 10000000) return `${(count / 1000000).toFixed(1)}M`;
-  return `${Math.round(count / 1000000)}M`;
-}
-
-function estimateTokensFromCharCount(charCount: number): number {
-  // Keep this estimate intentionally identical to pi-extension-git-footer-status.
-  // Provider tokenizers differ, so chars/4 is the shared rough display estimate.
-  return Math.max(0, Math.round(charCount / 4));
-}
-
-function estimatePromptInjectionTokens(systemPrompt: string): number {
-  return estimateTokensFromCharCount(systemPrompt.length);
-}
-
 function addPromptSource(sources: PromptInjectionSource[], label: string, content: string | undefined): number {
   if (!content) return 0;
   const chars = content.length;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firstpick/pi-extension-stats",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "Token and cost usage analytics command for Pi session history.",
   "license": "MIT",
   "keywords": [
@@ -14,6 +14,9 @@
       "./index.ts"
     ]
   },
+  "dependencies": {
+    "@firstpick/pi-utils": "^0.1.3"
+  },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*"
   },