@acarmisc/backstage-plugin-litellm 0.21.0 → 0.21.2

package/dist/budget.d.ts

@@ -41,12 +41,13 @@ export interface BudgetSummary {
 export declare function buildBudgetSummary(user: UserInfo | null, teams: TeamInfo[], keys: VirtualKey[], maxKeys?: number): BudgetSummary;
 /**
  * One-line summary of a `BudgetSummary`, for a collapsed/compact header:
- * how many concrete limits the user is subject to, and how close the
- * closest one is to its cap. `closestPct` is null when there are no limits.
+ * how many concrete limits the user is subject to, and the one closest to
+ * its cap — kept as the whole `BudgetLimit` so the caller can name its
+ * level (key / personal / team). `closest` is null when there are no limits.
  */
 export declare function budgetHeadline(summary: BudgetSummary): {
     count: number;
-    closestPct: number | null;
+    closest: BudgetLimit | null;
 };
 /** "30d" → "every 30 days"; "1d" → "daily". Returns null when duration is unset. */
 export declare function fmtBudgetDuration(duration?: string): string | null;

package/dist/index.cjs.js

@@ -2809,16 +2809,14 @@ function buildBudgetSummary(user, teams, keys, maxKeys = 3) {
   };
 }
 function budgetHeadline(summary) {
-  const pcts = [
-    ...summary.keys.map((k) => k.pct),
-    ...summary.user ? [summary.user.pct] : [],
-    ...summary.teams.map((t) => t.pct)
+  const all = [
+    ...summary.keys,
+    ...summary.user ? [summary.user] : [],
+    ...summary.teams
   ];
   const count = summary.keys.length + summary.hiddenBudgetedKeys + (summary.user ? 1 : 0) + summary.teams.length;
-  return {
-    count,
-    closestPct: pcts.length ? Math.max(...pcts) : null
-  };
+  const closest = all.length ? all.reduce((a, b) => b.pct > a.pct ? b : a) : null;
+  return { count, closest };
 }
 function fmtBudgetDuration(duration) {
   if (!duration) return null;
@@ -2857,7 +2855,7 @@ function levelToneColor(tone, theme) {
       return theme.palette.primary.main;
   }
 }
-var import_react7, import_Paper5, import_Box7, import_Typography7, import_CircularProgress4, import_Alert3, import_Collapse, import_IconButton4, import_icons_material5, import_styles4, import_core_plugin_api3, LevelFrame, EmptyLimitCard, LimitCard, TaggedLimit, FullBody, CompactBody, LiteLLMBudgetWidget;
+var import_react7, import_Paper5, import_Box7, import_Typography7, import_CircularProgress4, import_Alert3, import_Collapse, import_IconButton4, import_icons_material5, import_styles4, import_core_plugin_api3, import_core_components, LevelFrame, EmptyLimitCard, LimitCard, CLOSEST_LEVEL_LABEL, KEYS_PATH, MoreKeysNote, HierarchyNote, TaggedLimit, FullBody, CompactBody, LiteLLMBudgetWidget;
 var init_LiteLLMBudgetWidget = __esm({
   "src/components/LiteLLMBudgetWidget.tsx"() {
     "use strict";
@@ -2872,6 +2870,7 @@ var init_LiteLLMBudgetWidget = __esm({
     import_icons_material5 = require("@mui/icons-material");
     import_styles4 = require("@mui/material/styles");
     import_core_plugin_api3 = require("@backstage/core-plugin-api");
+    import_core_components = require("@backstage/core-components");
     init_api();
     init_format();
     init_ui();
@@ -2980,6 +2979,26 @@ var init_LiteLLMBudgetWidget = __esm({
         /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "caption", color: "text.secondary" }, fmtBudgetDuration(limit.budgetDuration) ? `resets ${fmtBudgetDuration(limit.budgetDuration)}` : "never resets")
       ));
     };
+    CLOSEST_LEVEL_LABEL = {
+      key: "key",
+      user: "personal",
+      team: "team"
+    };
+    KEYS_PATH = "/litellm?tab=keys";
+    MoreKeysNote = ({ count }) => /* @__PURE__ */ import_react7.default.createElement(import_core_components.Link, { to: KEYS_PATH, variant: "caption" }, "+", count, " more budgeted key", count > 1 ? "s" : "", " further from the cap \u2192");
+    HierarchyNote = ({ withReset }) => /* @__PURE__ */ import_react7.default.createElement(
+      import_Box7.default,
+      {
+        sx: (theme) => ({
+          px: 1.5,
+          py: 1.25,
+          borderRadius: 1.5,
+          border: "1px dashed",
+          borderColor: (0, import_styles4.alpha)(theme.palette.text.primary, 0.18)
+        })
+      },
+      /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "caption", color: "text.secondary", display: "block" }, /* @__PURE__ */ import_react7.default.createElement("b", null, "Order:"), " key \u2192 personal \u2192 team \u2192 global. The first cap you reach blocks the request; a key that belongs to a team uses the team's cap, not your personal one.", withReset && " A cap with a reset window drops to $0 when the window closes; without one it never resets.")
+    );
     TaggedLimit = ({ limit }) => {
       const tone = budgetTone2(limit.pct);
       return /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, null, /* @__PURE__ */ import_react7.default.createElement(
@@ -3009,7 +3028,7 @@ var init_LiteLLMBudgetWidget = __esm({
         },
         !hasBudgetedKeys && /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "No key has a budget, so no key-level cap applies."),
         summary.keys.map((k) => /* @__PURE__ */ import_react7.default.createElement(LimitCard, { key: k.sublabel ?? k.label, limit: k })),
-        summary.hiddenBudgetedKeys > 0 && /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "caption", color: "text.secondary" }, "+", summary.hiddenBudgetedKeys, " more budgeted key", summary.hiddenBudgetedKeys > 1 ? "s" : "", " further from the cap")
+        summary.hiddenBudgetedKeys > 0 && /* @__PURE__ */ import_react7.default.createElement(MoreKeysNote, { count: summary.hiddenBudgetedKeys })
       ), /* @__PURE__ */ import_react7.default.createElement(
         LevelFrame,
         {
@@ -3019,27 +3038,14 @@ var init_LiteLLMBudgetWidget = __esm({
           note: "skipped for team-bound keys"
         },
         summary.user ? /* @__PURE__ */ import_react7.default.createElement(LimitCard, { limit: summary.user }) : /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "No personal budget is set on your account.")
-      ), /* @__PURE__ */ import_react7.default.createElement(LevelFrame, { rank: 3, name: "Team", tagline: "shared cap for all keys in a team (if any)" }, summary.teams.length > 0 ? summary.teams.map((t) => /* @__PURE__ */ import_react7.default.createElement(LimitCard, { key: t.sublabel ?? t.label, limit: t })) : /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "No team you belong to has a budget.")), /* @__PURE__ */ import_react7.default.createElement(LevelFrame, { rank: 4, name: "Global", tagline: "proxy-wide cap set by your admin", isLast: true }, /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "Configured by your admin in the proxy config \u2014 it isn't visible from this page. When set, it limits spend across every request on the proxy.")), /* @__PURE__ */ import_react7.default.createElement(
-        import_Box7.default,
-        {
-          sx: (theme) => ({
-            mt: 2,
-            px: 1.5,
-            py: 1.25,
-            borderRadius: 1.5,
-            border: "1px dashed",
-            borderColor: (0, import_styles4.alpha)(theme.palette.text.primary, 0.18)
-          })
-        },
-        /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "caption", color: "text.secondary", display: "block" }, "The first cap a request runs into is the one that stops it. A limit with a reset window (e.g. daily or every 30 days) clears its spend to $0 when the window closes; without a window it never resets.")
-      ));
+      ), /* @__PURE__ */ import_react7.default.createElement(LevelFrame, { rank: 3, name: "Team", tagline: "shared cap for all keys in a team (if any)" }, summary.teams.length > 0 ? summary.teams.map((t) => /* @__PURE__ */ import_react7.default.createElement(LimitCard, { key: t.sublabel ?? t.label, limit: t })) : /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "No team you belong to has a budget.")), /* @__PURE__ */ import_react7.default.createElement(LevelFrame, { rank: 4, name: "Global", tagline: "proxy-wide cap set by your admin", isLast: true }, /* @__PURE__ */ import_react7.default.createElement(EmptyLimitCard, null, "Configured by your admin in the proxy config \u2014 it isn't visible from this page. When set, it limits spend across every request on the proxy.")), /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { mt: 2 } }, /* @__PURE__ */ import_react7.default.createElement(HierarchyNote, { withReset: true })));
     };
     CompactBody = ({ summary }) => {
       const { count } = budgetHeadline(summary);
       if (count === 0) {
         return /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "body2", color: "text.secondary", sx: { fontSize: 13 } }, "No budget limits apply to you right now.");
       }
-      return /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { display: "flex", flexDirection: "column", gap: 1.25 } }, summary.keys.map((k) => /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { key: k.sublabel ?? k.label, limit: k })), summary.hiddenBudgetedKeys > 0 && /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "caption", color: "text.secondary" }, "+", summary.hiddenBudgetedKeys, " more budgeted key", summary.hiddenBudgetedKeys > 1 ? "s" : "", " further from the cap"), summary.user && /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { limit: summary.user }), summary.teams.map((t) => /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { key: t.sublabel ?? t.label, limit: t })));
+      return /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { display: "flex", flexDirection: "column", gap: 1.25 } }, /* @__PURE__ */ import_react7.default.createElement(HierarchyNote, null), summary.keys.map((k) => /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { key: k.sublabel ?? k.label, limit: k })), summary.hiddenBudgetedKeys > 0 && /* @__PURE__ */ import_react7.default.createElement(MoreKeysNote, { count: summary.hiddenBudgetedKeys }), summary.user && /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { limit: summary.user }), summary.teams.map((t) => /* @__PURE__ */ import_react7.default.createElement(TaggedLimit, { key: t.sublabel ?? t.label, limit: t })));
     };
     LiteLLMBudgetWidget = ({
       title = "Budget Policy",
@@ -3086,7 +3092,9 @@ var init_LiteLLMBudgetWidget = __esm({
       );
       const headline = (0, import_react7.useMemo)(() => budgetHeadline(summary), [summary]);
       const limitsWord = `${headline.count} limit${headline.count === 1 ? "" : "s"}`;
-      const closestWord = headline.closestPct === null ? "" : ` \xB7 closest ${Math.round(headline.closestPct)}%`;
+      const closestWord = headline.closest === null ? "" : ` \xB7 closest: ${CLOSEST_LEVEL_LABEL[headline.closest.kind]} ${Math.round(
+        headline.closest.pct
+      )}%`;
       const summaryText = headline.count === 0 ? "no limits apply" : `${limitsWord}${closestWord}`;
       const body = /* @__PURE__ */ import_react7.default.createElement(import_react7.default.Fragment, null, loading && /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { display: "flex", justifyContent: "center", alignItems: "center", minHeight: 140 }, /* @__PURE__ */ import_react7.default.createElement(import_CircularProgress4.default, { size: 32 })), !loading && error && /* @__PURE__ */ import_react7.default.createElement(import_Alert3.default, { severity: "error", sx: { mt: 0.5 } }, error), !loading && !error && (compact ? /* @__PURE__ */ import_react7.default.createElement(CompactBody, { summary }) : /* @__PURE__ */ import_react7.default.createElement(FullBody, { summary })));
       return /* @__PURE__ */ import_react7.default.createElement(import_Paper5.default, { sx: { p: 2 } }, collapsible ? /* @__PURE__ */ import_react7.default.createElement(
@@ -3102,7 +3110,13 @@ var init_LiteLLMBudgetWidget = __esm({
             mb: expanded ? 1.5 : 0
           }
         },
-        /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { minWidth: 0, flex: 1, display: "flex", alignItems: "center", gap: 1, flexWrap: "wrap" } }, /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "h6", lineHeight: 1.2 }, title), !loading && !error && /* @__PURE__ */ import_react7.default.createElement(StatusPill, { label: summaryText, tone: budgetTone2(headline.closestPct ?? 0) })),
+        /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { minWidth: 0, flex: 1 } }, /* @__PURE__ */ import_react7.default.createElement(import_Typography7.default, { variant: "h6", lineHeight: 1.2 }, title), !loading && !error && /* @__PURE__ */ import_react7.default.createElement(import_Box7.default, { sx: { mt: 0.5 } }, /* @__PURE__ */ import_react7.default.createElement(
+          StatusPill,
+          {
+            label: summaryText,
+            tone: budgetTone2(headline.closest?.pct ?? 0)
+          }
+        ))),
         /* @__PURE__ */ import_react7.default.createElement(
           import_IconButton4.default,
           {
@@ -3836,7 +3850,7 @@ function initDateRange() {
   else start.setDate(start.getDate() - 7);
   return { start, end };
 }
-var import_react11, import_Box11, import_Snackbar2, import_Alert6, import_CircularProgress6, import_Typography11, import_Paper7, import_Tabs2, import_Tab2, import_Grid2, import_react_use3, import_core_plugin_api4, import_plugin_permission_react, PERIOD_LS_KEY2, LiteLLMPage;
+var import_react11, import_Box11, import_Snackbar2, import_Alert6, import_CircularProgress6, import_Typography11, import_Paper7, import_Tabs2, import_Tab2, import_Grid2, import_react_router_dom, import_react_use3, import_core_plugin_api4, import_plugin_permission_react, PERIOD_LS_KEY2, PAGE_TABS, isPageTab, LiteLLMPage;
 var init_LiteLLMPage = __esm({
   "src/components/LiteLLMPage.tsx"() {
     "use strict";
@@ -3850,6 +3864,7 @@ var init_LiteLLMPage = __esm({
     import_Tabs2 = __toESM(require("@mui/material/Tabs"));
     import_Tab2 = __toESM(require("@mui/material/Tab"));
     import_Grid2 = __toESM(require("@mui/material/Grid"));
+    import_react_router_dom = require("react-router-dom");
     import_react_use3 = require("react-use");
     import_core_plugin_api4 = require("@backstage/core-plugin-api");
     import_plugin_permission_react = require("@backstage/plugin-permission-react");
@@ -3865,6 +3880,8 @@ var init_LiteLLMPage = __esm({
     init_api();
     init_permissions();
     PERIOD_LS_KEY2 = "litellm_usage_period";
+    PAGE_TABS = ["overview", "keys", "teams", "models", "audit"];
+    isPageTab = (v) => !!v && PAGE_TABS.includes(v);
     LiteLLMPage = () => {
       const api = (0, import_core_plugin_api4.useApi)(liteLlmApiRef);
       const [dateRange, setDateRange] = (0, import_react11.useState)(initDateRange);
@@ -3875,7 +3892,25 @@ var init_LiteLLMPage = __esm({
           return "7d";
         }
       });
-      const [activeTab, setActiveTab] = (0, import_react11.useState)("overview");
+      const [searchParams, setSearchParams] = (0, import_react_router_dom.useSearchParams)();
+      const [activeTab, setActiveTab] = (0, import_react11.useState)(() => {
+        const t = searchParams.get("tab");
+        return isPageTab(t) ? t : "overview";
+      });
+      const selectTab = (0, import_react11.useCallback)(
+        (tab) => {
+          setActiveTab(tab);
+          setSearchParams(
+            (prev) => {
+              const next = new URLSearchParams(prev);
+              next.set("tab", tab);
+              return next;
+            },
+            { replace: true }
+          );
+        },
+        [setSearchParams]
+      );
       const [snackbar, setSnackbar] = (0, import_react11.useState)(null);
       const [generateDialogOpen, setGenerateDialogOpen] = (0, import_react11.useState)(false);
       const [manageTeam, setManageTeam] = (0, import_react11.useState)(null);
@@ -4082,7 +4117,7 @@ var init_LiteLLMPage = __esm({
         import_Tabs2.default,
         {
           value: activeTab,
-          onChange: (_, v) => setActiveTab(v),
+          onChange: (_, v) => selectTab(v),
           variant: "scrollable",
           scrollButtons: "auto",
           sx: {