v3.5.0

dist/modules/ai/azureTts.service.js

@@ -0,0 +1,32 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var AzureTtsService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AzureTtsService = void 0;
+const common_1 = require("@nestjs/common");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+const globalConfig_service_1 = require("../globalConfig/globalConfig.service");
+const upload_service_1 = require("../upload/upload.service");
+let AzureTtsService = AzureTtsService_1 = class AzureTtsService {
+    constructor(uploadService, globalConfigService, chatLogService) {
+        this.uploadService = uploadService;
+        this.globalConfigService = globalConfigService;
+        this.chatLogService = chatLogService;
+        this.logger = new common_1.Logger(AzureTtsService_1.name);
+    }
+};
+AzureTtsService = AzureTtsService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [upload_service_1.UploadService,
+        globalConfig_service_1.GlobalConfigService,
+        chatLog_service_1.ChatLogService])
+], AzureTtsService);
+exports.AzureTtsService = AzureTtsService;

dist/modules/ai/lumaVideo.service.js

@@ -0,0 +1,185 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LumaVideoService = void 0;
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+let LumaVideoService = class LumaVideoService {
+    constructor(chatLogService) {
+        this.chatLogService = chatLogService;
+    }
+    async lumaVideo(inputs) {
+        var _a, _b, _c;
+        const { apiKey, proxyUrl, fileInfo, prompt, timeout, assistantLogId, extraParam, } = inputs;
+        let result = {
+            text: '',
+            fileInfo: '',
+            taskId: '',
+            taskData: '',
+            status: 2,
+        };
+        let response = null;
+        let url = '';
+        let payloadJson = {};
+        const headers = { Authorization: `Bearer ${apiKey}` };
+        url = `${proxyUrl}/luma/generations/`;
+        const aspectRatio = '16:9';
+        payloadJson = {
+            user_prompt: prompt,
+            aspect_ratio: aspectRatio,
+            expand_prompt: true,
+        };
+        common_1.Logger.log(`正在准备发送请求到 ${url}，payload: ${JSON.stringify(payloadJson)}, headers: ${JSON.stringify(headers)}`, 'LumaService');
+        try {
+            response = await axios_1.default.post(url, payloadJson, { headers });
+        }
+        catch (error) {
+            common_1.Logger.error(`任务提交失败: ${error.message}`, 'LumaService');
+            throw new Error('任务提交失败');
+        }
+        if ((_a = response === null || response === void 0 ? void 0 : response.data) === null || _a === void 0 ? void 0 : _a.id) {
+            result.taskId = (_b = response === null || response === void 0 ? void 0 : response.data) === null || _b === void 0 ? void 0 : _b.id;
+            common_1.Logger.log(`任务提交成功, 任务ID: ${(_c = response === null || response === void 0 ? void 0 : response.data) === null || _c === void 0 ? void 0 : _c.id}`, 'LumaService');
+        }
+        else {
+            throw new Error('未能获取结果数据, 即将重试');
+        }
+        try {
+            await this.pollLumaVideoResult({
+                proxyUrl,
+                apiKey,
+                taskId: response.data.id,
+                timeout,
+                prompt,
+                onSuccess: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            videoUrl: data === null || data === void 0 ? void 0 : data.videoUrl,
+                            audioUrl: data === null || data === void 0 ? void 0 : data.audioUrl,
+                            fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                            answer: (data === null || data === void 0 ? void 0 : data.answer) || prompt,
+                            progress: '100%',
+                            status: 3,
+                            taskId: data === null || data === void 0 ? void 0 : data.taskId,
+                            taskData: data === null || data === void 0 ? void 0 : data.taskData,
+                        });
+                        common_1.Logger.log('视频任务已完成', 'LumaService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'LumaService');
+                    }
+                },
+                onGenerating: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            videoUrl: data === null || data === void 0 ? void 0 : data.videoUrl,
+                            audioUrl: data === null || data === void 0 ? void 0 : data.audioUrl,
+                            fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                            answer: (data === null || data === void 0 ? void 0 : data.answer) || '视频生成中...',
+                            progress: data === null || data === void 0 ? void 0 : data.progress,
+                            status: data.status,
+                        });
+                        common_1.Logger.log('视频生成中...', 'LumaService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'LumaService');
+                    }
+                },
+                onFailure: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            answer: '视频生成失败',
+                            status: data.status,
+                        });
+                        common_1.Logger.log('生成失败', 'Lum aService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'LumaService');
+                    }
+                },
+            });
+        }
+        catch (error) {
+            common_1.Logger.error('查询生成结果时发生错误:', error.message, 'LumaService');
+            throw new Error('查询生成结果时发生错误');
+        }
+        return result;
+    }
+    async pollLumaVideoResult(inputs) {
+        const { proxyUrl, apiKey, taskId, timeout, onSuccess, onFailure, onGenerating, action, } = inputs;
+        let result = {
+            videoUrl: '',
+            audioUrl: '',
+            fileInfo: '',
+            drawId: '',
+            taskData: '',
+            status: 2,
+            progress: 0,
+            answer: '',
+        };
+        const headers = { Authorization: `Bearer ${apiKey}` };
+        const url = `${proxyUrl}/luma/generations/${taskId}`;
+        const startTime = Date.now();
+        const totalDuration = 300000;
+        const POLL_INTERVAL = 5000;
+        let retryCount = 0;
+        try {
+            while (Date.now() - startTime < timeout) {
+                await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL));
+                try {
+                    const res = await axios_1.default.get(url, { headers });
+                    const interval = setInterval(() => {
+                        const elapsed = Date.now() - startTime;
+                        let percentage = Math.floor((elapsed / totalDuration) * 100);
+                        if (percentage >= 99)
+                            percentage = 99;
+                        result.answer = `视频生成中 （${percentage}%）`;
+                    }, 1000);
+                    const responses = res.data;
+                    common_1.Logger.debug(`轮询结果: ${JSON.stringify(responses)}`, 'LumaService');
+                    if (responses.state === 'completed') {
+                        result.taskId = responses.id;
+                        result.taskData = JSON.stringify(responses);
+                        result.fileInfo = responses.video.url;
+                        result.answer = `提示词: "${responses.prompt}"`;
+                        onSuccess(result);
+                        clearInterval(interval);
+                        return;
+                    }
+                    else {
+                        onGenerating(result);
+                    }
+                    if (result.progress) {
+                    }
+                }
+                catch (error) {
+                    retryCount++;
+                    common_1.Logger.error(`轮询失败，重试次数: ${retryCount}`, 'LumaService');
+                }
+            }
+            common_1.Logger.error('轮询超时，请稍后再试！', 'LumaService');
+            result.status = 4;
+            onFailure(result);
+            throw new Error('查询超时，请稍后再试！');
+        }
+        catch (error) {
+            common_1.Logger.error(`轮询过程中发生错误: ${error}`, 'LumaService');
+            result.status = 5;
+            onFailure(result);
+        }
+    }
+};
+LumaVideoService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [chatLog_service_1.ChatLogService])
+], LumaVideoService);
+exports.LumaVideoService = LumaVideoService;

dist/modules/ai/midjourneyDraw.service.js

@@ -0,0 +1,211 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MidjourneyService = void 0;
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+const globalConfig_service_1 = require("../globalConfig/globalConfig.service");
+const upload_service_1 = require("../upload/upload.service");
+let MidjourneyService = class MidjourneyService {
+    constructor(uploadService, globalConfigService, chatLogService) {
+        this.uploadService = uploadService;
+        this.globalConfigService = globalConfigService;
+        this.chatLogService = chatLogService;
+    }
+    async midjourneyDraw(inputs) {
+        var _a, _b;
+        const { id, apiKey, proxyUrl, action, drawId, prompt, usePrompt, customId, timeout, assistantLogId, } = inputs;
+        let result = {
+            text: '',
+            fileInfo: '',
+            drawId: '',
+            customId: '',
+            status: 2,
+        };
+        let response;
+        let retryCount = 0;
+        let url = '';
+        while (retryCount < 3) {
+            let payloadJson = {};
+            try {
+                if (action === 'IMAGINE') {
+                    url = `${proxyUrl}/mj/submit/imagine`;
+                    payloadJson = { prompt: usePrompt };
+                }
+                else {
+                    url = `${proxyUrl}/mj/submit/action`;
+                    payloadJson = { taskId: drawId, customId: customId };
+                }
+                const headers = { 'mj-api-secret': apiKey };
+                common_1.Logger.log(`正在准备发送请求到 ${url}，payload: ${JSON.stringify(payloadJson)}, headers: ${JSON.stringify(headers)}`, 'MidjourneyService');
+                response = await axios_1.default.post(url, payloadJson, { headers });
+                if ((_a = response === null || response === void 0 ? void 0 : response.data) === null || _a === void 0 ? void 0 : _a.result) {
+                    result.drawId = (_b = response === null || response === void 0 ? void 0 : response.data) === null || _b === void 0 ? void 0 : _b.result;
+                    break;
+                }
+                else {
+                    throw new Error('未能获取结果数据, 即将重试');
+                }
+            }
+            catch (error) {
+                retryCount++;
+                if (retryCount >= 3) {
+                    common_1.Logger.log(`绘画任务提交失败, 请检查后台配置或者稍后重试! ${error}`, 'MidjourneyService');
+                }
+            }
+        }
+        this.pollMjDrawingResult({
+            proxyUrl,
+            apiKey,
+            drawId: response.data.result,
+            timeout,
+            prompt,
+            onSuccess: async (data) => {
+                await this.chatLogService.updateChatLog(assistantLogId, {
+                    fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                    answer: (data === null || data === void 0 ? void 0 : data.answer) || prompt,
+                    progress: '100%',
+                    status: 3,
+                    drawId: data === null || data === void 0 ? void 0 : data.drawId,
+                    customId: data === null || data === void 0 ? void 0 : data.customId,
+                });
+                common_1.Logger.log('绘图成功！');
+            },
+            onDrawing: async (data) => {
+                await this.chatLogService.updateChatLog(assistantLogId, {
+                    answer: (data === null || data === void 0 ? void 0 : data.answer) || '绘制中',
+                    progress: data === null || data === void 0 ? void 0 : data.progress,
+                    status: 2,
+                });
+                common_1.Logger.log(`绘制中！绘制进度${data === null || data === void 0 ? void 0 : data.progress}`);
+            },
+            onFailure: async (data) => {
+                await this.chatLogService.updateChatLog(assistantLogId, {
+                    answer: '绘图失败',
+                    status: data.status,
+                });
+                common_1.Logger.log('绘图失败');
+            },
+        }).catch((error) => {
+            common_1.Logger.error('查询绘图结果时发生错误:', error, 'MidjourneyService');
+        });
+        common_1.Logger.log(`绘画任务提交成功, 绘画ID: ${response.data.result}`, 'MidjourneyService');
+        return result;
+    }
+    async pollMjDrawingResult(inputs) {
+        const { proxyUrl, apiKey, drawId, timeout, onSuccess, prompt, onFailure, onDrawing, } = inputs;
+        const { mjNotSaveImg, mjProxyImgUrl, mjNotUseProxy } = await this.globalConfigService.getConfigs([
+            'mjNotSaveImg',
+            'mjProxyImgUrl',
+            'mjNotUseProxy',
+        ]);
+        let response;
+        let result = {
+            fileInfo: '',
+            drawId: '',
+            customId: '',
+            status: 2,
+            progress: 0,
+            answer: '',
+        };
+        let payloadJson = {};
+        const startTime = Date.now();
+        const POLL_INTERVAL = 5000;
+        let retryCount = 0;
+        let pollingCount = 0;
+        try {
+            while (Date.now() - startTime < timeout) {
+                await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL));
+                try {
+                    const headers = {
+                        'Content-Type': 'application/x-www-form-urlencoded',
+                        'mj-api-secret': apiKey,
+                    };
+                    const url = `${proxyUrl}/mj/task/${drawId}/fetch`;
+                    const res = await axios_1.default.get(url, { headers });
+                    const responses = res.data;
+                    if (responses.status === 'SUCCESS') {
+                        common_1.Logger.log(`绘制成功, 获取到的URL: ${responses.imageUrl}`, 'MidjourneyService');
+                        let processedUrl = responses.imageUrl;
+                        const shouldReplaceUrl = mjNotUseProxy === '0' && mjProxyImgUrl;
+                        let logMessage = '';
+                        if (shouldReplaceUrl) {
+                            const newUrlBase = new URL(mjProxyImgUrl);
+                            const parsedUrl = new URL(responses.imageUrl);
+                            parsedUrl.protocol = newUrlBase.protocol;
+                            parsedUrl.hostname = newUrlBase.hostname;
+                            parsedUrl.port = newUrlBase.port ? newUrlBase.port : '';
+                            processedUrl = parsedUrl.toString();
+                            logMessage = `使用代理替换后的 URL: ${processedUrl}`;
+                            common_1.Logger.log(logMessage, 'MidjourneyService');
+                        }
+                        if (mjNotSaveImg !== '1') {
+                            try {
+                                common_1.Logger.log(`------> 开始上传图片！！！`);
+                                const now = new Date();
+                                const year = now.getFullYear();
+                                const month = String(now.getMonth() + 1).padStart(2, '0');
+                                const day = String(now.getDate()).padStart(2, '0');
+                                const currentDate = `${year}${month}/${day}`;
+                                processedUrl = await this.uploadService.uploadFileFromUrl({
+                                    url: processedUrl,
+                                    dir: `images/midjourney/${currentDate}`,
+                                });
+                                logMessage = `上传成功 URL: ${processedUrl}`;
+                            }
+                            catch (uploadError) {
+                                common_1.Logger.error('存储图片失败，使用原始/代理图片链接');
+                                logMessage = `存储图片失败，使用原始/代理图片链接 ${processedUrl}`;
+                            }
+                            common_1.Logger.log(logMessage, 'MidjourneyService');
+                        }
+                        else {
+                            logMessage = `不保存图片，使用 URL: ${processedUrl}`;
+                            common_1.Logger.log(logMessage, 'MidjourneyService');
+                        }
+                        result.fileInfo = processedUrl;
+                        result.drawId = responses.id;
+                        result.customId = JSON.stringify(responses.buttons);
+                        result.answer = `${prompt}\n${responses.finalPrompt || responses.properties.finalPrompt || ''}`;
+                        onSuccess(result);
+                        return;
+                    }
+                    result.progress = responses === null || responses === void 0 ? void 0 : responses.progress;
+                    result.answer = `当前绘制进度 ${responses === null || responses === void 0 ? void 0 : responses.progress}`;
+                    if (result.progress) {
+                        onDrawing(result);
+                    }
+                }
+                catch (error) {
+                    retryCount++;
+                    common_1.Logger.error(`轮询过程中发生错误: ${error}`, 'MidjourneyService');
+                }
+            }
+            common_1.Logger.error(`超过 ${startTime / 1000} s 未完成绘画, 请稍后再试! MidjourneyService`);
+            result.status = 4;
+            onFailure(result);
+            throw new common_1.HttpException('绘画超时，请稍后再试！', common_1.HttpStatus.BAD_REQUEST);
+        }
+        catch (error) {
+            common_1.Logger.error(`绘画失败: ${error} MidjourneyService`);
+            result.status = 5;
+            onFailure(result);
+        }
+    }
+};
+MidjourneyService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [upload_service_1.UploadService,
+        globalConfig_service_1.GlobalConfigService,
+        chatLog_service_1.ChatLogService])
+], MidjourneyService);
+exports.MidjourneyService = MidjourneyService;

dist/modules/ai/openaiChat.service.js

@@ -0,0 +1,145 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIChatService = void 0;
+const utils_1 = require("../../common/utils");
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const globalConfig_service_1 = require("../globalConfig/globalConfig.service");
+let OpenAIChatService = class OpenAIChatService {
+    constructor(globalConfigService) {
+        this.globalConfigService = globalConfigService;
+    }
+    async openAIChat(messagesHistory, inputs) {
+        const { onFailure, onProgress, apiKey, model, proxyUrl, modelName, timeout, chatId, isFileUpload, modelAvatar, temperature, abortController, } = inputs;
+        let result = {
+            text: '',
+            model: '',
+            modelName: modelName,
+            chatId: chatId,
+            answer: '',
+            errMsg: '',
+            modelAvatar: modelAvatar,
+        };
+        const options = {
+            method: 'POST',
+            url: `${proxyUrl}/v1/chat/completions`,
+            responseType: 'stream',
+            timeout: timeout,
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${apiKey}`,
+            },
+            data: Object.assign({ stream: true, model, temperature: temperature, messages: messagesHistory }, (isFileUpload === 2 && { max_tokens: 2048 })),
+        };
+        console.log('请求配置:', JSON.stringify(options, null, 2), 'ChatService');
+        try {
+            const response = await (0, axios_1.default)(options);
+            const stream = response.data;
+            let buffer = '';
+            await new Promise((resolve, reject) => {
+                stream.on('data', (chunk) => {
+                    buffer += chunk.toString();
+                    let lines = buffer.split('\n');
+                    buffer = lines.pop();
+                    lines.forEach((line) => {
+                        var _a, _b;
+                        if (line.trim() === 'data: [DONE]') {
+                            console.log('处理结束信号 [DONE]');
+                            resolve(result);
+                            return;
+                        }
+                        if (line.startsWith('data: ')) {
+                            try {
+                                const cleanedLine = line.slice(6).trim();
+                                if (cleanedLine) {
+                                    const jsonLine = JSON.parse(cleanedLine);
+                                    const content = ((_b = (_a = jsonLine.choices[0]) === null || _a === void 0 ? void 0 : _a.delta) === null || _b === void 0 ? void 0 : _b.content) || '';
+                                    result.answer += content;
+                                    onProgress === null || onProgress === void 0 ? void 0 : onProgress({ text: content, answer: result.answer });
+                                }
+                            }
+                            catch (error) {
+                                console.error('Error parsing line:', line, error);
+                            }
+                        }
+                    });
+                });
+                stream.on('end', () => {
+                    resolve(result);
+                });
+                stream.on('error', (error) => {
+                    reject(error);
+                });
+                abortController.signal.addEventListener('abort', () => {
+                    resolve(result);
+                });
+            });
+            return result;
+        }
+        catch (error) {
+            result.errMsg = (0, utils_1.handleError)(error);
+            common_1.Logger.error(result.errMsg);
+            onFailure(result);
+            return result;
+        }
+    }
+    async chatFree(prompt, systemMessage, messagesHistory) {
+        const { openaiBaseUrl = '', openaiBaseKey = '', openaiBaseModel, } = await this.globalConfigService.getConfigs([
+            'openaiBaseKey',
+            'openaiBaseUrl',
+            'openaiBaseModel',
+        ]);
+        const key = openaiBaseKey;
+        const proxyUrl = openaiBaseUrl;
+        let requestData = [];
+        if (systemMessage) {
+            requestData.push({
+                role: 'system',
+                content: systemMessage,
+            });
+        }
+        if (messagesHistory && messagesHistory.length > 0) {
+            requestData = requestData.concat(messagesHistory);
+        }
+        else {
+            requestData.push({
+                role: 'user',
+                content: prompt,
+            });
+        }
+        const options = {
+            method: 'POST',
+            url: `${proxyUrl}/v1/chat/completions`,
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${key}`,
+            },
+            data: {
+                model: openaiBaseModel || 'gpt-3.5-turbo-0125',
+                messages: requestData,
+            },
+        };
+        try {
+            const response = await (0, axios_1.default)(options);
+            common_1.Logger.log(`全局模型调用成功, 返回结果: ${response === null || response === void 0 ? void 0 : response.data.choices[0].message.content}`, 'ChatService');
+            return response === null || response === void 0 ? void 0 : response.data.choices[0].message.content;
+        }
+        catch (error) {
+            console.log('error: ', error);
+        }
+    }
+};
+OpenAIChatService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [globalConfig_service_1.GlobalConfigService])
+], OpenAIChatService);
+exports.OpenAIChatService = OpenAIChatService;

dist/modules/ai/openaiDraw.service.js

@@ -0,0 +1,121 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var OpenAIDrawService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIDrawService = void 0;
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+const globalConfig_service_1 = require("../globalConfig/globalConfig.service");
+const upload_service_1 = require("../upload/upload.service");
+const openaiChat_service_1 = require("./openaiChat.service");
+let OpenAIDrawService = OpenAIDrawService_1 = class OpenAIDrawService {
+    constructor(uploadService, globalConfigService, chatLogService, openAIChatService) {
+        this.uploadService = uploadService;
+        this.globalConfigService = globalConfigService;
+        this.chatLogService = chatLogService;
+        this.openAIChatService = openAIChatService;
+        this.logger = new common_1.Logger(OpenAIDrawService_1.name);
+    }
+    async dalleDraw(inputs, messagesHistory) {
+        var _a, _b, _c, _d;
+        common_1.Logger.log('开始提交 Dalle 绘图任务 ', 'DrawService');
+        const { apiKey, model, proxyUrl, prompt, extraParam, timeout, onSuccess, onFailure, } = inputs;
+        const size = (extraParam === null || extraParam === void 0 ? void 0 : extraParam.size) || '1024x1024';
+        let result = { answer: '', fileInfo: '', status: 2 };
+        try {
+            const options = {
+                method: 'POST',
+                url: `${proxyUrl}/v1/images/generations`,
+                timeout: timeout,
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${apiKey}`,
+                },
+                data: {
+                    model: model,
+                    prompt: prompt,
+                    size,
+                },
+            };
+            const response = await (0, axios_1.default)(options);
+            const url = response.data.data[0].url;
+            try {
+                common_1.Logger.log(`------> 开始上传图片！！！`, 'DrawService');
+                const now = new Date();
+                const year = now.getFullYear();
+                const month = String(now.getMonth() + 1).padStart(2, '0');
+                const day = String(now.getDate()).padStart(2, '0');
+                const currentDate = `${year}${month}/${day}`;
+                result.fileInfo = await this.uploadService.uploadFileFromUrl({
+                    url: url,
+                    dir: `images/dalle/${currentDate}`,
+                });
+                common_1.Logger.log(`图片上传成功，URL: ${result.fileInfo}`, 'DrawService');
+            }
+            catch (error) {
+                common_1.Logger.error(`上传图片过程中出现错误: ${error}`, 'DrawService');
+            }
+            let revised_prompt_cn;
+            try {
+                revised_prompt_cn = await this.openAIChatService.chatFree(`根据提示词{${response.data.data[0].revised_prompt}}, 模拟AI绘画机器人的语气，用中文回复，告诉用户已经画好了`);
+            }
+            catch (error) {
+                revised_prompt_cn = `${prompt} 绘制成功`;
+                common_1.Logger.error('翻译失败: ', error);
+            }
+            result.answer = revised_prompt_cn;
+            result.status = 3;
+            onSuccess(result);
+            return;
+        }
+        catch (error) {
+            result.status = 5;
+            onFailure(result);
+            const status = ((_a = error === null || error === void 0 ? void 0 : error.response) === null || _a === void 0 ? void 0 : _a.status) || 500;
+            console.log('draw error: ', JSON.stringify(error), status);
+            const message = (_d = (_c = (_b = error === null || error === void 0 ? void 0 : error.response) === null || _b === void 0 ? void 0 : _b.data) === null || _c === void 0 ? void 0 : _c.error) === null || _d === void 0 ? void 0 : _d.message;
+            if (status === 429) {
+                result.text = '当前请求已过载、请稍等会儿再试试吧！';
+                return result;
+            }
+            if (status === 400 &&
+                message.includes('This request has been blocked by our content filters')) {
+                result.text = '您的请求已被系统拒绝。您的提示可能存在一些非法的文本。';
+                return result;
+            }
+            if (status === 400 &&
+                message.includes('Billing hard limit has been reached')) {
+                result.text =
+                    '当前模型key已被封禁、已冻结当前调用Key、尝试重新对话试试吧！';
+                return result;
+            }
+            if (status === 500) {
+                result.text = '绘制图片失败，请检查你的提示词是否有非法描述！';
+                return result;
+            }
+            if (status === 401) {
+                result.text = '绘制图片失败，此次绘画被拒绝了！';
+                return result;
+            }
+            result.text = '绘制图片失败，请稍后试试吧！';
+            return result;
+        }
+    }
+};
+OpenAIDrawService = OpenAIDrawService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [upload_service_1.UploadService,
+        globalConfig_service_1.GlobalConfigService,
+        chatLog_service_1.ChatLogService,
+        openaiChat_service_1.OpenAIChatService])
+], OpenAIDrawService);
+exports.OpenAIDrawService = OpenAIDrawService;

dist/modules/ai/stableDiffusion.service.js

@@ -0,0 +1,91 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var StableDiffusionService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StableDiffusionService = void 0;
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+const globalConfig_service_1 = require("../globalConfig/globalConfig.service");
+const upload_service_1 = require("../upload/upload.service");
+let StableDiffusionService = StableDiffusionService_1 = class StableDiffusionService {
+    constructor(uploadService, globalConfigService, chatLogService) {
+        this.uploadService = uploadService;
+        this.globalConfigService = globalConfigService;
+        this.chatLogService = chatLogService;
+        this.logger = new common_1.Logger(StableDiffusionService_1.name);
+    }
+    async sdxl(messagesHistory, inputs) {
+        const { onGenerate, onSuccess, onFailure, apiKey, model, proxyUrl, modelName, timeout, chatId, isFileUpload, prompt, } = inputs;
+        let result = {
+            answer: '',
+            model: model,
+            modelName: modelName,
+            chatId: chatId,
+            fileInfo: '',
+            status: 2,
+        };
+        console.log('开始处理', { model, modelName, prompt });
+        const options = {
+            method: 'POST',
+            url: `${proxyUrl}/v1/chat/completions`,
+            timeout: timeout,
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${apiKey}`,
+            },
+            data: {
+                model,
+                messages: [{ role: 'user', content: prompt }],
+            },
+        };
+        try {
+            const response = await (0, axios_1.default)(options);
+            console.log('API响应接收', response.data);
+            if (response.data.choices && response.data.choices.length > 0) {
+                const choice = response.data.choices[0];
+                const content = choice.message.content;
+                console.log('处理内容', content);
+                const regex = /\]\((https?:\/\/[^\)]+)\)/;
+                const match = content.match(regex);
+                if (match && match[1]) {
+                    result.fileInfo = match[1];
+                    console.log('找到链接', match[1]);
+                }
+                else {
+                    console.log('没有找到链接');
+                }
+                let revised_prompt_cn;
+                result.answer = `${prompt} 绘制成功`;
+                if (result.fileInfo) {
+                    onSuccess(result);
+                    return;
+                }
+                else {
+                    onFailure('No link found.');
+                }
+            }
+            else {
+                onFailure('No choices returned.');
+            }
+        }
+        catch (error) {
+            common_1.Logger.error('服务器错误，请求失败：', error);
+        }
+    }
+};
+StableDiffusionService = StableDiffusionService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [upload_service_1.UploadService,
+        globalConfig_service_1.GlobalConfigService,
+        chatLog_service_1.ChatLogService])
+], StableDiffusionService);
+exports.StableDiffusionService = StableDiffusionService;

dist/modules/ai/suno.service.js

@@ -0,0 +1,247 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SunoService = void 0;
+const common_1 = require("@nestjs/common");
+const axios_1 = require("axios");
+const chatLog_service_1 = require("../chatLog/chatLog.service");
+let SunoService = class SunoService {
+    constructor(chatLogService) {
+        this.chatLogService = chatLogService;
+    }
+    async suno(inputs) {
+        var _a, _b, _c;
+        const { apiKey, proxyUrl, action, prompt, timeout, assistantLogId, taskData, extraParam, } = inputs;
+        let result = {
+            text: '',
+            fileInfo: '',
+            taskId: '',
+            taskData: '',
+            status: 2,
+        };
+        common_1.Logger.log('开始生成音乐', 'SunoService');
+        let response = null;
+        let url = '';
+        let payloadJson = {};
+        const headers = { Authorization: `Bearer ${apiKey}` };
+        if (action === 'LYRICS') {
+            url = `${proxyUrl}/task/suno/v1/submit/lyrics`;
+            payloadJson = { prompt: prompt };
+        }
+        if (action === 'MUSIC') {
+            url = `${proxyUrl}/task/suno/v1/submit/music`;
+            try {
+                payloadJson = JSON.parse(taskData);
+            }
+            catch (error) {
+                common_1.Logger.error(`解析taskData失败: ${error.message}`, 'SunoService');
+                throw new Error('taskData格式错误');
+            }
+        }
+        common_1.Logger.log(`正在准备发送请求到 ${url}，payload: ${JSON.stringify(payloadJson)}, headers: ${JSON.stringify(headers)}`);
+        try {
+            response = await axios_1.default.post(url, payloadJson, { headers });
+        }
+        catch (error) {
+            common_1.Logger.error(`任务提交失败: ${error.message}`, 'SunoService');
+            throw new Error('任务提交失败');
+        }
+        if ((_a = response === null || response === void 0 ? void 0 : response.data) === null || _a === void 0 ? void 0 : _a.data) {
+            result.taskId = (_b = response === null || response === void 0 ? void 0 : response.data) === null || _b === void 0 ? void 0 : _b.data;
+            common_1.Logger.log(`任务提交成功, 任务ID: ${(_c = response === null || response === void 0 ? void 0 : response.data) === null || _c === void 0 ? void 0 : _c.data}`, 'SunoService');
+        }
+        else {
+            throw new Error('未能获取结果数据, 即将重试');
+        }
+        try {
+            await this.pollSunoMusicResult({
+                proxyUrl,
+                apiKey,
+                taskId: response.data.data,
+                timeout,
+                prompt,
+                action,
+                onSuccess: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            videoUrl: data === null || data === void 0 ? void 0 : data.videoUrl,
+                            audioUrl: data === null || data === void 0 ? void 0 : data.audioUrl,
+                            fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                            answer: (data === null || data === void 0 ? void 0 : data.answer) || prompt,
+                            progress: '100%',
+                            status: 3,
+                            taskId: data === null || data === void 0 ? void 0 : data.taskId,
+                            taskData: data === null || data === void 0 ? void 0 : data.taskData,
+                        });
+                        common_1.Logger.log('音乐任务已完成', 'SunoService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'SunoService');
+                    }
+                },
+                onAudioSuccess: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            videoUrl: data === null || data === void 0 ? void 0 : data.videoUrl,
+                            audioUrl: data === null || data === void 0 ? void 0 : data.audioUrl,
+                            fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                            answer: (data === null || data === void 0 ? void 0 : data.answer) || prompt,
+                            progress: data === null || data === void 0 ? void 0 : data.progress,
+                            status: data.status,
+                            taskId: data === null || data === void 0 ? void 0 : data.taskId,
+                            taskData: data === null || data === void 0 ? void 0 : data.taskData,
+                        });
+                        common_1.Logger.log('音频生成成功，等待视频生成...', 'SunoService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'SunoService');
+                    }
+                },
+                onGenerating: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            videoUrl: data === null || data === void 0 ? void 0 : data.videoUrl,
+                            audioUrl: data === null || data === void 0 ? void 0 : data.audioUrl,
+                            fileInfo: data === null || data === void 0 ? void 0 : data.fileInfo,
+                            answer: (data === null || data === void 0 ? void 0 : data.answer) || '音乐生成中...',
+                            progress: data === null || data === void 0 ? void 0 : data.progress,
+                            status: data.status,
+                        });
+                        common_1.Logger.log('音乐生成中...', 'SunoService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'SunoService');
+                    }
+                },
+                onFailure: async (data) => {
+                    try {
+                        await this.chatLogService.updateChatLog(assistantLogId, {
+                            answer: '音乐生成失败',
+                            status: data.status,
+                        });
+                        common_1.Logger.log('生成失败', 'SunoService');
+                    }
+                    catch (error) {
+                        common_1.Logger.error(`更新日志失败: ${error.message}`, 'SunoService');
+                    }
+                },
+            });
+        }
+        catch (error) {
+            common_1.Logger.error('查询生成结果时发生错误:', error.message, 'SunoService');
+            throw new Error('查询生成结果时发生错误');
+        }
+        return result;
+    }
+    async pollSunoMusicResult(inputs) {
+        const { proxyUrl, apiKey, taskId, timeout, onSuccess, onAudioSuccess, onFailure, onGenerating, action, } = inputs;
+        let result = {
+            videoUrl: '',
+            audioUrl: '',
+            fileInfo: '',
+            drawId: '',
+            taskData: '',
+            status: 2,
+            progress: 0,
+            answer: '',
+        };
+        const headers = { Authorization: `Bearer ${apiKey}` };
+        const url = `${proxyUrl}/task/suno/v1/fetch/${taskId}`;
+        const startTime = Date.now();
+        const POLL_INTERVAL = 5000;
+        let retryCount = 0;
+        try {
+            while (Date.now() - startTime < timeout) {
+                await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL));
+                try {
+                    const res = await axios_1.default.get(url, { headers });
+                    const responses = res.data.data;
+                    if (action === 'LYRICS') {
+                        if (responses.status === 'SUCCESS') {
+                            result.taskId = responses.data.id;
+                            result.taskData = JSON.stringify(responses.data);
+                            result.answer = responses.data.text;
+                            onSuccess(result);
+                            return;
+                        }
+                        result.progress = responses === null || responses === void 0 ? void 0 : responses.progress;
+                        result.answer = `歌词生成中`;
+                        if (result.progress) {
+                            onGenerating(result);
+                        }
+                    }
+                    if (action === 'MUSIC') {
+                        const data = responses.data;
+                        if (responses.data) {
+                            const data = responses.data;
+                            result.taskData = JSON.stringify(data);
+                            if (Array.isArray(data)) {
+                                const validAudioUrls = data
+                                    .map((item) => item.audio_url)
+                                    .filter((url) => url);
+                                const validVideoUrls = data
+                                    .map((item) => item.video_url)
+                                    .filter((url) => url);
+                                const validImageUrls = data
+                                    .map((item) => item.image_url)
+                                    .filter((url) => url);
+                                const titles = data.map((item) => item.title);
+                                const firstTitle = titles.length > 0 ? titles[0] : '音乐已生成';
+                                const audioUrls = validAudioUrls.join(',');
+                                const videoUrls = validVideoUrls.join(',');
+                                const imageUrls = validImageUrls.join(',');
+                                result.audioUrl = audioUrls;
+                                result.videoUrl = videoUrls;
+                                result.fileInfo = imageUrls;
+                                if (validAudioUrls.length === 2) {
+                                    result.status = 3;
+                                    result.answer = firstTitle;
+                                }
+                                else {
+                                    result.status = 2;
+                                    result.progress = responses === null || responses === void 0 ? void 0 : responses.progress;
+                                    result.answer = `当前生成进度 ${responses === null || responses === void 0 ? void 0 : responses.progress}`;
+                                }
+                                onAudioSuccess(result);
+                            }
+                        }
+                        if (responses.status === 'SUCCESS') {
+                            common_1.Logger.debug(`音乐生成成功: ${JSON.stringify(data)}`, 'SunoService');
+                            onSuccess(result);
+                            return;
+                        }
+                        if (result.progress && result.status === 2) {
+                            onGenerating(result);
+                        }
+                    }
+                }
+                catch (error) {
+                    retryCount++;
+                    common_1.Logger.error(`轮询失败，重试次数: ${retryCount}`, 'SunoService');
+                }
+            }
+            common_1.Logger.error('轮询超时，请稍后再试！', 'SunoService');
+            result.status = 4;
+            onFailure(result);
+            throw new Error('查询超时，请稍后再试！');
+        }
+        catch (error) {
+            common_1.Logger.error(`轮询过程中发生错误: ${error}`, 'SunoService');
+            result.status = 5;
+            onFailure(result);
+        }
+    }
+};
+SunoService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [chatLog_service_1.ChatLogService])
+], SunoService);
+exports.SunoService = SunoService;