koishi-plugin-aka-ai-generator 0.0.1 → 0.0.4

package/lib/providers/gptgod.d.ts

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

package/lib/providers/index.d.ts

@@ -0,0 +1,20 @@
+import { ImageProvider } from './types';
+export type ProviderType = 'yunwu' | 'gptgod';
+export interface ProviderFactoryConfig {
+    provider: ProviderType;
+    yunwuApiKey: string;
+    yunwuModelId: string;
+    gptgodApiKey: string;
+    gptgodModelId: string;
+    apiTimeout: number;
+    logLevel: 'info' | 'debug';
+    logger: any;
+    ctx: any;
+}
+/**
+ * 创建图像生成供应商实例
+ */
+export declare function createImageProvider(config: ProviderFactoryConfig): ImageProvider;
+export { ImageProvider } from './types';
+export { YunwuProvider } from './yunwu';
+export { GptGodProvider } from './gptgod';

package/lib/providers/types.d.ts

@@ -0,0 +1,16 @@
+export interface ImageProvider {
+    /**
+     * 生成图像
+     * @param prompt 提示词
+     * @param imageUrls 输入图片 URL 数组
+     * @param numImages 需要生成的图片数量
+     * @returns 生成的图片 URL 数组（data: URL 或 http URL）
+     */
+    generateImages(prompt: string, imageUrls: string | string[], numImages: number): Promise<string[]>;
+}
+export interface ProviderConfig {
+    apiTimeout: number;
+    logLevel: 'info' | 'debug';
+    logger: any;
+    ctx: any;
+}

package/lib/providers/yunwu.d.ts

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

package/package.json

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