Merge pull request #999 from mendableai/nsc/credit-usage-endpoint

Credit usage endpoint

apps/api/src/controllers/v1/credit-usage.ts

@@ -0,0 +1,45 @@
+import { Request, Response } from "express";
+import { RequestWithAuth } from "./types";
+import { getACUC } from "../auth";
+import { logger } from "../../lib/logger";
+
+export async function creditUsageController(
+  req: RequestWithAuth,
+  res: Response,
+): Promise<void> {
+  try {
+    // If we already have the credit usage info from auth, use it
+    if (req.acuc) {
+      res.json({
+        success: true,
+        data: {
+          remaining_credits: req.acuc.remaining_credits,
+        },
+      });
+      return;
+    }
+
+    // Otherwise fetch fresh data
+    const chunk = await getACUC(req.auth.team_id);
+    if (!chunk) {
+      res.status(404).json({
+        success: false,
+        error: "Could not find credit usage information",
+      });
+      return;
+    }
+
+    res.json({
+      success: true,
+      data: {
+        remaining_credits: chunk.remaining_credits,
+      },
+    });
+  } catch (error) {
+    logger.error("Error in credit usage controller:", error);
+    res.status(500).json({
+      success: false,
+      error: "Internal server error while fetching credit usage",
+    });
+  }
+}

apps/api/src/routes/v1.ts

@@ -31,6 +31,7 @@ import { extractController } from "../controllers/v1/extract";
 // import { keyAuthController } from "../../src/controllers/v1/keyAuth";
 // import { livenessController } from "../controllers/v1/liveness";
 // import { readinessController } from "../controllers/v1/readiness";
+import { creditUsageController } from "../controllers/v1/credit-usage";
 
 function checkCreditsMiddleware(
   minimum?: number,
@@ -224,3 +225,10 @@ v1Router.delete(
 // Health/Probe routes
 // v1Router.get("/health/liveness", livenessController);
 // v1Router.get("/health/readiness", readinessController);
+
+v1Router.get(
+  "/team/credit-usage",
+  authMiddleware(RateLimiterMode.CrawlStatus),
+  wrap(creditUsageController),
+);
+