llm-content-creator 0.1.0

package/dist/schedulers/TaskScheduler.js

@@ -0,0 +1,233 @@
+/**
+ * TaskScheduler - 任务调度器
+ *
+ * 负责任务的创建和调度
+ */
+import { v4 as uuidv4 } from 'uuid';
+import { createTaskQueue } from '../infrastructure/queue/TaskQueue.js';
+import { createTaskRepository } from '../infrastructure/database/index.js';
+import { createLogger } from '../infrastructure/logging/logger.js';
+const logger = createLogger('TaskScheduler');
+/**
+ * TaskScheduler 类
+ *
+ * 负责任务的创建、验证和调度
+ */
+export class TaskScheduler {
+    queue = createTaskQueue();
+    repository = createTaskRepository();
+    initialized = false;
+    /**
+     * 初始化调度器
+     */
+    async initialize() {
+        if (this.initialized) {
+            return;
+        }
+        try {
+            await this.queue;
+            this.initialized = true;
+            logger.info('TaskScheduler initialized');
+        }
+        catch (error) {
+            logger.error('Failed to initialize TaskScheduler', error);
+            throw error;
+        }
+    }
+    /**
+     * 确保调度器已初始化
+     */
+    async ensureInitialized() {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+    }
+    /**
+     * 创建并调度单个任务
+     */
+    async scheduleTask(request) {
+        await this.ensureInitialized();
+        try {
+            // 1. 验证请求参数
+            this.validateTaskRequest(request);
+            // 2. 生成任务 ID
+            const taskId = uuidv4();
+            // 3. 创建任务实体
+            const task = {
+                id: taskId,
+                taskId: taskId,
+                mode: request.mode,
+                type: 'article',
+                topic: request.topic,
+                requirements: request.requirements,
+                hardConstraints: request.hardConstraints,
+                status: 'pending',
+                priority: 2,
+                version: 1,
+                textRetryCount: 0,
+                imageRetryCount: 0,
+                targetAudience: 'general',
+                createdAt: new Date(),
+                updatedAt: new Date(),
+            };
+            // 4. 保存到数据库
+            await this.repository.create(task);
+            logger.info('Task created', {
+                taskId,
+                mode: request.mode,
+                topic: request.topic,
+            });
+            // 5. 准备队列数据
+            const jobData = {
+                taskId: task.id,
+                mode: task.mode,
+                topic: task.topic,
+                requirements: task.requirements,
+                hardConstraints: task.hardConstraints,
+            };
+            // 6. 添加到队列
+            if (request.scheduleAt) {
+                // 延迟任务
+                const delay = request.scheduleAt.getTime() - Date.now();
+                if (delay > 0) {
+                    await (await this.queue).addDelayedTask(jobData, delay);
+                    logger.info('Task scheduled for delayed execution', {
+                        taskId,
+                        delay,
+                        scheduleAt: request.scheduleAt,
+                    });
+                }
+                else {
+                    // 时间已过，立即执行
+                    await (await this.queue).addTask(jobData, {
+                        priority: request.priority,
+                    });
+                    logger.info('Task scheduled for immediate execution', {
+                        taskId,
+                    });
+                }
+            }
+            else {
+                // 立即执行
+                await (await this.queue).addTask(jobData, {
+                    priority: request.priority,
+                });
+                logger.info('Task scheduled for immediate execution', {
+                    taskId,
+                });
+            }
+            return taskId;
+        }
+        catch (error) {
+            logger.error('Failed to schedule task', error, {
+                topic: request.topic,
+            });
+            throw error;
+        }
+    }
+    /**
+     * 批量创建任务
+     */
+    async scheduleBatchTasks(request) {
+        await this.ensureInitialized();
+        const taskIds = [];
+        try {
+            logger.info('Scheduling batch tasks', {
+                count: request.tasks.length,
+            });
+            for (const taskRequest of request.tasks) {
+                const taskId = await this.scheduleTask({
+                    ...taskRequest,
+                    userId: request.userId,
+                });
+                taskIds.push(taskId);
+            }
+            logger.info('Batch tasks scheduled', {
+                count: taskIds.length,
+            });
+            return taskIds;
+        }
+        catch (error) {
+            logger.error('Failed to schedule batch tasks', error);
+            throw error;
+        }
+    }
+    /**
+     * 验证任务请求参数
+     */
+    validateTaskRequest(request) {
+        if (!request.topic || request.topic.trim().length === 0) {
+            throw new Error('Topic is required');
+        }
+        if (!request.requirements || request.requirements.trim().length === 0) {
+            throw new Error('Requirements are required');
+        }
+        if (!['sync', 'async'].includes(request.mode)) {
+            throw new Error('Mode must be either sync or async');
+        }
+        if (request.hardConstraints) {
+            if (request.hardConstraints.minWords && request.hardConstraints.maxWords) {
+                if (request.hardConstraints.minWords > request.hardConstraints.maxWords) {
+                    throw new Error('minWords cannot be greater than maxWords');
+                }
+            }
+        }
+    }
+    /**
+     * 取消任务
+     */
+    async cancelTask(taskId) {
+        try {
+            // 更新数据库中的任务状态
+            const task = await this.repository.findById(taskId);
+            if (!task) {
+                logger.warn('Task not found', { taskId });
+                return false;
+            }
+            // 只有 pending 状态的任务可以取消
+            if (task.status !== 'pending') {
+                logger.warn('Task cannot be cancelled', {
+                    taskId,
+                    status: task.status,
+                });
+                return false;
+            }
+            await this.repository.update(taskId, {
+                status: 'cancelled',
+                updatedAt: new Date().toISOString(),
+            });
+            logger.info('Task cancelled', { taskId });
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to cancel task', error, { taskId });
+            return false;
+        }
+    }
+    /**
+     * 获取队列统计信息
+     */
+    async getQueueStats() {
+        await this.ensureInitialized();
+        return await (await this.queue).getStats();
+    }
+    /**
+     * 关闭调度器
+     */
+    async close() {
+        if (this.initialized) {
+            await (await this.queue).close();
+            this.initialized = false;
+            logger.info('TaskScheduler closed');
+        }
+    }
+}
+/**
+ * 创建 TaskScheduler 实例
+ */
+export async function createTaskScheduler() {
+    const scheduler = new TaskScheduler();
+    await scheduler.initialize();
+    return scheduler;
+}
+//# sourceMappingURL=TaskScheduler.js.map

package/dist/schedulers/TaskScheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TaskScheduler.js","sourceRoot":"","sources":["../../src/schedulers/TaskScheduler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,sCAAsC,CAAC;AACvE,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,qCAAqC,CAAC;AAInE,MAAM,MAAM,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;AA6C7C;;;;GAIG;AACH,MAAM,OAAO,aAAa;IAChB,KAAK,GAAG,eAAe,EAAE,CAAC;IAC1B,UAAU,GAAG,oBAAoB,EAAE,CAAC;IACpC,WAAW,GAAG,KAAK,CAAC;IAE5B;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC;YACjB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAc,CAAC,CAAC;YACnE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,OAA0B;QAC3C,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACH,YAAY;YACZ,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAElC,aAAa;YACb,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;YAExB,YAAY;YACZ,MAAM,IAAI,GAAkB;gBAC1B,EAAE,EAAE,MAAM;gBACV,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,OAAO,CAAC,IAAW;gBACzB,IAAI,EAAE,SAAgB;gBACtB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,eAAe,EAAE,OAAO,CAAC,eAAe;gBACxC,MAAM,EAAE,SAAgB;gBACxB,QAAQ,EAAE,CAAC;gBACX,OAAO,EAAE,CAAC;gBACV,cAAc,EAAE,CAAC;gBACjB,eAAe,EAAE,CAAC;gBAClB,cAAc,EAAE,SAAS;gBACzB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;YAEF,YAAY;YACZ,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAY,CAAC,CAAC;YAE3C,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE;gBAC1B,MAAM;gBACN,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;aACrB,CAAC,CAAC;YAEH,YAAY;YACZ,MAAM,OAAO,GAAgB;gBAC3B,MAAM,EAAE,IAAI,CAAC,EAAG;gBAChB,IAAI,EAAE,IAAI,CAAC,IAAK;gBAChB,KAAK,EAAE,IAAI,CAAC,KAAM;gBAClB,YAAY,EAAE,IAAI,CAAC,YAAa;gBAChC,eAAe,EAAE,IAAI,CAAC,eAAe;aACtC,CAAC;YAEF,WAAW;YACX,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,OAAO;gBACP,MAAM,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACxD,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBACxD,MAAM,CAAC,IAAI,CAAC,sCAAsC,EAAE;wBAClD,MAAM;wBACN,KAAK;wBACL,UAAU,EAAE,OAAO,CAAC,UAAU;qBAC/B,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,YAAY;oBACZ,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE;wBACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ;qBAC3B,CAAC,CAAC;oBACH,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;wBACpD,MAAM;qBACP,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO;gBACP,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE;oBACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;oBACpD,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,EAAE;gBACtD,KAAK,EAAE,OAAO,CAAC,KAAK;aACrB,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CACtB,OAAgC;QAEhC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,KAAK,MAAM,WAAW,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC;oBACrC,GAAG,WAAW;oBACd,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE;gBACnC,KAAK,EAAE,OAAO,CAAC,MAAM;aACtB,CAAC,CAAC;YAEH,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAc,CAAC,CAAC;YAC/D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,OAA0B;QACpD,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,eAAe,CAAC,QAAQ,IAAI,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;gBACzE,IAAI,OAAO,CAAC,eAAe,CAAC,QAAQ,GAAG,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;oBACxE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,IAAI,CAAC;YACH,cAAc;YACd,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAEpD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC1C,OAAO,KAAK,CAAC;YACf,CAAC;YAED,uBAAuB;YACvB,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE;oBACtC,MAAM;oBACN,MAAM,EAAE,IAAI,CAAC,MAAM;iBACpB,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE;gBACnC,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAE1C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAClE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa;QAQjB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;IACtC,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC;IAC7B,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/schedulers/index.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Schedulers 导出
+ */
+export { TaskScheduler, createTaskScheduler, type CreateTaskRequest, type CreateBatchTasksRequest, type ScheduleResult, } from './TaskScheduler.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/schedulers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/schedulers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,KAAK,iBAAiB,EACtB,KAAK,uBAAuB,EAC5B,KAAK,cAAc,GACpB,MAAM,oBAAoB,CAAC"}

package/dist/schedulers/index.js

@@ -0,0 +1,5 @@
+/**
+ * Schedulers 导出
+ */
+export { TaskScheduler, createTaskScheduler, } from './TaskScheduler.js';
+//# sourceMappingURL=index.js.map

package/dist/schedulers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/schedulers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,aAAa,EACb,mBAAmB,GAIpB,MAAM,oBAAoB,CAAC"}

package/dist/services/image/ImageService.d.ts

@@ -0,0 +1,68 @@
+/**
+ * Image 服务
+ *
+ * 封装 Doubao API (火山引擎) 进行图片生成
+ *
+ * 参考实现: @tonychaos/mcp-server-doubao
+ */
+/**
+ * Doubao 图片生成模型
+ */
+export declare const DOUBAO_IMAGE_MODELS: {
+    readonly SEEDREAM_4_5: "doubao-seedream-4-5-251128";
+    readonly SEEDREAM_4_0: "doubao-seedream-4-0-250428";
+    readonly SEEDREAM_3_0_T2I: "doubao-seedream-3-0-t2i";
+};
+/**
+ * 图片生成请求参数
+ */
+export interface ImageGenerationRequest {
+    prompt: string;
+    model?: keyof typeof DOUBAO_IMAGE_MODELS;
+    size?: string;
+    seed?: number;
+    stream?: boolean;
+    watermark?: boolean;
+}
+/**
+ * 图片生成响应
+ */
+export interface ImageGenerationResponse {
+    imageUrl: string;
+    model: string;
+    prompt?: string;
+    seed?: number;
+}
+/**
+ * Image 服务类
+ */
+export declare class ImageService {
+    private apiKey;
+    private baseURL;
+    constructor();
+    /**
+     * 生成图片
+     */
+    generateImage(request: ImageGenerationRequest): Promise<ImageGenerationResponse>;
+    /**
+     * 批量生成图片
+     */
+    batchGenerateImages(requests: ImageGenerationRequest[]): Promise<ImageGenerationResponse[]>;
+    /**
+     * 检查 API 连接
+     */
+    healthCheck(): Promise<boolean>;
+    /**
+     * 获取支持的模型列表
+     */
+    getSupportedModels(): typeof DOUBAO_IMAGE_MODELS;
+    /**
+     * 获取默认模型
+     */
+    getDefaultModel(): string;
+}
+/**
+ * Image 服务单例
+ */
+export declare const imageService: ImageService;
+//# sourceMappingURL=ImageService.d.ts.map

package/dist/services/image/ImageService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ImageService.d.ts","sourceRoot":"","sources":["../../../src/services/image/ImageService.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAQH;;GAEG;AACH,eAAO,MAAM,mBAAmB;;;;CAItB,CAAC;AAOX;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,OAAO,mBAAmB,CAAC;IACzC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAS;;IAWxB;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IA+EtF;;OAEG;IACG,mBAAmB,CAAC,QAAQ,EAAE,sBAAsB,EAAE,GAAG,OAAO,CAAC,uBAAuB,EAAE,CAAC;IAgCjG;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAgBrC;;OAEG;IACH,kBAAkB,IAAI,OAAO,mBAAmB;IAIhD;;OAEG;IACH,eAAe,IAAI,MAAM;CAG1B;AAED;;GAEG;AACH,eAAO,MAAM,YAAY,cAAqB,CAAC"}

package/dist/services/image/ImageService.js

@@ -0,0 +1,166 @@
+/**
+ * Image 服务
+ *
+ * 封装 Doubao API (火山引擎) 进行图片生成
+ *
+ * 参考实现: @tonychaos/mcp-server-doubao
+ */
+import axios, { AxiosError } from 'axios';
+import { config } from '../../config/index.js';
+import { createLogger } from '../../infrastructure/logging/logger.js';
+const logger = createLogger('Image');
+/**
+ * Doubao 图片生成模型
+ */
+export const DOUBAO_IMAGE_MODELS = {
+    SEEDREAM_4_5: 'doubao-seedream-4-5-251128', // 最新版本（推荐）
+    SEEDREAM_4_0: 'doubao-seedream-4-0-250428',
+    SEEDREAM_3_0_T2I: 'doubao-seedream-3-0-t2i',
+};
+/**
+ * 默认模型
+ */
+const DEFAULT_MODEL = DOUBAO_IMAGE_MODELS.SEEDREAM_4_5;
+/**
+ * Image 服务类
+ */
+export class ImageService {
+    apiKey;
+    baseURL;
+    constructor() {
+        this.apiKey = config.doubao.apiKey;
+        this.baseURL = config.doubao.apiURL || 'https://ark.cn-beijing.volces.com';
+        if (!this.apiKey) {
+            throw new Error('ARK_API_KEY is required for Doubao Image Service');
+        }
+    }
+    /**
+     * 生成图片
+     */
+    async generateImage(request) {
+        try {
+            logger.debug('Sending image generation request', {
+                prompt: request.prompt.substring(0, 100),
+                model: request.model || DEFAULT_MODEL,
+                size: request.size,
+            });
+            // 构建请求体（参考 @tonychaos/mcp-server-doubao）
+            const requestBody = {
+                model: request.model || DEFAULT_MODEL,
+                prompt: request.prompt,
+                size: request.size || '1920x1920', // Doubao 要求至少 3686400 像素 (1920x1920)
+                seed: request.seed,
+                stream: request.stream || false,
+                response_format: 'url',
+                watermark: request.watermark !== undefined ? request.watermark : true,
+            };
+            // 移除 undefined 的字段
+            Object.keys(requestBody).forEach((key) => {
+                if (requestBody[key] === undefined) {
+                    delete requestBody[key];
+                }
+            });
+            // 发送请求
+            const response = await axios.post(`${this.baseURL}/api/v3/images/generations`, requestBody, {
+                headers: {
+                    'Authorization': `Bearer ${this.apiKey}`,
+                    'Content-Type': 'application/json',
+                },
+                timeout: 120000, // 2 分钟超时（图片生成可能较慢）
+            });
+            // 解析响应
+            const data = response.data;
+            // Doubao API 返回格式: { data: [{ url: "..." }] }
+            if (!data.data || !data.data[0] || !data.data[0].url) {
+                throw new Error('Invalid response format from Doubao API');
+            }
+            const imageUrl = data.data[0].url;
+            logger.info('Image generation completed', {
+                imageUrl: imageUrl.substring(0, 50) + '...',
+                model: requestBody.model,
+            });
+            return {
+                imageUrl,
+                model: requestBody.model,
+                prompt: request.prompt,
+                seed: request.seed,
+            };
+        }
+        catch (error) {
+            if (axios.isAxiosError(error)) {
+                const axiosError = error;
+                logger.error('Image generation API request failed', {
+                    status: axiosError.response?.status,
+                    data: axiosError.response?.data,
+                    message: axiosError.message,
+                });
+                throw new Error(`Image generation API error: ${axiosError.response?.status} - ${JSON.stringify(axiosError.response?.data)}`);
+            }
+            logger.error('Image service error', error);
+            throw error;
+        }
+    }
+    /**
+     * 批量生成图片
+     */
+    async batchGenerateImages(requests) {
+        logger.info('Batch generating images', { count: requests.length });
+        const results = await Promise.allSettled(requests.map(async (req) => {
+            try {
+                return await this.generateImage(req);
+            }
+            catch (error) {
+                logger.error(`Image generation failed for prompt: ${req.prompt.substring(0, 50)}`, error);
+                throw error;
+            }
+        }));
+        const successful = results.filter((r) => r.status === 'fulfilled');
+        const failed = results.filter((r) => r.status === 'rejected');
+        logger.info('Batch image generation completed', {
+            total: requests.length,
+            successful: successful.length,
+            failed: failed.length,
+        });
+        // 如果全部失败，抛出错误
+        if (successful.length === 0) {
+            throw new Error('All image generation requests failed');
+        }
+        // 返回成功的结果
+        return successful.map((r) => r.value);
+    }
+    /**
+     * 检查 API 连接
+     */
+    async healthCheck() {
+        try {
+            // 验证 API key 格式
+            if (!this.apiKey || this.apiKey.length < 10) {
+                logger.error('Invalid API key format');
+                return false;
+            }
+            logger.info('Image service health check passed (API key format validated)');
+            return true;
+        }
+        catch (error) {
+            logger.error('Image service health check failed', error);
+            return false;
+        }
+    }
+    /**
+     * 获取支持的模型列表
+     */
+    getSupportedModels() {
+        return DOUBAO_IMAGE_MODELS;
+    }
+    /**
+     * 获取默认模型
+     */
+    getDefaultModel() {
+        return DEFAULT_MODEL;
+    }
+}
+/**
+ * Image 服务单例
+ */
+export const imageService = new ImageService();
+//# sourceMappingURL=ImageService.js.map

package/dist/services/image/ImageService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ImageService.js","sourceRoot":"","sources":["../../../src/services/image/ImageService.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AAEtE,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;AAErC;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,YAAY,EAAE,4BAA4B,EAAG,WAAW;IACxD,YAAY,EAAE,4BAA4B;IAC1C,gBAAgB,EAAE,yBAAyB;CACnC,CAAC;AAEX;;GAEG;AACH,MAAM,aAAa,GAAG,mBAAmB,CAAC,YAAY,CAAC;AAwBvD;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,CAAS;IACf,OAAO,CAAS;IAExB;QACE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,IAAI,mCAAmC,CAAC;QAE3E,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE;gBAC/C,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;gBACxC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,aAAa;gBACrC,IAAI,EAAE,OAAO,CAAC,IAAI;aACnB,CAAC,CAAC;YAEH,yCAAyC;YACzC,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,aAAa;gBACrC,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,WAAW,EAAG,qCAAqC;gBACzE,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,KAAK;gBAC/B,eAAe,EAAE,KAAK;gBACtB,SAAS,EAAE,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;aACtE,CAAC;YAEF,mBAAmB;YACnB,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACvC,IAAK,WAAmB,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE,CAAC;oBAC5C,OAAQ,WAAmB,CAAC,GAAG,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO;YACP,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,OAAO,4BAA4B,EAC3C,WAAW,EACX;gBACE,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,OAAO,EAAE,MAAM,EAAE,mBAAmB;aACrC,CACF,CAAC;YAEF,OAAO;YACP,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;YAE3B,8CAA8C;YAC9C,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBACrD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;YAElC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;gBACxC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;gBAC3C,KAAK,EAAE,WAAW,CAAC,KAAK;aACzB,CAAC,CAAC;YAEH,OAAO;gBACL,QAAQ;gBACR,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,OAAO,CAAC,IAAI;aACnB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9B,MAAM,UAAU,GAAG,KAAwB,CAAC;gBAC5C,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE;oBAClD,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,MAAM;oBACnC,IAAI,EAAE,UAAU,CAAC,QAAQ,EAAE,IAAI;oBAC/B,OAAO,EAAE,UAAU,CAAC,OAAO;iBAC5B,CAAC,CAAC;gBAEH,MAAM,IAAI,KAAK,CACb,+BAA+B,UAAU,CAAC,QAAQ,EAAE,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAC5G,CAAC;YACJ,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAc,CAAC,CAAC;YACpD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CAAC,QAAkC;QAC1D,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAEnE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACzB,IAAI,CAAC;gBACH,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;YACvC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,uCAAuC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,KAAc,CAAC,CAAC;gBACnG,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC;QAE9D,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC9C,KAAK,EAAE,QAAQ,CAAC,MAAM;YACtB,UAAU,EAAE,UAAU,CAAC,MAAM;YAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,cAAc;QACd,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,UAAU;QACV,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAqD,CAAC,KAAK,CAAC,CAAC;IAC7F,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,gBAAgB;YAChB,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;gBAC5C,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;gBACvC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;YAC5E,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAc,CAAC,CAAC;YAClE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/services/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * 服务层统一导出
+ */
+export * from './llm/LLMService.js';
+export * from './search/SearchService.js';
+export * from './image/ImageService.js';
+export * from './quality/index.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/services/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,cAAc,qBAAqB,CAAC;AAGpC,cAAc,2BAA2B,CAAC;AAG1C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,oBAAoB,CAAC"}

package/dist/services/index.js

@@ -0,0 +1,12 @@
+/**
+ * 服务层统一导出
+ */
+// LLM 服务
+export * from './llm/LLMService.js';
+// Search 服务
+export * from './search/SearchService.js';
+// Image 服务
+export * from './image/ImageService.js';
+// Quality 服务（新的质量检查服务）
+export * from './quality/index.js';
+//# sourceMappingURL=index.js.map

package/dist/services/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,SAAS;AACT,cAAc,qBAAqB,CAAC;AAEpC,YAAY;AACZ,cAAc,2BAA2B,CAAC;AAE1C,WAAW;AACX,cAAc,yBAAyB,CAAC;AAExC,uBAAuB;AACvB,cAAc,oBAAoB,CAAC"}