@skydiveai/pi-server 0.1.0-beta.267 → 0.1.0-beta.445

package/dist/index.mjs

@@ -1379,6 +1379,51 @@ function create$2(options) {
 	};
 }
 //#endregion
+//#region src/billing-blocked-provider-signal.ts
+const BILLING_BLOCKED_PROVIDER_SIGNAL_PREFIX = "ANYONE_BILLING_BLOCKED_V1:";
+const payloadSchema = z.object({
+	blockReason: z.literal("insufficient_balance"),
+	message: z.string()
+}).strict();
+const nestedProxyEnvelopeSchema = z.object({
+	error: z.object({
+		type: z.literal("billing_blocked"),
+		code: z.literal("billing_blocked"),
+		message: z.string()
+	}).strict(),
+	blockReason: z.literal("insufficient_balance"),
+	message: z.string()
+}).strict();
+function parsePrefixedSignal(message) {
+	const normalized = message.startsWith("402 ") ? message.slice(4) : message;
+	if (!normalized.startsWith(BILLING_BLOCKED_PROVIDER_SIGNAL_PREFIX)) return null;
+	try {
+		const parsed = payloadSchema.safeParse(JSON.parse(normalized.slice(26)));
+		return parsed.success ? parsed.data : null;
+	} catch {
+		return null;
+	}
+}
+/**
+* Decode only the versioned platform envelope. Anthropic and OpenAI prefix the
+* nested error message with HTTP 402. Google's SDK instead preserves the full
+* proxy response as JSON, so that outer shape is validated separately.
+*/
+function parseBillingBlockedProviderSignal(message) {
+	const direct = parsePrefixedSignal(message);
+	if (direct) return direct;
+	try {
+		const envelope = nestedProxyEnvelopeSchema.safeParse(JSON.parse(message));
+		if (!envelope.success) return null;
+		const nested = parsePrefixedSignal(envelope.data.error.message);
+		const topLevel = parsePrefixedSignal(envelope.data.message);
+		if (!nested || nested.blockReason !== envelope.data.blockReason || envelope.data.message !== nested.message && (!topLevel || topLevel.blockReason !== nested.blockReason || topLevel.message !== nested.message)) return null;
+		return nested;
+	} catch {
+		return null;
+	}
+}
+//#endregion
 //#region src/protocols/chat-completions.ts
 /**
 * OpenAI Chat Completions–compatible protocol handler.
@@ -1922,6 +1967,22 @@ function runStream$1({ session, sm, prompt, images, sessionId, created, reqStart
 				postPrompt
 			});
 			if (capturedModelError !== null) {
+				const billingBlocked = parseBillingBlockedProviderSignal(capturedModelError);
+				if (billingBlocked) {
+					log.warn({
+						event: "billing_blocked",
+						chatcmpl_id: sessionId,
+						source: "stop_reason_error"
+					}, "forwarding platform billing block to client");
+					emitBillingBlocked({
+						writer,
+						sessionId,
+						created,
+						billingBlocked
+					});
+					writer.write("data: [DONE]\n\n");
+					return;
+				}
 				const providerError = toModelProviderError(new Error(capturedModelError));
 				log.warn({
 					event: "model_provider_error",
@@ -1957,6 +2018,17 @@ function runStream$1({ session, sm, prompt, images, sessionId, created, reqStart
 				err
 			}, "chat error");
 			if (isModelProviderError(err)) {
+				const billingBlocked = parseBillingBlockedProviderSignal(err instanceof Error ? err.message : String(err));
+				if (billingBlocked) {
+					emitBillingBlocked({
+						writer,
+						sessionId,
+						created,
+						billingBlocked
+					});
+					writer.write("data: [DONE]\n\n");
+					return;
+				}
 				const providerError = toModelProviderError(err);
 				log.warn({
 					event: "model_provider_error",
@@ -2030,6 +2102,23 @@ function emitModelProviderError({ writer, sessionId, created, providerError }) {
 	};
 	writer.write(`data: ${JSON.stringify(chunk)}\n\n`);
 }
+function emitBillingBlocked({ writer, sessionId, created, billingBlocked }) {
+	const chunk = {
+		id: sessionId,
+		object: "chat.completion.chunk",
+		created,
+		choices: [{
+			index: 0,
+			delta: {},
+			finish_reason: "stop"
+		}],
+		x_billing_blocked: {
+			block_reason: billingBlocked.blockReason,
+			message: billingBlocked.message
+		}
+	};
+	writer.write(`data: ${JSON.stringify(chunk)}\n\n`);
+}
 async function runBlocking$1({ session, sm, prompt, images, sessionId, created, reqStart, upstreamStartMs, log, postPrompt, registry, pendingSteerIds }) {
 	const baselineMessageCount = sm.buildSessionContext().messages.length;
 	let content = "";

package/package.json

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