@jxgame2020/dsh-token-quota 0.1.39 → 0.1.40

package/lib/client.js

@@ -7,14 +7,6 @@ window.__ModuleLoader__.load({
 		let _deepseek_ai_dsh_client_store = require("@deepseek-ai/dsh-client-store");
 		let react_jsx_runtime = require("react/jsx-runtime");
 		let react = require("react");
-		//#region lib/types/types.js
-		/**
-		* Provider route keys the host knows how to query for account balance. A
-		* directory provider outside this list renders as "not integrated for query"
-		* in the balance window.
-		*/
-		const TOKEN_QUOTA_BALANCE_PROVIDERS = ["deepseek"];
-		//#endregion
 		//#region lib/types/client/store.js
 		/**
 		* Token-quota panel store: the shared, remount-surviving view state. The
@@ -123,7 +115,7 @@ window.__ModuleLoader__.load({
 						d.musicOnlyCurrentSession = only;
 					},
 					setUpgrade: (d, upgrade) => {
-						const normalized = upgrade !== null && upgrade.latestVersion !== "" && upgrade.latestVersion !== "0.1.39" ? upgrade : null;
+						const normalized = upgrade !== null && upgrade.latestVersion !== "" && upgrade.latestVersion !== "0.1.40" ? upgrade : null;
 						if (d.upgrade === null || d.upgrade.latestVersion !== normalized?.latestVersion) d.upgradeDismissed = false;
 						d.upgrade = normalized;
 					},
@@ -253,142 +245,142 @@ window.__ModuleLoader__.load({
 			document.head.appendChild(tag);
 		}
 		var TokenQuotaPanel_module_css_default = {
-			"logPagerBtn": "NPLLIq_logPagerBtn",
-			"selectBtn": "NPLLIq_selectBtn",
-			"dialogOverlay": "NPLLIq_dialogOverlay",
-			"fillIdle": "NPLLIq_fillIdle",
-			"balanceBarText": "NPLLIq_balanceBarText",
-			"balanceWinBody": "NPLLIq_balanceWinBody",
-			"monitorRow": "NPLLIq_monitorRow",
-			"resetSelect": "NPLLIq_resetSelect",
-			"upgradeCommandRow": "NPLLIq_upgradeCommandRow",
-			"rowDrag": "NPLLIq_rowDrag",
-			"logClearDate": "NPLLIq_logClearDate",
-			"headerText": "NPLLIq_headerText",
-			"noteWinBtn": "NPLLIq_noteWinBtn",
-			"fillOver": "NPLLIq_fillOver",
-			"copyBtn": "NPLLIq_copyBtn",
-			"balanceBarLow": "NPLLIq_balanceBarLow",
-			"switch": "NPLLIq_switch",
-			"dialogHeader": "NPLLIq_dialogHeader",
-			"logSearch": "NPLLIq_logSearch",
+			"noteWinResize": "NPLLIq_noteWinResize",
 			"dialogClose": "NPLLIq_dialogClose",
-			"logPagerInfo": "NPLLIq_logPagerInfo",
-			"balanceToggleOpen": "NPLLIq_balanceToggleOpen",
-			"logPagerControls": "NPLLIq_logPagerControls",
-			"versionNew": "NPLLIq_versionNew",
-			"headerActions": "NPLLIq_headerActions",
-			"logPager": "NPLLIq_logPager",
-			"upgradeBannerFlash": "NPLLIq_upgradeBannerFlash",
-			"logScroll": "NPLLIq_logScroll",
-			"balanceRefreshAll": "NPLLIq_balanceRefreshAll",
-			"noteWinTitle": "NPLLIq_noteWinTitle",
-			"logDialog": "NPLLIq_logDialog",
-			"upgradeCommands": "NPLLIq_upgradeCommands",
-			"resetRow": "NPLLIq_resetRow",
-			"noteWinText": "NPLLIq_noteWinText",
-			"save": "NPLLIq_save",
-			"balanceBarError": "NPLLIq_balanceBarError",
-			"row": "NPLLIq_row",
-			"checkBtn": "NPLLIq_checkBtn",
-			"balanceValueLow": "NPLLIq_balanceValueLow",
-			"notesBar": "NPLLIq_notesBar",
-			"dialogActions": "NPLLIq_dialogActions",
-			"logDayCol": "NPLLIq_logDayCol",
-			"bar": "NPLLIq_bar",
-			"upgradeBannerText": "NPLLIq_upgradeBannerText",
-			"noteNew": "NPLLIq_noteNew",
-			"logModelCol": "NPLLIq_logModelCol",
-			"logTable": "NPLLIq_logTable",
+			"balanceStateHint": "NPLLIq_balanceStateHint",
+			"panel": "NPLLIq_panel",
 			"noteMenuBackdrop": "NPLLIq_noteMenuBackdrop",
-			"noteChip": "NPLLIq_noteChip",
-			"noteWinCollapsed": "NPLLIq_noteWinCollapsed",
-			"currentBadge": "NPLLIq_currentBadge",
-			"input": "NPLLIq_input",
-			"balanceSortBtn": "NPLLIq_balanceSortBtn",
+			"notesBar": "NPLLIq_notesBar",
+			"dialogTabActive": "NPLLIq_dialogTabActive",
+			"balanceWinHeader": "NPLLIq_balanceWinHeader",
+			"balanceRowValue": "NPLLIq_balanceRowValue",
+			"fullNotice": "NPLLIq_fullNotice",
+			"logUsedCol": "NPLLIq_logUsedCol",
+			"upgradeCommands": "NPLLIq_upgradeCommands",
+			"rowHeader": "NPLLIq_rowHeader",
 			"fillWarn": "NPLLIq_fillWarn",
-			"gearBtn": "NPLLIq_gearBtn",
-			"monitorList": "NPLLIq_monitorList",
-			"checkUpdatesRow": "NPLLIq_checkUpdatesRow",
-			"dialogTab": "NPLLIq_dialogTab",
-			"subtitle": "NPLLIq_subtitle",
-			"dialog": "NPLLIq_dialog",
-			"controls": "NPLLIq_controls",
-			"noticeError": "NPLLIq_noticeError",
-			"tabLabel": "NPLLIq_tabLabel",
-			"checkErrorHint": "NPLLIq_checkErrorHint",
-			"balanceStateError": "NPLLIq_balanceStateError",
-			"logHeader": "NPLLIq_logHeader",
-			"logClearDialog": "NPLLIq_logClearDialog",
-			"logClearBtn": "NPLLIq_logClearBtn",
-			"noteWinGripBar": "NPLLIq_noteWinGripBar",
-			"noteWinResize": "NPLLIq_noteWinResize",
-			"collapse": "NPLLIq_collapse",
-			"monitorHint": "NPLLIq_monitorHint",
+			"logDialog": "NPLLIq_logDialog",
+			"balanceRowHeader": "NPLLIq_balanceRowHeader",
+			"monitorRow": "NPLLIq_monitorRow",
+			"logPager": "NPLLIq_logPager",
 			"title": "NPLLIq_title",
-			"noteChipOpen": "NPLLIq_noteChipOpen",
+			"logPageSizeSelect": "NPLLIq_logPageSizeSelect",
+			"balanceRowRefresh": "NPLLIq_balanceRowRefresh",
+			"balanceToggle": "NPLLIq_balanceToggle",
+			"upgradeBannerFlash": "NPLLIq_upgradeBannerFlash",
+			"upgradeBannerText": "NPLLIq_upgradeBannerText",
 			"copyBtnDone": "NPLLIq_copyBtnDone",
+			"checkBtn": "NPLLIq_checkBtn",
+			"logPagerBtn": "NPLLIq_logPagerBtn",
+			"dialogOverlay": "NPLLIq_dialogOverlay",
 			"noteWinTitleText": "NPLLIq_noteWinTitleText",
-			"balanceLastLine": "NPLLIq_balanceLastLine",
-			"noteNewPlus": "NPLLIq_noteNewPlus",
-			"rowDropTarget": "NPLLIq_rowDropTarget",
-			"balanceValueError": "NPLLIq_balanceValueError",
-			"upgradeBanner": "NPLLIq_upgradeBanner",
-			"logToolbar": "NPLLIq_logToolbar",
-			"version": "NPLLIq_version",
-			"balanceRowRefresh": "NPLLIq_balanceRowRefresh",
-			"balanceStateHint": "NPLLIq_balanceStateHint",
-			"noteWinBar": "NPLLIq_noteWinBar",
-			"rowHeader": "NPLLIq_rowHeader",
-			"balanceWin": "NPLLIq_balanceWin",
-			"balanceWinTitle": "NPLLIq_balanceWinTitle",
-			"balanceBar": "NPLLIq_balanceBar",
-			"balanceRowName": "NPLLIq_balanceRowName",
-			"dialogTabs": "NPLLIq_dialogTabs",
-			"noteMenu": "NPLLIq_noteMenu",
+			"balanceBarBadge": "NPLLIq_balanceBarBadge",
+			"controls": "NPLLIq_controls",
+			"upgradeHint": "NPLLIq_upgradeHint",
+			"resetRow": "NPLLIq_resetRow",
+			"dialogHint": "NPLLIq_dialogHint",
+			"rowDrag": "NPLLIq_rowDrag",
+			"balanceBarLow": "NPLLIq_balanceBarLow",
+			"bar": "NPLLIq_bar",
+			"panelDim": "NPLLIq_panelDim",
+			"noteMenuItem": "NPLLIq_noteMenuItem",
+			"currentBadge": "NPLLIq_currentBadge",
 			"rowDragging": "NPLLIq_rowDragging",
+			"upgradeCommand": "NPLLIq_upgradeCommand",
+			"rowMeta": "NPLLIq_rowMeta",
 			"panelDimStrong": "NPLLIq_panelDimStrong",
-			"dialogSection": "NPLLIq_dialogSection",
-			"dialogLabel": "NPLLIq_dialogLabel",
-			"monitorName": "NPLLIq_monitorName",
+			"checkErrorHint": "NPLLIq_checkErrorHint",
+			"logDayCol": "NPLLIq_logDayCol",
+			"balanceBarRefresh": "NPLLIq_balanceBarRefresh",
+			"header": "NPLLIq_header",
+			"checkUpdatesLabel": "NPLLIq_checkUpdatesLabel",
 			"logPagerPages": "NPLLIq_logPagerPages",
-			"balanceValueOk": "NPLLIq_balanceValueOk",
-			"body": "NPLLIq_body",
+			"noteNewPlus": "NPLLIq_noteNewPlus",
+			"balanceToggleOpen": "NPLLIq_balanceToggleOpen",
+			"logTable": "NPLLIq_logTable",
+			"version": "NPLLIq_version",
+			"settingsBtn": "NPLLIq_settingsBtn",
 			"settingsBtnActive": "NPLLIq_settingsBtnActive",
-			"logPageSizeSelect": "NPLLIq_logPageSizeSelect",
-			"panel": "NPLLIq_panel",
-			"balanceRowHeader": "NPLLIq_balanceRowHeader",
-			"balanceBarDim": "NPLLIq_balanceBarDim",
+			"balanceValueOk": "NPLLIq_balanceValueOk",
+			"resetSelect": "NPLLIq_resetSelect",
+			"noteWinTitle": "NPLLIq_noteWinTitle",
+			"logPagerInfo": "NPLLIq_logPagerInfo",
+			"noteMenu": "NPLLIq_noteMenu",
+			"upgradeCommandRow": "NPLLIq_upgradeCommandRow",
+			"dialogTabs": "NPLLIq_dialogTabs",
+			"balanceRow": "NPLLIq_balanceRow",
+			"logPagerControls": "NPLLIq_logPagerControls",
+			"monitorHint": "NPLLIq_monitorHint",
+			"noticeError": "NPLLIq_noticeError",
+			"tabLabel": "NPLLIq_tabLabel",
+			"tab": "NPLLIq_tab",
+			"dialog": "NPLLIq_dialog",
+			"balanceSortBtn": "NPLLIq_balanceSortBtn",
+			"fillOver": "NPLLIq_fillOver",
 			"dialogBtn": "NPLLIq_dialogBtn",
-			"panelDim": "NPLLIq_panelDim",
-			"balanceWinHeader": "NPLLIq_balanceWinHeader",
-			"radioRow": "NPLLIq_radioRow",
+			"noteWin": "NPLLIq_noteWin",
+			"save": "NPLLIq_save",
+			"dialogTab": "NPLLIq_dialogTab",
+			"noteWinText": "NPLLIq_noteWinText",
+			"balanceWin": "NPLLIq_balanceWin",
+			"logClearDate": "NPLLIq_logClearDate",
+			"monitorName": "NPLLIq_monitorName",
+			"tabIcon": "NPLLIq_tabIcon",
+			"logClearBtn": "NPLLIq_logClearBtn",
 			"dialogBtnPrimary": "NPLLIq_dialogBtnPrimary",
-			"header": "NPLLIq_header",
-			"fullNotice": "NPLLIq_fullNotice",
-			"upgradeCommand": "NPLLIq_upgradeCommand",
-			"dialogTabActive": "NPLLIq_dialogTabActive",
-			"balanceRowValue": "NPLLIq_balanceRowValue",
-			"balanceToggle": "NPLLIq_balanceToggle",
-			"rowMeta": "NPLLIq_rowMeta",
-			"balanceBarRefresh": "NPLLIq_balanceBarRefresh",
-			"checkUpdatesLabel": "NPLLIq_checkUpdatesLabel",
+			"body": "NPLLIq_body",
 			"rowName": "NPLLIq_rowName",
-			"logUsedCol": "NPLLIq_logUsedCol",
-			"noteWin": "NPLLIq_noteWin",
-			"notice": "NPLLIq_notice",
-			"settingsDialog": "NPLLIq_settingsDialog",
-			"dialogHint": "NPLLIq_dialogHint",
-			"balanceBarBadge": "NPLLIq_balanceBarBadge",
-			"noteMenuItem": "NPLLIq_noteMenuItem",
-			"upgradeClose": "NPLLIq_upgradeClose",
-			"tab": "NPLLIq_tab",
+			"collapse": "NPLLIq_collapse",
+			"dialogHeader": "NPLLIq_dialogHeader",
+			"balanceWinBody": "NPLLIq_balanceWinBody",
+			"balanceBarDim": "NPLLIq_balanceBarDim",
+			"noteWinCollapsed": "NPLLIq_noteWinCollapsed",
+			"switch": "NPLLIq_switch",
+			"rowDropTarget": "NPLLIq_rowDropTarget",
 			"dialogTitle": "NPLLIq_dialogTitle",
-			"balanceRow": "NPLLIq_balanceRow",
+			"checkUpdatesRow": "NPLLIq_checkUpdatesRow",
 			"dialogBody": "NPLLIq_dialogBody",
-			"settingsBtn": "NPLLIq_settingsBtn",
-			"upgradeHint": "NPLLIq_upgradeHint",
-			"tabIcon": "NPLLIq_tabIcon"
+			"balanceBarText": "NPLLIq_balanceBarText",
+			"settingsDialog": "NPLLIq_settingsDialog",
+			"logModelCol": "NPLLIq_logModelCol",
+			"logHeader": "NPLLIq_logHeader",
+			"balanceBar": "NPLLIq_balanceBar",
+			"upgradeBanner": "NPLLIq_upgradeBanner",
+			"balanceBarError": "NPLLIq_balanceBarError",
+			"monitorList": "NPLLIq_monitorList",
+			"balanceRefreshAll": "NPLLIq_balanceRefreshAll",
+			"dialogSection": "NPLLIq_dialogSection",
+			"balanceValueError": "NPLLIq_balanceValueError",
+			"balanceLastLine": "NPLLIq_balanceLastLine",
+			"upgradeClose": "NPLLIq_upgradeClose",
+			"notice": "NPLLIq_notice",
+			"balanceStateError": "NPLLIq_balanceStateError",
+			"fillIdle": "NPLLIq_fillIdle",
+			"noteChipOpen": "NPLLIq_noteChipOpen",
+			"versionNew": "NPLLIq_versionNew",
+			"balanceRowName": "NPLLIq_balanceRowName",
+			"headerText": "NPLLIq_headerText",
+			"row": "NPLLIq_row",
+			"input": "NPLLIq_input",
+			"balanceWinTitle": "NPLLIq_balanceWinTitle",
+			"headerActions": "NPLLIq_headerActions",
+			"noteWinBtn": "NPLLIq_noteWinBtn",
+			"radioRow": "NPLLIq_radioRow",
+			"balanceValueLow": "NPLLIq_balanceValueLow",
+			"gearBtn": "NPLLIq_gearBtn",
+			"dialogLabel": "NPLLIq_dialogLabel",
+			"logScroll": "NPLLIq_logScroll",
+			"noteWinGripBar": "NPLLIq_noteWinGripBar",
+			"noteWinBar": "NPLLIq_noteWinBar",
+			"subtitle": "NPLLIq_subtitle",
+			"copyBtn": "NPLLIq_copyBtn",
+			"selectBtn": "NPLLIq_selectBtn",
+			"logSearch": "NPLLIq_logSearch",
+			"logClearDialog": "NPLLIq_logClearDialog",
+			"noteNew": "NPLLIq_noteNew",
+			"dialogActions": "NPLLIq_dialogActions",
+			"logToolbar": "NPLLIq_logToolbar",
+			"noteChip": "NPLLIq_noteChip"
 		};
 		//#endregion
 		//#region lib/types/client/TokenQuotaPanel.js
@@ -879,7 +871,7 @@ window.__ModuleLoader__.load({
 				for (const provider of providers) {
 					const bal = balances.find((b) => b.provider === provider);
 					if (bal !== void 0) map.set(provider, bal);
-					else if (TOKEN_QUOTA_BALANCE_PROVIDERS.includes(provider)) map.set(provider, "loading");
+					else if (balances.length === 0) map.set(provider, "loading");
 					else map.set(provider, "unsupported");
 				}
 				return map;
@@ -1058,13 +1050,13 @@ window.__ModuleLoader__.load({
 										title: t("upgradeAvailableHint"),
 										children: [
 											"v",
-											"0.1.39",
+											"0.1.40",
 											" → v",
 											upgrade.latestVersion
 										]
 									}) : (0, react_jsx_runtime.jsxs)("span", {
 										className: TokenQuotaPanel_module_css_default.version,
-										children: ["v", "0.1.39"]
+										children: ["v", "0.1.40"]
 									})]
 								})
 							}),

package/lib/index.js

@@ -276,23 +276,8 @@ function defaultStoragePath() {
 function usageTokens(usage) {
 	return usage.inputTokens + (usage.cacheReadTokens ?? 0) + (usage.cacheWriteTokens ?? 0) + usage.outputTokens;
 }
-/**
-* Daily token-quota service.
-*
-* Mount it beside the other rows (`@jxgame2020/dsh-token-quota`) and write
-* per-model limits through the `token-quota` settings namespace.
-*/
-/**
-* Providers the host can query for account balance, keyed by route key. Each
-* entry names the credential reference (environment-variable name), the
-* balance endpoint base, and the optional `*_BASE_URL` override variable.
-* Directory providers outside this map render as "not integrated for query".
-*/
-const BALANCE_SOURCES = { deepseek: {
-	keyEnv: "DEEPSEEK_API_KEY",
-	baseURL: "https://api.deepseek.com",
-	baseEnv: "DEEPSEEK_BASE_URL"
-} };
+/** DeepSeek's public API host; used to recognise DeepSeek-family providers. */
+const DEEPSEEK_BALANCE_HOST = "api.deepseek.com";
 var TokenQuotaService = class TokenQuotaService extends Service {
 	static Config = z.object({ storagePath: z.string().default("") });
 	storagePath;
@@ -525,7 +510,7 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 							if (typeof parsed === "object" && parsed !== null && typeof parsed.provider === "string") provider = parsed.provider;
 						}
 					} catch {}
-					(provider !== void 0 && provider in BALANCE_SOURCES ? this.refreshProviderBalance(provider) : this.refreshAllBalances()).finally(() => {
+					(provider !== void 0 ? this.refreshProviderBalance(provider) : this.refreshAllBalances()).finally(() => {
 						const body = JSON.stringify(this.readSnapshot());
 						res.writeHead(200, {
 							"content-type": "application/json",
@@ -619,7 +604,7 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 		return this.upgradeCheck;
 	}
 	async doRefreshUpgrade() {
-		const current = "0.1.39";
+		const current = "0.1.40";
 		let latest;
 		try {
 			const response = await fetch(REGISTRY_LATEST_URL, {
@@ -692,9 +677,46 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 		};
 		schedule();
 	}
-	/** Refresh every provider the host knows how to query. */
+	/** Refresh every DeepSeek-family provider found in the model directory. */
 	refreshAllBalances() {
-		return Promise.all(Object.keys(BALANCE_SOURCES).map((provider) => this.refreshProviderBalance(provider))).then(() => void 0);
+		return Promise.all([...this.discoverBalanceSources().values()].map((source) => this.refreshProviderBalance(source.provider, source))).then(() => void 0);
+	}
+	/**
+	* Discover every provider in the model directory that looks like DeepSeek,
+	* together with its own credential reference and endpoint base. This is what
+	* decides which providers are queried: a provider qualifies when its base URL
+	* points at DeepSeek's API host or its credential reference is named with the
+	* `DEEPSEEK` prefix, and it must declare an `apiKeyEnv`.
+	*/
+	discoverBalanceSources() {
+		const found = /* @__PURE__ */ new Map();
+		const llm = this.ctx.get("llm");
+		const settings = this.ctx.get("settings");
+		const fallback = () => ({
+			provider: "deepseek",
+			keyEnv: "DEEPSEEK_API_KEY",
+			baseURL: process.env.DEEPSEEK_BASE_URL ?? `https://${DEEPSEEK_BALANCE_HOST}`
+		});
+		if (llm?.listConfigurableProviders === void 0) {
+			found.set("deepseek", fallback());
+			return found;
+		}
+		for (const entry of llm.listConfigurableProviders()) try {
+			let profile = settings?.get(entry.settingsNs);
+			for (const segment of entry.settingsPath ?? []) profile = profile !== null && typeof profile === "object" ? profile[segment] : void 0;
+			if (profile === null || typeof profile !== "object") continue;
+			const record = profile;
+			const keyEnv = typeof record.apiKeyEnv === "string" && record.apiKeyEnv !== "" ? record.apiKeyEnv : void 0;
+			const baseURL = typeof record.baseURL === "string" && record.baseURL !== "" ? record.baseURL : void 0;
+			if (!(baseURL !== void 0 && baseURL.includes(DEEPSEEK_BALANCE_HOST) || keyEnv !== void 0 && keyEnv.startsWith("DEEPSEEK") || entry.provider.startsWith("deepseek")) || keyEnv === void 0) continue;
+			found.set(entry.provider, {
+				provider: entry.provider,
+				keyEnv,
+				baseURL: baseURL ?? `https://${DEEPSEEK_BALANCE_HOST}`
+			});
+		} catch {}
+		if (found.size === 0) found.set("deepseek", fallback());
+		return found;
 	}
 	/**
 	* Refresh one provider's cached account balance. The API key is resolved
@@ -702,9 +724,9 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 	* environment as a fallback, so the Models page key is used with nothing to
 	* type. A failed fetch keeps the previous value and marks it as an error.
 	*/
-	async refreshProviderBalance(provider) {
-		const source = BALANCE_SOURCES[provider];
-		if (source === void 0) return;
+	async refreshProviderBalance(provider, source) {
+		const resolved = source ?? this.discoverBalanceSources().get(provider);
+		if (resolved === void 0) return;
 		const existing = this.balanceFetches.get(provider);
 		if (existing !== void 0) return existing;
 		const run = async () => {
@@ -712,9 +734,9 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 			const credentials = this.ctx.get("credentials");
 			let key;
 			if (credentials !== void 0) try {
-				key = (await credentials.resolve(source.keyEnv))?.value;
+				key = (await credentials.resolve(resolved.keyEnv))?.value;
 			} catch {}
-			if (key === void 0 || key === "") key = process.env[source.keyEnv];
+			if (key === void 0 || key === "") key = process.env[resolved.keyEnv];
 			if (key === void 0 || key === "") {
 				this.balances.set(provider, {
 					provider,
@@ -726,7 +748,7 @@ var TokenQuotaService = class TokenQuotaService extends Service {
 				});
 				return;
 			}
-			const base = source.baseEnv !== void 0 ? process.env[source.baseEnv] ?? source.baseURL : source.baseURL;
+			const base = resolved.baseURL;
 			try {
 				const controller = new AbortController();
 				const timer = setTimeout(() => {

package/lib/types/client/TokenQuotaPanel.js

@@ -10,7 +10,7 @@ import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-run
  * callbacks.
  */
 import { useEffect, useMemo, useRef, useState } from 'react';
-import { TOKEN_QUOTA_BALANCE_PROVIDERS, TOKEN_QUOTA_MAX_NOTES } from '@jxgame2020/dsh-token-quota/types';
+import { TOKEN_QUOTA_MAX_NOTES } from '@jxgame2020/dsh-token-quota/types';
 import { mergeModelRows, reorderKeys } from "./store.js";
 import css from './TokenQuotaPanel.module.css';
 /** Compact a token count for display. */
@@ -543,9 +543,10 @@ export function TokenQuotaPanel({ t, load, setLimit, selectModel, setMonitored,
             list.push('deepseek');
         return list;
     }, [groups]);
-    // Balance by provider key, built from the snapshot (status ok/error/unconfigured)
-    // plus a client-side "unsupported" sentinel for providers the Host does not
-    // yet know how to query, and "loading" for supported ones not yet returned.
+    // Balance by provider key. The Host returns a balances entry for every
+    // provider it can query (DeepSeek-family), so a directory provider with no
+    // entry is "not integrated for query" — except during the very first fetch,
+    // when the whole list is still empty and reads as "loading" instead.
     const balanceByProvider = useMemo(() => {
         const map = new Map();
         for (const provider of providers) {
@@ -553,7 +554,7 @@ export function TokenQuotaPanel({ t, load, setLimit, selectModel, setMonitored,
             if (bal !== undefined) {
                 map.set(provider, bal);
             }
-            else if (TOKEN_QUOTA_BALANCE_PROVIDERS.includes(provider)) {
+            else if (balances.length === 0) {
                 map.set(provider, 'loading');
             }
             else {

package/lib/types/index.d.ts

@@ -112,8 +112,16 @@ export declare class TokenQuotaService extends Service {
      * the enabled-transition in `onChange` fetch immediately).
      */
     private syncBalancePolling;
-    /** Refresh every provider the host knows how to query. */
+    /** Refresh every DeepSeek-family provider found in the model directory. */
     private refreshAllBalances;
+    /**
+     * Discover every provider in the model directory that looks like DeepSeek,
+     * together with its own credential reference and endpoint base. This is what
+     * decides which providers are queried: a provider qualifies when its base URL
+     * points at DeepSeek's API host or its credential reference is named with the
+     * `DEEPSEEK` prefix, and it must declare an `apiKeyEnv`.
+     */
+    private discoverBalanceSources;
     /**
      * Refresh one provider's cached account balance. The API key is resolved
      * through the harness credential seam (`credentials.resolve`) with the

package/lib/types/index.js

@@ -271,21 +271,8 @@ function usageTokens(usage) {
         + (usage.cacheWriteTokens ?? 0)
         + usage.outputTokens;
 }
-/**
- * Daily token-quota service.
- *
- * Mount it beside the other rows (`@jxgame2020/dsh-token-quota`) and write
- * per-model limits through the `token-quota` settings namespace.
- */
-/**
- * Providers the host can query for account balance, keyed by route key. Each
- * entry names the credential reference (environment-variable name), the
- * balance endpoint base, and the optional `*_BASE_URL` override variable.
- * Directory providers outside this map render as "not integrated for query".
- */
-const BALANCE_SOURCES = {
-    deepseek: { keyEnv: 'DEEPSEEK_API_KEY', baseURL: 'https://api.deepseek.com', baseEnv: 'DEEPSEEK_BASE_URL' },
-};
+/** DeepSeek's public API host; used to recognise DeepSeek-family providers. */
+const DEEPSEEK_BALANCE_HOST = 'api.deepseek.com';
 export class TokenQuotaService extends Service {
     static Config = z.object({
         storagePath: z.string().default(''),
@@ -551,7 +538,7 @@ export class TokenQuotaService extends Service {
                     catch {
                         // Malformed body: fall back to refreshing everything.
                     }
-                    const task = provider !== undefined && provider in BALANCE_SOURCES
+                    const task = provider !== undefined
                         ? this.refreshProviderBalance(provider)
                         : this.refreshAllBalances();
                     void task.finally(() => {
@@ -745,9 +732,66 @@ export class TokenQuotaService extends Service {
         };
         schedule();
     }
-    /** Refresh every provider the host knows how to query. */
+    /** Refresh every DeepSeek-family provider found in the model directory. */
     refreshAllBalances() {
-        return Promise.all(Object.keys(BALANCE_SOURCES).map(provider => this.refreshProviderBalance(provider))).then(() => undefined);
+        return Promise.all([...this.discoverBalanceSources().values()]
+            .map(source => this.refreshProviderBalance(source.provider, source))).then(() => undefined);
+    }
+    /**
+     * Discover every provider in the model directory that looks like DeepSeek,
+     * together with its own credential reference and endpoint base. This is what
+     * decides which providers are queried: a provider qualifies when its base URL
+     * points at DeepSeek's API host or its credential reference is named with the
+     * `DEEPSEEK` prefix, and it must declare an `apiKeyEnv`.
+     */
+    discoverBalanceSources() {
+        const found = new Map();
+        const llm = this.ctx.get('llm');
+        const settings = this.ctx.get('settings');
+        const fallback = () => ({
+            provider: 'deepseek',
+            keyEnv: 'DEEPSEEK_API_KEY',
+            baseURL: process.env.DEEPSEEK_BASE_URL ?? `https://${DEEPSEEK_BALANCE_HOST}`,
+        });
+        if (llm?.listConfigurableProviders === undefined) {
+            found.set('deepseek', fallback());
+            return found;
+        }
+        for (const entry of llm.listConfigurableProviders()) {
+            try {
+                let profile = settings?.get(entry.settingsNs);
+                for (const segment of entry.settingsPath ?? []) {
+                    profile = profile !== null && typeof profile === 'object'
+                        ? profile[segment]
+                        : undefined;
+                }
+                if (profile === null || typeof profile !== 'object')
+                    continue;
+                const record = profile;
+                const keyEnv = typeof record.apiKeyEnv === 'string' && record.apiKeyEnv !== ''
+                    ? record.apiKeyEnv
+                    : undefined;
+                const baseURL = typeof record.baseURL === 'string' && record.baseURL !== ''
+                    ? record.baseURL
+                    : undefined;
+                const isDeepSeek = (baseURL !== undefined && baseURL.includes(DEEPSEEK_BALANCE_HOST))
+                    || (keyEnv !== undefined && keyEnv.startsWith('DEEPSEEK'))
+                    || entry.provider.startsWith('deepseek');
+                if (!isDeepSeek || keyEnv === undefined)
+                    continue;
+                found.set(entry.provider, {
+                    provider: entry.provider,
+                    keyEnv,
+                    baseURL: baseURL ?? `https://${DEEPSEEK_BALANCE_HOST}`,
+                });
+            }
+            catch {
+                // Malformed provider entry — skip it and keep the others.
+            }
+        }
+        if (found.size === 0)
+            found.set('deepseek', fallback());
+        return found;
     }
     /**
      * Refresh one provider's cached account balance. The API key is resolved
@@ -755,9 +799,9 @@ export class TokenQuotaService extends Service {
      * environment as a fallback, so the Models page key is used with nothing to
      * type. A failed fetch keeps the previous value and marks it as an error.
      */
-    async refreshProviderBalance(provider) {
-        const source = BALANCE_SOURCES[provider];
-        if (source === undefined)
+    async refreshProviderBalance(provider, source) {
+        const resolved = source ?? this.discoverBalanceSources().get(provider);
+        if (resolved === undefined)
             return;
         const existing = this.balanceFetches.get(provider);
         if (existing !== undefined)
@@ -769,14 +813,14 @@ export class TokenQuotaService extends Service {
             let key;
             if (credentials !== undefined) {
                 try {
-                    key = (await credentials.resolve(source.keyEnv))?.value;
+                    key = (await credentials.resolve(resolved.keyEnv))?.value;
                 }
                 catch {
                     // Seam unreachable: fall through to the environment.
                 }
             }
             if (key === undefined || key === '')
-                key = process.env[source.keyEnv];
+                key = process.env[resolved.keyEnv];
             if (key === undefined || key === '') {
                 this.balances.set(provider, {
                     provider,
@@ -788,9 +832,7 @@ export class TokenQuotaService extends Service {
                 });
                 return;
             }
-            const base = source.baseEnv !== undefined
-                ? (process.env[source.baseEnv] ?? source.baseURL)
-                : source.baseURL;
+            const base = resolved.baseURL;
             try {
                 const controller = new AbortController();
                 const timer = setTimeout(() => { controller.abort(); }, 10_000);

package/lib/types/types.d.ts

@@ -55,12 +55,6 @@ export interface TokenQuotaBalanceSettings {
     pollMinutes: number;
 }
 export declare const TOKEN_QUOTA_DEFAULT_BALANCE: TokenQuotaBalanceSettings;
-/**
- * Provider route keys the host knows how to query for account balance. A
- * directory provider outside this list renders as "not integrated for query"
- * in the balance window.
- */
-export declare const TOKEN_QUOTA_BALANCE_PROVIDERS: readonly string[];
 /** Fetch state of the provider account balance. */
 export type TokenQuotaBalanceStatus = 'ok' | 'unconfigured' | 'error';
 /**

package/lib/types/types.js

@@ -33,12 +33,6 @@ export const TOKEN_QUOTA_DEFAULT_BALANCE = {
     enabled: true,
     pollMinutes: 5,
 };
-/**
- * Provider route keys the host knows how to query for account balance. A
- * directory provider outside this list renders as "not integrated for query"
- * in the balance window.
- */
-export const TOKEN_QUOTA_BALANCE_PROVIDERS = ['deepseek'];
 /** Default music settings: silent until the user flips the panel switch. */
 export const TOKEN_QUOTA_DEFAULT_MUSIC = {
     enabled: false,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@jxgame2020/dsh-token-quota",
   "description": "Daily per-model token quota accounting, enforcement, and a floating quota panel for the DeepSeek Harness — a fully self-contained plugin (no core-Harness wiring, no generated RPC contract)",
-  "version": "0.1.39",
+  "version": "0.1.40",
   "publishConfig": {
     "access": "public"
   },