| 1 | /** |
| 2 | * The built-in transcript formats by provider, for engine and cache tests. |
| 3 | * |
| 4 | * @module usageTestFormats |
| 5 | */ |
| 6 | import type { UsageProviderKind } from "@t3tools/contracts"; |
| 7 | import type { TranscriptUsageFormat } from "@t3tools/provider-core/server/usage"; |
| 8 | import { piUsageFormat } from "@t3tools/provider-pi/server/usage"; |
| 9 | import { grokUsageFormat } from "@t3tools/provider-grok/server/usage"; |
| 10 | |
| 11 | import { claudeUsageFormat } from "../provider/Drivers/claudeUsage.ts"; |
| 12 | import { codexUsageFormat } from "../provider/Drivers/codexUsage.ts"; |
| 13 | |
| 14 | export const TEST_FORMATS: Record< |
| 15 | "claude" | "codex" | "grok" | "pi", |
| 16 | TranscriptUsageFormat<unknown> |
| 17 | > = { |
| 18 | claude: claudeUsageFormat, |
| 19 | codex: codexUsageFormat, |
| 20 | grok: grokUsageFormat, |
| 21 | pi: piUsageFormat, |
| 22 | }; |
| 23 | |
| 24 | export const TEST_FORMAT_MAP = new Map<UsageProviderKind, TranscriptUsageFormat<unknown>>( |
| 25 | Object.entries(TEST_FORMATS) as Array<[UsageProviderKind, TranscriptUsageFormat<unknown>]>, |
| 26 | ); |