improves display of large token numbers

2023-08-31 13:23:36 -05:00
parent 4b32130eaa
commit fe0f04ceb8
6 changed files with 66 additions and 45 deletions
@@ -16,11 +16,10 @@ import { logger } from "../../logger";

 const log = logger.child({ module: "users" });

-export type UserTokenCounts = {
-  turbo: number;
-  gpt4: number;
-  claude: number;
-  "gpt4-32k"?: number;
+type UserTokenCounts = {
+  [K in Exclude<ModelFamily, "gpt4-32k">]: number;
+} & {
+  [K in "gpt4-32k"]?: number; // Optional because it was added later
 };

 export interface User {