@acarmisc/backstage-plugin-litellm-backend 0.6.1 → 0.7.0

package/config.d.ts

@@ -127,6 +127,27 @@ export interface Config {
       group?: string;
     };
 
+    /**
+     * Controls for the "Generate New Key" form in the frontend.
+     */
+    keyGeneration?: {
+      /**
+       * When true, users can tick an "Unlimited budget" checkbox that skips
+       * the max-budget cap entirely. The checkbox is hidden from the form
+       * when this is false, so a budget is always required.
+       * @default false
+       */
+      allowUnlimitedBudget?: boolean;
+
+      /**
+       * When true (the default), a team must be selected before a key can
+       * be generated. Set to false to let users generate personal,
+       * team-less keys.
+       * @default true
+       */
+      teamRequired?: boolean;
+    };
+
     bridge?: {
       /**
        * When true, mount the /bridge/keys, /bridge/keys (POST), /bridge/models

package/dist/index.cjs.js

@@ -511,12 +511,16 @@ var openApiSpec = {
     "/config": {
       get: {
         tags: ["System"],
-        summary: "Public LiteLLM proxy base URL (for snippet generation)",
+        summary: "Public LiteLLM proxy base URL and key-generation form settings",
         responses: {
           "200": {
-            description: "Proxy URL",
+            description: "Frontend config",
             content: { "application/json": { schema: { type: "object", properties: {
-              baseUrl: { type: "string" }
+              baseUrl: { type: "string" },
+              keyGeneration: { type: "object", properties: {
+                allowUnlimitedBudget: { type: "boolean" },
+                teamRequired: { type: "boolean" }
+              } }
             } } } }
           }
         }
@@ -2465,6 +2469,8 @@ async function createRouter(options) {
   const { enabled: provisioningEnabled, defaults: provisioningDefaults } = readProvisioningDefaults(config);
   const roleConfigs = readRoleConfigs(config);
   const auditGroup = config.getOptionalString("litellm.audit.group");
+  const allowUnlimitedBudget = config.getOptionalBoolean("litellm.keyGeneration.allowUnlimitedBudget") ?? false;
+  const teamRequired = config.getOptionalBoolean("litellm.keyGeneration.teamRequired") ?? true;
   const catalogClient = new import_catalog_client.CatalogClient({ discoveryApi: discovery });
   if (provisioningEnabled) {
     logger.info(
@@ -2477,7 +2483,10 @@ async function createRouter(options) {
     res.json({ status: "ok", provisioning: provisioningEnabled });
   });
   router.get("/config", (_req, res) => {
-    res.json({ baseUrl: publicBaseUrl });
+    res.json({
+      baseUrl: publicBaseUrl,
+      keyGeneration: { allowUnlimitedBudget, teamRequired }
+    });
   });
   router.get("/openapi.json", (_req, res) => {
     res.json(openApiSpec);
@@ -2662,6 +2671,13 @@ async function createRouter(options) {
         res.status(error.status).json(error.body);
         return;
       }
+      if (typeof error.message === "string" && error.message.includes("Invalid duration format") && req.body?.team_id) {
+        res.status(502).json({
+          error: 'LiteLLM rejected the key duration for this team. The team has a "Team Member Key Duration" set in LiteLLM that is not in a valid <number><unit> format (e.g. "30d") and overrides whatever duration is requested. Fix or clear it in LiteLLM under Teams \u2192 this team \u2192 Team Settings, then retry.',
+          teamId: req.body.team_id
+        });
+        return;
+      }
       logger.error("Failed to generate key", error);
       res.status(500).json({ error: error.message });
     }