dsh-plugin-subscription-oauth 0.12.0 → 0.12.3

package/lib/protocol.d.ts

@@ -12,7 +12,7 @@
 /** Prefix of every RPC method this plugin serves. */
 export declare const RPC_PREFIX = "subscription-oauth.";
 /** Endpoints the Settings page may call. */
-export declare const RPC_ENDPOINTS: readonly ["status", "login", "cancel", "logout", "setDefault", "refresh", "discover", "setModelVisible", "setModelOrder", "addModel", "removeModel", "usage", "reload", "setAutoOn", "deepseek-balance", "ccswitch-balance", "model-variants"];
+export declare const RPC_ENDPOINTS: readonly ["status", "login", "cancel", "logout", "setDefault", "refresh", "discover", "setModelVisible", "setModelOrder", "addModel", "removeModel", "usage", "reload", "auto-state", "setAutoOn", "deepseek-balance", "ccswitch-balance", "model-variants"];
 /** One endpoint name. */
 export type RpcEndpoint = typeof RPC_ENDPOINTS[number];
 /** Client → host envelope. */
@@ -315,6 +315,10 @@ export interface RpcMap {
         payload: undefined;
         value: SubscriptionStatus;
     };
+    'auto-state': {
+        payload: undefined;
+        value: AutoStateResult;
+    };
     setAutoOn: {
         payload: SetAutoOnRequest;
         value: AutoStateResult;

package/lib/protocol.js

@@ -26,6 +26,7 @@ export const RPC_ENDPOINTS = [
     'removeModel',
     'usage',
     'reload',
+    'auto-state',
     'setAutoOn',
     'deepseek-balance',
     'ccswitch-balance',

package/lib/providers/mimo.js

@@ -303,12 +303,13 @@ export class MimoAdapter extends LlmAdapter {
                         }
                         yield { type: 'text-delta', index: 0, text: delta };
                     }
-                    if (chunk.usage !== undefined) {
-                        const input = chunk.usage.prompt_tokens ?? 0;
-                        const output = chunk.usage.completion_tokens ?? 0;
+                    const usage = chunk.usage;
+                    if (usage !== undefined && usage !== null) {
+                        const input = usage.prompt_tokens ?? 0;
+                        const output = usage.completion_tokens ?? 0;
                         yield {
                             type: 'usage',
-                            usage: { inputTokens: input, outputTokens: output, totalTokens: chunk.usage.total_tokens ?? input + output },
+                            usage: { inputTokens: input, outputTokens: output, totalTokens: usage.total_tokens ?? input + output },
                         };
                     }
                 }

package/lib/providers/workbuddy-provider.js

@@ -115,7 +115,14 @@ export function createWorkBuddyProvider(options) {
             }
             try {
                 const renewed = await refreshWorkBuddyToken(token, fetchFn);
-                return { accessToken: renewed.accessToken, expiresAt: renewed.expiresAt };
+                // WorkBuddy rotates the refresh token: keep the new one so the next restart can
+                // still refresh, otherwise the stale token signs the account out.
+                return {
+                    accessToken: renewed.accessToken,
+                    expiresAt: renewed.expiresAt,
+                    ...renewed.refreshToken === undefined ? {} : { refreshToken: renewed.refreshToken },
+                    ...renewed.refreshExpiresAt === undefined ? {} : { refreshExpiresAt: renewed.refreshExpiresAt },
+                };
             }
             catch (error) {
                 throw new ProviderError(`${WORKBUDDY_PROVIDER_NAME} token refresh failed: ${error instanceof Error ? error.message : String(error)}`, REAUTH_REQUIRED);

package/lib/providers/workbuddy.js

@@ -248,12 +248,13 @@ export class WorkBuddyAdapter extends LlmAdapter {
                         }
                         yield { type: 'text-delta', index: 0, text: delta };
                     }
-                    if (chunk.usage !== undefined) {
-                        const input = chunk.usage.prompt_tokens ?? 0;
-                        const output = chunk.usage.completion_tokens ?? 0;
+                    const usage = chunk.usage;
+                    if (usage !== undefined && usage !== null) {
+                        const input = usage.prompt_tokens ?? 0;
+                        const output = usage.completion_tokens ?? 0;
                         yield {
                             type: 'usage',
-                            usage: { inputTokens: input, outputTokens: output, totalTokens: chunk.usage.total_tokens ?? input + output },
+                            usage: { inputTokens: input, outputTokens: output, totalTokens: usage.total_tokens ?? input + output },
                         };
                     }
                 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-plugin-subscription-oauth",
-  "version": "0.12.0",
+  "version": "0.12.3",
   "description": "Use subscription accounts as DeepSeek Harness LLM providers — Kimi Code (OAuth device flow), Grok / X Premium (OIDC + PKCE), Google Antigravity (Google OAuth + Cloud Code Assist) and GLM / BigModel (ZCode's browser sign-in) — with per-model visibility, drag-to-reorder Auto fallback and plan quotas from the web Settings page",
   "license": "MIT",
   "author": "EricFL1998",