@acarmisc/backstage-plugin-litellm 0.3.0 → 0.3.2

package/dist/index.esm.js

@@ -143,7 +143,7 @@ var init_DashboardHeader = __esm({
 });
 
 // src/components/KeysTable.tsx
-import React2, { useEffect, useState } from "react";
+import React2, { useState } from "react";
 import {
   Paper as Paper2,
   Table,
@@ -166,8 +166,8 @@ import {
   CircularProgress,
   Autocomplete
 } from "@mui/material";
-import { ContentCopy, Delete, Add, Edit, Visibility, VisibilityOff } from "@mui/icons-material";
-var maskKey, formatDate, emptyForm, keyToEditForm, KeysTable;
+import { ContentCopy, Delete, Add, Edit } from "@mui/icons-material";
+var maskKey, shortKeyId, formatDate, emptyForm, keyToEditForm, KeysTable;
 var init_KeysTable = __esm({
   "src/components/KeysTable.tsx"() {
     "use strict";
@@ -175,6 +175,11 @@ var init_KeysTable = __esm({
       if (key.length <= 8) return "***";
       return `${key.slice(0, 4)}...${key.slice(-4)}`;
     };
+    shortKeyId = (token) => {
+      if (!token) return "-";
+      if (token.length <= 16) return token;
+      return `${token.slice(0, 12)}\u2026`;
+    };
     formatDate = (dateStr) => {
       try {
         return new Date(dateStr).toLocaleDateString();
@@ -186,7 +191,7 @@ var init_KeysTable = __esm({
       alias: "",
       models: [],
       duration: "30d",
-      max_budget: void 0,
+      max_budget: 100,
       tpm_limit: void 0,
       team_id: void 0,
       key_type: "llm_api"
@@ -208,16 +213,9 @@ var init_KeysTable = __esm({
       onDeleteKey
     }) => {
       const [generateModalOpen, setGenerateModalOpen] = useState(false);
-      const [showKeyValue, setShowKeyValue] = useState(null);
       const [newKeyValue, setNewKeyValue] = useState(null);
       const [formData, setFormData] = useState(emptyForm());
       const [submitting, setSubmitting] = useState(false);
-      useEffect(() => {
-        if (!generateModalOpen) return;
-        if (!formData.team_id && teams.length > 0) {
-          setFormData((f) => ({ ...f, team_id: teams[0].team_id }));
-        }
-      }, [generateModalOpen, teams, formData.team_id]);
       const canGenerate = true;
       const [editingKey, setEditingKey] = useState(null);
       const [editForm, setEditForm] = useState({});
@@ -231,10 +229,6 @@ var init_KeysTable = __esm({
           const response = await onGenerateKey(formData);
           setNewKeyValue(response.key);
           setFormData(emptyForm());
-          setTimeout(() => {
-            setGenerateModalOpen(false);
-            setNewKeyValue(null);
-          }, 1500);
         } catch (error) {
           console.error("Failed to generate key:", error);
         } finally {
@@ -278,32 +272,26 @@ var init_KeysTable = __esm({
           onClick: () => setGenerateModalOpen(true)
         },
         "Generate New Key"
-      )), /* @__PURE__ */ React2.createElement(TableContainer, null, /* @__PURE__ */ React2.createElement(Table, null, /* @__PURE__ */ React2.createElement(TableHead, null, /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, null, "Alias"), /* @__PURE__ */ React2.createElement(TableCell, null, "Key"), /* @__PURE__ */ React2.createElement(TableCell, null, "Created"), /* @__PURE__ */ React2.createElement(TableCell, null, "Spend"), /* @__PURE__ */ React2.createElement(TableCell, null, "Budget"), /* @__PURE__ */ React2.createElement(TableCell, null, "TPM Limit"), /* @__PURE__ */ React2.createElement(TableCell, null, "Models"), /* @__PURE__ */ React2.createElement(TableCell, { align: "right" }, "Actions"))), /* @__PURE__ */ React2.createElement(TableBody, null, loading ? /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, { colSpan: 8, align: "center" }, /* @__PURE__ */ React2.createElement(CircularProgress, { size: 24 }))) : keys.length === 0 ? /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, { colSpan: 8, align: "center" }, /* @__PURE__ */ React2.createElement(Typography2, { color: "text.secondary" }, "No keys found"))) : keys.map((key) => /* @__PURE__ */ React2.createElement(TableRow, { key: key.key }, /* @__PURE__ */ React2.createElement(TableCell, null, key.key_alias || "-"), /* @__PURE__ */ React2.createElement(TableCell, null, /* @__PURE__ */ React2.createElement(Box2, { display: "flex", alignItems: "center", gap: 0.5 }, /* @__PURE__ */ React2.createElement(
-        Typography2,
-        {
-          variant: "body2",
-          component: "code",
-          color: "primary",
-          sx: {
-            fontFamily: "monospace",
-            backgroundColor: "background.default",
-            px: 1,
-            py: 0.5,
-            borderRadius: 1,
-            maxWidth: "250px",
-            overflow: "hidden",
-            textOverflow: "ellipsis"
-          }
-        },
-        showKeyValue === key.key ? key.key : maskKey(key.key)
-      ), /* @__PURE__ */ React2.createElement(
-        IconButton,
-        {
-          size: "small",
-          onClick: () => setShowKeyValue(showKeyValue === key.key ? null : key.key)
-        },
-        showKeyValue === key.key ? /* @__PURE__ */ React2.createElement(VisibilityOff, null) : /* @__PURE__ */ React2.createElement(Visibility, null)
-      ), /* @__PURE__ */ React2.createElement(IconButton, { size: "small", onClick: () => copyToClipboard(key.key) }, /* @__PURE__ */ React2.createElement(ContentCopy, { fontSize: "small" })))), /* @__PURE__ */ React2.createElement(TableCell, null, formatDate(key.created_at)), /* @__PURE__ */ React2.createElement(TableCell, null, "$", key.spend?.toFixed(4) || "0.00"), /* @__PURE__ */ React2.createElement(TableCell, null, key.max_budget ? `$${key.max_budget}` : "-"), /* @__PURE__ */ React2.createElement(TableCell, null, key.tpm_limit || "-"), /* @__PURE__ */ React2.createElement(TableCell, null, /* @__PURE__ */ React2.createElement(Box2, { display: "flex", gap: 0.5, flexWrap: "wrap" }, key.models?.slice(0, 2).map((model) => /* @__PURE__ */ React2.createElement(Chip2, { key: model, label: model, size: "small" })), (key.models?.length || 0) > 2 && /* @__PURE__ */ React2.createElement(Chip2, { label: `+${(key.models?.length || 0) - 2}`, size: "small", variant: "outlined" }))), /* @__PURE__ */ React2.createElement(TableCell, { align: "right" }, /* @__PURE__ */ React2.createElement(IconButton, { onClick: () => handleOpenEdit(key) }, /* @__PURE__ */ React2.createElement(Edit, { fontSize: "small" })), /* @__PURE__ */ React2.createElement(IconButton, { color: "error", onClick: () => onDeleteKey(key.key) }, /* @__PURE__ */ React2.createElement(Delete, null))))))))), /* @__PURE__ */ React2.createElement(Dialog, { open: generateModalOpen, onClose: handleCloseModal, maxWidth: "sm", fullWidth: true }, /* @__PURE__ */ React2.createElement(DialogTitle, null, newKeyValue ? "Key Generated" : "Generate New Key"), /* @__PURE__ */ React2.createElement(DialogContent, null, newKeyValue ? /* @__PURE__ */ React2.createElement(Box2, null, /* @__PURE__ */ React2.createElement(Typography2, { variant: "body2", color: "text.secondary", gutterBottom: true }, "Copy this key now. You won't be able to see it again."), /* @__PURE__ */ React2.createElement(
+      )), /* @__PURE__ */ React2.createElement(TableContainer, null, /* @__PURE__ */ React2.createElement(Table, null, /* @__PURE__ */ React2.createElement(TableHead, null, /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, null, "Alias"), /* @__PURE__ */ React2.createElement(TableCell, null, "Key ID"), /* @__PURE__ */ React2.createElement(TableCell, null, "Created"), /* @__PURE__ */ React2.createElement(TableCell, null, "Spend"), /* @__PURE__ */ React2.createElement(TableCell, null, "Budget"), /* @__PURE__ */ React2.createElement(TableCell, null, "TPM Limit"), /* @__PURE__ */ React2.createElement(TableCell, null, "Models"), /* @__PURE__ */ React2.createElement(TableCell, { align: "right" }, "Actions"))), /* @__PURE__ */ React2.createElement(TableBody, null, loading ? /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, { colSpan: 8, align: "center" }, /* @__PURE__ */ React2.createElement(CircularProgress, { size: 24 }))) : keys.length === 0 ? /* @__PURE__ */ React2.createElement(TableRow, null, /* @__PURE__ */ React2.createElement(TableCell, { colSpan: 8, align: "center" }, /* @__PURE__ */ React2.createElement(Typography2, { color: "text.secondary" }, "No keys found"))) : keys.map((key) => {
+        const keyId = key.token ?? key.key;
+        return /* @__PURE__ */ React2.createElement(TableRow, { key: keyId }, /* @__PURE__ */ React2.createElement(TableCell, null, key.key_alias || "-"), /* @__PURE__ */ React2.createElement(TableCell, null, /* @__PURE__ */ React2.createElement(Box2, { display: "flex", alignItems: "center", gap: 0.5 }, /* @__PURE__ */ React2.createElement(
+          Typography2,
+          {
+            variant: "body2",
+            component: "code",
+            color: "text.secondary",
+            title: keyId,
+            sx: {
+              fontFamily: "monospace",
+              backgroundColor: "background.default",
+              px: 1,
+              py: 0.5,
+              borderRadius: 1
+            }
+          },
+          shortKeyId(keyId)
+        ), /* @__PURE__ */ React2.createElement(IconButton, { size: "small", onClick: () => copyToClipboard(keyId), title: "Copy Key ID" }, /* @__PURE__ */ React2.createElement(ContentCopy, { fontSize: "small" })))), /* @__PURE__ */ React2.createElement(TableCell, null, formatDate(key.created_at)), /* @__PURE__ */ React2.createElement(TableCell, null, "$", key.spend?.toFixed(4) || "0.00"), /* @__PURE__ */ React2.createElement(TableCell, null, key.max_budget ? `$${key.max_budget}` : "-"), /* @__PURE__ */ React2.createElement(TableCell, null, key.tpm_limit || "-"), /* @__PURE__ */ React2.createElement(TableCell, null, /* @__PURE__ */ React2.createElement(Box2, { display: "flex", gap: 0.5, flexWrap: "wrap" }, key.models?.slice(0, 2).map((model) => /* @__PURE__ */ React2.createElement(Chip2, { key: model, label: model, size: "small" })), (key.models?.length || 0) > 2 && /* @__PURE__ */ React2.createElement(Chip2, { label: `+${(key.models?.length || 0) - 2}`, size: "small", variant: "outlined" }))), /* @__PURE__ */ React2.createElement(TableCell, { align: "right" }, /* @__PURE__ */ React2.createElement(IconButton, { onClick: () => handleOpenEdit(key) }, /* @__PURE__ */ React2.createElement(Edit, { fontSize: "small" })), /* @__PURE__ */ React2.createElement(IconButton, { color: "error", onClick: () => onDeleteKey(key.key) }, /* @__PURE__ */ React2.createElement(Delete, null))));
+      }))))), /* @__PURE__ */ React2.createElement(Dialog, { open: generateModalOpen, onClose: handleCloseModal, maxWidth: "sm", fullWidth: true }, /* @__PURE__ */ React2.createElement(DialogTitle, null, newKeyValue ? "Key Generated" : "Generate New Key"), /* @__PURE__ */ React2.createElement(DialogContent, null, newKeyValue ? /* @__PURE__ */ React2.createElement(Box2, null, /* @__PURE__ */ React2.createElement(Typography2, { variant: "body2", color: "text.secondary", gutterBottom: true }, "Copy this key now. You won't be able to see it again."), /* @__PURE__ */ React2.createElement(
         Box2,
         {
           display: "flex",
@@ -312,7 +300,9 @@ var init_KeysTable = __esm({
           mt: 2,
           p: 2,
           sx: {
-            backgroundColor: "grey.100",
+            backgroundColor: "action.hover",
+            border: "1px solid",
+            borderColor: "divider",
             borderRadius: 1
           }
         },
@@ -332,6 +322,7 @@ var init_KeysTable = __esm({
           label: "Alias",
           value: formData.alias || "",
           onChange: (e) => setFormData({ ...formData, alias: e.target.value }),
+          required: true,
           fullWidth: true
         }
       ), /* @__PURE__ */ React2.createElement(
@@ -375,7 +366,7 @@ var init_KeysTable = __esm({
           value: selectedModels,
           onChange: (_e, selected) => setFormData({ ...formData, models: selected.map((m) => m.model_name) }),
           renderOption: (props, m) => /* @__PURE__ */ React2.createElement("li", { ...props }, m.model_name, m.supports_function_calling && " \u{1F527}", m.supports_vision && " \u{1F441}\uFE0F"),
-          renderInput: (params) => /* @__PURE__ */ React2.createElement(TextField, { ...params, label: "Models", fullWidth: true })
+          renderInput: (params) => /* @__PURE__ */ React2.createElement(TextField, { ...params, label: "Models", helperText: "Leave empty to allow all models", fullWidth: true })
         }
       ), /* @__PURE__ */ React2.createElement(
         TextField,
@@ -384,6 +375,7 @@ var init_KeysTable = __esm({
           type: "number",
           value: formData.max_budget ?? "",
           onChange: (e) => setFormData({ ...formData, max_budget: e.target.value ? Number(e.target.value) : void 0 }),
+          required: true,
           fullWidth: true
         }
       ), /* @__PURE__ */ React2.createElement(
@@ -958,7 +950,7 @@ init_TeamUsage();
 
 // src/components/LiteLLMHomeWidget.tsx
 init_api();
-import React7, { useState as useState5, useEffect as useEffect2 } from "react";
+import React7, { useState as useState5, useEffect } from "react";
 import {
   Paper as Paper6,
   Box as Box6,
@@ -997,7 +989,7 @@ var LiteLLMHomeWidget = ({
   const [error, setError] = useState5(null);
   const [usage, setUsage] = useState5(null);
   const [keys, setKeys] = useState5([]);
-  useEffect2(() => {
+  useEffect(() => {
     let cancelled = false;
     setLoading(true);
     setError(null);