@teamflojo/floimg-google 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,65 @@
+import type { ImageGenerator, GeneratorSchema } from "@teamflojo/floimg";
+/**
+ * Supported Imagen models
+ */
+declare const IMAGEN_MODELS: readonly ["imagen-4.0-fast-generate-001", "imagen-4.0-generate-001", "imagen-4.0-ultra-generate-001", "imagen-3.0-generate-002"];
+type ImagenModel = (typeof IMAGEN_MODELS)[number];
+/**
+ * Supported aspect ratios
+ */
+declare const ASPECT_RATIOS: readonly ["1:1", "3:4", "4:3", "9:16", "16:9"];
+type AspectRatio = (typeof ASPECT_RATIOS)[number];
+/**
+ * Schema for the Google Imagen generator
+ */
+export declare const googleImagenSchema: GeneratorSchema;
+/**
+ * Configuration for the Google Imagen generator
+ */
+export interface GoogleImagenConfig {
+    /** API key for Google AI / Gemini API */
+    apiKey?: string;
+    /** Default model to use */
+    model?: ImagenModel;
+    /** Default aspect ratio */
+    aspectRatio?: AspectRatio;
+}
+/**
+ * Parameters for image generation
+ */
+export interface GoogleImagenParams extends Record<string, unknown> {
+    /** Text prompt describing the desired image */
+    prompt?: string;
+    /** Imagen model to use */
+    model?: ImagenModel;
+    /** Image aspect ratio */
+    aspectRatio?: AspectRatio;
+    /** Number of images to generate (1-4) */
+    numberOfImages?: number;
+}
+/**
+ * Create a Google Imagen image generator instance
+ *
+ * Generates images using Google's Imagen models via the Gemini API.
+ *
+ * @example
+ * ```typescript
+ * import createClient from '@teamflojo/floimg';
+ * import googleImagen from '@teamflojo/floimg-google';
+ *
+ * const floimg = createClient();
+ * floimg.registerGenerator(googleImagen({ apiKey: process.env.GOOGLE_AI_API_KEY }));
+ *
+ * const image = await floimg.generate({
+ *   generator: 'google-imagen',
+ *   params: {
+ *     prompt: 'A serene mountain landscape at sunset',
+ *     model: 'imagen-4.0-generate-001',
+ *     aspectRatio: '16:9',
+ *   }
+ * });
+ * ```
+ */
+export default function googleImagen(config?: GoogleImagenConfig): ImageGenerator;
+export { googleImagen };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAa,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpF;;GAEG;AACH,QAAA,MAAM,aAAa,kIAKT,CAAC;AAEX,KAAK,WAAW,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,CAAC,CAAC;AAElD;;GAEG;AACH,QAAA,MAAM,aAAa,gDAAiD,CAAC;AAErE,KAAK,WAAW,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,CAAC,CAAC;AAElD;;GAEG;AACH,eAAO,MAAM,kBAAkB,EAAE,eAsChC,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,yCAAyC;IACzC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2BAA2B;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,2BAA2B;IAC3B,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACjE,+CAA+C;IAC/C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,0BAA0B;IAC1B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,yBAAyB;IACzB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,yCAAyC;IACzC,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAsBD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,OAAO,UAAU,YAAY,CAAC,MAAM,GAAE,kBAAuB,GAAG,cAAc,CA+EpF;AAGD,OAAO,EAAE,YAAY,EAAE,CAAC"}

package/dist/index.js

@@ -0,0 +1,156 @@
+import { GoogleGenAI } from "@google/genai";
+/**
+ * Supported Imagen models
+ */
+const IMAGEN_MODELS = [
+    "imagen-4.0-fast-generate-001",
+    "imagen-4.0-generate-001",
+    "imagen-4.0-ultra-generate-001",
+    "imagen-3.0-generate-002",
+];
+/**
+ * Supported aspect ratios
+ */
+const ASPECT_RATIOS = ["1:1", "3:4", "4:3", "9:16", "16:9"];
+/**
+ * Schema for the Google Imagen generator
+ */
+export const googleImagenSchema = {
+    name: "google-imagen",
+    description: "Generate images using Google's Imagen models via the Gemini API",
+    category: "AI",
+    parameters: {
+        prompt: {
+            type: "string",
+            title: "Prompt",
+            description: "Describe the image you want to generate (max 480 tokens)",
+        },
+        model: {
+            type: "string",
+            title: "Model",
+            description: "Imagen model variant",
+            enum: [...IMAGEN_MODELS],
+            default: "imagen-4.0-generate-001",
+        },
+        aspectRatio: {
+            type: "string",
+            title: "Aspect Ratio",
+            description: "Image aspect ratio",
+            enum: [...ASPECT_RATIOS],
+            default: "1:1",
+        },
+        numberOfImages: {
+            type: "number",
+            title: "Number of Images",
+            description: "Number of images to generate (1-4)",
+            default: 1,
+            minimum: 1,
+            maximum: 4,
+        },
+    },
+    requiredParameters: ["prompt"],
+    // AI metadata
+    isAI: true,
+    requiresApiKey: true,
+    apiKeyEnvVar: "GOOGLE_AI_API_KEY",
+};
+/**
+ * Map aspect ratio to approximate dimensions
+ */
+function getApproximateDimensions(aspectRatio) {
+    switch (aspectRatio) {
+        case "1:1":
+            return { width: 1024, height: 1024 };
+        case "3:4":
+            return { width: 768, height: 1024 };
+        case "4:3":
+            return { width: 1024, height: 768 };
+        case "9:16":
+            return { width: 576, height: 1024 };
+        case "16:9":
+            return { width: 1024, height: 576 };
+        default:
+            return { width: 1024, height: 1024 };
+    }
+}
+/**
+ * Create a Google Imagen image generator instance
+ *
+ * Generates images using Google's Imagen models via the Gemini API.
+ *
+ * @example
+ * ```typescript
+ * import createClient from '@teamflojo/floimg';
+ * import googleImagen from '@teamflojo/floimg-google';
+ *
+ * const floimg = createClient();
+ * floimg.registerGenerator(googleImagen({ apiKey: process.env.GOOGLE_AI_API_KEY }));
+ *
+ * const image = await floimg.generate({
+ *   generator: 'google-imagen',
+ *   params: {
+ *     prompt: 'A serene mountain landscape at sunset',
+ *     model: 'imagen-4.0-generate-001',
+ *     aspectRatio: '16:9',
+ *   }
+ * });
+ * ```
+ */
+export default function googleImagen(config = {}) {
+    const { apiKey = process.env.GOOGLE_AI_API_KEY, model: defaultModel = "imagen-4.0-generate-001", aspectRatio: defaultAspectRatio = "1:1", } = config;
+    if (!apiKey) {
+        throw new Error("Google AI API key is required. Set GOOGLE_AI_API_KEY environment variable or pass apiKey in config.");
+    }
+    const client = new GoogleGenAI({ apiKey });
+    return {
+        name: "google-imagen",
+        schema: googleImagenSchema,
+        async generate(params) {
+            const { prompt, model = defaultModel, aspectRatio = defaultAspectRatio, numberOfImages = 1, } = params;
+            if (!prompt) {
+                throw new Error("prompt is required for Google Imagen image generation");
+            }
+            // Validate number of images
+            if (numberOfImages < 1 || numberOfImages > 4) {
+                throw new Error("numberOfImages must be between 1 and 4");
+            }
+            // Build generation config
+            const imageConfig = {
+                numberOfImages,
+                aspectRatio,
+            };
+            // Generate image using Imagen
+            const response = await client.models.generateImages({
+                model,
+                prompt,
+                config: imageConfig,
+            });
+            if (!response.generatedImages || response.generatedImages.length === 0) {
+                throw new Error("No image data returned from Google Imagen");
+            }
+            const generatedImage = response.generatedImages[0];
+            if (!generatedImage.image?.imageBytes) {
+                throw new Error("No image bytes returned from Google Imagen");
+            }
+            // Convert base64 to Buffer
+            const bytes = Buffer.from(generatedImage.image.imageBytes, "base64");
+            // Get approximate dimensions from aspect ratio
+            const dimensions = getApproximateDimensions(aspectRatio);
+            return {
+                bytes,
+                mime: "image/png",
+                width: dimensions.width,
+                height: dimensions.height,
+                source: `ai:google-imagen:${model}`,
+                metadata: {
+                    prompt,
+                    model,
+                    aspectRatio,
+                },
+            };
+        },
+    };
+}
+// Named export for convenience
+export { googleImagen };
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAA6B,MAAM,eAAe,CAAC;AAGvE;;GAEG;AACH,MAAM,aAAa,GAAG;IACpB,8BAA8B;IAC9B,yBAAyB;IACzB,+BAA+B;IAC/B,yBAAyB;CACjB,CAAC;AAIX;;GAEG;AACH,MAAM,aAAa,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAU,CAAC;AAIrE;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAoB;IACjD,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,iEAAiE;IAC9E,QAAQ,EAAE,IAAI;IACd,UAAU,EAAE;QACV,MAAM,EAAE;YACN,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,QAAQ;YACf,WAAW,EAAE,0DAA0D;SACxE;QACD,KAAK,EAAE;YACL,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,OAAO;YACd,WAAW,EAAE,sBAAsB;YACnC,IAAI,EAAE,CAAC,GAAG,aAAa,CAAC;YACxB,OAAO,EAAE,yBAAyB;SACnC;QACD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,cAAc;YACrB,WAAW,EAAE,oBAAoB;YACjC,IAAI,EAAE,CAAC,GAAG,aAAa,CAAC;YACxB,OAAO,EAAE,KAAK;SACf;QACD,cAAc,EAAE;YACd,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,kBAAkB;YACzB,WAAW,EAAE,oCAAoC;YACjD,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;SACX;KACF;IACD,kBAAkB,EAAE,CAAC,QAAQ,CAAC;IAC9B,cAAc;IACd,IAAI,EAAE,IAAI;IACV,cAAc,EAAE,IAAI;IACpB,YAAY,EAAE,mBAAmB;CAClC,CAAC;AA4BF;;GAEG;AACH,SAAS,wBAAwB,CAAC,WAAwB;IACxD,QAAQ,WAAW,EAAE,CAAC;QACpB,KAAK,KAAK;YACR,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACvC,KAAK,KAAK;YACR,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACtC,KAAK,KAAK;YACR,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;QACtC,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACtC,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;QACtC;YACE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACzC,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,OAAO,UAAU,YAAY,CAAC,SAA6B,EAAE;IAClE,MAAM,EACJ,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,EACtC,KAAK,EAAE,YAAY,GAAG,yBAAyB,EAC/C,WAAW,EAAE,kBAAkB,GAAG,KAAK,GACxC,GAAG,MAAM,CAAC;IAEX,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAE3C,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,MAAM,EAAE,kBAAkB;QAE1B,KAAK,CAAC,QAAQ,CAAC,MAA+B;YAC5C,MAAM,EACJ,MAAM,EACN,KAAK,GAAG,YAAY,EACpB,WAAW,GAAG,kBAAkB,EAChC,cAAc,GAAG,CAAC,GACnB,GAAG,MAA4B,CAAC;YAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;YAC3E,CAAC;YAED,4BAA4B;YAC5B,IAAI,cAAc,GAAG,CAAC,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;gBAC7C,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAC5D,CAAC;YAED,0BAA0B;YAC1B,MAAM,WAAW,GAAyB;gBACxC,cAAc;gBACd,WAAW;aACZ,CAAC;YAEF,8BAA8B;YAC9B,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC;gBAClD,KAAK;gBACL,MAAM;gBACN,MAAM,EAAE,WAAW;aACpB,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,eAAe,IAAI,QAAQ,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvE,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,cAAc,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEnD,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YAChE,CAAC;YAED,2BAA2B;YAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAErE,+CAA+C;YAC/C,MAAM,UAAU,GAAG,wBAAwB,CAAC,WAAW,CAAC,CAAC;YAEzD,OAAO;gBACL,KAAK;gBACL,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM,EAAE,oBAAoB,KAAK,EAAE;gBACnC,QAAQ,EAAE;oBACR,MAAM;oBACN,KAAK;oBACL,WAAW;iBACZ;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,+BAA+B;AAC/B,OAAO,EAAE,YAAY,EAAE,CAAC"}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "@teamflojo/floimg-google",
+  "version": "0.1.0",
+  "description": "Google Imagen image generation plugin for floimg",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build",
+    "test": "vitest --run"
+  },
+  "keywords": [
+    "floimg",
+    "google",
+    "imagen",
+    "gemini",
+    "image-generation",
+    "ai"
+  ],
+  "author": "Brett Cooke",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/TeamFlojo/floimg.git",
+    "directory": "packages/floimg-google"
+  },
+  "peerDependencies": {
+    "@teamflojo/floimg": "^0.2.0"
+  },
+  "dependencies": {
+    "@google/genai": "^0.7.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.2",
+    "@teamflojo/floimg": "workspace:*",
+    "typescript": "^5.7.2",
+    "vitest": "^2.1.8"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}