Add initial Portal version

src/app/api/usage/route.ts

@@ -0,0 +1,107 @@
+import { NextRequest, NextResponse } from "next/server";
+import { auth } from "@/lib/auth";
+import { getTeamInfo, getTeamSpendLogs } from "@/lib/litellm";
+
+// Pricing constants (CHF)
+const INPUT_RATE = 3; // CHF per MTok
+const OUTPUT_RATE = 15; // CHF per MTok
+
+export async function GET(req: NextRequest) {
+  const session = await auth();
+  if (!session?.user) {
+    return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
+  }
+
+  const { orgId } = session.user as any;
+  if (!orgId) {
+    return NextResponse.json(
+      { error: "No org context" },
+      { status: 400 }
+    );
+  }
+
+  // The LiteLLM team_id maps to the tenant name, which is derived from orgId
+  // Convention: team_id = "pieced-{orgId}" or looked up from the tenant CR
+  const searchParams = req.nextUrl.searchParams;
+  const teamId = searchParams.get("teamId");
+
+  if (!teamId) {
+    return NextResponse.json(
+      { error: "teamId query param required" },
+      { status: 400 }
+    );
+  }
+
+  try {
+    // Current period info
+    const teamInfo = await getTeamInfo(teamId);
+
+    // Historical spend logs (last 30 days)
+    const endDate = new Date();
+    const startDate = new Date();
+    startDate.setDate(startDate.getDate() - 30);
+
+    const spendLogs = await getTeamSpendLogs(
+      teamId,
+      startDate.toISOString().split("T")[0],
+      endDate.toISOString().split("T")[0]
+    );
+
+    // Calculate CHF costs from token counts
+    const dailyUsage = (spendLogs || []).map((day: any) => ({
+      date: day.date || day.day,
+      inputTokens: day.prompt_tokens || 0,
+      outputTokens: day.completion_tokens || 0,
+      inputCostCHF:
+        ((day.prompt_tokens || 0) / 1_000_000) * INPUT_RATE,
+      outputCostCHF:
+        ((day.completion_tokens || 0) / 1_000_000) * OUTPUT_RATE,
+      totalCostCHF:
+        ((day.prompt_tokens || 0) / 1_000_000) * INPUT_RATE +
+        ((day.completion_tokens || 0) / 1_000_000) * OUTPUT_RATE,
+    }));
+
+    // Totals for current period
+    const totalInputTokens = dailyUsage.reduce(
+      (s: number, d: any) => s + d.inputTokens,
+      0
+    );
+    const totalOutputTokens = dailyUsage.reduce(
+      (s: number, d: any) => s + d.outputTokens,
+      0
+    );
+    const totalCostCHF = dailyUsage.reduce(
+      (s: number, d: any) => s + d.totalCostCHF,
+      0
+    );
+
+    return NextResponse.json({
+      teamId,
+      currentPeriod: {
+        inputTokens: totalInputTokens,
+        outputTokens: totalOutputTokens,
+        inputCostCHF: (totalInputTokens / 1_000_000) * INPUT_RATE,
+        outputCostCHF: (totalOutputTokens / 1_000_000) * OUTPUT_RATE,
+        totalCostCHF,
+      },
+      budget: {
+        maxBudget: teamInfo?.max_budget ?? null,
+        spend: teamInfo?.spend ?? 0,
+        remaining: teamInfo?.max_budget
+          ? teamInfo.max_budget - (teamInfo.spend ?? 0)
+          : null,
+      },
+      rateLimits: {
+        rpm: teamInfo?.rpm_limit ?? null,
+        tpm: teamInfo?.tpm_limit ?? null,
+      },
+      dailyUsage,
+    });
+  } catch (err: any) {
+    console.error("Usage fetch error:", err.message);
+    return NextResponse.json(
+      { error: "Failed to fetch usage data" },
+      { status: 500 }
+    );
+  }
+}