dsh-plugin-subscription-oauth 0.8.12 → 0.8.14

package/lib/auth/provider-controller.d.ts

@@ -69,6 +69,8 @@ export interface ProviderSpec<S extends SessionBase> {
     applyDiscovery?(ids: readonly string[]): void;
     /** Read one account's plan and metered allowances, when the provider states them. */
     usage?(accessToken: string, account: string): Promise<ProviderUsageRead | undefined>;
+    /** The fast/lightspeed variant wire id for a base model, when the provider reported one. */
+    fastVariant?(baseId: string): string | undefined;
 }
 /** What a provider's usage read returns, before the controller stamps it. */
 export interface ProviderUsageRead {

package/lib/auth/provider-controller.js

@@ -168,16 +168,20 @@ export class ProviderController {
         // A hand-added id the adapter has not merged yet is appended here, with the
         // conservative metadata this build can honestly state. The adapter carries the
         // same ids for the picker; this list must not depend on its async refresh.
-        const rows = listed.map((entry) => ({
-            id: entry.id,
-            name: entry.name,
-            ...entry.description === undefined ? {} : { description: entry.description },
-            contextWindow: entry.contextWindow,
-            ...entry.inputModalities === undefined ? {} : { inputModalities: entry.inputModalities },
-            ...entry.defaultEffort === undefined ? {} : { defaultEffort: entry.defaultEffort },
-            visible: !hidden.has(entry.id),
-            ...userAdded.has(entry.id) ? { userAdded: true } : {},
-        }));
+        const rows = listed.map((entry) => {
+            const fast = this.options.spec.fastVariant?.(entry.id);
+            return {
+                id: entry.id,
+                name: entry.name,
+                ...entry.description === undefined ? {} : { description: entry.description },
+                contextWindow: entry.contextWindow,
+                ...entry.inputModalities === undefined ? {} : { inputModalities: entry.inputModalities },
+                ...entry.defaultEffort === undefined ? {} : { defaultEffort: entry.defaultEffort },
+                visible: !hidden.has(entry.id),
+                ...userAdded.has(entry.id) ? { userAdded: true } : {},
+                ...fast === undefined ? {} : { fastVariant: fast },
+            };
+        });
         for (const id of added) {
             if (known.has(id) || own.has(id))
                 continue;

package/lib/auth/registry.d.ts

@@ -8,7 +8,7 @@
  *
  * @module dsh-plugin-subscription-oauth/auth/registry
  */
-import type { LoginStart, ProviderStatus, RefreshResult, SubscriptionStatus, UsageResult } from '../protocol.js';
+import type { LoginStart, ModelVariants, ProviderStatus, RefreshResult, SubscriptionStatus, UsageResult } from '../protocol.js';
 import type { LlmAdapter } from '@deepseek-ai/dsh-llm';
 import type { ProviderLoginStart, ProviderRefreshResult, ProviderUsageResult } from './provider-controller.js';
 /** Everything the registry needs from one provider route. */
@@ -31,6 +31,8 @@ export interface ProviderRoute {
     removeModel(id: string): Promise<ProviderStatus>;
     usage(force?: boolean): Promise<ProviderUsageResult>;
     reload(): Promise<ProviderStatus>;
+    /** Fast and image variants for one model, when the route's discovery reported them. */
+    modelVariants?(id: string): Promise<ModelVariants | undefined> | ModelVariants | undefined;
 }
 /**
  * Routes every Settings call to the provider it names.
@@ -88,4 +90,6 @@ export declare class SubscriptionRegistry {
     usage(provider: string, force?: boolean): Promise<UsageResult>;
     /** Re-read every credential file and the shared preferences. */
     reload(): Promise<SubscriptionStatus>;
+    /** The fast and image variants for one model, as the route's discovery reported them. */
+    modelVariants(provider: string, id: string): Promise<ModelVariants | undefined>;
 }

package/lib/auth/registry.js

@@ -146,4 +146,9 @@ export class SubscriptionRegistry {
         await Promise.all([...this.byId.values()].map((route) => route.reload()));
         return this.status();
     }
+    /** The fast and image variants for one model, as the route's discovery reported them. */
+    async modelVariants(provider, id) {
+        const route = this.route(provider);
+        return route.modelVariants?.(id);
+    }
 }

package/lib/auth/rpc.js

@@ -185,6 +185,15 @@ async function runEndpoint(registry, endpoint, payload, balance) {
             return await queryUsageBalance({ apiKey, baseUrl })
                 ?? await queryCcSwitchBalance({ apiKey, baseUrl });
         }
+        case 'model-variants': {
+            // The fast variant for one model, as the route's discovery reported it. The picker
+            // uses this to show a quick-mode button without listing the variant as its own row.
+            const provider = requiredProvider(payload, endpoint);
+            const id = typeof payload.id === 'string' ? payload.id : undefined;
+            if (id === undefined)
+                return undefined;
+            return registry.modelVariants(provider, id);
+        }
         case 'reload':
             return registry.reload();
         default: