dsh-workbuddy-connect 0.5.2 → 0.5.4

package/README.en.md

@@ -22,13 +22,15 @@ Both the CN **WorkBuddy** and the international **WorkBuddy AI** apps are suppor
 
 - **Status and detection**: Settings → Plugins → the matching card shows the account, token validity, remaining credit, and model offers. It also lets you refresh the model list manually and shows whether the current list came from the upstream or from the built-in fallback, and provides manual reasoning-level detection for eligible models.
 
+- **Enterprise credit**: on the CN product, enterprise accounts (non-empty `enterpriseId`) read their cycle quota from the enterprise billing endpoint, and the card shows an "enterprise quota" row with the cycle reset time.
+
 - **Rate**: every model name carries its credits multiplier (e.g. `GLM-5.2 · x0.79`, `Hy3 · x0.00`) in both the `/model` popup and the composer's model dropdown. The rate is display-only and never affects requests.
 
 - **Promo badges**: promo badges (`限时免费`, `夜间折扣`) ride the model name itself (e.g. `Hy4 preview · x0.00 · 限时免费`), visible wherever you pick a model; the status card also collects currently-discounted models. Per the WorkBuddy service data, synced each time DSH starts. The international version's promotions come from the service's `modelPromotions` (which carry an effective window). Once a promotion lapses its badge is withdrawn; because the service writes the discounted value into the model's own rate field, the original price cannot be reconstructed, so that model then reports "price unavailable — refresh to update" rather than repeating the discounted rate or claiming the model is free.
 
 ![Settings card showing the plugin](assets/2.png)
 
-The expanded card has three tabs: **Status** shows the account, token validity, total credit, catalog source, and reasoning-level detection; **Context** lists each model's context window (the international version distinguishes the default window from a larger selectable one); **Details** shows per-package credit and model offers. The CN and international versions each get their own card, showing their own account's information.
+The expanded card has three tabs: **Status** shows the account, token validity, total credit, catalog source, and reasoning-level detection; **Context** lists each model's context window (where the international version offers a larger declared window, the "Use the largest declared context window" switch lives here — it is **on by default**, so DSH sizes context compression to the largest window the upstream declares; turn it off to follow the upstream default instead, and the preference persists across restarts); **Details** shows per-package credit and model offers. The CN and international versions each get their own card, showing their own account's information.
 
 ![Settings card showing account and remaining credit](assets/3.png)
 
@@ -110,6 +112,7 @@ dsh plugin --profile web exec dsh-workbuddy-connect doctor --provider workbuddy-
 - **The international version's model catalog comes from the app's own interface**: the service splits it by User-Agent, which is a private implementation detail that a server-side change can break. When that happens the plugin degrades to this account's last successful catalog and then to its built-in roster, showing the source (live / saved / built-in), the fetch time, and the failure reason on the card — but long-term compatibility is not guaranteed. The CN version's catalog uses the same interface as the official CLI and is unaffected.
 - **International-version environments not yet covered**: on Windows / WSL / Linux no reliable source for the international app's version has been located yet, so the saved value or the built-in default is used. On macOS, real-shim checks covered complete GPT-family replies, tool calls, and continued turns.
 - **Behaviour change with no credentials**: a version whose app was never signed in — and that left no plugin-owned copy — no longer shows a model group. The CN version used to display a built-in fallback list, but every model on it failed when selected.
+- **The enterprise credit path currently covers the CN product only**: the international enterprise billing interface is unverified, so those accounts still read through the personal endpoint pending measurement. The enterprise branch could not be tested locally (the development machine holds a personal account); it was implemented from the official app's interface contract, and reports from enterprise users are welcome.
 - Relies on WorkBuddy client interfaces (not a public API); the plugin may need updates as WorkBuddy changes.
 
 ## Disclaimer

package/README.md

@@ -30,6 +30,8 @@
 
 - **信息查看与检测**：设置 → 插件 → 对应卡片可查看账号、令牌有效期、剩余积分和模型优惠；也可以手动刷新模型列表，并在卡片上看到当前列表来自上游还是内置兜底。对于可检测模型，也可以在这里手动检测推理档位。
 
+- **企业账号积分**：国内版企业账号（`enterpriseId` 非空）走企业专用计费接口读取周期额度，卡片显示「企业额度」与周期重置时间。
+
 - **费率比例**：模型选择列表里每个模型名后直接显示积分倍率（如 `GLM-5.2 · x0.79`、`Hy3 · x0.00`），`/model` 弹窗与输入框的模型下拉都能看到。倍率只是显示，不影响实际请求。
 
 
@@ -37,7 +39,7 @@
 
 ![设置卡片显示插件](assets/2.png)
 
-卡片展开后分为「状态 / 上下文 / 明细」三个标签：状态页展示账号、令牌有效期、合计积分、模型列表来源与推理档位检测；上下文页列出各模型的上下文窗口（国际版会区分默认窗口与可选的更大窗口）；明细页展示各套餐余量与模型优惠。国内版与国际版各有一张自己的卡片，各显示自己账号的信息。
+卡片展开后分为「状态 / 上下文 / 明细」三个标签：状态页展示账号、令牌有效期、合计积分、模型列表来源与推理档位检测；上下文页列出各模型的上下文窗口。国际版在上游声明了更大可选窗口时，可在这里切换「使用上游声明的最大上下文窗口」；该开关**默认开启**，DSH 会按上游声明的最大窗口安排上下文压缩；想改用上游的默认窗口就在这里关掉，偏好会持久化，重启后保持。明细页展示各套餐余量与模型优惠。国内版与国际版各有一张自己的卡片，各显示自己账号的信息。
 
 ![设置卡片显示账号与剩余积分](assets/3.png)
 
@@ -120,6 +122,7 @@ dsh plugin --profile web exec dsh-workbuddy-connect doctor --provider workbuddy-
 - **国际版的模型目录来自 App 界面接口**：服务端按 User-Agent 分流下发，属私有实现，上游改动可能使其失效。届时插件按「本账号上次成功目录 → 内置目录」降级，并在卡片上标明来源（实时 / 已保存 / 内置）、更新时间与失败原因，但不能保证长期兼容。国内版目录走官方 CLI 同款接口，不受此影响。
 - **国际版仍未覆盖的环境**：Windows / WSL / Linux 下国际版 App 的版本读取尚未找到可靠来源，会退回最近保存的版本或内置值。macOS 上已通过真实 shim 验证 GPT 系完整回复、工具调用与续轮。
 - **无凭据时的行为变化**：某版 App 从未登录、也没留下插件自留副本时，该版模型分组不再显示。此前国内版会显示一份内置兜底列表，但那些模型选了必然报错。
+- **企业账号积分目前仅覆盖国内版**：国际版企业账号的计费接口尚未验证，仍按个人版接口读取；待有实测结论后再扩展。企业账号分支在本机无法自测（开发机为个人账号），依据官方 App 的接口契约实现，欢迎企业账号用户反馈实测结果。
 - 依赖 WorkBuddy 客户端接口（非官方开放 API），WorkBuddy 更新后插件可能需要随之调整。
 
 ## 免责声明

package/lib/bin.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { E as WorkBuddyUpstreamClient, J as resolveAppVersion, b as WorkBuddyCredentialStore, f as WORKBUDDY_CONNECT_VERSION, g as FALLBACK_WORKBUDDY_MODELS, h as FALLBACK_WORKBUDDY_AI_MODELS, i as variantFor, l as readHostHeartbeat, n as CN_VARIANT, r as WORKBUDDY_VARIANTS, s as isHeartbeatProcessAlive, u as workbuddyHostHeartbeatPath } from "./variants-CExA7lJt.js";
+import { E as WorkBuddyUpstreamClient, J as resolveAppVersion, b as WorkBuddyCredentialStore, f as WORKBUDDY_CONNECT_VERSION, g as FALLBACK_WORKBUDDY_MODELS, h as FALLBACK_WORKBUDDY_AI_MODELS, i as variantFor, l as readHostHeartbeat, n as CN_VARIANT, r as WORKBUDDY_VARIANTS, s as isHeartbeatProcessAlive, u as workbuddyHostHeartbeatPath } from "./variants-XUStsapF.js";
 import { realpathSync } from "node:fs";
 import { fileURLToPath } from "node:url";
 //#region src/bin.ts
@@ -118,7 +118,13 @@ async function status(jsonOutput, variant) {
 	let credits;
 	try {
 		const credential = await store.current();
-		if (credential !== void 0) credits = { total: (await client.fetchCredits(credential)).total };
+		if (credential !== void 0) {
+			const fetched = await client.fetchCredits(credential);
+			credits = {
+				total: fetched.total,
+				...fetched.unlimited === true ? { unlimited: true } : {}
+			};
+		}
 	} catch (error) {
 		credits = {
 			total: 0,
@@ -138,6 +144,7 @@ async function status(jsonOutput, variant) {
 			...authStatus.domain === void 0 || authStatus.domain === "" ? {} : { domain: authStatus.domain },
 			source: authStatus.source,
 			credits: credits?.total,
+			...credits?.unlimited === true ? { creditsUnlimited: true } : {},
 			...credits?.error === void 0 ? {} : { creditsError: credits.error },
 			hostBundle: hostState
 		});
@@ -146,7 +153,7 @@ async function status(jsonOutput, variant) {
 	process.stdout.write([
 		`${variant.displayName} Connect: signed in${authStatus.nickname === void 0 ? "" : ` as ${authStatus.nickname}`}`,
 		...expiresAt === void 0 ? [] : [`Access token expires ${expiresAt} (refresh is automatic)`],
-		credits?.error === void 0 ? `Remaining credit: ${credits?.total ?? "unknown"}` : `Remaining credit: unavailable (${credits.error})`,
+		credits?.error !== void 0 ? `Remaining credit: unavailable (${credits.error})` : credits?.unlimited === true ? "Remaining credit: unlimited" : `Remaining credit: ${credits?.total ?? "unknown"}`,
 		`Host bundle: ${hostAlive ? `running (pid ${heartbeat.pid})` : hostState === "stale" ? "stale heartbeat (DSH process exited)" : "not started in this profile"}`,
 		"Client card: load failures are logged to the browser console only; the host provider is unaffected.",
 		""

package/lib/client.js

@@ -63,15 +63,17 @@ window.__ModuleLoader__.load({
 			statusPath: WORKBUDDY_STATUS_PATH,
 			probePath: WORKBUDDY_PROBE_PATH
 		};
-		/** Both cards, in display order. */
-		const CARD_VARIANTS = [CN_CARD_VARIANT, {
+		/** International WorkBuddy AI. */
+		const AI_CARD_VARIANT = {
 			id: "workbuddy-ai",
 			titleKey: "titleAI",
 			introKey: "introAI",
 			signedOutKey: "signedOutHintAI",
 			statusPath: WORKBUDDY_AI_STATUS_PATH,
 			probePath: WORKBUDDY_AI_PROBE_PATH
-		}];
+		};
+		/** Both cards, in display order. */
+		const CARD_VARIANTS = [CN_CARD_VARIANT, AI_CARD_VARIANT];
 		const POLL_INTERVAL_MS = 6e4;
 		const cardStyle = {
 			overflow: "hidden",
@@ -199,6 +201,22 @@ window.__ModuleLoader__.load({
 			lineHeight: "18px",
 			color: "var(--dsw-alias-label-tertiary)"
 		};
+		const contextPreferenceStyle = {
+			display: "flex",
+			alignItems: "flex-start",
+			gap: 9,
+			padding: "10px 12px",
+			border: "1px solid var(--dsw-alias-border-l2)",
+			borderRadius: 8,
+			color: "var(--dsw-alias-label-primary)",
+			fontSize: 13,
+			lineHeight: "20px"
+		};
+		const contextPreferenceCopyStyle = {
+			display: "flex",
+			flexDirection: "column",
+			gap: 2
+		};
 		const modelBadgeChipStyle = {
 			padding: "1px 8px",
 			borderRadius: 999,
@@ -336,6 +354,11 @@ window.__ModuleLoader__.load({
 				timeStyle: "short"
 			}).format(new Date(ms));
 		}
+		function formatCycleReset(time) {
+			const parsed = Date.parse(time);
+			if (!Number.isNaN(parsed)) return formatTime(parsed);
+			return time;
+		}
 		/**
 		* One billing package as a labeled progress bar.
 		*
@@ -345,7 +368,29 @@ window.__ModuleLoader__.load({
 		* honest "remaining N" line printed below it. Unknown size therefore renders the
 		* percent slot as unknown copy and an unfilled, indeterminate track.
 		*/
-		function CreditBar({ label, remain, size, t }) {
+		function CreditBar({ label, remain, size, unlimited, t }) {
+			if (unlimited === true) {
+				const quotaText = t("unlimitedQuota");
+				return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
+					style: quotaGroupStyle,
+					children: [
+						/* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
+							style: quotaLabelStyle,
+							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: label }), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: quotaText })]
+						}),
+						/* @__PURE__ */ (0, react_jsx_runtime.jsx)("div", {
+							style: progressTrackStyle,
+							role: "progressbar",
+							"aria-label": label,
+							"aria-valuetext": quotaText
+						}),
+						/* @__PURE__ */ (0, react_jsx_runtime.jsx)("p", {
+							style: bodyStyle,
+							children: quotaText
+						})
+					]
+				});
+			}
 			const sizeKnown = size > 0;
 			const detail = sizeKnown ? t("exactRemaining", {
 				remain: formatNumber(remain),
@@ -425,31 +470,56 @@ window.__ModuleLoader__.load({
 		* the default is the budget actually requested, while the larger value is a
 		* ceiling the upstream would accept.
 		*/
-		function ContextTable({ models, t }) {
+		function ContextTable({ models, t, useMaximumContextWindow, disabled, onUseMaximumContextWindow }) {
 			const known = (models ?? []).filter((model) => model.contextWindow !== void 0).sort((a, b) => b.contextWindow - a.contextWindow);
-			if (known.length === 0) return null;
+			const canSelectMaximum = known.some((model) => model.maxContextWindow !== void 0 && model.maxContextWindow > (model.defaultContextWindow ?? model.contextWindow ?? 0));
+			const showPreference = onUseMaximumContextWindow !== void 0 && (canSelectMaximum || useMaximumContextWindow === true);
+			if (known.length === 0 && !showPreference) return null;
 			return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
 				style: quotaListStyle,
-				children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("h3", {
-					style: quotaTitleStyle,
-					children: t("contextHeading")
-				}), known.map((model) => {
-					const capacity = model.contextWindow;
-					const alternative = model.maxContextWindow !== void 0 && model.maxContextWindow > capacity ? model.maxContextWindow : void 0;
-					return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
-						style: quotaLabelStyle,
-						children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: model.name }), /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
-							style: modelOfferStyle,
-							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
-								style: { textAlign: "right" },
-								children: formatTokens(capacity)
-							}), alternative === void 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+				children: [
+					/* @__PURE__ */ (0, react_jsx_runtime.jsx)("h3", {
+						style: quotaTitleStyle,
+						children: t("contextHeading")
+					}),
+					showPreference && onUseMaximumContextWindow !== void 0 ? /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("label", {
+						style: contextPreferenceStyle,
+						children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("input", {
+							type: "checkbox",
+							checked: useMaximumContextWindow === true,
+							disabled,
+							onChange: (event) => {
+								onUseMaximumContextWindow(event.currentTarget.checked);
+							}
+						}), /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+							style: contextPreferenceCopyStyle,
+							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: t("useMaximumContextWindow") }), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
 								style: modelRateStyle,
-								children: t("contextUpTo", { size: formatTokens(alternative) })
+								children: t("useMaximumContextWindowHint")
 							})]
 						})]
-					}, model.id);
-				})]
+					}) : null,
+					known.map((model) => {
+						const capacity = model.contextWindow;
+						const alternative = model.maxContextWindow !== void 0 && model.maxContextWindow > capacity ? model.maxContextWindow : void 0;
+						return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
+							style: quotaLabelStyle,
+							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: model.name }), /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+								style: modelOfferStyle,
+								children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+									style: { textAlign: "right" },
+									children: formatTokens(capacity)
+								}), alternative !== void 0 ? /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+									style: modelRateStyle,
+									children: t("contextUpTo", { size: formatTokens(alternative) })
+								}) : model.defaultContextWindow !== void 0 && model.defaultContextWindow < capacity ? /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+									style: modelRateStyle,
+									children: t("contextDefault", { size: formatTokens(model.defaultContextWindow) })
+								}) : null]
+							})]
+						}, model.id);
+					})
+				]
 			});
 		}
 		/**
@@ -791,6 +861,10 @@ window.__ModuleLoader__.load({
 						const message = typeof value === "object" && value !== null && "error" in value ? String(value["error"]) : `HTTP ${response.status}`;
 						throw new Error(message);
 					}
+					if (action.action === "set-maximum-context-window" && (typeof value !== "object" || value === null || value["state"] !== "updated")) {
+						const reason = typeof value === "object" && value !== null && "reason" in value ? String(value["reason"]) : t("requestFailed");
+						throw new Error(reason);
+					}
 					await refresh(controller.signal);
 				} catch (error) {
 					if (mounted.current && controller.signal.aborted !== true) setReadFailure(error instanceof Error ? error.message : t("requestFailed"));
@@ -923,18 +997,21 @@ window.__ModuleLoader__.load({
 							tab === "status" ? /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
 								style: tabPanelStyle,
 								children: [
-									status.credits === void 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsx)("div", {
+									status.credits === void 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
 										style: quotaListStyle,
-										children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
+										children: [/* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
 											style: rowStyle,
 											children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("h3", {
 												style: quotaTitleStyle,
 												children: t("creditsHeading")
 											}), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
 												style: bodyStyle,
-												children: t("creditsTotal", { total: formatNumber(status.credits.total) })
+												children: status.credits.unlimited === true ? t("creditsTotalUnlimited") : t("creditsTotal", { total: formatNumber(status.credits.total) })
 											})]
-										})
+										}), status.credits.cycleResetTime === void 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsx)("p", {
+											style: descriptionStyle,
+											children: t("cycleResetAt", { time: formatCycleReset(status.credits.cycleResetTime) })
+										})]
 									}),
 									status.creditsError === void 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsx)("p", {
 										style: errorStyle,
@@ -954,7 +1031,15 @@ window.__ModuleLoader__.load({
 								style: tabPanelStyle,
 								children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ContextTable, {
 									models: status.models,
-									t
+									t,
+									disabled: busy,
+									...status.useMaximumContextWindow === void 0 ? {} : { useMaximumContextWindow: status.useMaximumContextWindow },
+									...variant.id === AI_CARD_VARIANT.id ? { onUseMaximumContextWindow: (enabled) => {
+										control({
+											action: "set-maximum-context-window",
+											enabled
+										});
+									} } : {}
 								})
 							}) : /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
 								style: tabPanelStyle,
@@ -963,10 +1048,11 @@ window.__ModuleLoader__.load({
 									children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("h3", {
 										style: quotaTitleStyle,
 										children: t("creditsDetailHeading")
-									}), status.credits.accounts.filter((account) => account.remain > 0).map((account, index) => /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CreditBar, {
-										label: account.packageName,
+									}), status.credits.accounts.filter((account) => account.packageName === "enterprise" || account.remain > 0 || account.unlimited === true).map((account, index) => /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CreditBar, {
+										label: account.packageName === "enterprise" ? t("packageEnterprise") : account.packageName,
 										remain: account.remain,
 										size: account.size,
+										unlimited: account.unlimited,
 										t
 									}, `${account.packageName}-${String(index)}`))]
 								}), status.models === void 0 || status.models.length === 0 ? null : /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
@@ -1445,6 +1531,10 @@ window.__ModuleLoader__.load({
 			tabDetails: "Credit details",
 			creditsDetailHeading: "By package",
 			creditsTotal: "Total: {total}",
+			creditsTotalUnlimited: "Total: Unlimited",
+			unlimitedQuota: "Unlimited",
+			packageEnterprise: "Enterprise quota",
+			cycleResetAt: "Resets {time}",
 			percentRemaining: "{percent}% remaining",
 			percentUnknown: "Remaining share unknown",
 			exactRemaining: "{remain} / {size} remaining",
@@ -1466,6 +1556,9 @@ window.__ModuleLoader__.load({
 			modelsHeading: "Model offers",
 			contextHeading: "Context window",
 			contextUpTo: "up to {size}",
+			contextDefault: "default {size}",
+			useMaximumContextWindow: "Use the largest declared context window",
+			useMaximumContextWindowHint: "Applies to WorkBuddy AI models that offer a larger window.",
 			freeModel: "Free",
 			badgeLimitedFree: "Limited-time free",
 			badgeNightDiscount: "Night discount",
@@ -1521,6 +1614,10 @@ window.__ModuleLoader__.load({
 			tabDetails: "积分详情",
 			creditsDetailHeading: "按套餐",
 			creditsTotal: "合计：{total}",
+			creditsTotalUnlimited: "合计：不限额",
+			unlimitedQuota: "不限额",
+			packageEnterprise: "企业额度",
+			cycleResetAt: "重置时间：{time}",
 			percentRemaining: "剩余 {percent}%",
 			percentUnknown: "剩余占比未知",
 			exactRemaining: "剩余 {remain} / {size}",
@@ -1542,6 +1639,9 @@ window.__ModuleLoader__.load({
 			modelsHeading: "模型优惠",
 			contextHeading: "上下文窗口",
 			contextUpTo: "最高 {size}",
+			contextDefault: "默认 {size}",
+			useMaximumContextWindow: "使用上游声明的最大上下文窗口",
+			useMaximumContextWindowHint: "仅作用于 WorkBuddy AI 中声明了更大窗口的模型。",
 			freeModel: "免费",
 			badgeLimitedFree: "限时免费",
 			badgeNightDiscount: "夜间折扣",

package/lib/index.d.ts

@@ -225,6 +225,8 @@ interface WorkBuddyUpstreamModel {
   id: string;
   name: string;
   contextWindow: number;
+  /** The upstream's preferred window before an optional maximum is selected. */
+  defaultContextWindow?: number;
   maxInputTokens?: number;
   supportedContextWindows?: readonly number[];
   promotions?: readonly WorkBuddyPromotion[];
@@ -295,11 +297,24 @@ interface WorkBuddyCreditAccount {
   packageName: string;
   remain: number;
   size: number;
+  unlimited?: true;
 }
 /** Aggregated credit answer for one credential. */
 interface WorkBuddyCredits {
   total: number;
   accounts: readonly WorkBuddyCreditAccount[];
+  /**
+   * The account's cycle quota is uncapped (`limitNum === -1` on the CN
+   * enterprise endpoint).
+   *
+   * A separate flag rather than a `-1`/`0` sentinel in {@link total}: the two
+   * mean opposite things to a reader ("no limit" vs "nothing left"), and the
+   * existing negative-clamp in the personal branch would turn a sentinel into
+   * a plausible-looking zero. Every renderer must therefore test this flag
+   * first and not fall back to `total` when it is set.
+   */
+  unlimited?: true;
+  cycleResetTime?: string;
 }
 /** Token refresh answer; fields the upstream omits stay absent. */
 interface WorkBuddyRefreshOutcome {
@@ -409,8 +424,41 @@ declare class WorkBuddyUpstreamClient {
    * such rather than as a generic catalog failure.
    */
   fetchModels(credential: WorkBuddyCredential, signal?: AbortSignal): Promise<readonly WorkBuddyUpstreamModel[]>;
-  /** POST the billing endpoint for the aggregated remaining credit. */
+  /**
+   * POST the billing endpoint for the aggregated remaining credit.
+   *
+   * Two upstream shapes, chosen by account type:
+   *
+   * - **CN enterprise** (`regionOf === 'cn'` and `enterpriseId` non-empty) asks
+   *   `/v2/billing/meter/get-enterprise-user-usage`, which answers with a single
+   *   cycle quota. The personal endpoint serves these accounts an empty
+   *   `Accounts` list, which the card then renders as "0 credit" — a wrong
+   *   number rather than a visible failure (issue #31).
+   * - **Everyone else** keeps the personal endpoint unchanged.
+   *
+   * The region gate is load-bearing: the enterprise endpoint is unverified for
+   * the global region, so an international credential that happens to carry an
+   * `enterpriseId` must stay on the measured personal path instead of being
+   * moved onto an unmeasured one.
+   */
   fetchCredits(credential: WorkBuddyCredential): Promise<WorkBuddyCredits>;
+  /**
+   * CN enterprise credit read: a single cycle quota instead of a package list.
+   *
+   * Verified against the WorkBuddy desktop app (`app.asar`,
+   * `BackendProvider.getEnterpriseUsage` and `CloudAccountRepo.billing`): the
+   * body is an empty object and the account identity travels only in the
+   * headers. The two official call sites disagree on the field spelling
+   * (`limitNum`/`credit` vs `limit_num`/`used_num`), so both are accepted.
+   *
+   * A body carrying no recognisable quota field is a hard error rather than a
+   * zero. Rendering `0` for "we did not understand the answer" is exactly how
+   * issue #31 stayed invisible while users saw a plausible wrong number.
+   *
+   * The error names fields and types only: it reaches the browser, and the
+   * response body may describe the account's usage.
+   */
+  private fetchEnterpriseCredits;
   /**
    * One probe request: a real streaming chat call carrying the effort under
    * test.
@@ -704,6 +752,7 @@ declare const FALLBACK_WORKBUDDY_AI_MODELS: readonly WorkBuddyModelInfo[];
 declare class WorkBuddyCatalog {
   private models;
   private visible;
+  private useMaximumContextWindow;
   constructor(initial?: readonly WorkBuddyModelInfo[]);
   /** Current entries; empty while the variant has no usable credential. */
   current(): readonly WorkBuddyModelInfo[];
@@ -716,6 +765,8 @@ declare class WorkBuddyCatalog {
    * caller can skip an invalidation that would re-render an identical list.
    */
   setVisible(visible: boolean): boolean;
+  /** Select the largest declared international window where the upstream offers one. */
+  setUseMaximumContextWindow(useMaximum: boolean): boolean;
   /** Models to fall back to when the upstream fetch fails; ignores visibility. */
   fallback(): readonly WorkBuddyModelInfo[];
 }
@@ -750,12 +801,12 @@ interface WorkBuddyProbeRecord {
    *
    * An effort set is a fact about one account's entitlement as much as about
    * the model: the same model id can accept different levels under a different
-   * subscription. Without this a record outlived the account that produced it,
-   * so signing out and in as someone else inherited the previous account's
-   * detected levels. Records written before this field existed carry no
-   * identity and are therefore never reused.
+   * subscription. Records are stored under this identity and only ever served
+   * back to it, so one account never inherits another's detected levels — and
+   * because the store nests by this identity, switching back finds this
+   * account's own records intact rather than re-probing from scratch.
    */
-  account?: string;
+  account: string;
 }
 /**
  * Plugin-owned probe record path inside the Harness home.
@@ -788,8 +839,9 @@ interface WorkBuddyProbeStoreOptions {
   now?: () => number;
 }
 /**
- * The plugin's probe records: read once, written atomically, never trusted
- * across a fingerprint change or past the TTL.
+ * The plugin's probe records: read once, written atomically, keyed by the
+ * account that produced each observation, and never trusted across a
+ * fingerprint change or past the TTL.
  */
 declare class WorkBuddyProbeStore {
   private readonly path;
@@ -802,24 +854,25 @@ declare class WorkBuddyProbeStore {
   filePath(): string;
   private load;
   /**
-   * The usable record for a model, or `undefined` when there is none, it is
-   * expired, it was taken against a different catalog row, or it belongs to a
-   * different account.
+   * The usable record for one account and model, or `undefined` when there is
+   * none, it is expired, it was taken against a different catalog row, or it
+   * belongs to a different account.
    *
    * @param account - the account in effect, as `uid:enterpriseId`. Records are
    *   only returned for the account that produced them.
    */
   get(modelId: string, fingerprint: string, account: string): WorkBuddyProbeRecord | undefined;
   /**
-   * Store one observation. Only a decisive answer (`validating` /
-   * `non-validating`) replaces an existing decisive record: a transient
-   * `unknown` must not erase knowledge the user already paid for.
+   * Store one observation under the account stamped on it. Only a decisive
+   * answer (`validating` / `non-validating`) replaces an existing decisive
+   * record *of the same account*: a transient `unknown` must not erase
+   * knowledge the user already paid for.
    */
   set(modelId: string, record: WorkBuddyProbeRecord): void;
-  /** Drop every record; used by the card's explicit "clear" action. */
+  /** Drop every record of every account; used by the card's explicit "clear" action. */
   clear(): void;
-  /** Every record currently held, for status display. */
-  all(): Readonly<Record<string, WorkBuddyProbeRecord>>;
+  /** Every record currently held, grouped by account, for status display. */
+  all(): Readonly<Record<string, Readonly<Record<string, WorkBuddyProbeRecord>>>>;
   /** Build a record stamped with this store's clock, version, and account. */
   record(fingerprint: string, validation: WorkBuddyProbeValidation, efforts: readonly WorkBuddyEffort[], account: string): WorkBuddyProbeRecord;
   /**
@@ -981,8 +1034,8 @@ interface WorkBuddyProbeServiceOptions {
    * Records are read and written against this identity, and it is re-checked
    * after the sweep finishes: an observation produced under account A must not
    * be stored once account B is in effect, however long the probe took. The
-   * caller's `clear()` on an account switch is not sufficient on its own,
-   * because an in-flight probe completes *after* that clear.
+   * store's per-account keying alone cannot catch that, because an in-flight
+   * probe completes *after* the switch has already happened.
    */
   account: () => string | undefined;
   sentinel?: SentinelFactory;
@@ -1126,6 +1179,8 @@ interface Config {
    * until the user explicitly agrees.
    */
   probeConsent?: boolean;
+  /** Use the largest context window the international catalog explicitly offers. */
+  useMaximumContextWindow?: boolean;
 }
 declare const Config: z<Config>;
 /**

package/lib/index.js

@@ -1,4 +1,4 @@
-import { A as parseModelCatalog, B as fallbackChatIdentity, C as desktopAuthCandidatesFor, D as classifyUpstreamError, E as WorkBuddyUpstreamClient, F as probeModel, G as appUserAgent, H as resolveChatIdentity, I as randomSentinel, J as resolveAppVersion, K as installedAppVersion, L as CN_APP_VERSION_FILENAME, M as prepareInternationalChatBody, N as regionOf, O as modelWithCurrentPromotion, P as PROBE_EFFORT_CANDIDATES, R as FALLBACK_CN_APP_VERSION, S as defaultDesktopAuthPath, T as workbuddyOwnAuthPath, U as validCliVersion, V as readCliVersion, W as WORKBUDDY_APP_VERSION_FILENAME, Y as validAppVersion, _ as WorkBuddyCatalog, a as WORKBUDDY_HOST_HEARTBEAT_FILENAME, b as WorkBuddyCredentialStore, c as processStartTimeMs, d as writeHostHeartbeat, f as WORKBUDDY_CONNECT_VERSION, g as FALLBACK_WORKBUDDY_MODELS, h as FALLBACK_WORKBUDDY_AI_MODELS, i as variantFor, j as prepareChatBody, k as normalizeCredits, l as readHostHeartbeat, n as CN_VARIANT, o as clearHostHeartbeat, q as readBundleVersion, r as WORKBUDDY_VARIANTS, s as isHeartbeatProcessAlive, t as AI_VARIANT, u as workbuddyHostHeartbeatPath, v as WORKBUDDY_AUTH_FILENAME, w as parseWorkBuddyAuth, x as defaultDesktopAuthCandidates, y as WORKBUDDY_AUTH_FILE_ENV, z as chatUserAgent } from "./variants-CExA7lJt.js";
+import { A as parseModelCatalog, B as fallbackChatIdentity, C as desktopAuthCandidatesFor, D as classifyUpstreamError, E as WorkBuddyUpstreamClient, F as probeModel, G as appUserAgent, H as resolveChatIdentity, I as randomSentinel, J as resolveAppVersion, K as installedAppVersion, L as CN_APP_VERSION_FILENAME, M as prepareInternationalChatBody, N as regionOf, O as modelWithCurrentPromotion, P as PROBE_EFFORT_CANDIDATES, R as FALLBACK_CN_APP_VERSION, S as defaultDesktopAuthPath, T as workbuddyOwnAuthPath, U as validCliVersion, V as readCliVersion, W as WORKBUDDY_APP_VERSION_FILENAME, Y as validAppVersion, _ as WorkBuddyCatalog, a as WORKBUDDY_HOST_HEARTBEAT_FILENAME, b as WorkBuddyCredentialStore, c as processStartTimeMs, d as writeHostHeartbeat, f as WORKBUDDY_CONNECT_VERSION, g as FALLBACK_WORKBUDDY_MODELS, h as FALLBACK_WORKBUDDY_AI_MODELS, i as variantFor, j as prepareChatBody, k as normalizeCredits, l as readHostHeartbeat, n as CN_VARIANT, o as clearHostHeartbeat, q as readBundleVersion, r as WORKBUDDY_VARIANTS, s as isHeartbeatProcessAlive, t as AI_VARIANT, u as workbuddyHostHeartbeatPath, v as WORKBUDDY_AUTH_FILENAME, w as parseWorkBuddyAuth, x as defaultDesktopAuthCandidates, y as WORKBUDDY_AUTH_FILE_ENV, z as chatUserAgent } from "./variants-XUStsapF.js";
 import z from "@deepseek-ai/schemastery";
 import { dirname, join, resolve } from "node:path";
 import { resolveDshHome } from "@deepseek-ai/dsh-home-paths";
@@ -656,8 +656,17 @@ function createWorkBuddyShim(options) {
 */
 /** Basename of the probe record inside the Harness home. */
 const WORKBUDDY_PROBE_FILENAME = ".workbuddy-probe.json";
-/** On-disk format this reader accepts; other versions are discarded. */
-const PROBE_FORMAT_VERSION = 1;
+/**
+* On-disk format this reader accepts; other versions are discarded.
+*
+* Version 2 nested the records under the account that produced them
+* (`records[account][modelId]`), so two accounts no longer overwrite each
+* other's observations for the same model. Version 1 files (flat, one record
+* per model) are deliberately not migrated: they read as empty and the
+* affected models are re-probed on demand, which keeps the reader free of
+* half-understood compatibility paths.
+*/
+const PROBE_FORMAT_VERSION = 2;
 /**
 * How long an observation stays usable. Conservative on purpose: the plan's
 * whole argument is that upstream metadata moves fast, so a result that has
@@ -718,13 +727,15 @@ function isRecord(value) {
 	if (typeof wrapped["fingerprint"] !== "string") return false;
 	if (typeof wrapped["probedAtMs"] !== "number" || !Number.isFinite(wrapped["probedAtMs"])) return false;
 	if (typeof wrapped["pluginVersion"] !== "string") return false;
+	if (typeof wrapped["account"] !== "string" || wrapped["account"] === "") return false;
 	const efforts = wrapped["efforts"];
 	if (!Array.isArray(efforts) || efforts.some((effort) => typeof effort !== "string")) return false;
 	return true;
 }
 /**
-* The plugin's probe records: read once, written atomically, never trusted
-* across a fingerprint change or past the TTL.
+* The plugin's probe records: read once, written atomically, keyed by the
+* account that produced each observation, and never trusted across a
+* fingerprint change or past the TTL.
 */
 var WorkBuddyProbeStore = class {
 	path;
@@ -750,21 +761,26 @@ var WorkBuddyProbeStore = class {
 		if (this.records === void 0) {
 			const document = readDocument(this.path);
 			const records = {};
-			for (const [id, record] of Object.entries(document?.records ?? {})) if (isRecord(record)) records[id] = record;
+			for (const [account, bucket] of Object.entries(document?.records ?? {})) {
+				if (typeof bucket !== "object" || bucket === null || Array.isArray(bucket)) continue;
+				const parsed = {};
+				for (const [modelId, record] of Object.entries(bucket)) if (isRecord(record)) parsed[modelId] = record;
+				records[account] = parsed;
+			}
 			this.records = records;
 		}
 		return this.records;
 	}
 	/**
-	* The usable record for a model, or `undefined` when there is none, it is
-	* expired, it was taken against a different catalog row, or it belongs to a
-	* different account.
+	* The usable record for one account and model, or `undefined` when there is
+	* none, it is expired, it was taken against a different catalog row, or it
+	* belongs to a different account.
 	*
 	* @param account - the account in effect, as `uid:enterpriseId`. Records are
 	*   only returned for the account that produced them.
 	*/
 	get(modelId, fingerprint, account) {
-		const record = this.load()[modelId];
+		const record = this.load()[account]?.[modelId];
 		if (record === void 0) return void 0;
 		if (record.fingerprint !== fingerprint) return void 0;
 		if (record.account !== account) return void 0;
@@ -772,25 +788,28 @@ var WorkBuddyProbeStore = class {
 		return record;
 	}
 	/**
-	* Store one observation. Only a decisive answer (`validating` /
-	* `non-validating`) replaces an existing decisive record: a transient
-	* `unknown` must not erase knowledge the user already paid for.
+	* Store one observation under the account stamped on it. Only a decisive
+	* answer (`validating` / `non-validating`) replaces an existing decisive
+	* record *of the same account*: a transient `unknown` must not erase
+	* knowledge the user already paid for.
 	*/
 	set(modelId, record) {
 		const records = this.load();
-		const existing = records[modelId];
+		const bucket = records[record.account] ?? (records[record.account] = {});
+		const existing = bucket[modelId];
 		if (record.validation === "unknown" && existing !== void 0 && existing.fingerprint === record.fingerprint && existing.validation !== "unknown") return;
-		records[modelId] = record;
+		bucket[modelId] = record;
 		this.persist();
 	}
-	/** Drop every record; used by the card's explicit "clear" action. */
+	/** Drop every record of every account; used by the card's explicit "clear" action. */
 	clear() {
 		this.records = {};
 		this.persist();
 	}
-	/** Every record currently held, for status display. */
+	/** Every record currently held, grouped by account, for status display. */
 	all() {
-		return { ...this.load() };
+		const records = this.load();
+		return Object.fromEntries(Object.entries(records).map(([account, bucket]) => [account, { ...bucket }]));
 	}
 	/** Build a record stamped with this store's clock, version, and account. */
 	record(fingerprint, validation, efforts, account) {
@@ -1001,6 +1020,7 @@ async function workBuddyWebStatus(deps) {
 		const rate = normalizeCredits(model.billing?.credits);
 		const supported = model.supportedContextWindows ?? [];
 		const maxContextWindow = supported.length > 0 ? Math.max(...supported) : void 0;
+		const defaultContextWindow = model.defaultContextWindow ?? model.contextWindow;
 		return {
 			id: model.id,
 			name: model.name,
@@ -1009,7 +1029,8 @@ async function workBuddyWebStatus(deps) {
 			...rate === void 0 ? {} : { credits: rate },
 			...model.billing?.rateUnknown === true ? { rateUnknown: true } : {},
 			...typeof model.contextWindow === "number" && model.contextWindow > 0 ? { contextWindow: model.contextWindow } : {},
-			...maxContextWindow === void 0 || maxContextWindow === model.contextWindow ? {} : { maxContextWindow },
+			...typeof defaultContextWindow === "number" && defaultContextWindow > 0 && defaultContextWindow < model.contextWindow ? { defaultContextWindow } : {},
+			...maxContextWindow === void 0 || maxContextWindow <= defaultContextWindow ? {} : { maxContextWindow },
 			...typeof model.maxInputTokens === "number" && model.maxInputTokens > 0 ? { maxInputTokens: model.maxInputTokens } : {}
 		};
 	});
@@ -1025,7 +1046,8 @@ async function workBuddyWebStatus(deps) {
 	const probed = deps.probe === void 0 ? statusWithModels : {
 		...statusWithModels,
 		probe: deps.probe(),
-		...deps.probeKey === void 0 ? {} : { probeKey: deps.probeKey }
+		...deps.probeKey === void 0 ? {} : { probeKey: deps.probeKey },
+		...deps.useMaximumContextWindow === void 0 ? {} : { useMaximumContextWindow: deps.useMaximumContextWindow() }
 	};
 	try {
 		const credential = await deps.store.current();
@@ -1145,6 +1167,10 @@ function parseAction(text) {
 	const action = wrapped["action"];
 	if (action === "clear") return { action: "clear" };
 	if (action === "refresh") return { action: "refresh" };
+	if (action === "set-maximum-context-window") return typeof wrapped["enabled"] === "boolean" ? {
+		action: "set-maximum-context-window",
+		enabled: wrapped["enabled"]
+	} : void 0;
 	if (action === "probe") {
 		const model = wrapped["model"];
 		if (typeof model !== "string" || model.trim() === "") return void 0;
@@ -1196,6 +1222,14 @@ function workBuddyProbeHandler(deps, key) {
 				json(res, 200, await deps.refresh());
 				return;
 			}
+			if (action.action === "set-maximum-context-window") {
+				if (deps.setMaximumContextWindow === void 0) {
+					json(res, 404, { error: "context-window-setting-not-supported" });
+					return;
+				}
+				json(res, 200, await deps.setMaximumContextWindow(action.enabled === true));
+				return;
+			}
 			json(res, 200, await deps.probe(action.model));
 		} catch (error) {
 			json(res, 500, { error: error instanceof Error ? error.message : String(error) });
@@ -1292,10 +1326,12 @@ const AUTH_FILE_FIELD = z.string().description("WorkBuddy desktop auth file (def
 const AUTH_FILE_AI_FIELD = z.string().description("WorkBuddy AI desktop auth file (defaults to the app's own location)");
 /** Probe authorization (shared by the plugin schema and the CN section). */
 const PROBE_CONSENT_FIELD = z.boolean().default(false).description("Authorize reasoning-effort probes (each probe sends real requests that may consume credit)");
+const MAXIMUM_CONTEXT_WINDOW_FIELD = z.boolean().default(true).description("Use the largest context window declared by WorkBuddy AI when alternatives are available (on by default)");
 const Config = z.object({
 	authFile: AUTH_FILE_FIELD,
 	authFileAI: AUTH_FILE_AI_FIELD,
-	probeConsent: PROBE_CONSENT_FIELD
+	probeConsent: PROBE_CONSENT_FIELD,
+	useMaximumContextWindow: MAXIMUM_CONTEXT_WINDOW_FIELD
 });
 /**
 * The CN card's settings section: only the fields that card edits.
@@ -1310,8 +1346,11 @@ const CN_SECTION = z.object({
 	authFile: AUTH_FILE_FIELD,
 	probeConsent: PROBE_CONSENT_FIELD
 });
-/** The international card's settings section: only its own auth-file path. */
-const AI_SECTION = z.object({ authFileAI: AUTH_FILE_AI_FIELD });
+/** The international card's settings section and its context-window preference. */
+const AI_SECTION = z.object({
+	authFileAI: AUTH_FILE_AI_FIELD,
+	useMaximumContextWindow: MAXIMUM_CONTEXT_WINDOW_FIELD
+});
 /** Stable identity key used by credentials, probe records, and catalog entries. */
 function credentialIdentity(credential) {
 	return `${credential.uid}:${credential.enterpriseId ?? ""}`;
@@ -1345,6 +1384,7 @@ function createVariantRuntime(config, variant, current, identityOf) {
 	});
 	const fallback = fallbackFor(variant);
 	const catalog = new WorkBuddyCatalog(fallback);
+	if (variant.id !== CN_VARIANT.id) catalog.setUseMaximumContextWindow(config.useMaximumContextWindow === true);
 	catalog.setVisible(false);
 	const probeStore = new WorkBuddyProbeStore({
 		pluginVersion: WORKBUDDY_CONNECT_VERSION,
@@ -1520,6 +1560,7 @@ function apply(ctx, config) {
 	const lastIdentities = /* @__PURE__ */ new Map();
 	const runtimes = WORKBUDDY_VARIANTS.map((variant) => createVariantRuntime(config, variant, () => current(), (id) => lastIdentities.get(id)));
 	const probeKey = createProbeKey();
+	let setMaximumContextWindow;
 	/**
 	* Point a variant at an account identity, invalidating whatever the previous
 	* one left behind.
@@ -1530,13 +1571,12 @@ function apply(ctx, config) {
 	* transition bumps {@link VariantRuntime.catalogGeneration}, which is what
 	* makes an in-flight request from before the change refuse to write back.
 	*
-	* Probe observations are dropped whenever the account actually changes —
-	* including sign-out, and including the "signed out, then in as someone else"
-	* sequence that used to look like a first sighting and let the new account
-	* inherit the old one's detected levels. They are deliberately NOT cleared on
-	* a first sign-in: no previous account's data could leak there, and clearing
-	* would delete records this very account owns (written before a restart, or
-	* seeded while all of this is running).
+	* Probe observations are kept across an account change: the store nests them
+	* per account, so the departing account's records simply stop being served
+	* (every read is account-scoped) and are found intact if that account
+	* returns. The "signed out, then in as someone else" sequence that used to
+	* look like a first sighting is still safe — a record only ever answers for
+	* the account stamped on it, so the new account inherits nothing.
 	*
 	* @param identity - the account now in effect, or `undefined` when signed out.
 	*/
@@ -1550,10 +1590,7 @@ function apply(ctx, config) {
 		runtime.catalogGeneration += 1;
 		runtime.inflightFetch?.controller.abort();
 		runtime.inflightFetch = void 0;
-		if (hadCredential && known !== identity) {
-			runtime.probeStore.clear();
-			runtime.invalidate();
-		}
+		if (hadCredential && known !== identity) runtime.invalidate();
 		if (identity === void 0) {
 			if (known !== void 0) runtime.savedCatalogs.delete(known);
 			runtime.catalog.set(runtime.fallback);
@@ -1586,7 +1623,8 @@ function apply(ctx, config) {
 				models: () => runtime.catalog.current(),
 				catalog: () => catalogSection(runtime),
 				probe: () => probeSection(runtime, current().probeConsent === true),
-				probeKey
+				probeKey,
+				...runtime.variant.id === CN_VARIANT.id ? {} : { useMaximumContextWindow: () => current().useMaximumContextWindow === true }
 			});
 			registerWorkBuddyProbeRoute(webCtx, {
 				path: runtime.variant.probePath,
@@ -1627,7 +1665,14 @@ function apply(ctx, config) {
 						state: "failed",
 						reason: runtime.catalogError
 					};
-				}
+				},
+				...runtime.variant.id === CN_VARIANT.id ? {} : { setMaximumContextWindow: async (enabled) => {
+					if (setMaximumContextWindow === void 0) return {
+						state: "failed",
+						reason: "settings are unavailable"
+					};
+					return setMaximumContextWindow(enabled);
+				} }
 			}, probeKey);
 		}
 	});
@@ -1641,10 +1686,16 @@ function apply(ctx, config) {
 		const merged = () => ({
 			...sources.cn().authFile === void 0 ? {} : { authFile: sources.cn().authFile },
 			...sources.cn().probeConsent === void 0 ? {} : { probeConsent: sources.cn().probeConsent },
-			...sources.ai().authFileAI === void 0 ? {} : { authFileAI: sources.ai().authFileAI }
+			...sources.ai().authFileAI === void 0 ? {} : { authFileAI: sources.ai().authFileAI },
+			...sources.ai().useMaximumContextWindow === void 0 ? {} : { useMaximumContextWindow: sources.ai().useMaximumContextWindow }
 		});
+		const applyMaximumContextWindow = (next) => {
+			const runtime = runtimes.find((candidate) => candidate.variant.id !== CN_VARIANT.id);
+			if (runtime?.catalog.setUseMaximumContextWindow(next.useMaximumContextWindow === true)) runtime.invalidate();
+		};
 		const repointStores = () => {
 			const next = merged();
+			applyMaximumContextWindow(next);
 			for (const runtime of runtimes) runtime.store.setDesktopPath(configuredAuthFile(next, runtime.variant));
 		};
 		settingsCtx.settings.installSection(ctx, WORKBUDDY_SETTINGS_NS, CN_SECTION, config, {
@@ -1661,6 +1712,10 @@ function apply(ctx, config) {
 			},
 			onChange: repointStores
 		});
+		setMaximumContextWindow = async (enabled) => {
+			await settingsCtx.settings.update(WORKBUDDY_AI_SETTINGS_NS, { useMaximumContextWindow: enabled });
+			return { state: "updated" };
+		};
 	});
 	ctx.effect(() => () => {
 		stopped = true;

package/lib/{variants-CExA7lJt.js → variants-XUStsapF.js}

@@ -485,6 +485,33 @@ async function probeModel(options) {
 const CN_CHAT_BASE = "https://copilot.tencent.com";
 const CN_BILLING_BASE = "https://www.codebuddy.cn";
 const GLOBAL_BASE = "https://www.workbuddy.ai";
+/**
+* Display name for the single synthetic row the enterprise endpoint produces.
+*
+* The endpoint reports one cycle quota, not the personal endpoint's list of
+* named packages, so the card's "by package" table has exactly one row.
+*/
+const enterprisePackageName = "enterprise";
+/**
+* Field names and value types of a response document, for diagnostics.
+*
+* Names and `typeof` only. This string ends up in the status route and then in
+* the browser, and the response describes the account's own usage; the values
+* themselves must never travel. Only the document and its `data` member are
+* described, so the output stays small.
+*/
+function describeShape(document) {
+	if (typeof document !== "object" || document === null || Array.isArray(document)) return typeof document;
+	const record = document;
+	const at = (source) => {
+		const keys = Object.keys(source).slice(0, 24);
+		return keys.length === 0 ? "(empty)" : keys.map((key) => `${key}:${typeof source[key]}`).join(", ");
+	};
+	const top = `top-level { ${at(record)} }`;
+	const data = record["data"];
+	if (typeof data !== "object" || data === null || Array.isArray(data)) return top;
+	return `${top}; data { ${at(data)} }`;
+}
 /** Shared CLI-form User-Agent for refresh and the CN catalog; chat and probe present the desktop identity (client-identity.ts). */
 const CLIENT_UA = "CLI/2.63.2 CodeBuddy/2.63.2";
 const JSON_TIMEOUT_MS = 3e4;
@@ -882,8 +909,25 @@ var WorkBuddyUpstreamClient = class {
 		};
 		return models;
 	}
-	/** POST the billing endpoint for the aggregated remaining credit. */
+	/**
+	* POST the billing endpoint for the aggregated remaining credit.
+	*
+	* Two upstream shapes, chosen by account type:
+	*
+	* - **CN enterprise** (`regionOf === 'cn'` and `enterpriseId` non-empty) asks
+	*   `/v2/billing/meter/get-enterprise-user-usage`, which answers with a single
+	*   cycle quota. The personal endpoint serves these accounts an empty
+	*   `Accounts` list, which the card then renders as "0 credit" — a wrong
+	*   number rather than a visible failure (issue #31).
+	* - **Everyone else** keeps the personal endpoint unchanged.
+	*
+	* The region gate is load-bearing: the enterprise endpoint is unverified for
+	* the global region, so an international credential that happens to carry an
+	* `enterpriseId` must stay on the measured personal path instead of being
+	* moved onto an unmeasured one.
+	*/
 	async fetchCredits(credential) {
+		if (regionOf(credential.domain) === "cn" && credential.enterpriseId !== void 0 && credential.enterpriseId !== "") return await this.fetchEnterpriseCredits(credential);
 		const now = /* @__PURE__ */ new Date();
 		const format = (date) => [
 			date.getFullYear().toString().padStart(4, "0"),
@@ -941,6 +985,75 @@ var WorkBuddyUpstreamClient = class {
 		};
 	}
 	/**
+	* CN enterprise credit read: a single cycle quota instead of a package list.
+	*
+	* Verified against the WorkBuddy desktop app (`app.asar`,
+	* `BackendProvider.getEnterpriseUsage` and `CloudAccountRepo.billing`): the
+	* body is an empty object and the account identity travels only in the
+	* headers. The two official call sites disagree on the field spelling
+	* (`limitNum`/`credit` vs `limit_num`/`used_num`), so both are accepted.
+	*
+	* A body carrying no recognisable quota field is a hard error rather than a
+	* zero. Rendering `0` for "we did not understand the answer" is exactly how
+	* issue #31 stayed invisible while users saw a plausible wrong number.
+	*
+	* The error names fields and types only: it reaches the browser, and the
+	* response body may describe the account's usage.
+	*/
+	async fetchEnterpriseCredits(credential) {
+		const response = await fetch(`${CN_BILLING_BASE}/v2/billing/meter/get-enterprise-user-usage`, {
+			method: "POST",
+			headers: billingHeaders(credential),
+			body: JSON.stringify({}),
+			signal: AbortSignal.timeout(JSON_TIMEOUT_MS)
+		});
+		const envelope = await readEnvelope(response);
+		if (!response.ok || envelope.code !== 0) throw envelopeError(response.status, envelope);
+		const sources = [];
+		for (const candidate of [envelope.data, envelope.document]) {
+			if (typeof candidate !== "object" || candidate === null || Array.isArray(candidate)) continue;
+			const record = candidate;
+			if (typeof record["data"] === "object" && record["data"] !== null && !Array.isArray(record["data"])) sources.push(record["data"]);
+			sources.push(record);
+		}
+		const numberAt = (source, key) => typeof source[key] === "number" ? source[key] : void 0;
+		let limit;
+		let used;
+		let resetTime;
+		for (const source of sources) {
+			const candidate = numberAt(source, "limitNum") ?? numberAt(source, "limit_num");
+			if (candidate === void 0) continue;
+			limit = candidate;
+			used = numberAt(source, "credit") ?? numberAt(source, "used_num");
+			if (typeof source["cycleResetTime"] === "string" && source["cycleResetTime"] !== "") resetTime = source["cycleResetTime"];
+			break;
+		}
+		if (limit === void 0) throw new Error(`workbuddy enterprise billing response carried no recognised quota field (expected limitNum/limit_num + credit/used_num; received ${describeShape(envelope.document)})`);
+		if (limit === -1) return {
+			total: 0,
+			accounts: [{
+				packageName: enterprisePackageName,
+				remain: 0,
+				size: 0,
+				unlimited: true
+			}],
+			unlimited: true,
+			...resetTime === void 0 ? {} : { cycleResetTime: resetTime }
+		};
+		if (used === void 0) throw new Error(`workbuddy enterprise billing response carried a quota limit but no recognised usage field (expected credit/used_num alongside limitNum/limit_num; received ${describeShape(envelope.document)})`);
+		let remain = limit - used;
+		if (remain < 0) remain = 0;
+		return {
+			total: remain,
+			accounts: [{
+				packageName: enterprisePackageName,
+				remain,
+				size: limit
+			}],
+			...resetTime === void 0 ? {} : { cycleResetTime: resetTime }
+		};
+	}
+	/**
 	* One probe request: a real streaming chat call carrying the effort under
 	* test.
 	*
@@ -1092,6 +1205,7 @@ function parseModelCatalog(data, international = false) {
 			name: typeof wrapped["name"] === "string" && wrapped["name"] !== "" ? wrapped["name"] : id,
 			contextWindow: international && isObject(wrapped["contextWindow"]) && positive(wrapped["contextWindow"]["defaultLength"]) ? wrapped["contextWindow"]["defaultLength"] : input,
 			...international ? {
+				...isObject(wrapped["contextWindow"]) && positive(wrapped["contextWindow"]["defaultLength"]) ? { defaultContextWindow: wrapped["contextWindow"]["defaultLength"] } : {},
 				maxInputTokens: input,
 				supportedContextWindows: isObject(wrapped["contextWindow"]) && Array.isArray(wrapped["contextWindow"]["supportedLengths"]) ? wrapped["contextWindow"]["supportedLengths"].filter(positive) : [],
 				promotions: parsePromotions(data["modelPromotions"], id)
@@ -1971,6 +2085,8 @@ const FALLBACK_WORKBUDDY_AI_MODELS = [
 		id: "hy4-preview-f",
 		name: "Hy4 preview",
 		contextWindow: 3e5,
+		defaultContextWindow: 3e5,
+		supportedContextWindows: [3e5, 1e6],
 		maxTokens: 64e3,
 		supportsImages: true,
 		reasoning: {
@@ -2007,6 +2123,8 @@ const FALLBACK_WORKBUDDY_AI_MODELS = [
 		id: "deepseek-v4.1-flash",
 		name: "Deepseek-V4.1-Flash",
 		contextWindow: 3e5,
+		defaultContextWindow: 3e5,
+		supportedContextWindows: [3e5, 1e6],
 		maxTokens: 128e3,
 		supportsImages: true,
 		reasoning: {
@@ -2024,6 +2142,8 @@ const FALLBACK_WORKBUDDY_AI_MODELS = [
 		id: "gpt-6-astra",
 		name: "GPT-6-Astra",
 		contextWindow: 4e5,
+		defaultContextWindow: 4e5,
+		supportedContextWindows: [4e5, 1e6],
 		maxTokens: 128e3,
 		supportsImages: true,
 		reasoning: {
@@ -2287,13 +2407,22 @@ const FALLBACK_WORKBUDDY_AI_MODELS = [
 var WorkBuddyCatalog = class {
 	models;
 	visible = true;
+	useMaximumContextWindow = false;
 	constructor(initial = FALLBACK_WORKBUDDY_MODELS) {
 		this.models = initial;
 	}
 	/** Current entries; empty while the variant has no usable credential. */
 	current() {
 		if (!this.visible) return [];
-		return this.models.map((model) => modelWithCurrentPromotion(model));
+		return this.models.map((model) => {
+			const current = modelWithCurrentPromotion(model);
+			const maximum = current.supportedContextWindows === void 0 ? void 0 : Math.max(...current.supportedContextWindows);
+			return this.useMaximumContextWindow && maximum !== void 0 && maximum > current.contextWindow ? {
+				...current,
+				defaultContextWindow: current.defaultContextWindow ?? current.contextWindow,
+				contextWindow: maximum
+			} : current;
+		});
 	}
 	/** Replace the list; callers invalidate their adapter snapshot after this. */
 	set(models) {
@@ -2312,6 +2441,12 @@ var WorkBuddyCatalog = class {
 		this.visible = visible;
 		return true;
 	}
+	/** Select the largest declared international window where the upstream offers one. */
+	setUseMaximumContextWindow(useMaximum) {
+		if (this.useMaximumContextWindow === useMaximum) return false;
+		this.useMaximumContextWindow = useMaximum;
+		return true;
+	}
 	/** Models to fall back to when the upstream fetch fails; ignores visibility. */
 	fallback() {
 		return this.models;
@@ -2344,7 +2479,7 @@ const WORKBUDDY_AI_STATUS_PATH = "/plugins/dsh-workbuddy-connect/ai/status";
 const WORKBUDDY_AI_PROBE_PATH = "/plugins/dsh-workbuddy-connect/ai/probe";
 //#endregion
 //#region src/version.ts
-const WORKBUDDY_CONNECT_VERSION = "0.5.2";
+const WORKBUDDY_CONNECT_VERSION = "0.5.4";
 //#endregion
 //#region src/host-heartbeat.ts
 /**

package/package.json

@@ -2,7 +2,7 @@
   "name": "dsh-workbuddy-connect",
   "displayName": "DSH WorkBuddy Connect",
   "description": "将 WorkBuddy 桌面 App 包含的模型自动接入 DeepSeek Harness — bring WorkBuddy desktop app models into DeepSeek Harness with zero configuration.",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "author": "corrinehu",
   "keywords": [
     "dsh-plugin",