@acarmisc/backstage-plugin-litellm-backend 0.1.7 → 0.1.10

package/dist/index.cjs.js

@@ -5658,6 +5658,12 @@ var LiteLLMClient = class {
       body: JSON.stringify({ json: request })
     });
   }
+  async updateKey(request) {
+    return this.request("/key/update", {
+      method: "POST",
+      body: JSON.stringify(request)
+    });
+  }
   async deleteKeys(request) {
     return this.request("/key/delete", {
       method: "POST",
@@ -5671,22 +5677,90 @@ var LiteLLMClient = class {
   async getTeamInfo(teamId) {
     return this.request(`/team/info?team_id=${encodeURIComponent(teamId)}`);
   }
-  async getUsage(startDate, endDate, userId, groupBy) {
-    const params = new URLSearchParams({ start_date: startDate, end_date: endDate });
+  emptyUsage() {
+    return {
+      total_spend: 0,
+      total_tokens: 0,
+      prompt_tokens: 0,
+      completion_tokens: 0,
+      usage_by_model: {},
+      daily_usage: []
+    };
+  }
+  /**
+   * Transforms LiteLLM's SpendAnalyticsPaginatedResponse into the flatter
+   * UsageMetrics shape consumed by the frontend charts.
+   */
+  transformDailyActivity(response) {
+    const results = Array.isArray(response?.results) ? response.results : [];
+    const meta = response?.metadata ?? {};
+    const daily_usage = results.map((r) => ({
+      date: r.date,
+      spend: r.metrics?.spend ?? 0,
+      total_tokens: r.metrics?.total_tokens ?? 0,
+      prompt_tokens: r.metrics?.prompt_tokens ?? 0,
+      completion_tokens: r.metrics?.completion_tokens ?? 0
+    })).sort((a, b) => a.date.localeCompare(b.date));
+    const usage_by_model = {};
+    for (const r of results) {
+      const models = r.breakdown?.models ?? {};
+      for (const [name, entry] of Object.entries(models)) {
+        const m = entry?.metrics ?? {};
+        const bucket = usage_by_model[name] ?? {
+          total_spend: 0,
+          total_tokens: 0,
+          prompt_tokens: 0,
+          completion_tokens: 0
+        };
+        bucket.total_spend += m.spend ?? 0;
+        bucket.total_tokens += m.total_tokens ?? 0;
+        bucket.prompt_tokens += m.prompt_tokens ?? 0;
+        bucket.completion_tokens += m.completion_tokens ?? 0;
+        usage_by_model[name] = bucket;
+      }
+    }
+    return {
+      total_spend: meta.total_spend ?? 0,
+      total_tokens: meta.total_tokens ?? 0,
+      prompt_tokens: meta.total_prompt_tokens ?? 0,
+      completion_tokens: meta.total_completion_tokens ?? 0,
+      usage_by_model,
+      daily_usage
+    };
+  }
+  async getUsage(startDate, endDate, userId, _groupBy) {
+    const params = new URLSearchParams({
+      start_date: startDate,
+      end_date: endDate,
+      page_size: "100"
+    });
     if (userId) params.append("user_id", userId);
-    if (groupBy) params.append("group_by", groupBy);
     try {
-      return await this.request(`/usage/keys?${params.toString()}`);
+      const response = await this.request(`/user/daily/activity?${params.toString()}`);
+      return this.transformDailyActivity(response);
     } catch (err) {
       if (err.status === 404 || err.message.includes("not found")) {
-        return { total_spend: 0, total_tokens: 0, prompt_tokens: 0, completion_tokens: 0, usage_by_model: {}, daily_usage: [] };
+        return this.emptyUsage();
       }
       throw err;
     }
   }
   async getTeamUsage(teamId, startDate, endDate) {
-    const params = new URLSearchParams({ start_date: startDate, end_date: endDate, team_id: teamId });
-    return this.request(`/usage/keys?${params.toString()}`);
+    const params = new URLSearchParams({
+      start_date: startDate,
+      end_date: endDate,
+      team_ids: teamId,
+      page_size: "100"
+    });
+    try {
+      const response = await this.request(`/team/daily/activity?${params.toString()}`);
+      return this.transformDailyActivity(response);
+    } catch (err) {
+      if (err.status === 404 || err.message.includes("not found")) {
+        return this.emptyUsage();
+      }
+      throw err;
+    }
   }
 };
 
@@ -5860,6 +5934,21 @@ async function createRouter(options) {
       res.status(500).json({ error: error.message });
     }
   });
+  router.post("/keys/:keyId/update", async (req, res) => {
+    try {
+      const { keyId } = req.params;
+      if (!keyId) {
+        res.status(400).json({ error: "keyId is required" });
+        return;
+      }
+      const request = { ...req.body, key: keyId };
+      const result = await client.updateKey(request);
+      res.json(result);
+    } catch (error) {
+      logger.error("Failed to update key", error);
+      res.status(500).json({ error: error.message });
+    }
+  });
   router.delete("/keys/:keyId", async (req, res) => {
     try {
       const { keyId } = req.params;