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

package/dist/index.js

@@ -20677,8 +20677,38 @@ 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;
+    if (typeof editor?.provider?.update === "function") {
+      editor.provider.update(AGY_PROVIDER_ID, (p) => {
+        p.name = "Antigravity CLI";
+        p.activation = "enabled";
+        p.package = "@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;
+            }
+          });
+        }
+      }
+      return;
+    }
+    const catalog = editor;
     catalog.providers = catalog.providers || {};
     catalog.providers[AGY_PROVIDER_ID] = {
       id: AGY_PROVIDER_ID,
@@ -20723,7 +20753,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 +20789,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") {