@vibedev-si/dsh-vibedev 0.3.1 → 0.3.2

package/README.md

@@ -97,6 +97,7 @@
 - 图片按张计费，视频按秒计费（参考视频的时长也计入），音乐和播客按次计费，语音按原始秒计费（按服务商返回的原始时长，可能和播放时长不同），转写按分钟计费。`media_models` 会列出当前价格；目录没有给出价格的模型显示「未定价」，不代表免费。
 - 视频和音频只在成功时扣费，失败和取消的任务不扣。视频提交后无法取消。
 - 同一次操作只会生成一次：提交没有得到确认时，会用同一个请求键向网关确认结果；查询或下载失败只会重新查询、刷新链接，不会重新提交。结果已经生成、但费用还在结算（`settlement_pending`）时，结果照常保存，不会重新生成。
+- 余额在每次付费请求结束后（对话回复、生成、转写）马上重新读取，侧栏和设置页几秒内就能看到扣费后的余额。
 - 插件设置里的「每次付费生成前先询问」默认关闭。打开后，每次付费的媒体请求之前都会先弹出确认，并显示预估金额。
 
 ## 设置
@@ -162,6 +163,8 @@ Everything is paid from your VibeDev balance; the DeepSeek account stays DeepSee
 
 **Gateway audio (0.3.1)**: audio models are recognized from the catalog's structured `audio` block (music, podcast, speech, transcription), never from their names; speech generation (Seed) joins music and podcasts; prices show their unit (per generation, per second of original audio, per minute) and "not priced" when the catalog states none; the `vibedevMedia` service gains `startAudio` with an operation key and query-only result reading.
 
+**Balance without the wait (0.3.2)**: the balance is read again right after a paid request ends (a chat reply, a generation, a transcription) and the account pages hear it through a long poll, so a charge shows within seconds instead of after up to two minutes. A page that already has a balance is answered at once while a newer one is read behind it; reads after charges are at least five seconds apart.
+
 **Models**: the list comes from the VibeDev gateway for your account, so nothing is listed before you sign in; models appear right after, without a restart. Using a VibeDev model while signed out, or after the sign-in has ended, asks you to sign in; a used-up balance says so with the top-up link. These messages are separate from the DeepSeek account's. An expired access token is renewed and the request sent again, without repeating any output.
 
 **Web search**: the plugin registers the web search provider `vibedev-gateway`, signed in with your VibeDev account. Registering does not select it: DeepSeek Harness keeps its own search unless the profile sets `searchProvider: vibedev-gateway` on the `web` row (the VibeDev app does). Searching is free, but the gateway refuses every request, searches included, while the balance is zero.

package/THIRD-PARTY-NOTICES.txt

@@ -1,4 +1,4 @@
-THIRD-PARTY NOTICES · @vibedev-si/dsh-vibedev 0.3.1
+THIRD-PARTY NOTICES · @vibedev-si/dsh-vibedev 0.3.2
 
 This package bundles code from the projects below. Each is used under the licence that
 follows its name; the licence texts are reproduced as the projects ship them.

package/client/client.js

@@ -9,6 +9,8 @@ window.__ModuleLoader__.load({ id: "@vibedev-si/dsh-vibedev", factory: (require)
 		const ACCOUNT_ROUTE = "/api/dsh-vibedev/account";
 		const PENDING_POLL_MS = 2e3;
 		const IDLE_POLL_MS = 6e4;
+		/** After a failed long poll, the next one waits this long (the Host may be restarting). */
+		const WATCH_RETRY_MS = 5e3;
 		/** Who to show: the nickname, else the email. */
 		function displayName(view) {
 			return view?.user?.nickname ?? view?.user?.email;
@@ -33,6 +35,8 @@ window.__ModuleLoader__.load({ id: "@vibedev-si/dsh-vibedev", factory: (require)
 			reading;
 			stopped = true;
 			unsubscribeFocus;
+			/** The open long poll, when one is. */
+			watching;
 			/**
 			* @param options - fetch, timers, window opening and focus (test seams).
 			*/
@@ -82,17 +86,61 @@ window.__ModuleLoader__.load({ id: "@vibedev-si/dsh-vibedev", factory: (require)
 					globalThis.clearTimeout(handle);
 				}))(this.timer);
 				this.timer = void 0;
+				this.watching?.abort();
+				this.watching = void 0;
 			}
-			schedule() {
-				if (this.stopped) return;
+			later(callback, ms) {
 				if (this.timer !== void 0) (this.options.clearTimeout ?? ((handle) => {
 					globalThis.clearTimeout(handle);
 				}))(this.timer);
-				const delay = this.state.view?.pending !== void 0 ? PENDING_POLL_MS : IDLE_POLL_MS;
-				this.timer = (this.options.setTimeout ?? ((callback, ms) => globalThis.setTimeout(callback, ms)))(() => {
+				this.timer = (this.options.setTimeout ?? ((run, delay) => globalThis.setTimeout(run, delay)))(() => {
+					this.timer = void 0;
+					callback();
+				}, ms);
+			}
+			schedule() {
+				if (this.stopped) return;
+				const view = this.state.view;
+				if (view?.pending === void 0 && typeof view?.revision === "number") {
+					if (this.timer !== void 0) (this.options.clearTimeout ?? ((handle) => {
+						globalThis.clearTimeout(handle);
+					}))(this.timer);
 					this.timer = void 0;
+					this.watch(view.revision);
+					return;
+				}
+				this.watching?.abort();
+				this.watching = void 0;
+				this.later(() => {
 					this.refresh();
-				}, delay);
+				}, view?.pending !== void 0 ? PENDING_POLL_MS : IDLE_POLL_MS);
+			}
+			/** Keep one long poll open for changes past `revision`. */
+			watch(revision) {
+				if (this.watching !== void 0) return;
+				const controller = new AbortController();
+				this.watching = controller;
+				(async () => {
+					let failed = false;
+					try {
+						const response = await this.fetch(`${ACCOUNT_ROUTE}?since=${revision}`, { signal: controller.signal });
+						if (!response.ok) throw new Error(`HTTP ${response.status}`);
+						const view = await response.json();
+						if (!controller.signal.aborted && (view.revision ?? 0) >= (this.state.view?.revision ?? 0)) this.set({
+							view,
+							loadFailed: false
+						});
+					} catch {
+						failed = !controller.signal.aborted;
+					} finally {
+						if (this.watching === controller) this.watching = void 0;
+					}
+					if (this.stopped || controller.signal.aborted) return;
+					if (failed) this.later(() => {
+						this.refresh();
+					}, WATCH_RETRY_MS);
+					else this.schedule();
+				})();
 			}
 			/**
 			* Read the account now.

package/lib/account/index.d.ts

@@ -7,6 +7,9 @@
  *   app's account, the plugin's own sign-in, a development key, or none), who
  *   is signed in, the balance, a sign-in waiting for the browser, and how many
  *   VibeDev models the pickers list; `?fresh=1` reads the balance again;
+ *   `?since=<revision>` holds the request until the person or the balance
+ *   changes (a charge is read again right after it ends), so the pages show a
+ *   new balance within seconds instead of on their next minute poll;
  * - `POST /api/dsh-vibedev/account/sign-in` — `{ open?: boolean }`: start the
  *   plugin's own sign-in; answers the authorization link (`open: false` leaves
  *   opening it to the page);
@@ -58,6 +61,11 @@ export interface AccountView {
     };
     /** The VibeDev account is the app's main account (the VibeDev app): its row leads the sidebar foot and Settings. */
     readonly primary: boolean;
+    /**
+     * Grows whenever the person or the balance changes; `GET …/account?since=<revision>`
+     * answers as soon as it grows (or after a while with the same view).
+     */
+    readonly revision: number;
 }
 /** What the account routes need from the plugin. */
 export interface AccountServiceOptions {
@@ -74,16 +82,51 @@ export interface AccountServiceOptions {
     readonly primary?: boolean;
     readonly fetch?: typeof globalThis.fetch;
     readonly now?: () => number;
+    /** Timer seams for the reads after a charge and the long poll. */
+    readonly setTimeout?: (callback: () => void, ms: number) => unknown;
+    readonly clearTimeout?: (handle: unknown) => void;
 }
 /** The account state behind the pages. */
 export declare class AccountService {
     private readonly options;
     private balanceCache;
+    /** A background read of a stale balance, so only one runs at a time. */
+    private revalidating;
     private readonly now;
+    private revision;
+    /** What the last read showed, to tell whether the next one changed anything. */
+    private lastSeen;
+    private readonly waiters;
+    private settleTimers;
+    /** Grows with every credential change; a read started before one is dropped. */
+    private epoch;
+    /** When the last read after a charge started, to space them out. */
+    private lastSpendReadAt;
     /**
      * @param options - origin, credential chain, the plugin's sign-in and the route's model state.
      */
     constructor(options: AccountServiceOptions);
+    private timeout;
+    private cancelTimeout;
+    /** Something the pages show changed: answer every waiting `?since=` request. */
+    changed(): void;
+    /** The account changed hands (sign-in, sign-out, switch): forget the balance and tell the pages. */
+    credentialChanged(): void;
+    /**
+     * A request that may have cost money ended (a chat reply, a generation, a
+     * transcription): read the balance again shortly, and once more later if the
+     * charge had not settled yet. Repeated calls in a burst share the reads.
+     */
+    spent(): void;
+    /** Read the balance of the current credential now (the pages hear about a change through the long poll). */
+    private readNow;
+    /**
+     * Wait until the view changes past `since`, or a while.
+     * @param since - the revision the page shows.
+     * @param signal - the page went away.
+     * @returns once there is something new, or the wait ran out.
+     */
+    wait(since: number, signal?: AbortSignal): Promise<void>;
     /** Links to the gateway's pages. */
     links(): AccountView['links'];
     /**
@@ -92,8 +135,12 @@ export declare class AccountService {
      * @returns the view.
      */
     view(fresh?: boolean): Promise<AccountView>;
-    /** `GET /v1/account/auth/me` for one token, reused for a minute. */
+    /**
+     * `GET /v1/account/auth/me` for one token. A read younger than the TTL is
+     * reused; an older one is answered at once while a new read runs behind it.
+     */
     private me;
+    private read;
     /**
      * Start the plugin's own sign-in.
      * @param open - open the system browser (false: the page opens the link itself).
@@ -111,6 +158,8 @@ export declare class AccountService {
      * @returns false when the credential is the app's account, which is signed out in the app.
      */
     signOut(): Promise<boolean>;
+    /** Stop the pending reads after a charge (the plugin is going away). */
+    dispose(): void;
 }
 /**
  * The account routes.

package/lib/gateway/http.d.ts

@@ -31,6 +31,11 @@ export interface GatewayHttpOptions {
     readonly userAgent: string;
     readonly fetch?: typeof globalThis.fetch;
     readonly sleep?: (ms: number, signal?: AbortSignal) => Promise<void>;
+    /** Told about every request the gateway accepted (2xx), after the fact. */
+    readonly onAccepted?: (request: {
+        readonly method: string;
+        readonly path: string;
+    }) => void;
 }
 /** One request. */
 export interface GatewayRequest {

package/lib/index.js

@@ -19,15 +19,41 @@ import { defineTool } from "@deepseek-ai/dsh-tools";
 //#region src/account/index.ts
 /** Path prefix of the plugin's Host routes. */
 const ACCOUNT_ROUTE_PREFIX = "/api/dsh-vibedev/account";
-/** How long a balance read is reused before the gateway is asked again. */
-const BALANCE_TTL_MS = 6e4;
+/**
+* How long a balance read is reused as it is. After that the cached value is
+* still answered at once while a new read runs in the background (the page
+* hears the result through the long poll), so a page never waits on the gateway
+* when it already has a balance.
+*/
+const BALANCE_TTL_MS = 3e4;
+/** How long `?since=` holds a request open when nothing changes. */
+const WAIT_MS = 25e3;
+/**
+* When the balance is read again after a charge: the gateway settles a charge
+* shortly after the request ends, so it is read a moment later, and once more
+* if the first read still shows the old amount.
+*/
+const SETTLE_READS_MS = [1500, 8e3];
+/** Reads after charges are at least this far apart. */
+const MIN_READ_SPACING_MS = 5e3;
 const record$8 = (value) => typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
 const text$8 = (value) => typeof value === "string" && value.trim() !== "" ? value.trim() : void 0;
 /** The account state behind the pages. */
 var AccountService = class {
 	options;
 	balanceCache;
+	/** A background read of a stale balance, so only one runs at a time. */
+	revalidating;
 	now;
+	revision = 0;
+	/** What the last read showed, to tell whether the next one changed anything. */
+	lastSeen;
+	waiters = /* @__PURE__ */ new Set();
+	settleTimers = [];
+	/** Grows with every credential change; a read started before one is dropped. */
+	epoch = 0;
+	/** When the last read after a charge started, to space them out. */
+	lastSpendReadAt = Number.NEGATIVE_INFINITY;
 	/**
 	* @param options - origin, credential chain, the plugin's sign-in and the route's model state.
 	*/
@@ -35,6 +61,73 @@ var AccountService = class {
 		this.options = options;
 		this.now = options.now ?? Date.now;
 	}
+	timeout(callback, ms) {
+		return (this.options.setTimeout ?? ((run, delay) => {
+			const handle = globalThis.setTimeout(run, delay);
+			handle.unref?.();
+			return handle;
+		}))(callback, ms);
+	}
+	cancelTimeout(handle) {
+		(this.options.clearTimeout ?? ((value) => {
+			globalThis.clearTimeout(value);
+		}))(handle);
+	}
+	/** Something the pages show changed: answer every waiting `?since=` request. */
+	changed() {
+		this.revision++;
+		const waiting = [...this.waiters];
+		this.waiters.clear();
+		for (const wake of waiting) wake();
+	}
+	/** The account changed hands (sign-in, sign-out, switch): forget the balance and tell the pages. */
+	credentialChanged() {
+		this.epoch++;
+		this.balanceCache = void 0;
+		this.revalidating = void 0;
+		this.lastSeen = void 0;
+		this.changed();
+	}
+	/**
+	* A request that may have cost money ended (a chat reply, a generation, a
+	* transcription): read the balance again shortly, and once more later if the
+	* charge had not settled yet. Repeated calls in a burst share the reads.
+	*/
+	spent() {
+		for (const handle of this.settleTimers) this.cancelTimeout(handle);
+		const before = this.lastSeen;
+		const spacing = Math.max(0, this.lastSpendReadAt + MIN_READ_SPACING_MS - this.now());
+		this.settleTimers = SETTLE_READS_MS.map((delay, index) => this.timeout(() => {
+			if (index > 0 && this.lastSeen !== before) return;
+			this.readNow();
+		}, Math.max(delay, spacing + (index === 0 ? 0 : delay - SETTLE_READS_MS[0]))));
+	}
+	/** Read the balance of the current credential now (the pages hear about a change through the long poll). */
+	async readNow() {
+		this.lastSpendReadAt = this.now();
+		const credential = await this.options.chain.resolve().catch(() => void 0);
+		if (credential !== void 0) await this.me(credential.token, true);
+	}
+	/**
+	* Wait until the view changes past `since`, or a while.
+	* @param since - the revision the page shows.
+	* @param signal - the page went away.
+	* @returns once there is something new, or the wait ran out.
+	*/
+	async wait(since, signal) {
+		if (since !== this.revision || signal?.aborted === true) return;
+		await new Promise((resolve) => {
+			const done = () => {
+				this.waiters.delete(done);
+				this.cancelTimeout(timer);
+				signal?.removeEventListener("abort", done);
+				resolve();
+			};
+			const timer = this.timeout(done, WAIT_MS);
+			this.waiters.add(done);
+			signal?.addEventListener("abort", done, { once: true });
+		});
+	}
 	/** Links to the gateway's pages. */
 	links() {
 		const origin = this.options.origin;
@@ -62,20 +155,39 @@ var AccountService = class {
 			...pending === void 0 ? {} : { pending },
 			...attempt === void 0 ? {} : { attempt }
 		};
-		if (credential === void 0) return base;
-		const grantUser = credential.kind === "plugin" ? await this.options.login.user().catch(() => void 0) : void 0;
-		const me = await this.me(credential.token, fresh);
+		if (credential === void 0) return {
+			...base,
+			revision: this.revision
+		};
+		const [grantUser, me] = await Promise.all([credential.kind === "plugin" ? this.options.login.user().catch(() => void 0) : Promise.resolve(void 0), this.me(credential.token, fresh)]);
 		const user = me.user ?? userOf(grantUser);
 		return {
 			...base,
 			...user === void 0 ? {} : { user },
-			...me.balance === void 0 ? {} : { balance: me.balance }
+			...me.balance === void 0 ? {} : { balance: me.balance },
+			revision: this.revision
 		};
 	}
-	/** `GET /v1/account/auth/me` for one token, reused for a minute. */
+	/**
+	* `GET /v1/account/auth/me` for one token. A read younger than the TTL is
+	* reused; an older one is answered at once while a new read runs behind it.
+	*/
 	me(token, fresh) {
 		const cached = this.balanceCache;
-		if (!fresh && cached !== void 0 && cached.token === token && this.now() - cached.at < BALANCE_TTL_MS) return cached.value;
+		if (!fresh && cached !== void 0 && cached.token === token) {
+			if (this.now() - cached.at >= BALANCE_TTL_MS && this.revalidating === void 0) {
+				const next = this.read(token);
+				this.revalidating = next;
+				next.finally(() => {
+					if (this.revalidating === next) this.revalidating = void 0;
+				});
+			}
+			return cached.value;
+		}
+		return this.read(token);
+	}
+	read(token) {
+		const epoch = this.epoch;
 		const value = (async () => {
 			try {
 				const response = await (this.options.fetch ?? globalThis.fetch)(`${this.options.origin}/v1/account/auth/me`, {
@@ -111,13 +223,28 @@ var AccountService = class {
 				return {};
 			}
 		})();
-		this.balanceCache = {
+		const previous = this.balanceCache;
+		if (previous === void 0 || previous.token !== token) this.balanceCache = {
 			token,
 			at: this.now(),
 			value
 		};
 		value.then((read) => {
-			if (read.balance === void 0 && this.balanceCache?.value === value) this.balanceCache = void 0;
+			if (epoch !== this.epoch) return;
+			if (read.balance === void 0) {
+				if (this.balanceCache?.value === value) this.balanceCache = void 0;
+				return;
+			}
+			this.balanceCache = {
+				token,
+				at: this.now(),
+				value
+			};
+			const seen = JSON.stringify(read);
+			if (this.lastSeen !== void 0 && this.lastSeen !== seen) {
+				this.lastSeen = seen;
+				this.changed();
+			} else this.lastSeen = seen;
 		});
 		return value;
 	}
@@ -145,9 +272,17 @@ var AccountService = class {
 	async signOut() {
 		if ((await this.options.chain.resolve().catch(() => void 0))?.kind === "account") return false;
 		await this.options.login.signOut();
-		this.balanceCache = void 0;
+		this.credentialChanged();
 		return true;
 	}
+	/** Stop the pending reads after a charge (the plugin is going away). */
+	dispose() {
+		for (const handle of this.settleTimers) this.cancelTimeout(handle);
+		this.settleTimers = [];
+		const waiting = [...this.waiters];
+		this.waiters.clear();
+		for (const wake of waiting) wake();
+	}
 };
 function userOf(user) {
 	if (user === void 0) return void 0;
@@ -189,7 +324,10 @@ function accountRoutes(account) {
 			requestBody: "buffered",
 			fetch: async (request) => {
 				try {
-					return json(200, await account.view(new URL(request.url, "http://host").searchParams.get("fresh") === "1"));
+					const query = new URL(request.url, "http://host").searchParams;
+					const since = query.get("since");
+					if (since !== null && /^\d{1,15}$/.test(since)) await account.wait(Number(since), request.signal);
+					return json(200, await account.view(query.get("fresh") === "1"));
 				} catch (error) {
 					return failure(error);
 				}
@@ -2761,6 +2899,7 @@ var GatewayAdapter = class extends LlmAdapter {
 			const upstream = options.signal === void 0 ? consumer.signal : AbortSignal.any([options.signal, consumer.signal]);
 			const streamIdleTimeoutMs = this.config.streamIdleTimeoutMs ?? 3e5;
 			const watchdog = _usingCtx$1.u(idleWatchdog(upstream, streamIdleTimeoutMs, "LLM_STREAM_IDLE_TIMEOUT"));
+			let answered = false;
 			try {
 				const containsImage = options.messages.some((message) => contentHasImage(message.content));
 				if (containsImage && !model.input.includes("image")) throw new LlmError(`VibeDev model "${model.id}" does not support image input`, "UNSUPPORTED_CONTENT");
@@ -2797,6 +2936,7 @@ var GatewayAdapter = class extends LlmAdapter {
 							...init,
 							redirect: "error"
 						});
+						answered = true;
 						response = {
 							status: received.status,
 							headers: Object.fromEntries(received.headers.entries())
@@ -2882,6 +3022,9 @@ var GatewayAdapter = class extends LlmAdapter {
 				throw error;
 			} finally {
 				consumer.abort("gateway stream consumer stopped");
+				if (answered) try {
+					this.config.onRequestEnded?.();
+				} catch {}
 			}
 		} catch (_) {
 			_usingCtx$1.e = _;
@@ -3034,7 +3177,8 @@ function installGatewayModels(ctx, options) {
 		resolveImageAccess: (attachments, ref) => resolveImageAttachmentAccess(attachments, (hostPath) => ctx.get("fs")?.processPathFromHostPath(hostPath), ref),
 		onReplayDegrade: ({ provider, model, reason }) => {
 			ctx.logger.warn(`dsh-vibedev: unusable replay state on assistant history for "${provider}/${model}"; sending provider-neutral content (${reason})`);
-		}
+		},
+		...options.onRequestEnded === void 0 ? {} : { onRequestEnded: options.onRequestEnded }
 	});
 	const registration = ctx.llm.registerAdapter([VIBEDEV_ROUTE], adapter);
 	ctx.inject(["settings"], (child) => {
@@ -3380,12 +3524,21 @@ var GatewayHttp = class {
 	*/
 	async send(path, request = {}) {
 		const url = this.url(path);
-		let credential = request.anonymous !== true && this.isGatewayUrl(url) ? await this.credential() : void 0;
+		const authenticate = request.anonymous !== true && this.isGatewayUrl(url);
+		let credential = authenticate ? await this.credential() : void 0;
 		let refreshed = false;
 		let busyRetries = request.busyRetries ?? 2;
 		for (;;) {
 			const response = await this.attempt(url, request, credential);
-			if (response.ok) return response;
+			if (response.ok) {
+				if (authenticate) try {
+					this.options.onAccepted?.({
+						method: request.method ?? "GET",
+						path: new URL(url).pathname
+					});
+				} catch {}
+				return response;
+			}
 			const failure = parseGatewayFailure(await response.clone().json().catch(() => void 0));
 			if (response.status === 401 && credential !== void 0 && credential.kind !== "key" && !refreshed) {
 				refreshed = true;
@@ -7353,11 +7506,15 @@ var TaskTracker = class {
 	}
 	async finish(id, patch) {
 		this.downloadFailures.delete(id);
-		return this.update(id, {
+		const record = await this.update(id, {
 			...patch,
 			progress: void 0,
 			finishedAt: this.now()
 		});
+		try {
+			this.options.onFinished?.(record);
+		} catch {}
+		return record;
 	}
 	/** Hold (or keep) this process's claim on a task, renewing it while held. */
 	async holdClaim(id) {
@@ -8415,6 +8572,8 @@ function audioTranscribeTool(runtime) {
 * ```
 * @module dsh-vibedev
 */
+/** Requests that are charged: generations and transcriptions (uploads and reads are free). */
+const PAID_PATH = /^\/v1\/(?:images|videos|audio)(?:\/|$)/;
 /** The package version, sent in the `User-Agent` the gateway attributes plugin traffic by. */
 const version = JSON.parse(readFileSync(new URL("../package.json", import.meta.url), "utf8")).version;
 const name = "dsh-vibedev";
@@ -8509,17 +8668,25 @@ function apply(ctx, config) {
 		account: () => ctx.get("deepseekAccount"),
 		apiKey: () => developmentKey(origin, config.apiKeyEnv)
 	});
+	let account;
+	const spent = () => {
+		account?.spent();
+	};
 	const http = new GatewayHttp({
 		origin,
 		userAgent,
 		resolveCredential: () => chain.resolve(),
-		rejectToken: (credential) => chain.reject(credential)
+		rejectToken: (credential) => chain.reject(credential),
+		onAccepted: ({ method, path }) => {
+			if (method === "POST" && PAID_PATH.test(path)) spent();
+		}
 	});
 	const store = new TaskStore(gatewayStateDirectory(stateDir, origin));
 	const tracker = new TaskTracker({
 		http,
 		store,
-		log: warn
+		log: warn,
+		onFinished: spent
 	});
 	const slots = {
 		image: config.imageModel,
@@ -8573,7 +8740,8 @@ function apply(ctx, config) {
 			deviceId: () => storage.deviceId(),
 			preferredModels: () => config.preferredModels,
 			onCredentialChange: (listener) => login.onChange(listener),
-			catalogRefreshMinutes: config.catalogRefreshMinutes
+			catalogRefreshMinutes: config.catalogRefreshMinutes,
+			onRequestEnded: spent
 		});
 		llmCtx.effect(() => () => {
 			route = void 0;
@@ -8590,7 +8758,7 @@ function apply(ctx, config) {
 			maxRetryWaitMs: 1e4
 		}));
 	});
-	const account = new AccountService({
+	const accountService = new AccountService({
 		origin,
 		userAgent,
 		chain,
@@ -8601,8 +8769,9 @@ function apply(ctx, config) {
 			...route?.hidden() === void 0 ? {} : { hidden: route.hidden() }
 		})
 	});
+	account = accountService;
 	ctx.inject(["connection"], (scoped) => {
-		for (const accountRoute of accountRoutes(account)) scoped.effect(() => scoped.connection.fetch.register(accountRoute), `dsh-vibedev: ${accountRoute.path}`);
+		for (const accountRoute of accountRoutes(accountService)) scoped.effect(() => scoped.connection.fetch.register(accountRoute), `dsh-vibedev: ${accountRoute.path}`);
 	});
 	ctx.effect(() => {
 		tracker.resume().catch((error) => {
@@ -8610,9 +8779,11 @@ function apply(ctx, config) {
 		});
 		const unsubscribe = login.onChange(() => {
 			runtime.invalidateCatalog();
+			accountService.credentialChanged();
 		});
 		return () => {
 			unsubscribe();
+			accountService.dispose();
 			tracker.dispose();
 			login.dispose();
 		};
@@ -8622,7 +8793,10 @@ function apply(ctx, config) {
 			const controller = new AbortController();
 			(async () => {
 				try {
-					for await (const _view of accountCtx.deepseekAccount.watch(controller.signal)) runtime.invalidateCatalog();
+					for await (const _view of accountCtx.deepseekAccount.watch(controller.signal)) {
+						runtime.invalidateCatalog();
+						accountService.credentialChanged();
+					}
 				} catch {}
 			})();
 			return () => {

package/lib/llm/adapter.d.ts

@@ -87,6 +87,8 @@ export interface GatewayAdapterOptions {
         model: string;
         reason: string;
     }) => void;
+    /** A request reached the gateway and ended, however it ended: it may have been charged. */
+    readonly onRequestEnded?: () => void;
 }
 /**
  * The gateway as an LLM adapter. Each operation reads the current catalog, so a refresh reaches

package/lib/llm/index.d.ts

@@ -43,6 +43,8 @@ export interface GatewayModelsOptions {
     readonly catalogRefreshMinutes?: number;
     readonly catalogTimeoutMs?: number;
     readonly cacheRetention?: 'none' | 'short' | 'long';
+    /** A model request that may have been charged ended (the account reads the balance again). */
+    readonly onRequestEnded?: () => void;
 }
 /** The route's live state, for status lines and tests. */
 export interface GatewayModels {

package/lib/tasks/tracker.d.ts

@@ -47,6 +47,8 @@ export interface TrackerOptions {
     readonly maxDownloadBytes?: number;
     /** A claim not renewed for this long is taken over by another process. Defaults to 60 seconds. */
     readonly claimStaleMs?: number;
+    /** Told when a task ends (completed, failed or lost). */
+    readonly onFinished?: (record: TaskRecord) => void;
 }
 /**
  * How a failed submission may be taken further.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibedev-si/dsh-vibedev",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "VibeDev 账号与模型：在 DeepSeek Harness 里登录 VibeDev，与 DeepSeek 账号并存，使用 VibeDev 网关的模型（Claude、GPT 等），并生成图片、视频、音乐和播客 · VibeDev account and models for DeepSeek Harness: sign in to VibeDev next to the DeepSeek account and use the VibeDev gateway's models (Claude, GPT and more), plus image, video, music and podcast generation",
   "type": "module",
   "main": "lib/index.js",