@anthonyhaussman/opencode-agy-auth 1.2.7-alpha.0 → 1.2.7-alpha.1

package/dist/index.js

@@ -21085,97 +21085,140 @@ function createV2FetchInterceptor(getAuthSnapshot) {
   };
 }
 async function setupOpenCodeV2(ctx) {
-  ctx.catalog.transform(async (editor) => {
-    if (!editor) return;
-    const resolveVariants = (modelId) => {
-      const mapping = TIER_MAPPING[modelId];
-      if (!mapping) return void 0;
-      const variants = [];
-      if (mapping.minimal !== void 0) variants.push({ id: "minimal" });
-      if (mapping.low !== void 0) variants.push({ id: "low" });
-      if (mapping.medium !== void 0) variants.push({ id: "medium" });
-      if (mapping.high !== void 0) variants.push({ id: "high" });
-      return variants.length > 0 ? variants : void 0;
+  const resolveVariants = (modelId) => {
+    const mapping = TIER_MAPPING[modelId];
+    if (!mapping) return void 0;
+    const variants = [];
+    if (mapping.minimal !== void 0) variants.push({ id: "minimal" });
+    if (mapping.low !== void 0) variants.push({ id: "low" });
+    if (mapping.medium !== void 0) variants.push({ id: "medium" });
+    if (mapping.high !== void 0) variants.push({ id: "high" });
+    return variants.length > 0 ? variants : void 0;
+  };
+  const updateProviderRecord = (p) => {
+    p.name = "Antigravity CLI";
+    p.activation = "enabled";
+    p.package = "aisdk:@ai-sdk/google";
+    p.description = "Google Gemini Antigravity Code Assist OAuth provider";
+    p.settings = { ...p.settings || {}, apiKey: "dummy" };
+    p.integrationID = AGY_PROVIDER_ID;
+  };
+  const buildModelRecord = (modelId, simple, existingModel) => {
+    const isClaude = modelId.startsWith("claude-");
+    const isGpt = modelId.startsWith("gpt-");
+    const variants = resolveVariants(modelId);
+    return {
+      id: modelId,
+      providerID: AGY_PROVIDER_ID,
+      name: simple.name,
+      description: simple.description,
+      family: modelId.includes("gemini") ? "gemini" : isClaude ? "claude" : isGpt ? "gpt" : "unknown",
+      reasoning: simple.reasoning,
+      attachment: simple.attachment,
+      tool_call: simple.toolCall,
+      limit: {
+        context: simple.maxTokens,
+        output: simple.maxOutputTokens
+      },
+      cost: simple.cost || { input: 0, output: 0 },
+      ...variants ? { variants } : {},
+      ...existingModel || {}
     };
-    if (typeof editor?.provider?.update === "function") {
-      editor.provider.update(AGY_PROVIDER_ID, (p) => {
-        p.name = "Antigravity CLI";
-        p.activation = "enabled";
-        p.package = "aisdk:@ai-sdk/google";
-        p.description = "Google Gemini Antigravity Code Assist OAuth provider";
-        p.settings = { ...p.settings || {}, apiKey: "dummy" };
-        p.integrationID = AGY_PROVIDER_ID;
-      });
-      if (typeof editor?.model?.update === "function") {
+  };
+  const updateModelRecord = (m, modelId, simple) => {
+    m.name = simple.name;
+    m.capabilities = {
+      tools: true,
+      input: ["text", "image"],
+      output: ["text"]
+    };
+    m.limit = {
+      context: simple.maxTokens,
+      output: simple.maxOutputTokens
+    };
+    m.family = "gemini";
+    if (simple.cost) {
+      m.cost = simple.cost;
+    }
+    const variants = resolveVariants(modelId);
+    if (variants) {
+      m.variants = variants;
+    }
+  };
+  if (ctx.provider?.transform) {
+    ctx.provider.transform(async (editor) => {
+      if (!editor) return;
+      if (typeof editor.update === "function") {
+        editor.update(AGY_PROVIDER_ID, updateProviderRecord);
+      } else if (typeof editor.set === "function") {
+        const p = { id: AGY_PROVIDER_ID };
+        updateProviderRecord(p);
+        editor.set(AGY_PROVIDER_ID, p);
+      } else if (typeof editor === "object") {
+        const p = editor[AGY_PROVIDER_ID] || { id: AGY_PROVIDER_ID };
+        updateProviderRecord(p);
+        editor[AGY_PROVIDER_ID] = p;
+      }
+    });
+  }
+  if (ctx.model?.transform) {
+    ctx.model.transform(async (editor) => {
+      if (!editor) return;
+      if (typeof editor.update === "function") {
         for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
-          editor.model.update(AGY_PROVIDER_ID, modelId, (m) => {
-            m.name = simple.name;
-            m.capabilities = {
-              tools: true,
-              input: ["text", "image"],
-              output: ["text"]
-            };
-            m.limit = {
-              context: simple.maxTokens,
-              output: simple.maxOutputTokens
-            };
-            m.family = "gemini";
-            if (simple.cost) {
-              m.cost = simple.cost;
-            }
-            const variants = resolveVariants(modelId);
-            if (variants) {
-              m.variants = variants;
-            }
-          });
+          editor.update(AGY_PROVIDER_ID, modelId, (m) => updateModelRecord(m, modelId, simple));
+        }
+      } else if (typeof editor.set === "function") {
+        for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
+          const m = buildModelRecord(modelId, simple);
+          editor.set(`${AGY_PROVIDER_ID}:${modelId}`, m);
+        }
+      } else if (typeof editor === "object") {
+        for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
+          const key = `${AGY_PROVIDER_ID}:${modelId}`;
+          editor[key] = buildModelRecord(modelId, simple, editor[key]);
         }
       }
-      return;
-    }
-    const catalog = editor;
-    catalog.providers = catalog.providers || {};
-    catalog.providers[AGY_PROVIDER_ID] = {
-      id: AGY_PROVIDER_ID,
-      name: "Antigravity CLI",
-      npm: "aisdk:@ai-sdk/google",
-      settings: { apiKey: "dummy" },
-      integrationID: AGY_PROVIDER_ID,
-      models: { ...catalog.providers[AGY_PROVIDER_ID]?.models || {} },
-      ...catalog.providers[AGY_PROVIDER_ID]
-    };
-    if (catalog.providers[AGY_PROVIDER_ID].settings) {
-      catalog.providers[AGY_PROVIDER_ID].settings = {
-        ...catalog.providers[AGY_PROVIDER_ID].settings,
-        apiKey: "dummy"
-      };
-    } else {
-      catalog.providers[AGY_PROVIDER_ID].settings = { apiKey: "dummy" };
-    }
-    catalog.providers[AGY_PROVIDER_ID].integrationID = AGY_PROVIDER_ID;
-    const targetModels = catalog.providers[AGY_PROVIDER_ID].models;
-    for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
-      const isClaude = modelId.startsWith("claude-");
-      const isGpt = modelId.startsWith("gpt-");
-      const variants = resolveVariants(modelId);
-      targetModels[modelId] = {
-        id: modelId,
-        providerID: AGY_PROVIDER_ID,
-        name: simple.name,
-        description: simple.description,
-        family: modelId.includes("gemini") ? "gemini" : isClaude ? "claude" : isGpt ? "gpt" : "unknown",
-        reasoning: simple.reasoning,
-        attachment: simple.attachment,
-        tool_call: simple.toolCall,
-        limit: {
-          context: simple.maxTokens,
-          output: simple.maxOutputTokens
-        },
-        cost: simple.cost || { input: 0, output: 0 },
-        ...variants ? { variants } : {},
-        ...targetModels[modelId] || {}
+    });
+  }
+  if (ctx.catalog?.transform) {
+    ctx.catalog.transform(async (editor) => {
+      if (!editor) return;
+      if (typeof editor?.provider?.update === "function") {
+        editor.provider.update(AGY_PROVIDER_ID, updateProviderRecord);
+        if (typeof editor?.model?.update === "function") {
+          for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
+            editor.model.update(AGY_PROVIDER_ID, modelId, (m) => updateModelRecord(m, modelId, simple));
+          }
+        }
+        return;
+      }
+      const catalog = editor;
+      catalog.providers = catalog.providers || {};
+      catalog.providers[AGY_PROVIDER_ID] = {
+        id: AGY_PROVIDER_ID,
+        name: "Antigravity CLI",
+        npm: "aisdk:@ai-sdk/google",
+        settings: { apiKey: "dummy" },
+        integrationID: AGY_PROVIDER_ID,
+        models: { ...catalog.providers[AGY_PROVIDER_ID]?.models || {} },
+        ...catalog.providers[AGY_PROVIDER_ID]
       };
-    }
-  });
+      if (catalog.providers[AGY_PROVIDER_ID].settings) {
+        catalog.providers[AGY_PROVIDER_ID].settings = {
+          ...catalog.providers[AGY_PROVIDER_ID].settings,
+          apiKey: "dummy"
+        };
+      } else {
+        catalog.providers[AGY_PROVIDER_ID].settings = { apiKey: "dummy" };
+      }
+      catalog.providers[AGY_PROVIDER_ID].integrationID = AGY_PROVIDER_ID;
+      const targetModels = catalog.providers[AGY_PROVIDER_ID].models;
+      for (const [modelId, simple] of Object.entries(STATIC_MODELS_SIMPLE)) {
+        targetModels[modelId] = buildModelRecord(modelId, simple, targetModels[modelId]);
+      }
+    });
+  }
   ctx.tool.transform(async (editor) => {
     if (!editor) return;
     const quotaTool = createAgyQuotaTool({