@percygfx/opencode-antigravity-auth 0.0.0-stage → 2.0.0

package/dist/src/plugin/model-registry.d.ts

@@ -0,0 +1,92 @@
+/**
+ * Antigravity model registry.
+ *
+ * The models this plugin offers come from Antigravity itself: `fetchAvailableModels` returns a
+ * "Recommended" list with display names, limits and capabilities. The registry turns that list
+ * into catalog entries (one entry per model, Low/Medium/High tiers as variants) and answers which
+ * upstream model ID and User-Agent a request needs. New models therefore appear, and retired ones
+ * disappear, without code changes.
+ *
+ * Until a discovery has run (before sign-in, offline), a built-in snapshot is used. The latest
+ * discovery is cached on disk so later starts have it immediately.
+ */
+export type ModelTier = "minimal" | "low" | "medium" | "high";
+export type ModelVendor = "google" | "anthropic" | "openai" | "other";
+/** One entry of `fetchAvailableModels().models`, reduced to what the registry uses. */
+export interface AvailableModelInfo {
+    displayName?: string;
+    modelProvider?: string;
+    supportsImages?: boolean;
+    supportsThinking?: boolean;
+    maxTokens?: number;
+    maxOutputTokens?: number;
+    tagTitle?: string;
+    tagDescription?: string;
+}
+/** The parts of a `fetchAvailableModels` response the registry reads. */
+export interface AvailableModelsResponse {
+    models?: Record<string, AvailableModelInfo | undefined>;
+    agentModelSorts?: Array<{
+        groups?: Array<{
+            modelIds?: string[];
+        }>;
+    }>;
+}
+export interface RegistryVariant {
+    tier: ModelTier;
+    upstreamId: string;
+    /** Only served to the current Antigravity client (hub User-Agent). */
+    hubOnly: boolean;
+}
+export interface RegistryModel {
+    /** Catalog ID without the `antigravity-` prefix, e.g. `gemini-3.8-flash`, `claude-opus-4-6-thinking`. */
+    id: string;
+    /** Display name without the tier, e.g. `Gemini 3.8 Flash`. */
+    name: string;
+    vendor: ModelVendor;
+    /** Upstream ID used when no tier is requested (Antigravity's first-listed tier). */
+    upstreamId: string;
+    hubOnly: boolean;
+    /** Tiers that map to their own upstream IDs; empty for single-ID models. */
+    variants: RegistryVariant[];
+    supportsImages: boolean;
+    supportsThinking: boolean;
+    contextLimit: number;
+    outputLimit: number;
+    /** Antigravity's notice, e.g. "Leaving Soon: Start using Gemini 3.8 Flash". */
+    notice?: string;
+    leavingSoon: boolean;
+}
+export interface UpstreamModelResolution {
+    model: RegistryModel;
+    upstreamId: string;
+    tier?: ModelTier;
+    hubOnly: boolean;
+}
+/**
+ * Builds registry models from a `fetchAvailableModels` response made with the hub User-Agent.
+ * `legacyIds` are the model IDs the legacy User-Agent sees; IDs missing from it need the hub UA.
+ *
+ * Entries named "<Name> (Low|Medium|High)" that share a name become one model with variants.
+ * A name that appears with a single tier (e.g. "GPT-OSS 120B (Medium)") stays a plain model.
+ */
+export declare function buildRegistryModels(hubResponse: AvailableModelsResponse, legacyIds?: ReadonlySet<string>): RegistryModel[];
+export declare function builtinRegistryModels(): RegistryModel[];
+export declare function getRegistryCachePath(): string;
+/** The models currently offered: the last discovery (memory, then disk cache) or the snapshot. */
+export declare function getRegistryModels(): RegistryModel[];
+/** When the current list was discovered, or undefined for the built-in snapshot. */
+export declare function getRegistryFetchedAt(): number | undefined;
+/** Replaces the current list with a fresh discovery and caches it on disk. */
+export declare function setRegistryModels(models: RegistryModel[], fetchedAt?: number): void;
+/** Test hook: forget the in-memory state so the next read starts over. */
+export declare function resetRegistryForTests(models?: RegistryModel[]): void;
+/**
+ * Maps a requested model (catalog ID, with or without `antigravity-`, or an upstream ID) and an
+ * optional tier to the upstream model ID to send. Returns undefined for models the registry does
+ * not know, which keep the pipeline's own name handling.
+ */
+export declare function resolveUpstreamModel(requestedModel: string, tier?: string): UpstreamModelResolution | undefined;
+/** True when the upstream ID is only served to the current Antigravity client. */
+export declare function requiresHubUserAgent(upstreamId: string): boolean;
+//# sourceMappingURL=model-registry.d.ts.map

package/dist/src/plugin/model-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../../src/plugin/model-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAUH,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAC9D,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,OAAO,CAAC;AAKtE,uFAAuF;AACvF,MAAM,WAAW,kBAAkB;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,yEAAyE;AACzE,MAAM,WAAW,uBAAuB;IACtC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,GAAG,SAAS,CAAC,CAAC;IACxD,eAAe,CAAC,EAAE,KAAK,CAAC;QAAE,MAAM,CAAC,EAAE,KAAK,CAAC;YAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAA;SAAE,CAAC,CAAA;KAAE,CAAC,CAAC;CACtE;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,SAAS,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,sEAAsE;IACtE,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,aAAa;IAC5B,yGAAyG;IACzG,EAAE,EAAE,MAAM,CAAC;IACX,8DAA8D;IAC9D,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,WAAW,CAAC;IACpB,oFAAoF;IACpF,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,4EAA4E;IAC5E,QAAQ,EAAE,eAAe,EAAE,CAAC;IAC5B,cAAc,EAAE,OAAO,CAAC;IACxB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,+EAA+E;IAC/E,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,uBAAuB;IACtC,KAAK,EAAE,aAAa,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,SAAS,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;CAClB;AAgCD;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,uBAAuB,EACpC,SAAS,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,GAC9B,aAAa,EAAE,CAsDjB;AAuBD,wBAAgB,qBAAqB,IAAI,aAAa,EAAE,CAWvD;AAsBD,wBAAgB,oBAAoB,IAAI,MAAM,CAE7C;AAuBD,kGAAkG;AAClG,wBAAgB,iBAAiB,IAAI,aAAa,EAAE,CAKnD;AAED,oFAAoF;AACpF,wBAAgB,oBAAoB,IAAI,MAAM,GAAG,SAAS,CAGzD;AAED,8EAA8E;AAC9E,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,SAAS,SAAa,GAAG,IAAI,CAYvF;AAED,0EAA0E;AAC1E,wBAAgB,qBAAqB,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,GAAG,IAAI,CAEpE;AAWD;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,cAAc,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,uBAAuB,GAAG,SAAS,CAwB/G;AAED,kFAAkF;AAClF,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAOhE"}

package/dist/src/plugin/model-registry.js

@@ -0,0 +1,248 @@
+/**
+ * Antigravity model registry.
+ *
+ * The models this plugin offers come from Antigravity itself: `fetchAvailableModels` returns a
+ * "Recommended" list with display names, limits and capabilities. The registry turns that list
+ * into catalog entries (one entry per model, Low/Medium/High tiers as variants) and answers which
+ * upstream model ID and User-Agent a request needs. New models therefore appear, and retired ones
+ * disappear, without code changes.
+ *
+ * Until a discovery has run (before sign-in, offline), a built-in snapshot is used. The latest
+ * discovery is cached on disk so later starts have it immediately.
+ */
+import { existsSync, mkdirSync, readFileSync, renameSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+import { createLogger } from "./logger";
+const log = createLogger("model-registry");
+const TIER_ORDER = ["minimal", "low", "medium", "high"];
+const TIER_SUFFIX = /^(.*\S)\s+\((minimal|low|medium|high)\)$/i;
+const FALLBACK_CONTEXT = 200_000;
+const FALLBACK_OUTPUT = 32_000;
+function vendorOf(modelProvider, id) {
+    const provider = (modelProvider ?? "").toUpperCase();
+    if (provider.includes("GOOGLE") || id.startsWith("gemini"))
+        return "google";
+    if (provider.includes("ANTHROPIC") || id.startsWith("claude"))
+        return "anthropic";
+    if (provider.includes("OPENAI") || id.startsWith("gpt"))
+        return "openai";
+    return "other";
+}
+function slugify(name) {
+    return name
+        .toLowerCase()
+        .replace(/[^a-z0-9.]+/g, "-")
+        .replace(/^-+|-+$/g, "");
+}
+function recommendedIds(response) {
+    const ids = [];
+    for (const sort of response.agentModelSorts ?? []) {
+        for (const group of sort.groups ?? []) {
+            for (const id of group.modelIds ?? []) {
+                if (!ids.includes(id))
+                    ids.push(id);
+            }
+        }
+    }
+    return ids;
+}
+/**
+ * Builds registry models from a `fetchAvailableModels` response made with the hub User-Agent.
+ * `legacyIds` are the model IDs the legacy User-Agent sees; IDs missing from it need the hub UA.
+ *
+ * Entries named "<Name> (Low|Medium|High)" that share a name become one model with variants.
+ * A name that appears with a single tier (e.g. "GPT-OSS 120B (Medium)") stays a plain model.
+ */
+export function buildRegistryModels(hubResponse, legacyIds) {
+    const groups = new Map();
+    for (const upstreamId of recommendedIds(hubResponse)) {
+        const info = hubResponse.models?.[upstreamId];
+        if (!info)
+            continue;
+        const displayName = info.displayName?.trim() || upstreamId;
+        const match = TIER_SUFFIX.exec(displayName);
+        const base = match?.[1] ?? displayName;
+        const tier = match?.[2]?.toLowerCase();
+        const members = groups.get(base) ?? [];
+        members.push({ upstreamId, tier, info });
+        groups.set(base, members);
+    }
+    const isHubOnly = (upstreamId) => (legacyIds ? !legacyIds.has(upstreamId) : false);
+    const models = [];
+    for (const [base, members] of groups) {
+        const tiered = members.length > 1 && members.every((member) => member.tier);
+        const first = members[0];
+        const info = first.info;
+        const notice = [info.tagTitle, info.tagDescription].filter(Boolean).join(": ") || undefined;
+        const variants = tiered
+            ? members
+                .map((member) => ({
+                tier: member.tier,
+                upstreamId: member.upstreamId,
+                hubOnly: isHubOnly(member.upstreamId),
+            }))
+                .sort((a, b) => TIER_ORDER.indexOf(a.tier) - TIER_ORDER.indexOf(b.tier))
+            : [];
+        for (const member of tiered ? [first] : members) {
+            const memberInfo = member.info;
+            models.push({
+                id: tiered ? slugify(base) : member.upstreamId,
+                name: tiered ? base : memberInfo.displayName?.trim() || member.upstreamId,
+                vendor: vendorOf(memberInfo.modelProvider, member.upstreamId),
+                upstreamId: member.upstreamId,
+                hubOnly: isHubOnly(member.upstreamId),
+                variants,
+                supportsImages: Boolean(memberInfo.supportsImages),
+                supportsThinking: Boolean(memberInfo.supportsThinking),
+                contextLimit: memberInfo.maxTokens ?? FALLBACK_CONTEXT,
+                outputLimit: memberInfo.maxOutputTokens ?? FALLBACK_OUTPUT,
+                notice,
+                leavingSoon: /leaving soon|will be removed|no longer be available/i.test(notice ?? ""),
+            });
+        }
+    }
+    return models;
+}
+/**
+ * Snapshot of Antigravity's Recommended list (hub User-Agent), taken 2026-10-13. Used until the
+ * first discovery succeeds. Refresh it with `node scripts/list-models.mjs --json`.
+ */
+const BUILTIN_SNAPSHOT = [
+    { id: "gemini-3.8-flash-high", displayName: "Gemini 3.8 Flash (High)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true },
+    { id: "gemini-3.8-flash-medium", displayName: "Gemini 3.8 Flash (Medium)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true },
+    { id: "gemini-3.8-flash-low", displayName: "Gemini 3.8 Flash (Low)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true },
+    { id: "gemini-3.7-flash-high", displayName: "Gemini 3.7 Flash (High)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-3.7-flash-medium", displayName: "Gemini 3.7 Flash (Medium)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-3.7-flash-low", displayName: "Gemini 3.7 Flash (Low)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, hubOnly: true, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-3.6-flash-high", displayName: "Gemini 3.6 Flash (High)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-3.6-flash-medium", displayName: "Gemini 3.6 Flash (Medium)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-3.6-flash-low", displayName: "Gemini 3.6 Flash (Low)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, tagTitle: "Leaving Soon", tagDescription: "Start using Gemini 3.8 Flash, our best Flash model." },
+    { id: "gemini-pro-agent", displayName: "Gemini 3.1 Pro (High)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, tagTitle: "Leaving Soon", tagDescription: "Use Gemini 3.8 Flash! As a newer model, it delivers more intelligence at 2x the speed for 70% less cost than Gemini 3.1 Pro." },
+    { id: "gemini-3.1-pro-low", displayName: "Gemini 3.1 Pro (Low)", modelProvider: "MODEL_PROVIDER_GOOGLE", supportsImages: true, supportsThinking: true, maxTokens: 1048576, maxOutputTokens: 65536, tagTitle: "Leaving Soon", tagDescription: "Use Gemini 3.8 Flash! As a newer model, it delivers more intelligence at 2x the speed for 70% less cost than Gemini 3.1 Pro." },
+    { id: "claude-sonnet-4-6", displayName: "Claude Sonnet 4.6 (Thinking)", modelProvider: "MODEL_PROVIDER_ANTHROPIC", supportsImages: true, supportsThinking: true, maxTokens: 250000, maxOutputTokens: 64000, tagTitle: "Notice", tagDescription: "Sonnet 5.5 is now available on paid Pro and Ultra plans. Third-party model access will no longer be available on your current plan starting on November 2, 2026." },
+    { id: "claude-opus-4-6-thinking", displayName: "Claude Opus 4.6 (Thinking)", modelProvider: "MODEL_PROVIDER_ANTHROPIC", supportsImages: true, supportsThinking: true, maxTokens: 250000, maxOutputTokens: 64000, tagTitle: "Notice", tagDescription: "Opus 5.5 is now available on paid Pro and Ultra plans. Third-party model access will no longer be available on your current plan starting on November 2, 2026." },
+    { id: "gpt-oss-120b-medium", displayName: "GPT-OSS 120B (Medium)", modelProvider: "MODEL_PROVIDER_OPENAI", supportsImages: false, supportsThinking: true, maxTokens: 131072, maxOutputTokens: 32768, tagTitle: "Notice", tagDescription: "GPT-OSS will be removed from Antigravity on November 2, 2026." },
+];
+export function builtinRegistryModels() {
+    const models = {};
+    const legacyIds = new Set();
+    for (const { id, hubOnly, ...info } of BUILTIN_SNAPSHOT) {
+        models[id] = info;
+        if (!hubOnly)
+            legacyIds.add(id);
+    }
+    return buildRegistryModels({ models, agentModelSorts: [{ groups: [{ modelIds: BUILTIN_SNAPSHOT.map((entry) => entry.id) }] }] }, legacyIds);
+}
+// ---------------------------------------------------------------------------
+// Current registry state (process-wide) and its on-disk cache
+// ---------------------------------------------------------------------------
+const CACHE_FILENAME = "antigravity-models.json";
+const CACHE_VERSION = 1;
+let current;
+function configDir() {
+    if (process.env.OPENCODE_CONFIG_DIR)
+        return process.env.OPENCODE_CONFIG_DIR;
+    return join(process.env.XDG_CONFIG_HOME || join(homedir(), ".config"), "opencode");
+}
+export function getRegistryCachePath() {
+    return join(configDir(), CACHE_FILENAME);
+}
+function isRegistryModel(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const model = value;
+    return typeof model.id === "string" && typeof model.upstreamId === "string" && Array.isArray(model.variants);
+}
+function readCache() {
+    const path = getRegistryCachePath();
+    if (!existsSync(path))
+        return undefined;
+    try {
+        const parsed = JSON.parse(readFileSync(path, "utf8"));
+        if (parsed.version !== CACHE_VERSION || !Array.isArray(parsed.models))
+            return undefined;
+        const models = parsed.models.filter(isRegistryModel);
+        if (models.length === 0)
+            return undefined;
+        return { models, fetchedAt: typeof parsed.fetchedAt === "number" ? parsed.fetchedAt : 0 };
+    }
+    catch (error) {
+        log.warn("ignoring unreadable model cache", { path, error: String(error) });
+        return undefined;
+    }
+}
+/** The models currently offered: the last discovery (memory, then disk cache) or the snapshot. */
+export function getRegistryModels() {
+    if (!current) {
+        current = readCache() ?? { models: builtinRegistryModels() };
+    }
+    return current.models;
+}
+/** When the current list was discovered, or undefined for the built-in snapshot. */
+export function getRegistryFetchedAt() {
+    getRegistryModels();
+    return current?.fetchedAt;
+}
+/** Replaces the current list with a fresh discovery and caches it on disk. */
+export function setRegistryModels(models, fetchedAt = Date.now()) {
+    if (models.length === 0)
+        return;
+    current = { models, fetchedAt };
+    const path = getRegistryCachePath();
+    try {
+        mkdirSync(dirname(path), { recursive: true });
+        const temp = `${path}.${process.pid}.tmp`;
+        writeFileSync(temp, `${JSON.stringify({ version: CACHE_VERSION, fetchedAt, models }, null, 2)}\n`);
+        renameSync(temp, path);
+    }
+    catch (error) {
+        log.warn("could not write model cache", { path, error: String(error) });
+    }
+}
+/** Test hook: forget the in-memory state so the next read starts over. */
+export function resetRegistryForTests(models) {
+    current = models ? { models } : undefined;
+}
+// ---------------------------------------------------------------------------
+// Lookups used by the request pipeline
+// ---------------------------------------------------------------------------
+function normalizeTier(value) {
+    const lower = value?.toLowerCase();
+    return TIER_ORDER.find((tier) => tier === lower);
+}
+/**
+ * Maps a requested model (catalog ID, with or without `antigravity-`, or an upstream ID) and an
+ * optional tier to the upstream model ID to send. Returns undefined for models the registry does
+ * not know, which keep the pipeline's own name handling.
+ */
+export function resolveUpstreamModel(requestedModel, tier) {
+    const name = requestedModel.replace(/^antigravity-/i, "").toLowerCase();
+    const requestedTier = normalizeTier(tier);
+    for (const model of getRegistryModels()) {
+        if (model.id.toLowerCase() === name) {
+            if (model.variants.length === 0) {
+                return { model, upstreamId: model.upstreamId, hubOnly: model.hubOnly };
+            }
+            const variant = model.variants.find((entry) => entry.tier === requestedTier) ??
+                model.variants.find((entry) => entry.upstreamId === model.upstreamId) ??
+                model.variants[0];
+            return { model, upstreamId: variant.upstreamId, tier: variant.tier, hubOnly: variant.hubOnly };
+        }
+        const variant = model.variants.find((entry) => entry.upstreamId.toLowerCase() === name);
+        if (variant) {
+            return { model, upstreamId: variant.upstreamId, tier: variant.tier, hubOnly: variant.hubOnly };
+        }
+        if (model.variants.length === 0 && model.upstreamId.toLowerCase() === name) {
+            return { model, upstreamId: model.upstreamId, hubOnly: model.hubOnly };
+        }
+    }
+    return undefined;
+}
+/** True when the upstream ID is only served to the current Antigravity client. */
+export function requiresHubUserAgent(upstreamId) {
+    const lower = upstreamId.toLowerCase();
+    return getRegistryModels().some((model) => (model.variants.length === 0 && model.hubOnly && model.upstreamId.toLowerCase() === lower) ||
+        model.variants.some((variant) => variant.hubOnly && variant.upstreamId.toLowerCase() === lower));
+}
+//# sourceMappingURL=model-registry.js.map

package/dist/src/plugin/model-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../../../src/plugin/model-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACzF,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAE1C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAExC,MAAM,GAAG,GAAG,YAAY,CAAC,gBAAgB,CAAC,CAAC;AAK3C,MAAM,UAAU,GAAyB,CAAC,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;AAC9E,MAAM,WAAW,GAAG,2CAA2C,CAAC;AAsDhE,MAAM,gBAAgB,GAAG,OAAO,CAAC;AACjC,MAAM,eAAe,GAAG,MAAM,CAAC;AAE/B,SAAS,QAAQ,CAAC,aAAiC,EAAE,EAAU;IAC7D,MAAM,QAAQ,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IACrD,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC5E,IAAI,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,WAAW,CAAC;IAClF,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IACzE,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,OAAO,CAAC,IAAY;IAC3B,OAAO,IAAI;SACR,WAAW,EAAE;SACb,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC;SAC5B,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,cAAc,CAAC,QAAiC;IACvD,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,eAAe,IAAI,EAAE,EAAE,CAAC;QAClD,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YACtC,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;gBACtC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB,CACjC,WAAoC,EACpC,SAA+B;IAE/B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAqF,CAAC;IAE5G,KAAK,MAAM,UAAU,IAAI,cAAc,CAAC,WAAW,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,UAAU,CAAC;QAC3D,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC;QACvC,MAAM,IAAI,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAA2B,CAAC;QAChE,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACvC,OAAO,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,SAAS,GAAG,CAAC,UAAkB,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC3F,MAAM,MAAM,GAAoB,EAAE,CAAC;IAEnC,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5E,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACxB,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;QAE5F,MAAM,QAAQ,GAAsB,MAAM;YACxC,CAAC,CAAC,OAAO;iBACJ,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBAChB,IAAI,EAAE,MAAM,CAAC,IAAiB;gBAC9B,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC;aACtC,CAAC,CAAC;iBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC5E,CAAC,CAAC,EAAE,CAAC;QAEP,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;YAChD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU;gBAC9C,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,MAAM,CAAC,UAAU;gBACzE,MAAM,EAAE,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE,MAAM,CAAC,UAAU,CAAC;gBAC7D,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC;gBACrC,QAAQ;gBACR,cAAc,EAAE,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC;gBAClD,gBAAgB,EAAE,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC;gBACtD,YAAY,EAAE,UAAU,CAAC,SAAS,IAAI,gBAAgB;gBACtD,WAAW,EAAE,UAAU,CAAC,eAAe,IAAI,eAAe;gBAC1D,MAAM;gBACN,WAAW,EAAE,sDAAsD,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;aACvF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,gBAAgB,GAAkE;IACtF,EAAE,EAAE,EAAE,uBAAuB,EAAE,WAAW,EAAE,yBAAyB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE;IACxN,EAAE,EAAE,EAAE,yBAAyB,EAAE,WAAW,EAAE,2BAA2B,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE;IAC5N,EAAE,EAAE,EAAE,sBAAsB,EAAE,WAAW,EAAE,wBAAwB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE;IACtN,EAAE,EAAE,EAAE,uBAAuB,EAAE,WAAW,EAAE,yBAAyB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IACzT,EAAE,EAAE,EAAE,yBAAyB,EAAE,WAAW,EAAE,2BAA2B,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IAC7T,EAAE,EAAE,EAAE,sBAAsB,EAAE,WAAW,EAAE,wBAAwB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IACvT,EAAE,EAAE,EAAE,uBAAuB,EAAE,WAAW,EAAE,yBAAyB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IAC1S,EAAE,EAAE,EAAE,yBAAyB,EAAE,WAAW,EAAE,2BAA2B,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IAC9S,EAAE,EAAE,EAAE,sBAAsB,EAAE,WAAW,EAAE,wBAAwB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,qDAAqD,EAAE;IACxS,EAAE,EAAE,EAAE,kBAAkB,EAAE,WAAW,EAAE,uBAAuB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,8HAA8H,EAAE;IAC5W,EAAE,EAAE,EAAE,oBAAoB,EAAE,WAAW,EAAE,sBAAsB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,8HAA8H,EAAE;IAC7W,EAAE,EAAE,EAAE,mBAAmB,EAAE,WAAW,EAAE,8BAA8B,EAAE,aAAa,EAAE,0BAA0B,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,cAAc,EAAE,kKAAkK,EAAE;IACpZ,EAAE,EAAE,EAAE,0BAA0B,EAAE,WAAW,EAAE,4BAA4B,EAAE,aAAa,EAAE,0BAA0B,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,cAAc,EAAE,gKAAgK,EAAE;IACvZ,EAAE,EAAE,EAAE,qBAAqB,EAAE,WAAW,EAAE,uBAAuB,EAAE,aAAa,EAAE,uBAAuB,EAAE,cAAc,EAAE,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,cAAc,EAAE,+DAA+D,EAAE;CAC3S,CAAC;AAEF,MAAM,UAAU,qBAAqB;IACnC,MAAM,MAAM,GAAsC,EAAE,CAAC;IACrD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,gBAAgB,EAAE,CAAC;QACxD,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,OAAO;YAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,mBAAmB,CACxB,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EACpG,SAAS,CACV,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,8DAA8D;AAC9D,8EAA8E;AAE9E,MAAM,cAAc,GAAG,yBAAyB,CAAC;AACjD,MAAM,aAAa,GAAG,CAAC,CAAC;AAQxB,IAAI,OAAoE,CAAC;AAEzE,SAAS,SAAS;IAChB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;IAC5E,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,CAAC,EAAE,UAAU,CAAC,CAAC;AACrF,CAAC;AAED,MAAM,UAAU,oBAAoB;IAClC,OAAO,IAAI,CAAC,SAAS,EAAE,EAAE,cAAc,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,KAAK,GAAG,KAA+B,CAAC;IAC9C,OAAO,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;AAC/G,CAAC;AAED,SAAS,SAAS;IAChB,MAAM,IAAI,GAAG,oBAAoB,EAAE,CAAC;IACpC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IACxC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAA+B,CAAC;QACpF,IAAI,MAAM,CAAC,OAAO,KAAK,aAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;YAAE,OAAO,SAAS,CAAC;QACxF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QACrD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAC1C,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC5F,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC5E,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,kGAAkG;AAClG,MAAM,UAAU,iBAAiB;IAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,GAAG,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,qBAAqB,EAAE,EAAE,CAAC;IAC/D,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC;AACxB,CAAC;AAED,oFAAoF;AACpF,MAAM,UAAU,oBAAoB;IAClC,iBAAiB,EAAE,CAAC;IACpB,OAAO,OAAO,EAAE,SAAS,CAAC;AAC5B,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,iBAAiB,CAAC,MAAuB,EAAE,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE;IAC/E,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAChC,OAAO,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAChC,MAAM,IAAI,GAAG,oBAAoB,EAAE,CAAC;IACpC,IAAI,CAAC;QACH,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,GAAG,IAAI,IAAI,OAAO,CAAC,GAAG,MAAM,CAAC;QAC1C,aAAa,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,EAA8B,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QAC/H,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACzB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,IAAI,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,qBAAqB,CAAC,MAAwB;IAC5D,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5C,CAAC;AAED,8EAA8E;AAC9E,uCAAuC;AACvC,8EAA8E;AAE9E,SAAS,aAAa,CAAC,KAAyB;IAC9C,MAAM,KAAK,GAAG,KAAK,EAAE,WAAW,EAAE,CAAC;IACnC,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC;AACnD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAAC,cAAsB,EAAE,IAAa;IACxE,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IACxE,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IAE1C,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE,EAAE,CAAC;QACxC,IAAI,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;YACzE,CAAC;YACD,MAAM,OAAO,GACX,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,aAAa,CAAC;gBAC5D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU,CAAC;gBACrE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC;YACrB,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;QACjG,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,CAAC;QACxF,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;QACjG,CAAC;QACD,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC;YAC3E,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QACzE,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,kFAAkF;AAClF,MAAM,UAAU,oBAAoB,CAAC,UAAkB;IACrD,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;IACvC,OAAO,iBAAiB,EAAE,CAAC,IAAI,CAC7B,CAAC,KAAK,EAAE,EAAE,CACR,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC;QAC1F,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC,CAClG,CAAC;AACJ,CAAC"}

package/dist/src/plugin/project.d.ts

@@ -0,0 +1,33 @@
+import type { OAuthAuthDetails, ProjectContextResult } from "./types";
+interface AntigravityUserTier {
+    id?: string;
+    isDefault?: boolean;
+    userDefinedCloudaicompanionProject?: boolean;
+}
+interface LoadCodeAssistPayload {
+    cloudaicompanionProject?: string | {
+        id?: string;
+    };
+    currentTier?: {
+        id?: string;
+    };
+    allowedTiers?: AntigravityUserTier[];
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export declare function invalidateProjectContextCache(refresh?: string): void;
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export declare function loadManagedProject(accessToken: string, projectId?: string): Promise<LoadCodeAssistPayload | null>;
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export declare function onboardManagedProject(accessToken: string, tierId: string, projectId?: string, attempts?: number, delayMs?: number): Promise<string | undefined>;
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export declare function ensureProjectContext(auth: OAuthAuthDetails): Promise<ProjectContextResult>;
+export {};
+//# sourceMappingURL=project.d.ts.map

package/dist/src/plugin/project.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.d.ts","sourceRoot":"","sources":["../../../src/plugin/project.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAatE,UAAU,mBAAmB;IAC3B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,kCAAkC,CAAC,EAAE,OAAO,CAAC;CAC9C;AAED,UAAU,qBAAqB;IAC7B,uBAAuB,CAAC,EAAE,MAAM,GAAG;QAAE,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACnD,WAAW,CAAC,EAAE;QACZ,EAAE,CAAC,EAAE,MAAM,CAAC;KACb,CAAC;IACF,YAAY,CAAC,EAAE,mBAAmB,EAAE,CAAC;CACtC;AAuED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAQpE;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,MAAM,EACnB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAuCvC;AAGD;;GAEG;AACH,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,MAAM,EAClB,QAAQ,SAAK,EACb,OAAO,SAAO,GACb,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA6C7B;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CA+FhG"}

package/dist/src/plugin/project.js

@@ -0,0 +1,234 @@
+import { getAntigravityHeaders, ANTIGRAVITY_ENDPOINT_FALLBACKS, ANTIGRAVITY_LOAD_ENDPOINTS, ANTIGRAVITY_DEFAULT_PROJECT_ID, } from "../constants";
+import { formatRefreshParts, parseRefreshParts } from "./auth";
+import { createLogger } from "./logger";
+const log = createLogger("project");
+const projectContextResultCache = new Map();
+const projectContextPendingCache = new Map();
+const CODE_ASSIST_METADATA = {
+    ideType: "ANTIGRAVITY",
+    platform: process.platform === "win32" ? "WINDOWS" : "MACOS",
+    pluginType: "GEMINI",
+};
+function buildMetadata(projectId) {
+    const metadata = {
+        ideType: CODE_ASSIST_METADATA.ideType,
+        platform: CODE_ASSIST_METADATA.platform,
+        pluginType: CODE_ASSIST_METADATA.pluginType,
+    };
+    if (projectId) {
+        metadata.duetProject = projectId;
+    }
+    return metadata;
+}
+/**
+ * Selects the default tier ID from the allowed tiers list.
+ */
+function getDefaultTierId(allowedTiers) {
+    if (!allowedTiers || allowedTiers.length === 0) {
+        return undefined;
+    }
+    for (const tier of allowedTiers) {
+        if (tier?.isDefault) {
+            return tier.id;
+        }
+    }
+    return allowedTiers[0]?.id;
+}
+/**
+ * Promise-based delay utility.
+ */
+function wait(ms) {
+    return new Promise(function (resolve) {
+        setTimeout(resolve, ms);
+    });
+}
+/**
+ * Extracts the cloudaicompanion project id from loadCodeAssist responses.
+ */
+function extractManagedProjectId(payload) {
+    if (!payload) {
+        return undefined;
+    }
+    if (typeof payload.cloudaicompanionProject === "string") {
+        return payload.cloudaicompanionProject;
+    }
+    if (payload.cloudaicompanionProject && typeof payload.cloudaicompanionProject.id === "string") {
+        return payload.cloudaicompanionProject.id;
+    }
+    return undefined;
+}
+/**
+ * Generates a cache key for project context based on refresh token.
+ */
+function getCacheKey(auth) {
+    const refresh = auth.refresh?.trim();
+    return refresh ? refresh : undefined;
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export function invalidateProjectContextCache(refresh) {
+    if (!refresh) {
+        projectContextPendingCache.clear();
+        projectContextResultCache.clear();
+        return;
+    }
+    projectContextPendingCache.delete(refresh);
+    projectContextResultCache.delete(refresh);
+}
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export async function loadManagedProject(accessToken, projectId) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = { metadata };
+    const loadHeaders = {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${accessToken}`,
+        "User-Agent": "google-api-nodejs-client/9.15.1",
+        "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
+        "Client-Metadata": getAntigravityHeaders()["Client-Metadata"],
+    };
+    const loadEndpoints = Array.from(new Set([...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]));
+    for (const baseEndpoint of loadEndpoints) {
+        try {
+            const response = await fetch(`${baseEndpoint}/v1internal:loadCodeAssist`, {
+                method: "POST",
+                headers: loadHeaders,
+                body: JSON.stringify(requestBody),
+            });
+            if (!response.ok) {
+                continue;
+            }
+            return (await response.json());
+        }
+        catch (error) {
+            log.debug("Failed to load managed project", { endpoint: baseEndpoint, error: String(error) });
+            continue;
+        }
+    }
+    return null;
+}
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export async function onboardManagedProject(accessToken, tierId, projectId, attempts = 10, delayMs = 5000) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = {
+        tierId,
+        metadata,
+    };
+    for (const baseEndpoint of ANTIGRAVITY_ENDPOINT_FALLBACKS) {
+        for (let attempt = 0; attempt < attempts; attempt += 1) {
+            try {
+                const response = await fetch(`${baseEndpoint}/v1internal:onboardUser`, {
+                    method: "POST",
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: `Bearer ${accessToken}`,
+                        ...getAntigravityHeaders(),
+                    },
+                    body: JSON.stringify(requestBody),
+                });
+                if (!response.ok) {
+                    break;
+                }
+                const payload = (await response.json());
+                const managedProjectId = payload.response?.cloudaicompanionProject?.id;
+                if (payload.done && managedProjectId) {
+                    return managedProjectId;
+                }
+                if (payload.done && projectId) {
+                    return projectId;
+                }
+            }
+            catch (error) {
+                log.debug("Failed to onboard managed project", { endpoint: baseEndpoint, error: String(error) });
+                break;
+            }
+            await wait(delayMs);
+        }
+    }
+    return undefined;
+}
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export async function ensureProjectContext(auth) {
+    const accessToken = auth.access;
+    if (!accessToken) {
+        return { auth, effectiveProjectId: "" };
+    }
+    const cacheKey = getCacheKey(auth);
+    if (cacheKey) {
+        const cached = projectContextResultCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        const pending = projectContextPendingCache.get(cacheKey);
+        if (pending) {
+            return pending;
+        }
+    }
+    const resolveContext = async () => {
+        const parts = parseRefreshParts(auth.refresh);
+        if (parts.managedProjectId) {
+            return { auth, effectiveProjectId: parts.managedProjectId };
+        }
+        const fallbackProjectId = ANTIGRAVITY_DEFAULT_PROJECT_ID;
+        const persistManagedProject = async (managedProjectId) => {
+            const updatedAuth = {
+                ...auth,
+                refresh: formatRefreshParts({
+                    refreshToken: parts.refreshToken,
+                    projectId: parts.projectId,
+                    managedProjectId,
+                }),
+            };
+            return { auth: updatedAuth, effectiveProjectId: managedProjectId };
+        };
+        // Try to resolve a managed project from Antigravity if possible.
+        const loadPayload = await loadManagedProject(accessToken, parts.projectId ?? fallbackProjectId);
+        const resolvedManagedProjectId = extractManagedProjectId(loadPayload);
+        if (resolvedManagedProjectId) {
+            return persistManagedProject(resolvedManagedProjectId);
+        }
+        // No managed project found - try to auto-provision one via onboarding.
+        // This handles accounts that were added before managed project provisioning was required.
+        const tierId = getDefaultTierId(loadPayload?.allowedTiers) ?? "FREE";
+        log.debug("Auto-provisioning managed project", { tierId, projectId: parts.projectId });
+        const provisionedProjectId = await onboardManagedProject(accessToken, tierId, parts.projectId);
+        if (provisionedProjectId) {
+            log.debug("Successfully provisioned managed project", { provisionedProjectId });
+            return persistManagedProject(provisionedProjectId);
+        }
+        log.warn("Failed to provision managed project - account may not work correctly", {
+            hasProjectId: !!parts.projectId,
+        });
+        if (parts.projectId) {
+            return { auth, effectiveProjectId: parts.projectId };
+        }
+        // No project id present in auth; fall back to the hardcoded id for requests.
+        return { auth, effectiveProjectId: fallbackProjectId };
+    };
+    if (!cacheKey) {
+        return resolveContext();
+    }
+    const promise = resolveContext()
+        .then((result) => {
+        const nextKey = getCacheKey(result.auth) ?? cacheKey;
+        projectContextPendingCache.delete(cacheKey);
+        projectContextResultCache.set(nextKey, result);
+        if (nextKey !== cacheKey) {
+            projectContextResultCache.delete(cacheKey);
+        }
+        return result;
+    })
+        .catch((error) => {
+        projectContextPendingCache.delete(cacheKey);
+        throw error;
+    });
+    projectContextPendingCache.set(cacheKey, promise);
+    return promise;
+}
+//# sourceMappingURL=project.js.map