@anthonyhaussman/opencode-agy-auth 1.2.3-alpha.0 → 1.2.3-alpha.2

package/dist/index.js

@@ -20677,13 +20677,57 @@ Immediately call \`${AGY_QUOTA_SUMMARY_TOOL_NAME}\` with no arguments and return
 Do not call other tools.
 `;
 async function setupOpenCodeV2(ctx) {
-  ctx.catalog.transform(async (catalog) => {
-    if (!catalog) return;
+  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;
+    };
+    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";
+      });
+      if (typeof editor?.model?.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;
+            }
+          });
+        }
+      }
+      return;
+    }
+    const catalog = editor;
     catalog.providers = catalog.providers || {};
     catalog.providers[AGY_PROVIDER_ID] = {
       id: AGY_PROVIDER_ID,
       name: "Antigravity CLI",
-      npm: "@ai-sdk/google",
+      npm: "aisdk:@ai-sdk/google",
       models: { ...catalog.providers[AGY_PROVIDER_ID]?.models || {} },
       ...catalog.providers[AGY_PROVIDER_ID]
     };
@@ -20691,6 +20735,7 @@ async function setupOpenCodeV2(ctx) {
     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,
@@ -20705,6 +20750,7 @@ async function setupOpenCodeV2(ctx) {
           output: simple.maxOutputTokens
         },
         cost: simple.cost || { input: 0, output: 0 },
+        ...variants ? { variants } : {},
         ...targetModels[modelId] || {}
       };
     }
@@ -20723,7 +20769,20 @@ async function setupOpenCodeV2(ctx) {
       getConfiguredProjectId: () => void 0,
       getUserAgentModel: () => void 0
     });
-    if (typeof editor.set === "function") {
+    if (typeof editor.add === "function") {
+      editor.add({
+        name: AGY_QUOTA_TOOL_NAME,
+        description: quotaTool.description,
+        input: {},
+        execute: async (args, context) => quotaTool.execute(args, context)
+      });
+      editor.add({
+        name: AGY_QUOTA_SUMMARY_TOOL_NAME,
+        description: quotaSummaryTool.description,
+        input: {},
+        execute: async (args, context) => quotaSummaryTool.execute(args, context)
+      });
+    } else if (typeof editor.set === "function") {
       editor.set(AGY_QUOTA_TOOL_NAME, quotaTool);
       editor.set(AGY_QUOTA_SUMMARY_TOOL_NAME, quotaSummaryTool);
     } else if (editor.tools && typeof editor.tools === "object") {
@@ -20746,7 +20805,18 @@ async function setupOpenCodeV2(ctx) {
         template: AGY_V2_QUOTA_SUMMARY_COMMAND_TEMPLATE
       }
     };
-    if (typeof editor.set === "function") {
+    if (typeof editor.add === "function") {
+      editor.add({
+        name: AGY_V2_QUOTA_COMMAND,
+        description: "Display current Antigravity quota usage",
+        template: AGY_V2_QUOTA_COMMAND_TEMPLATE
+      });
+      editor.add({
+        name: AGY_V2_QUOTA_SUMMARY_COMMAND,
+        description: "Display Antigravity quota summary grouped by model family",
+        template: AGY_V2_QUOTA_SUMMARY_COMMAND_TEMPLATE
+      });
+    } else if (typeof editor.set === "function") {
       editor.set(AGY_V2_QUOTA_COMMAND, commands[AGY_V2_QUOTA_COMMAND]);
       editor.set(AGY_V2_QUOTA_SUMMARY_COMMAND, commands[AGY_V2_QUOTA_SUMMARY_COMMAND]);
     } else if (editor.commands && typeof editor.commands === "object") {