import { CostEstimateResponse } from "../types"; import { formatNumberWithCommas } from "@/utils/dataUtils"; import { MultiModelResult } from "./types"; const formatCostForExport = (value: number | null | undefined): string => { if (value === null || value === undefined) return "-"; if (value === 0) return "$0.00"; if (value < 0.01) return `$${value.toFixed(6)}`; if (value < 1) return `$${value.toFixed(4)}`; return `$${formatNumberWithCommas(value, 2)}`; }; const formatRequestsForExport = (value: number | null | undefined): string => { if (value === null || value === undefined) return "-"; return formatNumberWithCommas(value, 0); }; const generateModelSection = (result: CostEstimateResponse): string => { return `

${result.model} ${result.provider ? `(${result.provider})` : ""}

Input Tokens per Request: ${formatRequestsForExport(result.input_tokens)}

Output Tokens per Request: ${formatRequestsForExport(result.output_tokens)}

${result.num_requests_per_day ? `

Requests per Day: ${formatRequestsForExport(result.num_requests_per_day)}

` : ""} ${result.num_requests_per_month ? `

Requests per Month: ${formatRequestsForExport(result.num_requests_per_month)}

` : ""}

${result.daily_cost !== null ? "" : ""} ${result.monthly_cost !== null ? "" : ""} ${result.daily_cost !== null ? `` : ""} ${result.monthly_cost !== null ? `` : ""} ${result.daily_cost !== null ? `` : ""} ${result.monthly_cost !== null ? `` : ""} ${result.daily_cost !== null ? `` : ""} ${result.monthly_cost !== null ? `` : ""} ${result.daily_cost !== null ? `` : ""} ${result.monthly_cost !== null ? `` : ""}

Cost Type	Per Request	Daily	Monthly
Input Cost	${formatCostForExport(result.input_cost_per_request)}	${formatCostForExport(result.daily_input_cost)}	${formatCostForExport(result.monthly_input_cost)}
Output Cost	${formatCostForExport(result.output_cost_per_request)}	${formatCostForExport(result.daily_output_cost)}	${formatCostForExport(result.monthly_output_cost)}
Margin/Fee	${formatCostForExport(result.margin_cost_per_request)}	${formatCostForExport(result.daily_margin_cost)}	${formatCostForExport(result.monthly_margin_cost)}
Total	${formatCostForExport(result.cost_per_request)}	${formatCostForExport(result.daily_cost)}	${formatCostForExport(result.monthly_cost)}

`; }; export const exportMultiToPDF = (multiResult: MultiModelResult): void => { const printWindow = window.open("", "_blank"); if (!printWindow) { alert("Please allow popups to export PDF"); return; } const validEntries = multiResult.entries.filter((e) => e.result !== null); const modelCount = validEntries.length; const html = ` Multi-Model Cost Estimate Report

LLM Cost Estimate Report

${modelCount} model${modelCount !== 1 ? "s" : ""} configured

Combined Totals

Total Per Request

${formatCostForExport(multiResult.totals.cost_per_request)}

Total Daily

${formatCostForExport(multiResult.totals.daily_cost)}

Total Monthly

${formatCostForExport(multiResult.totals.monthly_cost)}

${multiResult.totals.margin_per_request > 0 ? `

Margin/Request

${formatCostForExport(multiResult.totals.margin_per_request)}

Daily Margin

${formatCostForExport(multiResult.totals.daily_margin)}

Monthly Margin

${formatCostForExport(multiResult.totals.monthly_margin)}

` : ""}

Model Breakdown

${validEntries.map((e) => generateModelSection(e.result!)).join("")} `; printWindow.document.write(html); printWindow.document.close(); printWindow.onload = () => { printWindow.print(); }; }; export const exportMultiToCSV = (multiResult: MultiModelResult): void => { const validEntries = multiResult.entries.filter((e) => e.result !== null); const rows: string[][] = [ ["LLM Multi-Model Cost Estimate Report"], ["Generated", new Date().toLocaleString()], [""], ]; // Summary section rows.push( ["COMBINED TOTALS"], ["Total Per Request", multiResult.totals.cost_per_request.toString()], ["Total Daily", multiResult.totals.daily_cost?.toString() || "-"], ["Total Monthly", multiResult.totals.monthly_cost?.toString() || "-"], ["Margin Per Request", multiResult.totals.margin_per_request.toString()], ["Daily Margin", multiResult.totals.daily_margin?.toString() || "-"], ["Monthly Margin", multiResult.totals.monthly_margin?.toString() || "-"], [""] ); // Summary table header rows.push([ "Model", "Provider", "Input Tokens", "Output Tokens", "Requests/Day", "Requests/Month", "Cost/Request", "Daily Cost", "Monthly Cost", "Input Cost/Req", "Output Cost/Req", "Margin/Req", ]); // Add each model's data for (const entry of validEntries) { const r = entry.result!; rows.push([ r.model, r.provider || "-", r.input_tokens.toString(), r.output_tokens.toString(), r.num_requests_per_day?.toString() || "-", r.num_requests_per_month?.toString() || "-", r.cost_per_request.toString(), r.daily_cost?.toString() || "-", r.monthly_cost?.toString() || "-", r.input_cost_per_request.toString(), r.output_cost_per_request.toString(), r.margin_cost_per_request.toString(), ]); } const csv = rows.map((row) => row.map((cell) => `"${cell}"`).join(",")).join("\n"); const blob = new Blob([csv], { type: "text/csv;charset=utf-8;" }); const url = window.URL.createObjectURL(blob); const a = document.createElement("a"); a.href = url; a.download = `cost_estimate_multi_model_${new Date().toISOString().split("T")[0]}.csv`; document.body.appendChild(a); a.click(); document.body.removeChild(a); window.URL.revokeObjectURL(url); };