@burdenoff/fe-libs 2026.1027.14 → 2026.1028.2

package/dist/shared/assistant/api.js

@@ -2,19 +2,20 @@ import { SandboxProxyError as e, sandboxProxyErrorFromBody as t, sandboxProxyErr
 import { extensionObjectsIn as r } from "./quota.js";
 import { abandonAssistantSandboxTurns as i, withAssistantProxyResponse as a } from "./promptDelivery.js";
 import { buildSandboxContextInput as o, executeAssistantGraphQLOnce as s, executeSandboxGraphQL as c } from "./gatewayClient.js";
-import { CHECK_AI_CREDITS as l, CREATE_SANDBOX as u, EXTEND_MY_ASSISTANT_SANDBOX_TTL as d, EXTEND_SANDBOX_TTL as f, GET_ACTIVE_SANDBOXES as p, GET_SANDBOX as ee, PROVISION_ASSISTANT_SANDBOX as te, PROXY_SANDBOX_REQUEST as ne, REPORT_AI_TOKEN_USAGE as re, STOP_ASSISTANT_SANDBOX as ie, STOP_MY_ASSISTANT_SANDBOX as ae, WAIT_FOR_ASSISTANT_READY as oe, WAIT_FOR_SANDBOX_RUNNING as se } from "./operations.js";
-import { ASSISTANT_API_AUTH_STRATEGY as ce, getImageForMode as m, getImageTagForMode as le, getResourcesForMode as ue, resolveSandboxAssistantMode as h } from "./images.js";
-import { STREAM_MINT_BACKOFF_ERROR_MS as g, STREAM_MINT_BACKOFF_UNAVAILABLE_MS as _, backOffStreamMint as v, clearStreamMintBackoff as de, getCapability as y, isMissingGatewayFieldError as b, isUnknownFieldError as fe, markSupported as x, markUnsupported as S, streamMintBackedOff as pe } from "./capabilities.js";
+import { CHARGE_AI_INVOCATION as l, CHECK_AI_CREDITS as u, CHECK_AI_CREDITS_FOR as d, CREATE_SANDBOX as f, EXTEND_MY_ASSISTANT_SANDBOX_TTL as p, EXTEND_SANDBOX_TTL as m, GET_ACTIVE_SANDBOXES as h, GET_SANDBOX as g, PROVISION_ASSISTANT_SANDBOX as ee, PROXY_SANDBOX_REQUEST as te, REPORT_AI_TOKEN_USAGE as ne, STOP_ASSISTANT_SANDBOX as re, STOP_MY_ASSISTANT_SANDBOX as ie, WAIT_FOR_ASSISTANT_READY as ae, WAIT_FOR_SANDBOX_RUNNING as oe } from "./operations.js";
+import { ASSISTANT_TURN_OPERATION as se } from "./productAiCredits.js";
+import { ASSISTANT_API_AUTH_STRATEGY as ce, getImageForMode as _, getImageTagForMode as le, getResourcesForMode as ue, resolveSandboxAssistantMode as v } from "./images.js";
+import { STREAM_MINT_BACKOFF_ERROR_MS as y, STREAM_MINT_BACKOFF_UNAVAILABLE_MS as b, backOffStreamMint as x, clearStreamMintBackoff as de, getCapability as S, isMissingGatewayFieldError as C, isUnknownFieldError as fe, markSupported as w, markUnsupported as T, streamMintBackedOff as pe } from "./capabilities.js";
 import { filterAssistantPageContext as me } from "./pageContextKeys.js";
-import { useAssistantStore as C } from "./store.js";
+import { useAssistantStore as E } from "./store.js";
 import { REQUEST_TOO_LARGE_REFUSAL as he } from "./gatewayLimits.js";
 //#region src/shared/assistant/api.ts
-function w(e) {
+function D(e) {
 	return e === "assistant" || e === "api-calls";
 }
-function T(e, t, n, r, i, a, s) {
+function O(e, t, n, r, i, a, s) {
 	return {
-		query: ne,
+		query: te,
 		operationName: "ProxySandboxRequest",
 		variables: { input: {
 			sandboxId: e,
@@ -22,16 +23,16 @@ function T(e, t, n, r, i, a, s) {
 			method: r,
 			body: i,
 			context: o(t, s),
-			...w(a) ? {
+			...D(a) ? {
 				forwardedAuthToken: s?.accessToken ?? void 0,
 				forwardedWorkspaceToken: s?.workspaceToken ?? void 0
 			} : {}
 		} }
 	};
 }
-async function E(e, n, r, i, o, l, u, d) {
+async function k(e, n, r, i, o, l, u, d) {
 	let f = (await (d === "transport" ? c : s)({
-		...T(e, n, r, i, o, l, u),
+		...O(e, n, r, i, o, l, u),
 		workspaceId: n,
 		authContext: u
 	})).proxySandboxRequest;
@@ -41,9 +42,9 @@ async function E(e, n, r, i, o, l, u, d) {
 	}
 	return f.body;
 }
-async function D(e, t, n, r, i, a, o) {
+async function A(e, t, n, r, i, a, o) {
 	let c = (await s({
-		...T(e, t, n, r, i, a, o),
+		...O(e, t, n, r, i, a, o),
 		workspaceId: t,
 		authContext: o
 	})).proxySandboxRequest;
@@ -54,43 +55,43 @@ async function D(e, t, n, r, i, a, o) {
 	};
 }
 async function ge(e, t, n, r, i, a = /* @__PURE__ */ new Set(), o = "legacy") {
-	return (await _e(e, t, n, r, i, a, o))?.id ?? null;
+	return (await j(e, t, n, r, i, a, o))?.id ?? null;
 }
-async function _e(e, t, n, r, i, a = /* @__PURE__ */ new Set(), s = "legacy") {
+async function j(e, t, n, r, i, a = /* @__PURE__ */ new Set(), s = "legacy") {
 	let l = (await c({
-		query: p,
+		query: h,
 		operationName: "GetActiveSandboxes",
 		variables: { context: o(e, i) },
 		workspaceId: e,
 		authContext: i
 	})).getActiveSandboxes ?? [];
-	L(l);
-	let u = k(l, t, n, r, i, a, B(s));
+	V(l);
+	let u = N(l, t, n, r, i, a, W(s));
 	return u ? {
 		id: u,
-		status: O(l, u)
+		status: M(l, u)
 	} : null;
 }
-function O(e, t) {
+function M(e, t) {
 	return e.find((e) => e.id === t)?.status ?? "UNKNOWN";
 }
-function k(e, t, n, r, i, a = /* @__PURE__ */ new Set(), o = "legacy") {
-	return e.find((e) => !a.has(e.id) && (e.status === "RUNNING" || e.status === "PROVISIONING" || e.status === "PENDING") && e.template === "AI_BUILDER" && h(e, [n]) === n && (!i.userId || e.createdBy === i.userId) && e.metadata?.authStrategy === "sandbox-app-client-headers-v1" && e.metadata?.assistantProduct === t && (!N(n) || M(e.metadata, o)) && A(e, n, o))?.id ?? null;
+function N(e, t, n, r, i, a = /* @__PURE__ */ new Set(), o = "legacy") {
+	return e.find((e) => !a.has(e.id) && (e.status === "RUNNING" || e.status === "PROVISIONING" || e.status === "PENDING") && e.template === "AI_BUILDER" && v(e, [n]) === n && (!i.userId || e.createdBy === i.userId) && e.metadata?.authStrategy === "sandbox-app-client-headers-v1" && e.metadata?.assistantProduct === t && (!L(n) || I(e.metadata, o)) && P(e, n, o))?.id ?? null;
 }
-function A(e, t, n) {
-	return t !== "assistant" || n !== "proposal-v1" ? !0 : I(e);
+function P(e, t, n) {
+	return t !== "assistant" || n !== "proposal-v1" ? !0 : B(e);
 }
-function j(e) {
+function F(e) {
 	return typeof e == "object" && !!e && !Array.isArray(e);
 }
-function M(e, t) {
+function I(e, t) {
 	let n = e?.assistantAuth;
-	return j(n) && n.accessProfile === "member" ? !0 : (e?.writeApproval ?? "legacy") === t ? e?.confirmWrites === (t === "proposal-v1" ? "proposal-v1" : !0) : !1;
+	return F(n) && n.accessProfile === "member" ? !0 : (e?.writeApproval ?? "legacy") === t ? e?.confirmWrites === (t === "proposal-v1" ? "proposal-v1" : !0) : !1;
 }
-function N(e) {
+function L(e) {
 	return e === "api-calls";
 }
-function P(e, t, n, r) {
+function R(e, t, n, r) {
 	let i = [
 		{
 			name: "AI_ASSISTANT_MODE",
@@ -118,46 +119,46 @@ function P(e, t, n, r) {
 	}), r.workspaceToken && i.push({
 		name: "WORKSPACE_AUTH_TOKEN",
 		value: r.workspaceToken
-	})), N(t) && i.push({
+	})), L(t) && i.push({
 		name: "AI_ASSISTANT_CONFIRM_WRITES",
 		value: "1"
 	}), i;
 }
-var ve = "assistant.errors.assistantProvisioningUnavailable", ye = "provisionAssistantSandbox", be = [
+var _e = "assistant.errors.assistantProvisioningUnavailable", ve = "provisionAssistantSandbox", ye = [
 	"ProvisionAssistantSandboxInput",
 	"SandboxAssistantMode",
 	"SandboxAssistantWriteApproval"
-], F = /* @__PURE__ */ new Set();
-function I(e) {
+], z = /* @__PURE__ */ new Set();
+function B(e) {
 	let t = e.metadata?.assistantAuth;
-	return j(t) && t.provisionedVia === ye;
+	return F(t) && t.provisionedVia === ve;
 }
-function L(e) {
-	for (let t of e) I(t) && F.add(t.id);
+function V(e) {
+	for (let t of e) B(t) && z.add(t.id);
 }
-function R(e) {
-	return F.has(e);
+function H(e) {
+	return z.has(e);
 }
-function z(e, t) {
-	return q(t) === "proposal-v1" && (e === "assistant" || e === "api-calls") && y("provisionAssistantSandbox") !== "unsupported";
+function U(e, t) {
+	return Y(t) === "proposal-v1" && (e === "assistant" || e === "api-calls") && S("provisionAssistantSandbox") !== "unsupported";
 }
-function B(e) {
-	return q(e) === "proposal-v1" && y("provisionAssistantSandbox") !== "unsupported" ? "proposal-v1" : "legacy";
+function W(e) {
+	return Y(e) === "proposal-v1" && S("provisionAssistantSandbox") !== "unsupported" ? "proposal-v1" : "legacy";
 }
-function V(e) {
-	return r(e).some((e) => e.i18nKey === ve || e.code === "SERVICE_UNAVAILABLE");
+function G(e) {
+	return r(e).some((e) => e.i18nKey === _e || e.code === "SERVICE_UNAVAILABLE");
 }
-function H(e, t) {
+function K(e, t) {
 	if (t <= 600) return !1;
 	let n = e instanceof Error ? e.message.toLowerCase() : String(e);
 	return n.includes("ttl") || n.includes("validation") || n.includes("maximum");
 }
-function U(e, t, n, r, i, a) {
+function q(e, t, n, r, i, a) {
 	if (t !== "assistant" && t !== "api-calls") throw Error(`Mode ${t} is not provisioned through the assistant path`);
 	return {
 		mode: t === "assistant" ? "ASSISTANT" : "API_CALLS",
 		product: e,
-		image: m(t, r),
+		image: _(t, r),
 		ttlSeconds: i,
 		...t === "api-calls" ? { writeApproval: "PROPOSAL_V1" } : {},
 		env: [
@@ -177,47 +178,47 @@ function U(e, t, n, r, i, a) {
 		context: o(n, a)
 	};
 }
-async function W(e, t, n, r, i, a) {
+async function J(e, t, n, r, i, a) {
 	let o = (await c({
-		query: te,
+		query: ee,
 		operationName: "ProvisionAssistantSandbox",
-		variables: { input: U(e, t, n, r, i, a) },
+		variables: { input: q(e, t, n, r, i, a) },
 		workspaceId: n,
 		authContext: a
 	}))?.provisionAssistantSandbox?.id;
 	if (!o) throw Error("provisionAssistantSandbox returned empty response");
 	return o;
 }
-async function xe(e, t, n, r, i, a) {
+async function be(e, t, n, r, i, a) {
 	try {
-		return await W(e, t, n, r, i, a);
+		return await J(e, t, n, r, i, a);
 	} catch (o) {
-		if (V(o) || G(o) || !H(o, i)) throw o;
-		return W(e, t, n, r, 600, a);
+		if (G(o) || xe(o) || !K(o, i)) throw o;
+		return J(e, t, n, r, 600, a);
 	}
 }
-function G(e) {
-	return b(e, "provisionAssistantSandbox", be);
+function xe(e) {
+	return C(e, "provisionAssistantSandbox", ye);
 }
 async function Se(e, t, n, r, i, a, o = {}) {
-	if (z(t, o.writeApproval)) try {
-		let o = await xe(e, t, n, r, i, a);
-		return x("provisionAssistantSandbox"), F.add(o), o;
+	if (U(t, o.writeApproval)) try {
+		let o = await be(e, t, n, r, i, a);
+		return w("provisionAssistantSandbox"), z.add(o), o;
 	} catch (e) {
-		if (!G(e)) throw e;
-		S("provisionAssistantSandbox");
+		if (!xe(e)) throw e;
+		T("provisionAssistantSandbox");
 	}
 	return Ce(e, t, n, r, i, a);
 }
 async function Ce(e, t, n, r, i, a) {
-	let s = ue(t), l = P(e, t, n, a), d = (i) => ({
+	let s = ue(t), l = R(e, t, n, a), u = (i) => ({
 		name: `ai-assistant-${t}-${Date.now()}`,
 		description: `AI Assistant sandbox (${t} mode)`,
 		template: "AI_BUILDER",
 		mode: "DEPLOYMENT",
 		containers: [{
 			name: "ai-assistant",
-			image: m(t, r),
+			image: _(t, r),
 			ports: [{
 				containerPort: 8080,
 				name: "http"
@@ -233,20 +234,20 @@ async function Ce(e, t, n, r, i, a) {
 			assistantOrganizationId: a.organizationId ?? null,
 			authStrategy: ce,
 			imageTag: le(t, r),
-			confirmWrites: N(t)
+			confirmWrites: L(t)
 		},
 		context: o(n, a)
-	}), f = (e) => c({
-		query: u,
+	}), d = (e) => c({
+		query: f,
 		operationName: "CreateSandbox",
-		variables: { input: d(e) },
+		variables: { input: u(e) },
 		workspaceId: n,
 		authContext: a
 	}), p;
 	try {
-		p = await f(i);
+		p = await d(i);
 	} catch (e) {
-		if (H(e, i)) p = await f(600);
+		if (K(e, i)) p = await d(600);
 		else throw e;
 	}
 	if (!p?.createSandbox?.id) throw Error("createSandbox returned empty response");
@@ -254,7 +255,7 @@ async function Ce(e, t, n, r, i, a) {
 }
 async function we(e, t, n) {
 	return (await c({
-		query: ee,
+		query: g,
 		operationName: "GetSandbox",
 		variables: {
 			id: e,
@@ -264,14 +265,14 @@ async function we(e, t, n) {
 		authContext: n
 	})).getSandbox;
 }
-function K(e, t) {
+function Te(e, t) {
 	if (e === "FAILED" || e === "STOPPED" || e === "EXPIRED") throw Error(t ? `Sandbox ${e.toLowerCase()}: ${t}` : `Sandbox ${e.toLowerCase()}`);
 }
-async function Te(e, t, n, r = 3e5) {
+async function Ee(e, t, n, r = 3e5) {
 	let i = Date.now();
-	if (y("waitForSandboxRunning") !== "unsupported") for (; Date.now() - i < r;) try {
+	if (S("waitForSandboxRunning") !== "unsupported") for (; Date.now() - i < r;) try {
 		let r = await c({
-			query: se,
+			query: oe,
 			operationName: "WaitForSandboxRunning",
 			variables: {
 				id: e,
@@ -281,13 +282,13 @@ async function Te(e, t, n, r = 3e5) {
 			workspaceId: t,
 			authContext: n
 		});
-		x("waitForSandboxRunning");
+		w("waitForSandboxRunning");
 		let i = r.waitForSandboxRunning;
 		if (i?.status === "RUNNING") return;
-		K(i?.status, i?.errorMessage);
+		Te(i?.status, i?.errorMessage);
 	} catch (e) {
 		if (fe(e)) {
-			S("waitForSandboxRunning");
+			T("waitForSandboxRunning");
 			break;
 		}
 		throw e;
@@ -295,17 +296,17 @@ async function Te(e, t, n, r = 3e5) {
 	for (; Date.now() - i < r;) {
 		let r = await we(e, t, n);
 		if (r?.status === "RUNNING") return;
-		K(r?.status, r?.errorMessage), await new Promise((e) => setTimeout(e, 2e3));
+		Te(r?.status, r?.errorMessage), await new Promise((e) => setTimeout(e, 2e3));
 	}
 	throw Error("Sandbox startup timed out");
 }
-async function Ee(e, t, n, r, i = 9e4, a) {
-	let s = Date.now(), l = (t) => (C.getState().setSandboxWriteCapabilities(e, Y(t)), t);
-	if (y("waitForAssistantReady") !== "unsupported") for (; Date.now() - s < i;) {
+async function De(e, t, n, r, i = 9e4, a) {
+	let s = Date.now(), l = (t) => (E.getState().setSandboxWriteCapabilities(e, X(t)), t);
+	if (S("waitForAssistantReady") !== "unsupported") for (; Date.now() - s < i;) {
 		let n = Date.now();
 		try {
 			let u = await c({
-				query: oe,
+				query: ae,
 				operationName: "WaitForAssistantReady",
 				variables: {
 					id: e,
@@ -315,20 +316,20 @@ async function Ee(e, t, n, r, i = 9e4, a) {
 				workspaceId: t,
 				authContext: r
 			});
-			x("waitForAssistantReady");
+			w("waitForAssistantReady");
 			let d = u.waitForAssistantReady;
 			if ((d.waited || !d.ready) && a?.(), d.ready && d.health) return l(d.health);
 			let f = Date.now(), p = i - (f - s);
 			f - n < 400 && p > 0 && await new Promise((e) => setTimeout(e, Math.min(400, p)));
 		} catch (e) {
-			if (!b(e, "waitForAssistantReady")) throw e;
-			S("waitForAssistantReady");
+			if (!C(e, "waitForAssistantReady")) throw e;
+			T("waitForAssistantReady");
 			break;
 		}
 	}
 	for (; Date.now() - s < i;) {
 		try {
-			return l(await E(e, t, "/health", "GET", void 0, n, r, "transport"));
+			return l(await k(e, t, "/health", "GET", void 0, n, r, "transport"));
 		} catch (e) {
 			let t = e instanceof Error ? e.message : String(e);
 			if (!/proxy error: 502|proxy error: 503|proxy error: 404|service not available|starting/i.test(t)) throw e;
@@ -338,69 +339,69 @@ async function Ee(e, t, n, r, i = 9e4, a) {
 	}
 	throw Error("Assistant service did not become healthy before timeout");
 }
-function q(e) {
+function Y(e) {
 	return e === "proposal-v1" ? "proposal-v1" : "legacy";
 }
-function J(e, t) {
+function Oe(e, t) {
 	return e === "enabled" || e === void 0 && !t ? "enabled" : "fenced";
 }
-var De = new Set([
+var ke = new Set([
 	"relay",
 	"delegated",
 	"direct",
 	"unconfigured"
 ]);
-function Y(e) {
+function X(e) {
 	let t = typeof e == "object" && e && !Array.isArray(e) ? e.capabilities : void 0, n = typeof t == "object" && t && !Array.isArray(t) ? t : {}, r = n.writeProposals !== void 0 || n.writeApprovalMode !== void 0 || n.writeExecution !== void 0, i = n.writeRiskVersion, a = n.sessionCredentials;
 	return {
 		proposals: n.writeProposals === "v1",
-		writeApprovalMode: q(n.writeApprovalMode),
-		writeExecution: J(n.writeExecution, r),
+		writeApprovalMode: Y(n.writeApprovalMode),
+		writeExecution: Oe(n.writeExecution, r),
 		replyApproval: "off",
 		...typeof i == "string" && i ? { writeRiskVersion: i } : {},
-		...typeof a == "string" && De.has(a) ? { sessionCredentials: a } : {},
+		...typeof a == "string" && ke.has(a) ? { sessionCredentials: a } : {},
 		...n.uiOffers === 1 ? { uiOffers: !0 } : {},
 		...n.clientTools === 1 ? { clientTools: !0 } : {}
 	};
 }
-async function Oe(e, t, n, r) {
+async function Ae(e, t, n, r) {
 	let i;
 	try {
-		i = await D(e, t, "/health", "GET", void 0, n, r);
+		i = await A(e, t, "/health", "GET", void 0, n, r);
 	} catch {
 		return null;
 	}
 	let a = i.body;
 	if (i.status < 200 || i.status >= 300 || typeof a != "object" || !a || Array.isArray(a)) return null;
-	let o = Y(a);
-	return C.getState().setSandboxWriteCapabilities(e, o), o;
+	let o = X(a);
+	return E.getState().setSandboxWriteCapabilities(e, o), o;
 }
-async function ke(e, t, n, r) {
-	return { sessionId: (await E(e, t, "/sessions", "POST", { mode: n }, n, r, "never")).session.id };
+async function je(e, t, n, r) {
+	return { sessionId: (await k(e, t, "/sessions", "POST", { mode: n }, n, r, "never")).session.id };
 }
-async function Ae(e, t, n, r, i) {
-	let a = await E(e, t, `/write-confirmations/${encodeURIComponent(n)}/approve`, "POST", {}, r, i, "never");
+async function Me(e, t, n, r, i) {
+	let a = await k(e, t, `/write-confirmations/${encodeURIComponent(n)}/approve`, "POST", {}, r, i, "never");
 	if (a?.approved !== !0) throw Error(typeof a?.error == "string" ? a.error : "The change could not be approved. It may have expired — ask the assistant to try again.");
 }
-function X(e, t, n, r, i, a = /* @__PURE__ */ new Set(), o = "legacy") {
-	return i.userId ? e.filter((e) => e.metadata?.authStrategy === "sandbox-app-client-headers-v1" && e.createdBy === i.userId && !(!a.has(e.id) && e.metadata?.assistantProduct === t && h(e, [n]) === n && (!N(n) || M(e.metadata, o)) && A(e, n, o))).map((e) => e.id) : [];
+function Ne(e, t, n, r, i, a = /* @__PURE__ */ new Set(), o = "legacy") {
+	return i.userId ? e.filter((e) => e.metadata?.authStrategy === "sandbox-app-client-headers-v1" && e.createdBy === i.userId && !(!a.has(e.id) && e.metadata?.assistantProduct === t && v(e, [n]) === n && (!L(n) || I(e.metadata, o)) && P(e, n, o))).map((e) => e.id) : [];
 }
-async function je(e, t, n, r, i, a = /* @__PURE__ */ new Set(), s = "legacy") {
+async function Pe(e, t, n, r, i, a = /* @__PURE__ */ new Set(), s = "legacy") {
 	let l, u = /* @__PURE__ */ new Set();
 	try {
 		let d = (await c({
-			query: p,
+			query: h,
 			operationName: "GetActiveSandboxes",
 			variables: { context: o(e, i) },
 			workspaceId: e,
 			authContext: i
 		})).getActiveSandboxes ?? [];
-		L(d);
-		let f = B(s);
-		if (l = X(d, t, n, r, i, a, f), f === "proposal-v1") {
+		V(d);
+		let f = W(s);
+		if (l = Ne(d, t, n, r, i, a, f), f === "proposal-v1") {
 			let e = new Set(l);
 			for (let t of d) {
-				let n = h(t);
+				let n = v(t);
 				e.has(t.id) && (n === "assistant" || n === "api-calls") && u.add(t.id);
 			}
 		}
@@ -409,16 +410,16 @@ async function je(e, t, n, r, i, a = /* @__PURE__ */ new Set(), s = "legacy") {
 	}
 	return (await Promise.all(l.map(async (t) => {
 		try {
-			return await Z(t, e, i, { ownerScoped: u.has(t) }), t;
+			return await Fe(t, e, i, { ownerScoped: u.has(t) }), t;
 		} catch {
 			return null;
 		}
 	}))).filter((e) => e !== null);
 }
-async function Z(e, t, n, r = {}) {
-	if (i(e), (r.ownerScoped === !0 || R(e)) && y("stopMyAssistantSandbox") !== "unsupported") try {
+async function Fe(e, t, n, r = {}) {
+	if (i(e), (r.ownerScoped === !0 || H(e)) && S("stopMyAssistantSandbox") !== "unsupported") try {
 		await c({
-			query: ae,
+			query: ie,
 			operationName: "StopMyAssistantSandbox",
 			variables: {
 				id: e,
@@ -426,14 +427,14 @@ async function Z(e, t, n, r = {}) {
 			},
 			workspaceId: t,
 			authContext: n
-		}), x("stopMyAssistantSandbox");
+		}), w("stopMyAssistantSandbox");
 		return;
 	} catch (e) {
-		if (!b(e, "stopMyAssistantSandbox")) throw e;
-		S("stopMyAssistantSandbox");
+		if (!C(e, "stopMyAssistantSandbox")) throw e;
+		T("stopMyAssistantSandbox");
 	}
 	await c({
-		query: ie,
+		query: re,
 		operationName: "StopAssistantSandbox",
 		variables: {
 			id: e,
@@ -443,46 +444,46 @@ async function Z(e, t, n, r = {}) {
 		authContext: n
 	});
 }
-var Q = /^[A-Za-z0-9_-]{1,128}$/, Me = new Set([
+var Ie = /^[A-Za-z0-9_-]{1,128}$/, Le = new Set([
 	"affirm",
 	"decline",
 	"other"
-]), Ne = new Set([
+]), Re = new Set([
 	"typed",
 	"spoken",
 	"chip",
 	"host_queued",
 	"retry",
 	"proposal_result"
-]), Pe = /^wp_[0-9a-f]{32}$/;
-function Fe(e) {
-	return !e || !Me.has(e.kind) || !Ne.has(e.origin) ? null : e.proposalId === void 0 ? {
+]), ze = /^wp_[0-9a-f]{32}$/;
+function Be(e) {
+	return !e || !Le.has(e.kind) || !Re.has(e.origin) ? null : e.proposalId === void 0 ? {
 		kind: e.kind,
 		origin: e.origin
-	} : Pe.test(e.proposalId) ? {
+	} : ze.test(e.proposalId) ? {
 		kind: e.kind,
 		origin: e.origin,
 		proposalId: e.proposalId
 	} : null;
 }
-function $(e) {
+function Z(e) {
 	let { mode: t, prompt: n, parts: r } = e, i = {
 		mode: t,
 		prompt: n,
 		pageContext: me(e.pageContext)
 	};
-	r && r.length > 0 && (i.parts = r), typeof e.conversationId == "string" && Q.test(e.conversationId) && (i.conversationId = e.conversationId);
-	let a = Fe(e.writeReply);
+	r && r.length > 0 && (i.parts = r), typeof e.conversationId == "string" && Ie.test(e.conversationId) && (i.conversationId = e.conversationId);
+	let a = Be(e.writeReply);
 	return a && (i.writeReply = a), e.uiCatalogue && (i.uiCatalogue = e.uiCatalogue), e.clientTools && e.clientTools.length > 0 && (i.clientTools = e.clientTools), i;
 }
-function Ie(e) {
-	return T(e.sandboxId, e.workspaceId, `/sessions/${e.sessionId}/prompt-async`, "POST", $(e), e.mode, e.authContext);
+function Ve(e) {
+	return O(e.sandboxId, e.workspaceId, `/sessions/${e.sessionId}/prompt-async`, "POST", Z(e), e.mode, e.authContext);
 }
-function Le(e) {
-	return new TextEncoder().encode(JSON.stringify(Ie(e))).byteLength;
+function Q(e) {
+	return new TextEncoder().encode(JSON.stringify(Ve(e))).byteLength;
 }
-async function Re(e, t, n, r, i, a, o, s) {
-	if (Le({
+async function He(e, t, n, r, i, a, o, s) {
+	if (Q({
 		sandboxId: e,
 		workspaceId: t,
 		sessionId: n,
@@ -496,7 +497,7 @@ async function Re(e, t, n, r, i, a, o, s) {
 		uiCatalogue: s?.uiCatalogue,
 		clientTools: s?.clientTools
 	}) > 262144) throw Object.assign(Error(he), { assistantPromptNotDispatched: !0 });
-	let c = await E(e, t, `/sessions/${n}/prompt-async`, "POST", $({
+	let c = await k(e, t, `/sessions/${n}/prompt-async`, "POST", Z({
 		mode: i,
 		prompt: r,
 		pageContext: a,
@@ -508,16 +509,16 @@ async function Re(e, t, n, r, i, a, o, s) {
 	}), i, o, "never");
 	return { messageID: typeof c.messageID == "string" ? c.messageID : void 0 };
 }
-async function ze(e, t, r, i, a, o) {
-	let s = `/sessions/${r}/messages?limit=${o?.limit ?? 50}`, c = Number.isFinite(o?.sinceMs) && (o?.sinceMs ?? 0) > 0 && y("sinceParam") !== "unsupported", l = c, u = (n) => E(e, t, n, "GET", void 0, i, a, "transport"), d;
+async function Ue(e, t, r, i, a, o) {
+	let s = `/sessions/${r}/messages?limit=${o?.limit ?? 50}`, c = Number.isFinite(o?.sinceMs) && (o?.sinceMs ?? 0) > 0 && S("sinceParam") !== "unsupported", l = c, u = (n) => k(e, t, n, "GET", void 0, i, a, "transport"), d;
 	try {
 		d = await u(c ? `${s}&since=${o?.sinceMs}` : s);
 	} catch (e) {
 		let t = typeof e == "object" && e && "assistantProxyResponse" in e ? e.assistantProxyResponse : null;
 		if (!c || typeof t != "object" || !t || !("status" in t) || ![400, 422].includes(Number(t.status)) || !("errorBody" in t) || typeof t.errorBody != "string" || !/\b(?:unknown|unsupported|unrecognized)\b.{0,40}\b(?:parameter|argument)\b.{0,20}["']?since\b/i.test(t.errorBody)) throw e;
-		S("sinceParam"), l = !1, d = await u(s);
+		T("sinceParam"), l = !1, d = await u(s);
 	}
-	if (l && (typeof d.now == "number" && Number.isFinite(d.now) ? x("sinceParam") : S("sinceParam")), o?.surfaceUiOffers) try {
+	if (l && (typeof d.now == "number" && Number.isFinite(d.now) ? w("sinceParam") : T("sinceParam")), o?.surfaceUiOffers) try {
 		o.surfaceUiOffers(d.uiOffers, typeof d.now == "number" ? d.now : NaN);
 	} catch {}
 	if (o?.surfaceClientToolRequests) try {
@@ -529,41 +530,41 @@ async function ze(e, t, r, i, a, o) {
 	}
 	return d.messages ?? [];
 }
-async function Be(t, n, r, i, a, o, s) {
+async function We(t, n, r, i, a, o, s) {
 	try {
-		await E(t, n, `/sessions/${r}/client-tools/${i}/result`, "POST", { result: a }, o, s, "never");
+		await k(t, n, `/sessions/${r}/client-tools/${i}/result`, "POST", { result: a }, o, s, "never");
 	} catch (t) {
 		if (t instanceof e && t.status === 409) return;
 		throw t;
 	}
 }
-async function Ve(e, t, n, r, i) {
-	if (y("sse") === "unsupported") return null;
+async function Ge(e, t, n, r, i) {
+	if (S("sse") === "unsupported") return null;
 	try {
-		let a = await E(e, t, `/sessions/${n}/stream-url`, "GET", void 0, r, i, "transport");
-		return typeof a.url == "string" && typeof a.expiresAt == "number" ? (x("sse"), {
+		let a = await k(e, t, `/sessions/${n}/stream-url`, "GET", void 0, r, i, "transport");
+		return typeof a.url == "string" && typeof a.expiresAt == "number" ? (w("sse"), {
 			url: a.url,
 			expiresAt: a.expiresAt
-		}) : (S("sse"), null);
+		}) : (T("sse"), null);
 	} catch {
-		return S("sse"), null;
+		return T("sse"), null;
 	}
 }
-async function He(e, t, n, r, i, a = Date.now) {
+async function Ke(e, t, n, r, i, a = Date.now) {
 	if (pe(a())) return null;
 	try {
-		let { status: o, body: s } = await D(e, t, `/sessions/${n}/stream-url`, "GET", void 0, r, i);
-		if (o === 404) return v(_, a()), null;
+		let { status: o, body: s } = await A(e, t, `/sessions/${n}/stream-url`, "GET", void 0, r, i);
+		if (o === 404) return x(b, a()), null;
 		let c = o >= 200 && o < 300 && typeof s == "object" && s ? s : null;
-		return c && c.transport === "relay" && typeof c.url == "string" && typeof c.expiresAt == "number" && Ue(c.url) ? (de(), {
+		return c && c.transport === "relay" && typeof c.url == "string" && typeof c.expiresAt == "number" && qe(c.url) ? (de(), {
 			url: c.url,
 			expiresAt: c.expiresAt
-		}) : (v(c ? _ : g, a()), null);
+		}) : (x(c ? b : y, a()), null);
 	} catch {
-		return v(g, a()), null;
+		return x(y, a()), null;
 	}
 }
-function Ue(e) {
+function qe(e) {
 	try {
 		let { protocol: t } = new URL(e);
 		return t === "https:" || t === "http:";
@@ -571,10 +572,10 @@ function Ue(e) {
 		return !1;
 	}
 }
-async function We(e, t, n, r) {
-	if (R(e) && y("extendMyAssistantSandboxTTL") !== "unsupported") try {
+async function Je(e, t, n, r) {
+	if (H(e) && S("extendMyAssistantSandboxTTL") !== "unsupported") try {
 		await c({
-			query: d,
+			query: p,
 			operationName: "ExtendMyAssistantSandboxTtl",
 			variables: {
 				id: e,
@@ -583,14 +584,14 @@ async function We(e, t, n, r) {
 			},
 			workspaceId: t,
 			authContext: r
-		}), x("extendMyAssistantSandboxTTL");
+		}), w("extendMyAssistantSandboxTTL");
 		return;
 	} catch (e) {
-		if (!b(e, "extendMyAssistantSandboxTTL")) throw e;
-		S("extendMyAssistantSandboxTTL");
+		if (!C(e, "extendMyAssistantSandboxTTL")) throw e;
+		T("extendMyAssistantSandboxTTL");
 	}
 	await c({
-		query: f,
+		query: m,
 		operationName: "ExtendSandboxTtl",
 		variables: {
 			id: e,
@@ -601,28 +602,28 @@ async function We(e, t, n, r) {
 		authContext: r
 	});
 }
-async function Ge(e, t, n, r, i) {
-	await E(e, t, `/sessions/${n}/heartbeat`, "POST", void 0, r, i, "transport").catch(() => void 0);
+async function Ye(e, t, n, r, i) {
+	await k(e, t, `/sessions/${n}/heartbeat`, "POST", void 0, r, i, "transport").catch(() => void 0);
 }
-async function Ke(e, t, n, r, i) {
+async function Xe(e, t, n, r, i) {
 	try {
-		return (await E(e, t, `/sessions/${n}/abort`, "POST", void 0, r, i, "never"))?.aborted === !0;
+		return (await k(e, t, `/sessions/${n}/abort`, "POST", void 0, r, i, "never"))?.aborted === !0;
 	} catch {
 		return !1;
 	}
 }
-async function qe(e, t, n, r, i) {
-	return ((await E(e, t, `/sessions/${n}/artifacts`, "GET", void 0, r, i, "transport")).artifacts ?? []).filter((e) => !e.name.startsWith("."));
+async function Ze(e, t, n, r, i) {
+	return ((await k(e, t, `/sessions/${n}/artifacts`, "GET", void 0, r, i, "transport")).artifacts ?? []).filter((e) => !e.name.startsWith("."));
 }
-async function Je(e, t, n, r, i) {
-	let a = await E(e, t, `/sessions/${n}`, "GET", void 0, r, i, "transport");
+async function Qe(e, t, n, r, i) {
+	let a = await k(e, t, `/sessions/${n}`, "GET", void 0, r, i, "transport");
 	return {
 		id: a.session?.id ?? n,
 		mode: a.mode ?? "unknown"
 	};
 }
-async function Ye(e, t, n, r, i, a) {
-	let o = (await E(e, t, `/sessions/${n}/artifacts/download?path=${encodeURIComponent(r)}`, "GET", void 0, i, a, "transport")).artifact;
+async function $e(e, t, n, r, i, a) {
+	let o = (await k(e, t, `/sessions/${n}/artifacts/download?path=${encodeURIComponent(r)}`, "GET", void 0, i, a, "transport")).artifact;
 	if (!o) throw Error("Artifact download returned an empty response");
 	return {
 		filename: o.filename,
@@ -630,21 +631,52 @@ async function Ye(e, t, n, r, i, a) {
 		contentBase64: o.contentBase64
 	};
 }
-async function Xe(e, t) {
-	return (await c({
-		query: l,
+function $(e, t) {
+	return !e || t === "product-service" ? null : e;
+}
+async function et(e, t, n, r) {
+	let i = $(n, r);
+	return i ? !(await c({
+		query: d,
+		operationName: "CheckAiCreditsFor",
+		variables: { productKey: i },
+		workspaceId: e,
+		authContext: t
+	})).checkAiCreditsFor.exhausted : (await c({
+		query: u,
 		operationName: "CheckAiCredits",
 		variables: {},
 		workspaceId: e,
 		authContext: t
 	})).checkAiCredits;
 }
-async function Ze(e, t, n, r, i, a) {
-	let o = Math.max(0, Math.round(i.input ?? 0)), l = Math.max(0, Math.round(i.output ?? 0)), u = Math.max(0, Math.round(i.cache?.read ?? 0)), d = Math.max(0, Math.round(i.cache?.write ?? 0));
-	if (o + l + u + d === 0 || !r) return;
-	let f = !1;
+async function tt(e, t, n, r, i, a, o, u) {
+	let d = Math.max(0, Math.round(i.input ?? 0)), f = Math.max(0, Math.round(i.output ?? 0)), p = Math.max(0, Math.round(i.cache?.read ?? 0)), m = Math.max(0, Math.round(i.cache?.write ?? 0));
+	if (d + f + p + m === 0 || !r) return;
+	let h = $(o, u);
+	if (h) {
+		await c({
+			query: l,
+			operationName: "ChargeAiInvocation",
+			variables: { input: {
+				productKey: h,
+				sessionId: t,
+				messageId: n,
+				model: r,
+				inputTokens: d,
+				outputTokens: f,
+				cacheReadTokens: p,
+				cacheWriteTokens: m,
+				operationKey: se
+			} },
+			workspaceId: e,
+			authContext: a
+		});
+		return;
+	}
+	let g = !1;
 	if (i.cache) try {
-		f = (await s({
+		g = (await s({
 			query: "query AiCacheTokensSupported { aiCacheTokensSupported }",
 			operationName: "AiCacheTokensSupported",
 			variables: {},
@@ -653,17 +685,17 @@ async function Ze(e, t, n, r, i, a) {
 		})).aiCacheTokensSupported === !0;
 	} catch {}
 	await c({
-		query: re,
+		query: ne,
 		operationName: "ReportAiTokenUsage",
 		variables: { input: {
 			sessionId: t,
 			messageId: n,
 			model: r,
-			inputTokens: o,
-			outputTokens: l,
-			...f ? {
-				cacheReadTokens: u,
-				cacheWriteTokens: d
+			inputTokens: d,
+			outputTokens: f,
+			...g ? {
+				cacheReadTokens: p,
+				cacheWriteTokens: m
 			} : {}
 		} },
 		workspaceId: e,
@@ -671,4 +703,4 @@ async function Ze(e, t, n, r, i, a) {
 	});
 }
 //#endregion
-export { Q as PROMPT_CONVERSATION_ID_PATTERN, Ke as abortAssistantTurn, Ae as approvePendingWrite, A as askReuseMatchesProvisioning, P as buildAssistantSandboxEnv, Ie as buildPromptGatewayRequest, $ as buildPromptRequestBody, U as buildProvisionAssistantSandboxInput, Xe as checkAssistantAiCredits, Se as createAssistantSandbox, ke as createAssistantSession, Ye as downloadAssistantArtifact, We as extendAssistantSandboxTTL, ge as findExistingSandbox, _e as findExistingSandboxWithStatus, w as forwardsUserTokensToSandbox, qe as getAssistantArtifacts, ze as getAssistantMessages, Je as getAssistantSession, Ve as getAssistantStreamAccess, V as isAssistantProvisioningUnavailable, R as isDedicatedAssistantSandbox, I as isDedicatedAssistantSandboxRow, He as mintAssistantTurnStream, N as modeRequiresConfirmWrites, Be as postClientToolResult, Le as promptGatewayRequestBytes, D as proxySandboxRaw, Y as readSandboxWriteCapabilities, Oe as readSandboxWriteCapabilitiesOnce, je as releaseSupersededAssistantSandboxes, Ze as reportAssistantAiTokenUsage, O as sandboxStatusOf, k as selectExistingAssistantSandbox, X as selectSupersededAssistantSandboxes, Re as sendAssistantPromptAsync, Ge as sendHeartbeat, Z as stopReapedAssistantSandbox, z as usesDedicatedAssistantProvisioning, Ee as waitForAssistantServiceReady, Te as waitForSandboxRunning, q as writeApprovalModeOf, M as writeApprovalReuseMatches, J as writeExecutionOf };
+export { Ie as PROMPT_CONVERSATION_ID_PATTERN, Xe as abortAssistantTurn, Me as approvePendingWrite, P as askReuseMatchesProvisioning, R as buildAssistantSandboxEnv, Ve as buildPromptGatewayRequest, Z as buildPromptRequestBody, q as buildProvisionAssistantSandboxInput, et as checkAssistantAiCredits, Se as createAssistantSandbox, je as createAssistantSession, $e as downloadAssistantArtifact, Je as extendAssistantSandboxTTL, ge as findExistingSandbox, j as findExistingSandboxWithStatus, D as forwardsUserTokensToSandbox, Ze as getAssistantArtifacts, Ue as getAssistantMessages, Qe as getAssistantSession, Ge as getAssistantStreamAccess, G as isAssistantProvisioningUnavailable, H as isDedicatedAssistantSandbox, B as isDedicatedAssistantSandboxRow, Ke as mintAssistantTurnStream, L as modeRequiresConfirmWrites, We as postClientToolResult, Q as promptGatewayRequestBytes, A as proxySandboxRaw, X as readSandboxWriteCapabilities, Ae as readSandboxWriteCapabilitiesOnce, Pe as releaseSupersededAssistantSandboxes, tt as reportAssistantAiTokenUsage, M as sandboxStatusOf, N as selectExistingAssistantSandbox, Ne as selectSupersededAssistantSandboxes, He as sendAssistantPromptAsync, Ye as sendHeartbeat, $ as sharedAiCreditProduct, Fe as stopReapedAssistantSandbox, U as usesDedicatedAssistantProvisioning, De as waitForAssistantServiceReady, Ee as waitForSandboxRunning, Y as writeApprovalModeOf, I as writeApprovalReuseMatches, Oe as writeExecutionOf };