koishi-plugin-aka-ai-generator 0.2.16 → 0.2.17

package/lib/index.d.ts

@@ -1,6 +1,6 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "aka-ai-generator";
-export type ImageProvider = 'yunwu' | 'gptgod';
+export type ImageProvider = 'yunwu' | 'gptgod' | 'gemini';
 export interface ModelMappingConfig {
     suffix: string;
     modelId: string;
@@ -36,6 +36,9 @@ export interface Config {
     yunwuModelId: string;
     gptgodApiKey: string;
     gptgodModelId: string;
+    geminiApiKey: string;
+    geminiModelId: string;
+    geminiApiBase: string;
     modelMappings?: ModelMappingConfig[];
     apiTimeout: number;
     commandTimeout: number;

package/lib/index.js

@@ -561,6 +561,145 @@ var GptGodProvider = class {
   }
 };
 
+// src/providers/gemini.ts
+async function downloadImageAsBase643(ctx, url, timeout, logger) {
+  try {
+    const response = await ctx.http.get(url, {
+      responseType: "arraybuffer",
+      timeout: timeout * 1e3
+    });
+    const buffer = Buffer.from(response);
+    const base64 = buffer.toString("base64");
+    let mimeType = "image/jpeg";
+    const urlLower = url.toLowerCase();
+    if (urlLower.endsWith(".png")) {
+      mimeType = "image/png";
+    } else if (urlLower.endsWith(".webp")) {
+      mimeType = "image/webp";
+    } else if (urlLower.endsWith(".gif")) {
+      mimeType = "image/gif";
+    }
+    logger.debug("图片下载并转换为Base64", { url, mimeType, size: base64.length });
+    return { data: base64, mimeType };
+  } catch (error) {
+    logger.error("下载图片失败", { url, error });
+    throw new Error("下载图片失败，请检查图片链接是否有效");
+  }
+}
+__name(downloadImageAsBase643, "downloadImageAsBase64");
+function parseGeminiResponse(response) {
+  try {
+    const images = [];
+    if (response.candidates && response.candidates.length > 0) {
+      for (const candidate of response.candidates) {
+        if (candidate.content && candidate.content.parts) {
+          for (const part of candidate.content.parts) {
+            if (part.inlineData && part.inlineData.data) {
+              const base64Data = part.inlineData.data;
+              const mimeType = part.inlineData.mimeType || "image/jpeg";
+              const dataUrl = `data:${mimeType};base64,${base64Data}`;
+              images.push(dataUrl);
+            } else if (part.inline_data && part.inline_data.data) {
+              const base64Data = part.inline_data.data;
+              const mimeType = part.inline_data.mime_type || "image/jpeg";
+              const dataUrl = `data:${mimeType};base64,${base64Data}`;
+              images.push(dataUrl);
+            } else if (part.fileData && part.fileData.fileUri) {
+              images.push(part.fileData.fileUri);
+            }
+          }
+        }
+      }
+    }
+    return images;
+  } catch (error) {
+    return [];
+  }
+}
+__name(parseGeminiResponse, "parseGeminiResponse");
+var GeminiProvider = class {
+  static {
+    __name(this, "GeminiProvider");
+  }
+  config;
+  constructor(config) {
+    this.config = config;
+  }
+  async generateImages(prompt, imageUrls, numImages) {
+    const urls = Array.isArray(imageUrls) ? imageUrls : [imageUrls];
+    const logger = this.config.logger;
+    const ctx = this.config.ctx;
+    logger.debug("开始下载图片并转换为Base64", { urls });
+    const imageParts = [];
+    for (const url of urls) {
+      const { data, mimeType } = await downloadImageAsBase643(
+        ctx,
+        url,
+        this.config.apiTimeout,
+        logger
+      );
+      imageParts.push({
+        inline_data: {
+          mime_type: mimeType,
+          data
+        }
+      });
+    }
+    const apiBase = this.config.apiBase?.replace(/\/$/, "") || "https://generativelanguage.googleapis.com";
+    const endpoint = `${apiBase}/v1beta/models/${this.config.modelId}:generateContent`;
+    const allImages = [];
+    for (let i = 0; i < numImages; i++) {
+      const requestData = {
+        contents: [
+          {
+            role: "user",
+            parts: [
+              { text: prompt },
+              ...imageParts
+            ]
+          }
+        ],
+        generationConfig: {
+          responseModalities: ["IMAGE"]
+        }
+      };
+      logger.debug("调用 Gemini API", { prompt, imageCount: urls.length, numImages, current: i + 1, endpoint });
+      try {
+        const response = await ctx.http.post(
+          endpoint,
+          requestData,
+          {
+            headers: {
+              "Content-Type": "application/json"
+            },
+            params: {
+              key: this.config.apiKey
+            },
+            timeout: this.config.apiTimeout * 1e3
+          }
+        );
+        const images = parseGeminiResponse(response);
+        allImages.push(...images);
+        logger.success("Gemini API 调用成功", { current: i + 1, total: numImages });
+      } catch (error) {
+        logger.error("Gemini API 调用失败", {
+          message: error?.message || "未知错误",
+          code: error?.code,
+          status: error?.response?.status,
+          current: i + 1,
+          total: numImages
+        });
+        if (allImages.length > 0) {
+          logger.warn("部分图片生成失败，返回已生成的图片", { generated: allImages.length, requested: numImages });
+          break;
+        }
+        throw new Error(`图像处理API调用失败: ${error?.message || "未知错误"}`);
+      }
+    }
+    return allImages;
+  }
+};
+
 // src/providers/index.ts
 function createImageProvider(config) {
   switch (config.provider) {
@@ -582,6 +721,16 @@ function createImageProvider(config) {
         logger: config.logger,
         ctx: config.ctx
       });
+    case "gemini":
+      return new GeminiProvider({
+        apiKey: config.geminiApiKey,
+        modelId: config.geminiModelId,
+        apiBase: config.geminiApiBase,
+        apiTimeout: config.apiTimeout,
+        logLevel: config.logLevel,
+        logger: config.logger,
+        ctx: config.ctx
+      });
     default:
       throw new Error(`不支持的供应商类型: ${config.provider}`);
   }
@@ -614,17 +763,22 @@ var Config = import_koishi.Schema.intersect([
   import_koishi.Schema.object({
     provider: import_koishi.Schema.union([
       import_koishi.Schema.const("yunwu").description("云雾 Gemini 服务"),
-      import_koishi.Schema.const("gptgod").description("GPTGod 服务")
+      import_koishi.Schema.const("gptgod").description("GPTGod 服务"),
+      import_koishi.Schema.const("gemini").description("Google Gemini 原生")
     ]).default("yunwu").description("图像生成供应商"),
     yunwuApiKey: import_koishi.Schema.string().description("云雾API密钥").role("secret").required(),
     yunwuModelId: import_koishi.Schema.string().default("gemini-2.5-flash-image").description("云雾图像生成模型ID"),
     gptgodApiKey: import_koishi.Schema.string().description("GPTGod API 密钥").role("secret").default(""),
     gptgodModelId: import_koishi.Schema.string().default("nano-banana").description("GPTGod 模型ID"),
+    geminiApiKey: import_koishi.Schema.string().description("Gemini API 密钥").role("secret").default(""),
+    geminiModelId: import_koishi.Schema.string().default("gemini-2.5-flash").description("Gemini 模型ID"),
+    geminiApiBase: import_koishi.Schema.string().default("https://generativelanguage.googleapis.com").description("Gemini API 基础地址"),
     modelMappings: import_koishi.Schema.array(import_koishi.Schema.object({
       suffix: import_koishi.Schema.string().required().description("指令后缀（例如 4K，对应输入 -4K）"),
       provider: import_koishi.Schema.union([
         import_koishi.Schema.const("yunwu").description("云雾 Gemini 服务"),
-        import_koishi.Schema.const("gptgod").description("GPTGod 服务")
+        import_koishi.Schema.const("gptgod").description("GPTGod 服务"),
+        import_koishi.Schema.const("gemini").description("Google Gemini 原生")
       ]).description("可选：覆盖供应商"),
       modelId: import_koishi.Schema.string().required().description("触发该后缀时使用的模型 ID")
     })).role("table").default([]).description("根据 -后缀切换模型/供应商"),
@@ -680,6 +834,9 @@ function apply(ctx, config) {
         yunwuModelId: providerType === "yunwu" ? modelId || config.yunwuModelId : config.yunwuModelId,
         gptgodApiKey: config.gptgodApiKey,
         gptgodModelId: providerType === "gptgod" ? modelId || config.gptgodModelId : config.gptgodModelId,
+        geminiApiKey: config.geminiApiKey,
+        geminiModelId: providerType === "gemini" ? modelId || config.geminiModelId : config.geminiModelId,
+        geminiApiBase: config.geminiApiBase,
         apiTimeout: config.apiTimeout,
         logLevel: config.logLevel,
         logger,

package/lib/providers/gemini.d.ts

@@ -0,0 +1,11 @@
+import { ImageProvider, ProviderConfig } from './types';
+export interface GeminiConfig extends ProviderConfig {
+    apiKey: string;
+    modelId: string;
+    apiBase?: string;
+}
+export declare class GeminiProvider implements ImageProvider {
+    private config;
+    constructor(config: GeminiConfig);
+    generateImages(prompt: string, imageUrls: string | string[], numImages: number): Promise<string[]>;
+}

package/lib/providers/index.d.ts

@@ -1,11 +1,14 @@
 import { ImageProvider } from './types';
-export type ProviderType = 'yunwu' | 'gptgod';
+export type ProviderType = 'yunwu' | 'gptgod' | 'gemini';
 export interface ProviderFactoryConfig {
     provider: ProviderType;
     yunwuApiKey: string;
     yunwuModelId: string;
     gptgodApiKey: string;
     gptgodModelId: string;
+    geminiApiKey: string;
+    geminiModelId: string;
+    geminiApiBase: string;
     apiTimeout: number;
     logLevel: 'info' | 'debug';
     logger: any;
@@ -18,3 +21,4 @@ export declare function createImageProvider(config: ProviderFactoryConfig): Imag
 export { ImageProvider } from './types';
 export { YunwuProvider } from './yunwu';
 export { GptGodProvider } from './gptgod';
+export { GeminiProvider } from './gemini';

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.2.16",
+  "version": "0.2.17",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [