From 6e20d7d8c730bcf96d23031b8ecde866c191bee1 Mon Sep 17 00:00:00 2001 From: Scrum4Me Agent <30029041+madhura68@users.noreply.github.com> Date: Wed, 6 May 2026 03:10:11 +0200 Subject: [PATCH] =?UTF-8?q?feat(ST-vmc7vpps):=20lib/insights/token-stats.t?= =?UTF-8?q?s=20=E2=80=94=20sprint=20KPI=20+=20per-job=20query?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit SQL-queries voor totale tokens/kosten (KPI) en per-job tabel met ModelPrice JOIN. Guard op lege sprintId. Tests voor empty guard, KPI-mapping en null token-data. Co-Authored-By: Claude Sonnet 4.6 --- __tests__/lib/insights/token-stats.test.ts | 67 +++++++++++ lib/insights/token-stats.ts | 129 +++++++++++++++++++++ 2 files changed, 196 insertions(+) create mode 100644 __tests__/lib/insights/token-stats.test.ts create mode 100644 lib/insights/token-stats.ts diff --git a/__tests__/lib/insights/token-stats.test.ts b/__tests__/lib/insights/token-stats.test.ts new file mode 100644 index 0000000..8614292 --- /dev/null +++ b/__tests__/lib/insights/token-stats.test.ts @@ -0,0 +1,67 @@ +import { describe, it, expect, vi, beforeEach } from 'vitest' + +const { mockQueryRaw } = vi.hoisted(() => ({ mockQueryRaw: vi.fn() })) + +vi.mock('@/lib/prisma', () => ({ + prisma: { $queryRaw: mockQueryRaw }, +})) + +import { getTokenStats } from '@/lib/insights/token-stats' + +beforeEach(() => { + vi.clearAllMocks() +}) + +describe('getTokenStats', () => { + it('returns empty result for empty sprintId', async () => { + const result = await getTokenStats('user-1', '') + + expect(result.kpi.totalTokens).toBe(0) + expect(result.kpi.totalCostUsd).toBe(0) + expect(result.kpi.avgCostPerJob).toBe(0) + expect(result.kpi.jobCount).toBe(0) + expect(result.jobs).toHaveLength(0) + expect(mockQueryRaw).not.toHaveBeenCalled() + }) + + it('maps kpi rows correctly', async () => { + const kpiRows = [{ total_tokens: BigInt(10000), total_cost: 0.15, avg_cost: 0.05, job_count: BigInt(3) }] + const jobRows: unknown[] = [] + mockQueryRaw.mockResolvedValueOnce(kpiRows).mockResolvedValueOnce(jobRows) + + const result = await getTokenStats('user-1', 'sprint-1') + + expect(result.kpi.totalTokens).toBe(10000) + expect(result.kpi.totalCostUsd).toBe(0.15) + expect(result.kpi.avgCostPerJob).toBe(0.05) + expect(result.kpi.jobCount).toBe(3) + }) + + it('maps job rows and handles null token data', async () => { + const kpiRows = [{ total_tokens: BigInt(0), total_cost: null, avg_cost: null, job_count: BigInt(0) }] + const jobRows = [ + { + job_id: 'job-1', + task_title: 'My Task', + idea_code: null, + model_id: 'claude-sonnet-4-6', + input_tokens: null, + output_tokens: null, + cache_read_tokens: null, + cache_write_tokens: null, + cost_usd: null, + duration_seconds: 42, + }, + ] + mockQueryRaw.mockResolvedValueOnce(kpiRows).mockResolvedValueOnce(jobRows) + + const result = await getTokenStats('user-1', 'sprint-1') + + expect(result.jobs).toHaveLength(1) + const job = result.jobs[0] + expect(job.jobId).toBe('job-1') + expect(job.taskTitle).toBe('My Task') + expect(job.costUsd).toBeNull() + expect(job.durationSeconds).toBe(42) + }) +}) diff --git a/lib/insights/token-stats.ts b/lib/insights/token-stats.ts new file mode 100644 index 0000000..312c822 --- /dev/null +++ b/lib/insights/token-stats.ts @@ -0,0 +1,129 @@ +import { prisma } from '@/lib/prisma' + +export interface TokenKpi { + totalTokens: number + totalCostUsd: number + avgCostPerJob: number + jobCount: number +} + +export interface TokenJobRow { + jobId: string + taskTitle: string | null + ideaCode: string | null + modelId: string | null + inputTokens: number | null + outputTokens: number | null + cacheReadTokens: number | null + cacheWriteTokens: number | null + costUsd: number | null + durationSeconds: number | null +} + +export interface TokenStatsResult { + kpi: TokenKpi + jobs: TokenJobRow[] +} + +type RawKpiRow = { + total_tokens: bigint + total_cost: number | null + avg_cost: number | null + job_count: bigint +} + +type RawJobRow = { + job_id: string + task_title: string | null + idea_code: string | null + model_id: string | null + input_tokens: number | null + output_tokens: number | null + cache_read_tokens: number | null + cache_write_tokens: number | null + cost_usd: number | null + duration_seconds: number | null +} + +const EMPTY_KPI: TokenKpi = { totalTokens: 0, totalCostUsd: 0, avgCostPerJob: 0, jobCount: 0 } + +export async function getTokenStats(userId: string, sprintId: string): Promise { + if (!sprintId) return { kpi: EMPTY_KPI, jobs: [] } + + const [kpiRows, jobRows] = await Promise.all([ + prisma.$queryRaw` + SELECT + COALESCE(SUM(cj.input_tokens + cj.output_tokens + cj.cache_read_tokens + cj.cache_write_tokens), 0) AS total_tokens, + SUM( + cj.input_tokens * mp.input_price_per_1m / 1000000.0 + + cj.output_tokens * mp.output_price_per_1m / 1000000.0 + + cj.cache_read_tokens * mp.cache_read_price_per_1m / 1000000.0 + + cj.cache_write_tokens * mp.cache_write_price_per_1m / 1000000.0 + ) FILTER (WHERE cj.input_tokens IS NOT NULL) AS total_cost, + AVG( + cj.input_tokens * mp.input_price_per_1m / 1000000.0 + + cj.output_tokens * mp.output_price_per_1m / 1000000.0 + + cj.cache_read_tokens * mp.cache_read_price_per_1m / 1000000.0 + + cj.cache_write_tokens * mp.cache_write_price_per_1m / 1000000.0 + ) FILTER (WHERE cj.input_tokens IS NOT NULL) AS avg_cost, + COUNT(*) FILTER (WHERE cj.input_tokens IS NOT NULL) AS job_count + FROM claude_jobs cj + JOIN tasks t ON cj.task_id = t.id + JOIN stories s ON t.story_id = s.id + LEFT JOIN model_prices mp ON mp.model_id = cj.model_id + WHERE cj.user_id = ${userId} + AND s.sprint_id = ${sprintId} + AND cj.status = 'DONE' + `, + prisma.$queryRaw` + SELECT + cj.id AS job_id, + t.title AS task_title, + i.code AS idea_code, + cj.model_id, + cj.input_tokens, + cj.output_tokens, + cj.cache_read_tokens, + cj.cache_write_tokens, + CASE WHEN cj.input_tokens IS NOT NULL THEN + cj.input_tokens * mp.input_price_per_1m / 1000000.0 + + cj.output_tokens * mp.output_price_per_1m / 1000000.0 + + cj.cache_read_tokens * mp.cache_read_price_per_1m / 1000000.0 + + cj.cache_write_tokens * mp.cache_write_price_per_1m / 1000000.0 + END AS cost_usd, + EXTRACT(EPOCH FROM (cj.finished_at - cj.claimed_at)) AS duration_seconds + FROM claude_jobs cj + LEFT JOIN tasks t ON cj.task_id = t.id + LEFT JOIN ideas i ON cj.idea_id = i.id + LEFT JOIN stories s ON t.story_id = s.id + LEFT JOIN model_prices mp ON mp.model_id = cj.model_id + WHERE cj.user_id = ${userId} + AND (s.sprint_id = ${sprintId} OR cj.idea_id IS NOT NULL) + AND cj.status = 'DONE' + ORDER BY cj.finished_at DESC + `, + ]) + + const kpi = kpiRows[0] + + return { + kpi: { + totalTokens: Number(kpi?.total_tokens ?? 0), + totalCostUsd: Number(kpi?.total_cost ?? 0), + avgCostPerJob: Number(kpi?.avg_cost ?? 0), + jobCount: Number(kpi?.job_count ?? 0), + }, + jobs: jobRows.map(r => ({ + jobId: r.job_id, + taskTitle: r.task_title, + ideaCode: r.idea_code, + modelId: r.model_id, + inputTokens: r.input_tokens, + outputTokens: r.output_tokens, + cacheReadTokens: r.cache_read_tokens, + cacheWriteTokens: r.cache_write_tokens, + costUsd: r.cost_usd != null ? Number(r.cost_usd) : null, + durationSeconds: r.duration_seconds != null ? Number(r.duration_seconds) : null, + })), + } +}