copilot-api-plus 1.2.36 → 1.2.38

package/dist/main.js

@@ -1046,6 +1046,7 @@ const DEFAULT_MAX_CONCURRENCY = 10;
 var ModelRouter = class {
 	config;
 	queues = /* @__PURE__ */ new Map();
+	requestCounts = /* @__PURE__ */ new Map();
 	constructor(config) {
 		this.config = config ? { ...config } : {
 			...DEFAULT_CONFIG,
@@ -1098,6 +1099,7 @@ var ModelRouter = class {
 		}
 		if (queue.active < maxConcurrency) {
 			queue.active++;
+			this.requestCounts.set(resolvedModel, (this.requestCounts.get(resolvedModel) ?? 0) + 1);
 			consola.debug(`Slot acquired for "${resolvedModel}": ${queue.active}/${maxConcurrency} active`);
 			return () => this.releaseSlot(resolvedModel);
 		}
@@ -1106,6 +1108,7 @@ var ModelRouter = class {
 		return new Promise((resolve) => {
 			currentQueue.waiters.push(() => {
 				currentQueue.active++;
+				this.requestCounts.set(resolvedModel, (this.requestCounts.get(resolvedModel) ?? 0) + 1);
 				consola.debug(`Queued slot acquired for "${resolvedModel}": ${currentQueue.active}/${maxConcurrency} active`);
 				resolve(() => this.releaseSlot(resolvedModel));
 			});
@@ -1126,16 +1129,22 @@ var ModelRouter = class {
 	*/
 	getStats() {
 		const stats = {};
-		for (const [model, queue] of this.queues) {
+		const allModels = new Set([...this.queues.keys(), ...this.requestCounts.keys()]);
+		for (const model of allModels) {
+			const queue = this.queues.get(model);
 			const maxConcurrency = this.config.concurrency[model] ?? this.config.concurrency["default"] ?? DEFAULT_MAX_CONCURRENCY;
 			stats[model] = {
-				active: queue.active,
-				queued: queue.waiters.length,
-				maxConcurrency
+				active: queue?.active ?? 0,
+				queued: queue?.waiters.length ?? 0,
+				maxConcurrency,
+				totalRequests: this.requestCounts.get(model) ?? 0
 			};
 		}
 		return stats;
 	}
+	resetStats() {
+		this.requestCounts.clear();
+	}
 	/**
 	* Update the model name mapping configuration.
 	*/
@@ -1659,6 +1668,10 @@ statsRoute.get("/", (c) => {
 		uptime: process.uptime()
 	});
 });
+statsRoute.delete("/", (c) => {
+	modelRouter.resetStats();
+	return c.json({ success: true });
+});
 //#endregion
 //#region src/routes/admin/route.ts
 const adminRoutes = new Hono();