opencode-antigravity-img 0.1.1 → 0.2.0

package/README.md

@@ -31,7 +31,7 @@ opencode
 # Use the authenticate command from antigravity-auth
 ```
 
-This creates `~/.config/opencode/antigravity-accounts.json` with your credentials.
+This creates `antigravity-accounts.json` in your opencode config directory.
 
 ## Tools
 
@@ -41,12 +41,14 @@ Generate an image from a text prompt.
 
 **Arguments:**
 - `prompt` (required): Text description of the image to generate
-- `filename` (optional): Output filename (default: `generated_<timestamp>.png`)
+- `filename` (optional): Output filename (default: `generated_<timestamp>.jpg`)
 - `output_dir` (optional): Output directory (default: current working directory)
+- `aspect_ratio` (optional): Image aspect ratio. Supported values: `1:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9` (default: `1:1`)
+- `image_size` (optional): Image resolution. Supported values: `1K`, `2K`, `4K` (default: `1K`)
 
 **Example:**
 ```
-Generate an image of a sunset over mountains with a lake in the foreground
+Generate a 16:9 landscape image of a sunset over mountains with a lake in the foreground
 ```
 
 **Output:**
@@ -66,13 +68,27 @@ Check the remaining quota for the Gemini 3 Pro Image model.
 ## Image Details
 
 - **Model**: Gemini 3 Pro Image
-- **Resolution**: 1408x768 pixels
-- **Format**: JPEG (typically 600KB - 1MB)
+- **Resolutions**: 
+  - `1K`: 1024x1024 (default)
+  - `2K`: 2048x2048
+  - `4K`: 4096x4096
+- **Aspect ratios**: `1:1`, `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9`
+- **Format**: JPEG (always, regardless of filename extension)
 - **Generation time**: 10-30 seconds
 
+## Unsupported Parameters
+
+The following parameters are documented in Google's API but are **not supported** by the Gemini 3 Pro Image model (they are Imagen-specific):
+
+- `personGeneration` - Control generation of people in images
+- `outputMimeType` - Output format selection (always returns JPEG)
+- `compressionQuality` - JPEG compression control
+
+These parameters may work with Imagen models but have no effect with Gemini 3 Pro Image.
+
 ## Quota
 
-Image generation uses a separate quota from text models. The quota typically resets daily. Use the `image_quota` tool to check your remaining quota before generating images.
+Image generation uses a separate quota from text models. The quota resets every 5 hours. Use the `image_quota` tool to check your remaining quota.
 
 ## Troubleshooting
 
@@ -81,7 +97,7 @@ Image generation uses a separate quota from text models. The quota typically res
 Make sure you've:
 1. Installed `opencode-antigravity-auth`
 2. Authenticated with your Google account
-3. The credentials file exists at `~/.config/opencode/antigravity-accounts.json`
+3. The credentials file `antigravity-accounts.json` exists in your opencode config directory
 
 ### "Rate limited" or generation fails
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-antigravity-img",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "description": "OpenCode plugin for Gemini image generation via Antigravity/CloudCode API",
   "main": "src/index.ts",
   "type": "module",

package/src/api.ts

@@ -15,6 +15,7 @@ import type {
   CloudCodeQuotaResponse,
   GenerateContentResponse,
   ImageGenerationResult,
+  ImageGenerationOptions,
   QuotaInfo,
 } from "./types";
 
@@ -149,12 +150,34 @@ export async function getImageModelQuota(account: Account): Promise<QuotaInfo |
   }
 }
 
+/**
+ * Build imageConfig from options
+ */
+function buildImageConfig(options?: ImageGenerationOptions): ImageGenerationOptions | undefined {
+  if (!options) return undefined;
+
+  const { aspectRatio, imageSize } = options;
+
+  // Only include if at least one option is set
+  if (!aspectRatio && !imageSize) {
+    return undefined;
+  }
+
+  const imageConfig: ImageGenerationOptions = {};
+
+  if (aspectRatio) imageConfig.aspectRatio = aspectRatio;
+  if (imageSize) imageConfig.imageSize = imageSize;
+
+  return imageConfig;
+}
+
 /**
  * Generate an image using the Gemini 3 Pro Image model
  */
 export async function generateImage(
   account: Account,
-  prompt: string
+  prompt: string,
+  options?: ImageGenerationOptions
 ): Promise<ImageGenerationResult> {
   try {
     // Get access token
@@ -172,6 +195,7 @@ export async function generateImage(
     }
 
     // Build request
+    const imageConfig = buildImageConfig(options);
     const requestBody = {
       project: projectId,
       requestId: `req_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
@@ -183,6 +207,7 @@ export async function generateImage(
         session_id: `sess_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
         generationConfig: {
           responseModalities: ["TEXT", "IMAGE"],
+          ...(imageConfig && { imageConfig }),
         },
       },
     };

package/src/index.ts

@@ -3,7 +3,7 @@ import * as fs from "fs/promises";
 import { existsSync, mkdirSync, writeFileSync, readFileSync } from "fs";
 import { join, dirname } from "path";
 import { CONFIG_PATHS, COMMAND_DIR, COMMAND_FILE, COMMAND_CONTENT, IMAGE_MODEL } from "./constants";
-import type { AccountsConfig, Account } from "./types";
+import type { AccountsConfig, Account, ImageGenerationOptions, AspectRatio, ImageSize } from "./types";
 import { generateImage, getImageModelQuota } from "./api";
 
 // Create command file for opencode discovery
@@ -83,9 +83,17 @@ export const plugin: Plugin = async (ctx) => {
             .string()
             .optional()
             .describe("Output directory (default: current working directory)"),
+          aspect_ratio: tool.schema
+            .string()
+            .optional()
+            .describe("Aspect ratio: 1:1, 2:3, 3:2, 3:4, 4:3, 4:5, 5:4, 9:16, 16:9, 21:9 (default: 1:1)"),
+          image_size: tool.schema
+            .string()
+            .optional()
+            .describe("Image resolution: 1K, 2K, 4K (default: 1K)"),
         },
         async execute(args, context) {
-          const { prompt, filename, output_dir } = args;
+          const { prompt, filename, output_dir, aspect_ratio, image_size } = args;
 
           if (!prompt?.trim()) {
             return "Error: Please provide a prompt describing the image to generate.";
@@ -105,8 +113,13 @@ export const plugin: Plugin = async (ctx) => {
 
           context.metadata({ title: "Generating image..." });
 
+          // Build generation options
+          const options: ImageGenerationOptions = {};
+          if (aspect_ratio) options.aspectRatio = aspect_ratio as AspectRatio;
+          if (image_size) options.imageSize = image_size as ImageSize;
+
           // Generate image
-          const result = await generateImage(account, prompt);
+          const result = await generateImage(account, prompt, Object.keys(options).length > 0 ? options : undefined);
 
           if (!result.success || !result.imageData) {
             return `Error generating image: ${result.error || "Unknown error"}`;

package/src/types.ts

@@ -37,7 +37,20 @@ export interface ModelInfo {
   };
 }
 
-// Image generation
+// Image generation options
+export type AspectRatio = "1:1" | "2:3" | "3:2" | "3:4" | "4:3" | "4:5" | "5:4" | "9:16" | "16:9" | "21:9";
+export type ImageSize = "1K" | "2K" | "4K";
+
+export interface ImageConfig {
+  aspectRatio?: AspectRatio;
+  imageSize?: ImageSize;
+}
+
+export interface ImageGenerationOptions {
+  aspectRatio?: AspectRatio;
+  imageSize?: ImageSize;
+}
+
 export interface GenerateContentRequest {
   project: string;
   requestId: string;
@@ -52,6 +65,7 @@ export interface GenerateContentRequest {
     session_id: string;
     generationConfig: {
       responseModalities: string[];
+      imageConfig?: ImageConfig;
     };
   };
 }