@skydiveai/pi-extensions 0.1.0-beta.2268 → 0.1.0-beta.2269

package/dist/index.mjs

@@ -1555,47 +1555,58 @@ var McpExtension = class {
 			if (outcome.error) summary.errors.push(outcome.error);
 			if (outcome.change === "added") summary.added.push(id);
 			else if (outcome.change === "refreshed") summary.refreshed.push(id);
-			if (outcome.tools) {
-				const advertised = outcome.tools.list.length;
-				const perServerRoom = Math.min(advertised, limits.maxToolsPerServer);
-				let registered = 0;
-				let serverTokens = 0;
-				let droppedReason = null;
-				for (const tool of outcome.tools.list) {
-					if (summary.totalTools >= limits.maxTotalTools) {
-						droppedReason = "total";
-						break;
-					}
-					if (registered >= perServerRoom) {
-						droppedReason = "per_server";
-						break;
-					}
-					const cost = estimateToolTokens(tool);
-					if (summary.tokensUsed + cost > tokenBudget && summary.totalTools > 0) {
-						droppedReason = "tokens";
-						break;
-					}
-					this.registerMcpTool({
-						pi,
-						serverId: id,
-						client: outcome.tools.client,
-						tool
-					});
-					registered++;
-					serverTokens += cost;
-					summary.totalTools++;
-					summary.tokensUsed += cost;
-				}
-				const dropped = advertised - registered;
-				if (dropped > 0) summary.droppedTools += dropped;
-				else droppedReason = null;
-				summary.toolCounts[id] = {
-					advertised,
-					registered,
-					tokens: serverTokens,
-					droppedReason
-				};
+		}
+		const alloc = outcomes.filter((o) => Boolean(o.tools)).map((o) => {
+			const advertised = o.tools.list.length;
+			return {
+				id: o.id,
+				client: o.tools.client,
+				list: o.tools.list,
+				advertised,
+				room: Math.min(advertised, limits.maxToolsPerServer),
+				registered: 0,
+				serverTokens: 0,
+				droppedReason: null
+			};
+		});
+		const maxAdvertised = alloc.reduce((m, s) => Math.max(m, s.advertised), 0);
+		outer: for (let i = 0; i < maxAdvertised; i++) for (const s of alloc) {
+			if (summary.totalTools >= limits.maxTotalTools) {
+				for (const t of alloc) if (t.registered < t.advertised && t.droppedReason === null) t.droppedReason = "total";
+				break outer;
+			}
+			if (i >= s.room) {
+				if (s.registered < s.advertised && s.droppedReason === null) s.droppedReason = "per_server";
+				continue;
+			}
+			const tool = s.list[i];
+			if (tool === void 0) continue;
+			const cost = estimateToolTokens(tool);
+			if (summary.tokensUsed + cost > tokenBudget && summary.totalTools > 0) {
+				for (const t of alloc) if (t.registered < t.advertised && t.droppedReason === null) t.droppedReason = "tokens";
+				break outer;
 			}
+			this.registerMcpTool({
+				pi,
+				serverId: s.id,
+				client: s.client,
+				tool
+			});
+			s.registered++;
+			s.serverTokens += cost;
+			summary.totalTools++;
+			summary.tokensUsed += cost;
+		}
+		for (const s of alloc) {
+			const dropped = s.advertised - s.registered;
+			const droppedReason = dropped > 0 ? s.droppedReason : null;
+			if (dropped > 0) summary.droppedTools += dropped;
+			summary.toolCounts[s.id] = {
+				advertised: s.advertised,
+				registered: s.registered,
+				tokens: s.serverTokens,
+				droppedReason
+			};
 		}
 		return summary;
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.0-beta.2268",
+  "version": "0.1.0-beta.2269",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",