@acarmisc/backstage-plugin-litellm 0.2.1 → 0.3.0

package/dist/components/LiteLLMHomeWidget.d.ts

@@ -0,0 +1,8 @@
+import React from 'react';
+export interface LiteLLMHomeWidgetProps {
+    /** Default period when the widget mounts. Defaults to '7d'. */
+    defaultPeriod?: 'today' | '7d' | '30d';
+    /** Optional title override. Defaults to 'LiteLLM Usage'. */
+    title?: string;
+}
+export declare const LiteLLMHomeWidget: React.FC<LiteLLMHomeWidgetProps>;

package/dist/index.cjs.js

@@ -868,6 +868,7 @@ var index_exports = {};
 __export(index_exports, {
   DashboardHeader: () => DashboardHeader,
   KeysTable: () => KeysTable,
+  LiteLLMHomeWidget: () => LiteLLMHomeWidget,
   LiteLLMPage: () => LiteLLMPage,
   LiteLlmApi: () => LiteLlmApi,
   TeamUsage: () => TeamUsage,
@@ -911,5 +912,90 @@ init_DashboardHeader();
 init_KeysTable();
 init_UsageStats();
 init_TeamUsage();
+
+// src/components/LiteLLMHomeWidget.tsx
+var import_react7 = __toESM(require("react"));
+var import_material6 = require("@mui/material");
+var import_recharts3 = require("recharts");
+var import_core_plugin_api3 = require("@backstage/core-plugin-api");
+init_api();
+var fmtUsd2 = (n) => `$${(n ?? 0).toFixed(n < 1 ? 4 : 2)}`;
+var fmtInt2 = (n) => (n ?? 0).toLocaleString();
+function presetToDateRange(preset) {
+  const end = /* @__PURE__ */ new Date();
+  const start = /* @__PURE__ */ new Date();
+  if (preset === "today") {
+    start.setHours(0, 0, 0, 0);
+  } else if (preset === "7d") {
+    start.setDate(start.getDate() - 7);
+  } else {
+    start.setDate(start.getDate() - 30);
+  }
+  return { start, end };
+}
+var Kpi = ({ label, value }) => /* @__PURE__ */ import_react7.default.createElement(import_material6.Box, null, /* @__PURE__ */ import_react7.default.createElement(import_material6.Typography, { variant: "caption", color: "text.secondary", display: "block" }, label), /* @__PURE__ */ import_react7.default.createElement(import_material6.Typography, { variant: "subtitle1", fontWeight: 600 }, value));
+var LiteLLMHomeWidget = ({
+  defaultPeriod = "7d",
+  title = "LiteLLM Usage"
+}) => {
+  const api = (0, import_core_plugin_api3.useApi)(liteLlmApiRef);
+  const [period, setPeriod] = (0, import_react7.useState)(defaultPeriod);
+  const [loading, setLoading] = (0, import_react7.useState)(true);
+  const [error, setError] = (0, import_react7.useState)(null);
+  const [usage, setUsage] = (0, import_react7.useState)(null);
+  const [keys, setKeys] = (0, import_react7.useState)([]);
+  (0, import_react7.useEffect)(() => {
+    let cancelled = false;
+    setLoading(true);
+    setError(null);
+    const { start, end } = presetToDateRange(period);
+    const startDate = start.toISOString().split("T")[0];
+    const endDate = end.toISOString().split("T")[0];
+    Promise.all([api.getUsage(startDate, endDate), api.listKeys()]).then(([usageData, keysData]) => {
+      if (!cancelled) {
+        setUsage(usageData);
+        setKeys(keysData);
+        setLoading(false);
+      }
+    }).catch((err) => {
+      if (!cancelled) {
+        setError(err.message ?? "Failed to load usage data");
+        setLoading(false);
+      }
+    });
+    return () => {
+      cancelled = true;
+    };
+  }, [api, period]);
+  const dailyData = (usage?.daily_usage ?? []).map((d) => ({
+    date: d.date,
+    spend: d.spend
+  }));
+  const hasSparkline = dailyData.length > 0;
+  return /* @__PURE__ */ import_react7.default.createElement(import_material6.Paper, { sx: { p: 2 } }, /* @__PURE__ */ import_react7.default.createElement(import_material6.Box, { display: "flex", justifyContent: "space-between", alignItems: "center", mb: 1.5 }, /* @__PURE__ */ import_react7.default.createElement(import_material6.Typography, { variant: "h6" }, title), /* @__PURE__ */ import_react7.default.createElement(import_material6.FormControl, { size: "small", sx: { minWidth: 90 } }, /* @__PURE__ */ import_react7.default.createElement(
+    import_material6.Select,
+    {
+      value: period,
+      onChange: (e) => setPeriod(e.target.value),
+      displayEmpty: true
+    },
+    /* @__PURE__ */ import_react7.default.createElement(import_material6.MenuItem, { value: "today" }, "Today"),
+    /* @__PURE__ */ import_react7.default.createElement(import_material6.MenuItem, { value: "7d" }, "7d"),
+    /* @__PURE__ */ import_react7.default.createElement(import_material6.MenuItem, { value: "30d" }, "30d")
+  ))), loading && /* @__PURE__ */ import_react7.default.createElement(import_material6.Box, { display: "flex", justifyContent: "center", alignItems: "center", minHeight: 120 }, /* @__PURE__ */ import_react7.default.createElement(import_material6.CircularProgress, { size: 32 })), !loading && error && /* @__PURE__ */ import_react7.default.createElement(import_material6.Alert, { severity: "error", sx: { mt: 1 } }, error), !loading && !error && /* @__PURE__ */ import_react7.default.createElement(import_react7.default.Fragment, null, /* @__PURE__ */ import_react7.default.createElement(import_material6.Grid, { container: true, spacing: 2, sx: { mb: hasSparkline ? 1.5 : 0 } }, /* @__PURE__ */ import_react7.default.createElement(import_material6.Grid, { item: true, xs: 6 }, /* @__PURE__ */ import_react7.default.createElement(Kpi, { label: "USD Spent", value: fmtUsd2(usage?.total_spend ?? 0) })), /* @__PURE__ */ import_react7.default.createElement(import_material6.Grid, { item: true, xs: 6 }, /* @__PURE__ */ import_react7.default.createElement(Kpi, { label: "Tokens In", value: fmtInt2(usage?.prompt_tokens ?? 0) })), /* @__PURE__ */ import_react7.default.createElement(import_material6.Grid, { item: true, xs: 6 }, /* @__PURE__ */ import_react7.default.createElement(Kpi, { label: "Tokens Out", value: fmtInt2(usage?.completion_tokens ?? 0) })), /* @__PURE__ */ import_react7.default.createElement(import_material6.Grid, { item: true, xs: 6 }, /* @__PURE__ */ import_react7.default.createElement(Kpi, { label: "Keys", value: fmtInt2(keys.length) }))), hasSparkline && /* @__PURE__ */ import_react7.default.createElement(import_material6.Box, { height: 120 }, /* @__PURE__ */ import_react7.default.createElement(import_recharts3.ResponsiveContainer, { width: "100%", height: "100%" }, /* @__PURE__ */ import_react7.default.createElement(import_recharts3.AreaChart, { data: dailyData, margin: { top: 4, right: 0, bottom: 0, left: 0 } }, /* @__PURE__ */ import_react7.default.createElement(
+    import_recharts3.Area,
+    {
+      type: "monotone",
+      dataKey: "spend",
+      stroke: "#8884d8",
+      fill: "#8884d8",
+      fillOpacity: 0.3,
+      dot: false,
+      isAnimationActive: false
+    }
+  ))))));
+};
+
+// src/index.ts
 init_api();
 //# sourceMappingURL=index.cjs.js.map