@acarmisc/backstage-plugin-litellm 0.6.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"
       ))));
     };
   }
@@ -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));
@@ -1355,7 +1420,8 @@ var init_LiteLLMPage = __esm({
           onBlockKey: handleBlockKey,
           onUnblockKey: handleUnblockKey,
           onResetKeySpend: handleResetKeySpend,
-          onDeleteKey: handleDeleteKey
+          onDeleteKey: handleDeleteKey,
+          onPruneExpiredKeys: handlePruneExpiredKeys
         }
       ), activeTab === "teams" && /* @__PURE__ */ React6.createElement(
         TeamUsage,