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

package/README.md

@@ -8,6 +8,7 @@ An [OpenCode](https://opencode.ai/) authentication plugin that enables seamless
 - **Dynamic Model Retrieval**: Fetches available models based on user tier and current allocations.
 - **Quota Tracking**: Injects the `agy_quota` tool into OpenCode to check usage limits directly.
 - **Traffic Simulation**: Maintains background heartbeat with `agy` servers.
+- **Dual OpenCode Compatibility**: Fully supports both OpenCode v1 (>= 1.18.29) and OpenCode v2 seamlessly without breaking changes.
 
 ## Installation
 

package/dist/index.d.ts

@@ -1,7 +1,8 @@
 import { AgyCLIOAuthPlugin, GoogleOAuthPlugin } from "./src/plugin";
+import { setupOpenCodeV2, v2PluginDefinition } from "./src/plugin/v2";
+import type { DualOpenCodePlugin } from "./src/plugin/types";
 export { AgyCLIOAuthPlugin, GoogleOAuthPlugin };
-declare const _default: {
-    id: string;
-    server: typeof AgyCLIOAuthPlugin;
-};
-export default _default;
+export { setupOpenCodeV2, v2PluginDefinition };
+export * from "./src/plugin/types";
+declare const dualPlugin: DualOpenCodePlugin;
+export default dualPlugin;

package/dist/index.js

@@ -579,6 +579,12 @@ async function resolveQuotaResetDelay(accessToken, projectId, model, userAgentMo
     return null;
   }
 }
+var retryInternals = {
+  parseRetryDelayValue,
+  parseRetryDelayFromMessage,
+  findResetTimeForModel,
+  resolveQuotaResetDelay
+};
 
 // src/sdk/retry/helpers.ts
 var DEFAULT_MAX_ATTEMPTS = 3;
@@ -20652,14 +20658,232 @@ function toUrlString(value) {
   return value.toString();
 }
 
+// src/plugin/types.ts
+function defineOpenCodeV2Plugin(def) {
+  return def;
+}
+
+// src/plugin/v2.ts
+var AGY_V2_QUOTA_COMMAND = "agy-quota";
+var AGY_V2_QUOTA_SUMMARY_COMMAND = "agy-quota-summary";
+var AGY_V2_QUOTA_COMMAND_TEMPLATE = `Retrieve Agy Code Assist quota usage for the current authenticated account.
+
+Immediately call \`${AGY_QUOTA_TOOL_NAME}\` with no arguments and return its output verbatim.
+Do not call other tools.
+`;
+var AGY_V2_QUOTA_SUMMARY_COMMAND_TEMPLATE = `Retrieve Agy Code Assist quota summary (weekly and 5-hour limits by model group) for the current authenticated account.
+
+Immediately call \`${AGY_QUOTA_SUMMARY_TOOL_NAME}\` with no arguments and return its output verbatim.
+Do not call other tools.
+`;
+async function setupOpenCodeV2(ctx) {
+  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,
+      name: "Antigravity CLI",
+      npm: "@ai-sdk/google",
+      models: { ...catalog.providers[AGY_PROVIDER_ID]?.models || {} },
+      ...catalog.providers[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-");
+      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 },
+        ...targetModels[modelId] || {}
+      };
+    }
+  });
+  ctx.tool.transform(async (editor) => {
+    if (!editor) return;
+    const quotaTool = createAgyQuotaTool({
+      client: void 0,
+      getAuthResolver: () => void 0,
+      getConfiguredProjectId: () => void 0,
+      getUserAgentModel: () => void 0
+    });
+    const quotaSummaryTool = createAgyQuotaSummaryTool({
+      client: void 0,
+      getAuthResolver: () => void 0,
+      getConfiguredProjectId: () => void 0,
+      getUserAgentModel: () => void 0
+    });
+    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") {
+      editor.tools[AGY_QUOTA_TOOL_NAME] = quotaTool;
+      editor.tools[AGY_QUOTA_SUMMARY_TOOL_NAME] = quotaSummaryTool;
+    } else if (typeof editor === "object") {
+      editor[AGY_QUOTA_TOOL_NAME] = quotaTool;
+      editor[AGY_QUOTA_SUMMARY_TOOL_NAME] = quotaSummaryTool;
+    }
+  });
+  ctx.command.transform(async (editor) => {
+    if (!editor) return;
+    const commands = {
+      [AGY_V2_QUOTA_COMMAND]: {
+        description: "Show Agy Code Assist quota usage",
+        template: AGY_V2_QUOTA_COMMAND_TEMPLATE
+      },
+      [AGY_V2_QUOTA_SUMMARY_COMMAND]: {
+        description: "Show Agy Code Assist quota summary with weekly and 5-hour limits",
+        template: AGY_V2_QUOTA_SUMMARY_COMMAND_TEMPLATE
+      }
+    };
+    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") {
+      editor.commands[AGY_V2_QUOTA_COMMAND] = commands[AGY_V2_QUOTA_COMMAND];
+      editor.commands[AGY_V2_QUOTA_SUMMARY_COMMAND] = commands[AGY_V2_QUOTA_SUMMARY_COMMAND];
+    } else if (typeof editor === "object") {
+      editor[AGY_V2_QUOTA_COMMAND] = commands[AGY_V2_QUOTA_COMMAND];
+      editor[AGY_V2_QUOTA_SUMMARY_COMMAND] = commands[AGY_V2_QUOTA_SUMMARY_COMMAND];
+    }
+  });
+  ctx.session.hook("http.request", async (event) => {
+    if (!event) return;
+    const url2 = typeof event.url === "string" ? event.url : event.request?.url || "";
+    const isGL = isGenerativeLanguageRequest(url2);
+    const isInternal = url2.includes("cloudcode-pa.googleapis.com");
+    if (!isGL && !isInternal) {
+      return;
+    }
+    event.headers = event.headers || {};
+    let modelName;
+    if (isGL) {
+      const parsed = parseGenerativeLanguageRequest(url2);
+      modelName = parsed?.effectiveModel;
+    }
+    const userAgent = buildAgyCliUserAgent(modelName);
+    if (typeof event.headers.set === "function") {
+      if (!event.headers.get("User-Agent")) {
+        event.headers.set("User-Agent", userAgent);
+      }
+    } else {
+      if (!event.headers["User-Agent"] && !event.headers["user-agent"]) {
+        event.headers["User-Agent"] = userAgent;
+      }
+    }
+  });
+  ctx.session.hook("http.response", async (event) => {
+    if (!event) return;
+    if (event.response?.status === 429) {
+      return;
+    }
+  });
+  ctx.session.hook("retry", async (event) => {
+    if (!event) return;
+    const response = event.response;
+    if (response instanceof Response) {
+      if (response.status === 429 || response.status === 503) {
+        const quota = await classifyQuotaResponse(response);
+        if (quota?.retryDelayMs) {
+          event.retryDelayMs = quota.retryDelayMs;
+        } else {
+          event.retryDelayMs = await resolveRetryDelayMs(response, event.attempt || 1);
+        }
+      }
+    } else if (event.details && Array.isArray(event.details)) {
+      const retryInfo = event.details.find(
+        (d) => d && d["@type"] === "type.googleapis.com/google.rpc.RetryInfo"
+      );
+      if (retryInfo?.retryDelay) {
+        const delayMs = retryInternals.parseRetryDelayValue(retryInfo.retryDelay);
+        if (delayMs !== null) {
+          event.retryDelayMs = delayMs;
+        }
+      }
+    }
+  });
+}
+var v2PluginDefinition = defineOpenCodeV2Plugin({
+  id: AGY_PROVIDER_ID,
+  setup: setupOpenCodeV2
+});
+
 // index.ts
-var index_default = {
-  id: "@anthonyhaussman/opencode-agy-auth",
+var dualPlugin = {
+  ...v2PluginDefinition,
   server: AgyCLIOAuthPlugin
 };
+var index_default = dualPlugin;
 export {
   AgyCLIOAuthPlugin,
   GoogleOAuthPlugin,
-  index_default as default
+  index_default as default,
+  defineOpenCodeV2Plugin,
+  setupOpenCodeV2,
+  v2PluginDefinition
 };
 //# sourceMappingURL=index.js.map