fix: add modalities support for input/output in configuration and improve toast notifications in plugin

Author: NoeFabris

README.md

@@ -68,41 +68,65 @@ Open the **same config file** you created in Step 1 (`~/.config/opencode/opencod
           "limit": {
             "context": 1048576,
             "output": 65535
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         },
         "gemini-3-pro-low": {
           "name": "Gemini 3 Pro Low (Antigravity)",
           "limit": {
             "context": 1048576,
             "output": 65535
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         },
         "claude-sonnet-4-5": {
           "name": "Claude Sonnet 4.5 (Antigravity)",
           "limit": {
             "context": 200000,
             "output": 64000
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         },
         "claude-sonnet-4-5-thinking": {
           "name": "Claude Sonnet 4.5 Thinking (Antigravity)",
           "limit": {
             "context": 200000,
             "output": 64000
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         },
         "claude-opus-4-5-thinking": {
           "name": "Claude Opus 4.5 Thinking (Antigravity)",
           "limit": {
             "context": 200000,
             "output": 64000
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         },
         "gpt-oss-120b-medium": {
           "name": "GPT-OSS 120B Medium (Antigravity)",
           "limit": {
             "context": 131072,
             "output": 32768
+          },
+          "modalities": {
+            "input": ["text", "image", "pdf"],
+            "output": ["text"]
           }
         }
       }
@@ -111,7 +135,7 @@ Open the **same config file** you created in Step 1 (`~/.config/opencode/opencod
 }
 ```
 
-> **Tip:** You only need to add the models you plan to use. The example above includes all available models, but you can remove any you don't need.
+> **Tip:** You only need to add the models you plan to use. The example above includes all available models, but you can remove any you don't need. The `modalities` field enables image and PDF support in the TUI.
 
 ### Step 4: Use a model
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-antigravity-auth",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "description": "Google Antigravity IDE OAuth auth plugin for Opencode - access Gemini 3 Pro and Claude 4.5 using Google credentials",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

src/plugin.ts

@@ -243,8 +243,6 @@ export const createAntigravityPlugin = (providerId: string) => async (
   auth: {
     provider: providerId,
     loader: async (getAuth: GetAuth, provider: Provider): Promise<LoaderResult | Record<string, unknown>> => {
-      // Track which account was used in the previous request for detecting switches
-      let previousAccountIndex: number | null = null;
       const auth = await getAuth();
 
       // If OpenCode has no valid OAuth auth, clear any stale account storage
@@ -334,10 +332,36 @@ export const createAntigravityPlugin = (providerId: string) => async (
           let lastFailure: FailureContext | null = null;
           let lastError: Error | null = null;
           const abortSignal = init?.signal ?? undefined;
+          
+          // Track which account was used in this request for detecting switches
+          // This is scoped to the fetch call so it resets per-request
+          let previousAccountIndex: number | null = null;
+
+          // Helper to check if request was aborted
+          const checkAborted = () => {
+            if (abortSignal?.aborted) {
+              throw abortSignal.reason instanceof Error ? abortSignal.reason : new Error("Aborted");
+            }
+          };
+
+          // Helper to show toast without blocking on abort
+          const showToast = async (message: string, variant: "info" | "warning" | "success" | "error") => {
+            if (abortSignal?.aborted) return;
+            try {
+              await client.tui.showToast({
+                body: { message, variant },
+              });
+            } catch {
+              // TUI may not be available
+            }
+          };
 
           // Use while(true) loop to handle rate limits with backoff
           // This ensures we wait and retry when all accounts are rate-limited
           while (true) {
+            // Check for abort at the start of each iteration
+            checkAborted();
+            
             const accountCount = accountManager.getAccountCount();
 
             if (accountCount === 0) {
@@ -351,16 +375,7 @@ export const createAntigravityPlugin = (providerId: string) => async (
               const waitMs = accountManager.getMinWaitTimeMs() || 60_000;
               const waitSec = Math.max(1, Math.ceil(waitMs / 1000));
 
-              try {
-                await client.tui.showToast({
-                  body: {
-                    message: `All ${accountCount} account(s) rate-limited. Waiting ${waitSec}s...`,
-                    variant: "warning",
-                  },
-                });
-              } catch {
-                // TUI may not be available
-              }
+              await showToast(`All ${accountCount} account(s) rate-limited. Waiting ${waitSec}s...`, "warning");
 
               // Wait for the cooldown to expire
               await sleep(waitMs, abortSignal);
@@ -371,16 +386,10 @@ export const createAntigravityPlugin = (providerId: string) => async (
             const isAccountSwitch = previousAccountIndex !== null && previousAccountIndex !== account.index;
             if (isAccountSwitch || previousAccountIndex === null) {
               const accountLabel = account.email || `Account ${account.index + 1}`;
-              try {
-                await client.tui.showToast({
-                  body: {
-                    message: `Using ${accountLabel}${accountCount > 1 ? ` (${account.index + 1}/${accountCount})` : ""}`,
-                    variant: "info",
-                  },
-                });
-              } catch {
-                // TUI may not be available
-              }
+              await showToast(
+                `Using ${accountLabel}${accountCount > 1 ? ` (${account.index + 1}/${accountCount})` : ""}`,
+                "info"
+              );
             }
             previousAccountIndex = account.index;
 
@@ -515,16 +524,7 @@ export const createAntigravityPlugin = (providerId: string) => async (
 
                   if (accountManager.getAccountCount() > 1) {
                     // Multiple accounts - switch to next
-                    try {
-                      await client.tui.showToast({
-                        body: {
-                          message: `Rate limited on ${accountLabel}. Switching...`,
-                          variant: "warning",
-                        },
-                      });
-                    } catch {
-                      // TUI may not be available
-                    }
+                    await showToast(`Rate limited on ${accountLabel}. Switching...`, "warning");
 
                     lastFailure = {
                       response,
@@ -543,16 +543,7 @@ export const createAntigravityPlugin = (providerId: string) => async (
                   } else {
                     // Single account - wait and retry
                     const waitSec = Math.max(1, Math.ceil(retryAfterMs / 1000));
-                    try {
-                      await client.tui.showToast({
-                        body: {
-                          message: `Rate limited. Waiting ${waitSec}s...`,
-                          variant: "warning",
-                        },
-                      });
-                    } catch {
-                      // TUI may not be available
-                    }
+                    await showToast(`Rate limited. Waiting ${waitSec}s...`, "warning");
 
                     lastFailure = {
                       response,