feat: add OpenRouter image generation and editing endpoints (#717)

will refine after.

Author: necrodome

call-ai/pkg/image.ts

@@ -1,19 +1,19 @@
 /**
  * Image generation API implementation for call-ai
- * Integration with custom image generation API
+ * Integration with OpenRouter image generation API
  */
 import { ImageGenOptions, ImageResponse } from "./types.js";
 import { callAiFetch, joinUrlParts } from "./utils.js";
 import { callAiEnv } from "./env.js";
 
 /**
- * Generate images using a custom API that mimics OpenAI's image generation capabilities
+ * Generate images using OpenRouter API with Gemini image generation
  * @param prompt Text prompt describing the image to generate
  * @param options Configuration options for the image generation request
  * @returns A Promise that resolves to the image response containing base64 encoded image data
  */
 export async function imageGen(prompt: string, options: ImageGenOptions = {}): Promise<ImageResponse> {
-  const { model = "gpt-image-1", apiKey = callAiEnv.CALLAI_API_KEY, size = "1024x1024" } = options;
+  const { model = "google/gemini-2.5-flash-image", apiKey = callAiEnv.CALLAI_API_KEY, size = "1024x1024" } = options;
 
   if (!apiKey) {
     throw new Error("API key is required for image generation. Provide via options.apiKey or set window.CALLAI_API_KEY");
@@ -27,7 +27,7 @@ export async function imageGen(prompt: string, options: ImageGenOptions = {}): P
     // Simple image generation with text prompt
     // Use explicit endpoint if provided (highest priority), otherwise construct from origin
     const generateEndpoint =
-      options.endpoint || joinUrlParts(customOrigin || callAiEnv.def.CALLAI_CHAT_URL, "/api/openai-image/generate");
+      options.endpoint || joinUrlParts(customOrigin || callAiEnv.def.CALLAI_CHAT_URL, "/api/openrouter-image/generate");
 
     if (!apiKey) {
       throw new Error("API key is required for image generation (simple)");
@@ -81,7 +81,8 @@ export async function imageGen(prompt: string, options: ImageGenOptions = {}): P
     if (options.style) formData.append("style", options.style);
 
     // Use explicit endpoint if provided (highest priority), otherwise construct from origin
-    const editEndpoint = options.endpoint || joinUrlParts(customOrigin || callAiEnv.def.CALLAI_CHAT_URL, "/api/openai-image/edit");
+    const editEndpoint =
+      options.endpoint || joinUrlParts(customOrigin || callAiEnv.def.CALLAI_CHAT_URL, "/api/openrouter-image/edit");
 
     if (!apiKey) {
       throw new Error("API key is required for image generation (edit)");

call-ai/tests/integration/imagegen.integration.test.ts

@@ -60,7 +60,7 @@ describe("Image Generation Integration Tests", () => {
     expect(mock.fetch).toHaveBeenCalledTimes(1);
     const [url, optionsArg] = mock.fetch.mock.calls[0];
 
-    expect(url).toMatch(/.*\/api\/openai-image\/generate$/);
+    expect(url).toMatch(/.*\/api\/openrouter-image\/generate$/);
     expect(optionsArg.method).toBe("POST");
 
     // Check headers
@@ -77,7 +77,7 @@ describe("Image Generation Integration Tests", () => {
     // Verify request body content
     const requestBody = JSON.parse(optionsArg.body);
     expect(requestBody.prompt).toBe(testPrompt);
-    expect(requestBody.model).toBe("gpt-image-1");
+    expect(requestBody.model).toBe("google/gemini-2.5-flash-image");
   });
 
   it("should handle image editing with multiple input images", async () => {
@@ -118,7 +118,7 @@ describe("Image Generation Integration Tests", () => {
     expect(mock.fetch).toHaveBeenCalledTimes(1);
     const [url, optionsArg] = mock.fetch.mock.calls[0];
 
-    expect(url).toMatch(/.*\/api\/openai-image\/edit$/);
+    expect(url).toMatch(/.*\/api\/openrouter-image\/edit$/);
     expect(optionsArg.method).toBe("POST");
 
     // Check headers

call-ai/tests/unit/imagegen.test.ts

@@ -95,7 +95,7 @@ describe("imageGen", () => {
     expect(mock.fetch).toHaveBeenCalledTimes(1);
     const [url, optionsArg] = mock.fetch.mock.calls[0];
 
-    expect(url).toContain("/api/openai-image/generate");
+    expect(url).toContain("/api/openrouter-image/generate");
     expect(optionsArg.method).toBe("POST");
 
     // Check headers
@@ -148,7 +148,7 @@ describe("imageGen", () => {
     expect(mock.fetch).toHaveBeenCalledTimes(1);
     const [url, optionsArg] = mock.fetch.mock.calls[0];
 
-    expect(url).toContain("/api/openai-image/edit");
+    expect(url).toContain("/api/openrouter-image/edit");
     expect(optionsArg.method).toBe("POST");
 
     // Check headers
@@ -236,7 +236,7 @@ describe("imageGen", () => {
     // Verify the custom endpoint was used directly, not the default vibes.diy endpoint
     expect(url).toBe(customEndpoint);
     expect(url).not.toContain("vibes-diy-api.com");
-    expect(url).not.toContain("/api/openai-image/generate");
+    expect(url).not.toContain("/api/openrouter-image/generate");
 
     // Verify the request was made with correct headers
     const headers = optionsArg.headers;
@@ -283,7 +283,7 @@ describe("imageGen", () => {
     // Verify the custom endpoint was used directly
     expect(url).toBe(customEndpoint);
     expect(url).not.toContain("vibes-diy-api.com");
-    expect(url).not.toContain("/api/openai-image/edit");
+    expect(url).not.toContain("/api/openrouter-image/edit");
 
     // Verify the request was made with FormData
     expect(optionsArg.body).toBeInstanceOf(FormData);
@@ -308,6 +308,6 @@ describe("imageGen", () => {
     const [url] = mock.fetch.mock.calls[0];
 
     // Should use the default vibes.diy endpoint
-    expect(url).toContain("/api/openai-image/generate");
+    expect(url).toContain("/api/openrouter-image/generate");
   });
 });