@acarmisc/backstage-plugin-litellm 0.5.0 → 0.7.0

package/dist/index.esm.js

@@ -15,6 +15,14 @@ var __export = (target, all) => {
 
 // src/api.ts
 import { createApiRef } from "@backstage/core-plugin-api";
+function expiryStatus(expiresAt) {
+  if (!expiresAt) return null;
+  const diff = new Date(expiresAt).getTime() - Date.now();
+  const days = diff / (1e3 * 60 * 60 * 24);
+  if (days < 0) return "expired";
+  if (days < 7) return "soon";
+  return "ok";
+}
 var ApiError, liteLlmApiRef, LiteLlmApi;
 var init_api = __esm({
   "src/api.ts"() {
@@ -100,6 +108,19 @@ var init_api = __esm({
       async resetKeySpend(keyId) {
         await this.post(`/keys/${encodeURIComponent(keyId)}/reset_spend`, {});
       }
+      async pruneExpiredKeys() {
+        const keys = await this.get("/keys");
+        const expired = keys.filter((k) => expiryStatus(k.expires_at) === "expired");
+        if (expired.length === 0) return { pruned: 0 };
+        for (const key of expired) {
+          try {
+            await this.del(`/keys/${encodeURIComponent(key.token ?? key.key)}`);
+          } catch (err) {
+            console.warn(`Failed to delete expired key ${key.token ?? key.key}:`, err);
+          }
+        }
+        return { pruned: expired.length };
+      }
       async getAuditLogs(params) {
         const strParams = {};
         if (params.page !== void 0) strParams.page = String(params.page);
@@ -183,14 +204,6 @@ import {
   InputAdornment
 } from "@mui/material";
 import { ContentCopy, Delete, Add, Edit, Autorenew, Search, Warning, Lock, LockOpen } from "@mui/icons-material";
-function expiryStatus(expiresAt) {
-  if (!expiresAt) return null;
-  const diff = new Date(expiresAt).getTime() - Date.now();
-  const days = diff / (1e3 * 60 * 60 * 24);
-  if (days < 0) return "expired";
-  if (days < 7) return "soon";
-  return "ok";
-}
 function ExpiryChip({ expiresAt }) {
   const status = expiryStatus(expiresAt);
   if (!status) return /* @__PURE__ */ React2.createElement(Typography2, { variant: "body2", color: "text.secondary" }, "-");
@@ -214,6 +227,7 @@ var maskKey, shortKeyId, formatDate, emptyForm, keyToEditForm, KeysTable;
 var init_KeysTable = __esm({
   "src/components/KeysTable.tsx"() {
     "use strict";
+    init_api();
     maskKey = (key) => {
       if (key.length <= 8) return "***";
       return `${key.slice(0, 4)}...${key.slice(-4)}`;
@@ -258,7 +272,8 @@ var init_KeysTable = __esm({
       onBlockKey,
       onUnblockKey,
       onResetKeySpend,
-      onDeleteKey
+      onDeleteKey,
+      onPruneExpiredKeys
     }) => {
       const [generateModalOpen, setGenerateModalOpen] = useState(false);
       const [newKeyValue, setNewKeyValue] = useState(null);
@@ -371,6 +386,23 @@ var init_KeysTable = __esm({
       const copyToClipboard = (text) => {
         navigator.clipboard.writeText(text);
       };
+      const [pruneConfirmCount, setPruneConfirmCount] = useState(null);
+      const [pruneSubmitting, setPruneSubmitting] = useState(false);
+      const expiredKeys = useMemo(() => {
+        return keys.filter((k) => expiryStatus(k.expires_at) === "expired");
+      }, [keys]);
+      const handlePruneExpired = async () => {
+        if (pruneConfirmCount === null || pruneConfirmCount === 0) return;
+        setPruneSubmitting(true);
+        try {
+          await onPruneExpiredKeys();
+        } catch (error) {
+          console.error("Failed to prune expired keys:", error);
+        } finally {
+          setPruneSubmitting(false);
+          setPruneConfirmCount(null);
+        }
+      };
       const modelOption = (m) => {
         const inCost = fmtCost(m.input_cost_per_token);
         const outCost = fmtCost(m.output_cost_per_token);
@@ -388,6 +420,17 @@ var init_KeysTable = __esm({
             startAdornment: /* @__PURE__ */ React2.createElement(InputAdornment, { position: "start" }, /* @__PURE__ */ React2.createElement(Search, { fontSize: "small" }))
           }
         }
+      ), expiredKeys.length > 0 && /* @__PURE__ */ React2.createElement(
+        Button,
+        {
+          variant: "outlined",
+          color: "error",
+          startIcon: /* @__PURE__ */ React2.createElement(Autorenew, null),
+          onClick: () => setPruneConfirmCount(expiredKeys.length)
+        },
+        "Prune expired (",
+        expiredKeys.length,
+        ")"
       ), /* @__PURE__ */ React2.createElement(
         Button,
         {
@@ -651,6 +694,15 @@ var init_KeysTable = __esm({
           disabled: deleteSubmitting
         },
         deleteSubmitting ? /* @__PURE__ */ React2.createElement(CircularProgress, { size: 20 }) : "Revoke"
+      ))), /* @__PURE__ */ React2.createElement(Dialog, { open: pruneConfirmCount !== null, onClose: () => setPruneConfirmCount(null), maxWidth: "xs", fullWidth: true }, /* @__PURE__ */ React2.createElement(DialogTitle, null, "Prune Expired Keys?"), /* @__PURE__ */ React2.createElement(DialogContent, null, /* @__PURE__ */ React2.createElement(Typography2, null, "This will permanently delete ", pruneConfirmCount, " expired key", pruneConfirmCount !== 1 ? "s" : "", " from LiteLLM. Any integrations using them will stop working immediately.")), /* @__PURE__ */ React2.createElement(DialogActions, null, /* @__PURE__ */ React2.createElement(Button, { onClick: () => setPruneConfirmCount(null), disabled: pruneSubmitting }, "Cancel"), /* @__PURE__ */ React2.createElement(
+        Button,
+        {
+          onClick: handlePruneExpired,
+          variant: "contained",
+          color: "error",
+          disabled: pruneSubmitting
+        },
+        pruneSubmitting ? /* @__PURE__ */ React2.createElement(CircularProgress, { size: 20 }) : "Prune"
       ))));
     };
   }
@@ -1121,7 +1173,7 @@ __export(LiteLLMPage_exports, {
   LiteLLMPage: () => LiteLLMPage
 });
 import React6, { useState as useState5, useCallback as useCallback2, useMemo as useMemo3 } from "react";
-import { Grid as Grid2, Box as Box6, Snackbar, Alert as Alert2, CircularProgress as CircularProgress4, Typography as Typography6, Paper as Paper6, Tabs as Tabs2, Tab as Tab2 } from "@mui/material";
+import { Box as Box6, Snackbar, Alert as Alert2, CircularProgress as CircularProgress4, Typography as Typography6, Paper as Paper6, Tabs as Tabs2, Tab as Tab2 } from "@mui/material";
 import { useAsync as useAsync2, useAsyncRetry } from "react-use";
 import { useApi } from "@backstage/core-plugin-api";
 function initDateRange() {
@@ -1312,6 +1364,19 @@ var init_LiteLLMPage = __esm({
         },
         [api, refreshKeys]
       );
+      const handlePruneExpiredKeys = useCallback2(
+        async () => {
+          try {
+            const result = await api.pruneExpiredKeys();
+            setSnackbar({ message: `Pruned ${result.pruned} expired key${result.pruned !== 1 ? "s" : ""}`, severity: "success" });
+            refreshKeys();
+          } catch (e) {
+            setSnackbar({ message: `Failed to prune expired keys: ${e.message}`, severity: "error" });
+          }
+          return { pruned: 0 };
+        },
+        [api, refreshKeys]
+      );
       const isInitialLoading = userLoading && !userInfo;
       if (isInitialLoading) {
         return /* @__PURE__ */ React6.createElement(Box6, { display: "flex", justifyContent: "center", alignItems: "center", minHeight: "50vh" }, /* @__PURE__ */ React6.createElement(CircularProgress4, null));
@@ -1321,7 +1386,7 @@ var init_LiteLLMPage = __esm({
         const hint = userError?.body?.hint;
         return /* @__PURE__ */ React6.createElement(Box6, { p: 3 }, /* @__PURE__ */ React6.createElement(Paper6, { sx: { p: 3 } }, /* @__PURE__ */ React6.createElement(Typography6, { variant: "h6", gutterBottom: true }, "Account not provisioned"), /* @__PURE__ */ React6.createElement(Typography6, { color: "text.secondary", paragraph: true }, "Your Backstage account is not linked to a LiteLLM user."), hint ? /* @__PURE__ */ React6.createElement(Typography6, { variant: "body2", color: "text.secondary" }, hint) : /* @__PURE__ */ React6.createElement(Typography6, { variant: "body2", color: "text.secondary" }, isProvisioningEnabled ? "Auto-provisioning is enabled but failed. Check the backend logs." : "Set litellm.provisioning.enabled: true in app-config.yaml to enable auto-provisioning, or ask your administrator to create the account manually.")));
       }
-      return /* @__PURE__ */ React6.createElement(Box6, { p: 3 }, userInfo.can_view_audit && /* @__PURE__ */ React6.createElement(
+      return /* @__PURE__ */ React6.createElement(Box6, { p: 3 }, /* @__PURE__ */ React6.createElement(Box6, { mb: 2 }, /* @__PURE__ */ React6.createElement(DashboardHeader, { userInfo, teams: teams ?? [], loading: userLoading || teamsLoading })), /* @__PURE__ */ React6.createElement(
         Tabs2,
         {
           value: activeTab,
@@ -1329,20 +1394,10 @@ var init_LiteLLMPage = __esm({
           sx: { mb: 2, borderBottom: 1, borderColor: "divider" }
         },
         /* @__PURE__ */ React6.createElement(Tab2, { label: "Overview", value: "overview" }),
-        /* @__PURE__ */ React6.createElement(Tab2, { label: "Audit Log", value: "audit" })
-      ), activeTab === "audit" ? /* @__PURE__ */ React6.createElement(AuditLog, { api }) : /* @__PURE__ */ React6.createElement(Grid2, { container: true, spacing: 2 }, /* @__PURE__ */ React6.createElement(Grid2, { item: true, xs: 12 }, /* @__PURE__ */ React6.createElement(DashboardHeader, { userInfo, teams: teams ?? [], loading: userLoading || teamsLoading })), /* @__PURE__ */ React6.createElement(Grid2, { item: true, xs: 12 }, /* @__PURE__ */ React6.createElement(
-        TeamUsage,
-        {
-          teams: teams ?? [],
-          loading: teamsLoading,
-          dateRange,
-          getTeamUsage: (teamId) => {
-            if (teamUsageCache[teamId] === void 0) loadTeamUsage(teamId);
-            return teamUsageCache[teamId] ?? null;
-          },
-          getTeamUsageLoading: (teamId) => teamUsageLoading[teamId] ?? false
-        }
-      )), /* @__PURE__ */ React6.createElement(Grid2, { item: true, xs: 12 }, /* @__PURE__ */ React6.createElement(
+        /* @__PURE__ */ React6.createElement(Tab2, { label: "Keys", value: "keys" }),
+        /* @__PURE__ */ React6.createElement(Tab2, { label: "Teams", value: "teams" }),
+        userInfo.can_view_audit && /* @__PURE__ */ React6.createElement(Tab2, { label: "Audit Log", value: "audit" })
+      ), activeTab === "overview" && /* @__PURE__ */ React6.createElement(
         UsageStats,
         {
           usage: usage ?? null,
@@ -1352,7 +1407,7 @@ var init_LiteLLMPage = __esm({
           loading: usageLoading,
           userInfo
         }
-      )), /* @__PURE__ */ React6.createElement(Grid2, { item: true, xs: 12 }, /* @__PURE__ */ React6.createElement(
+      ), activeTab === "keys" && /* @__PURE__ */ React6.createElement(
         KeysTable,
         {
           keys: keys ?? [],
@@ -1365,9 +1420,22 @@ var init_LiteLLMPage = __esm({
           onBlockKey: handleBlockKey,
           onUnblockKey: handleUnblockKey,
           onResetKeySpend: handleResetKeySpend,
-          onDeleteKey: handleDeleteKey
+          onDeleteKey: handleDeleteKey,
+          onPruneExpiredKeys: handlePruneExpiredKeys
+        }
+      ), activeTab === "teams" && /* @__PURE__ */ React6.createElement(
+        TeamUsage,
+        {
+          teams: teams ?? [],
+          loading: teamsLoading,
+          dateRange,
+          getTeamUsage: (teamId) => {
+            if (teamUsageCache[teamId] === void 0) loadTeamUsage(teamId);
+            return teamUsageCache[teamId] ?? null;
+          },
+          getTeamUsageLoading: (teamId) => teamUsageLoading[teamId] ?? false
         }
-      ))), /* @__PURE__ */ React6.createElement(
+      ), activeTab === "audit" && userInfo.can_view_audit && /* @__PURE__ */ React6.createElement(AuditLog, { api }), /* @__PURE__ */ React6.createElement(
         Snackbar,
         {
           open: !!snackbar,
@@ -1431,7 +1499,7 @@ import {
   FormControl as FormControl2,
   Select as Select2,
   MenuItem as MenuItem4,
-  Grid as Grid3,
+  Grid as Grid2,
   CircularProgress as CircularProgress5,
   Alert as Alert3
 } from "@mui/material";
@@ -1500,7 +1568,7 @@ var LiteLLMHomeWidget = ({
     /* @__PURE__ */ React8.createElement(MenuItem4, { value: "today" }, "Today"),
     /* @__PURE__ */ React8.createElement(MenuItem4, { value: "7d" }, "7d"),
     /* @__PURE__ */ React8.createElement(MenuItem4, { value: "30d" }, "30d")
-  ))), loading && /* @__PURE__ */ React8.createElement(Box7, { display: "flex", justifyContent: "center", alignItems: "center", minHeight: 120 }, /* @__PURE__ */ React8.createElement(CircularProgress5, { size: 32 })), !loading && error && /* @__PURE__ */ React8.createElement(Alert3, { severity: "error", sx: { mt: 1 } }, error), !loading && !error && /* @__PURE__ */ React8.createElement(React8.Fragment, null, /* @__PURE__ */ React8.createElement(Grid3, { container: true, spacing: 2, sx: { mb: hasSparkline ? 1.5 : 0 } }, /* @__PURE__ */ React8.createElement(Grid3, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "USD Spent", value: fmtUsd2(usage?.total_spend ?? 0) })), /* @__PURE__ */ React8.createElement(Grid3, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Tokens In", value: fmtInt2(usage?.prompt_tokens ?? 0) })), /* @__PURE__ */ React8.createElement(Grid3, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Tokens Out", value: fmtInt2(usage?.completion_tokens ?? 0) })), /* @__PURE__ */ React8.createElement(Grid3, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Keys", value: fmtInt2(keys.length) }))), hasSparkline && /* @__PURE__ */ React8.createElement(Box7, { height: 120 }, /* @__PURE__ */ React8.createElement(ResponsiveContainer3, { width: "100%", height: "100%" }, /* @__PURE__ */ React8.createElement(AreaChart3, { data: dailyData, margin: { top: 4, right: 0, bottom: 0, left: 0 } }, /* @__PURE__ */ React8.createElement(
+  ))), loading && /* @__PURE__ */ React8.createElement(Box7, { display: "flex", justifyContent: "center", alignItems: "center", minHeight: 120 }, /* @__PURE__ */ React8.createElement(CircularProgress5, { size: 32 })), !loading && error && /* @__PURE__ */ React8.createElement(Alert3, { severity: "error", sx: { mt: 1 } }, error), !loading && !error && /* @__PURE__ */ React8.createElement(React8.Fragment, null, /* @__PURE__ */ React8.createElement(Grid2, { container: true, spacing: 2, sx: { mb: hasSparkline ? 1.5 : 0 } }, /* @__PURE__ */ React8.createElement(Grid2, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "USD Spent", value: fmtUsd2(usage?.total_spend ?? 0) })), /* @__PURE__ */ React8.createElement(Grid2, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Tokens In", value: fmtInt2(usage?.prompt_tokens ?? 0) })), /* @__PURE__ */ React8.createElement(Grid2, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Tokens Out", value: fmtInt2(usage?.completion_tokens ?? 0) })), /* @__PURE__ */ React8.createElement(Grid2, { item: true, xs: 6 }, /* @__PURE__ */ React8.createElement(Kpi, { label: "Keys", value: fmtInt2(keys.length) }))), hasSparkline && /* @__PURE__ */ React8.createElement(Box7, { height: 120 }, /* @__PURE__ */ React8.createElement(ResponsiveContainer3, { width: "100%", height: "100%" }, /* @__PURE__ */ React8.createElement(AreaChart3, { data: dailyData, margin: { top: 4, right: 0, bottom: 0, left: 0 } }, /* @__PURE__ */ React8.createElement(
     Area3,
     {
       type: "monotone",