mirror of
https://github.com/vastxie/99AI.git
synced 2025-09-18 01:36:37 +08:00
220 lines
9.2 KiB
JavaScript
220 lines
9.2 KiB
JavaScript
"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 __param = (this && this.__param) || function (paramIndex, decorator) {
|
||
return function (target, key) { decorator(target, key, paramIndex); }
|
||
};
|
||
var __rest = (this && this.__rest) || function (s, e) {
|
||
var t = {};
|
||
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
|
||
t[p] = s[p];
|
||
if (s != null && typeof Object.getOwnPropertySymbols === "function")
|
||
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
|
||
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
|
||
t[p[i]] = s[p[i]];
|
||
}
|
||
return t;
|
||
};
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.ChatGroupService = void 0;
|
||
const common_1 = require("@nestjs/common");
|
||
const typeorm_1 = require("@nestjs/typeorm");
|
||
const axios_1 = require("axios");
|
||
const pdf = require("pdf-parse");
|
||
const typeorm_2 = require("typeorm");
|
||
const app_entity_1 = require("../app/app.entity");
|
||
const models_service_1 = require("../models/models.service");
|
||
const chatGroup_entity_1 = require("./chatGroup.entity");
|
||
let ChatGroupService = class ChatGroupService {
|
||
constructor(chatGroupEntity, appEntity, modelsService) {
|
||
this.chatGroupEntity = chatGroupEntity;
|
||
this.appEntity = appEntity;
|
||
this.modelsService = modelsService;
|
||
}
|
||
async create(body, req) {
|
||
const { id } = req.user;
|
||
const { appId, modelConfig: bodyModelConfig, params } = body;
|
||
let modelConfig = bodyModelConfig || (await this.modelsService.getBaseConfig());
|
||
if (!modelConfig) {
|
||
throw new common_1.HttpException('管理员未配置任何AI模型、请先联系管理员开通聊天模型配置！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
modelConfig = JSON.parse(JSON.stringify(modelConfig));
|
||
const groupParams = { title: '新对话', userId: id, appId, params };
|
||
if (appId) {
|
||
const appInfo = await this.appEntity.findOne({ where: { id: appId } });
|
||
if (!appInfo) {
|
||
throw new common_1.HttpException('非法操作、您在使用一个不存在的应用！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
const { status, name, isFixedModel, isGPTs, appModel, coverImg } = appInfo;
|
||
Object.assign(modelConfig.modelInfo, {
|
||
isGPTs,
|
||
isFixedModel,
|
||
modelAvatar: coverImg,
|
||
modelName: name,
|
||
});
|
||
if (isGPTs === 1 || isFixedModel === 1) {
|
||
const appModelKey = await this.modelsService.getCurrentModelKeyInfo(isFixedModel === 1 ? appModel : 'gpts');
|
||
Object.assign(modelConfig.modelInfo, {
|
||
deductType: appModelKey.deductType,
|
||
deduct: appModelKey.deduct,
|
||
model: appModel,
|
||
isFileUpload: appModelKey.isFileUpload,
|
||
});
|
||
}
|
||
if (![1, 3, 4, 5].includes(status)) {
|
||
throw new common_1.HttpException('非法操作、您在使用一个未启用的应用！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
if (name) {
|
||
groupParams.title = name;
|
||
}
|
||
}
|
||
const newGroup = await this.chatGroupEntity.save(Object.assign(Object.assign({}, groupParams), { config: JSON.stringify(modelConfig) }));
|
||
return newGroup;
|
||
}
|
||
async query(req) {
|
||
try {
|
||
const { id } = req.user;
|
||
const params = { userId: id, isDelete: false };
|
||
const res = await this.chatGroupEntity.find({
|
||
where: params,
|
||
order: { isSticky: 'DESC', updatedAt: 'DESC' },
|
||
});
|
||
return res;
|
||
const appIds = res.filter((t) => t.appId).map((t) => t.appId);
|
||
const appInfos = await this.appEntity.find({ where: { id: (0, typeorm_2.In)(appIds) } });
|
||
return res.map((item) => {
|
||
var _a;
|
||
item.appLogo = (_a = appInfos.find((t) => t.id === item.appId)) === null || _a === void 0 ? void 0 : _a.coverImg;
|
||
return item;
|
||
});
|
||
}
|
||
catch (error) {
|
||
console.log('error: ', error);
|
||
}
|
||
}
|
||
async update(body, req) {
|
||
const { title, isSticky, groupId, config, fileUrl } = body;
|
||
const { id } = req.user;
|
||
const g = await this.chatGroupEntity.findOne({
|
||
where: { id: groupId, userId: id },
|
||
});
|
||
if (!g) {
|
||
throw new common_1.HttpException('请先选择一个对话或者新加一个对话再操作！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
const { appId } = g;
|
||
if (appId && !title) {
|
||
try {
|
||
const parseData = JSON.parse(config);
|
||
if (Number(parseData.keyType) !== 1) {
|
||
throw new common_1.HttpException('应用对话名称不能修改哟！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
}
|
||
catch (error) {
|
||
}
|
||
}
|
||
const data = {};
|
||
title && (data['title'] = title);
|
||
typeof isSticky !== 'undefined' && (data['isSticky'] = isSticky);
|
||
config && (data['config'] = config);
|
||
fileUrl && (data['fileUrl'] = fileUrl);
|
||
const u = await this.chatGroupEntity.update({ id: groupId }, data);
|
||
if (u.affected) {
|
||
if (fileUrl) {
|
||
this.handlePdfExtraction(fileUrl, groupId);
|
||
}
|
||
return true;
|
||
}
|
||
else {
|
||
throw new common_1.HttpException('更新对话失败！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
}
|
||
async handlePdfExtraction(fileUrl, groupId) {
|
||
try {
|
||
const pdfText = await this.extractPdfText(fileUrl);
|
||
await this.chatGroupEntity.update({ id: groupId }, { pdfTextContent: pdfText });
|
||
}
|
||
catch (error) {
|
||
console.error('PDF 读取失败:', error);
|
||
}
|
||
}
|
||
async extractPdfText(fileUrl) {
|
||
try {
|
||
const response = await axios_1.default.get(fileUrl, {
|
||
responseType: 'arraybuffer',
|
||
});
|
||
const dataBuffer = Buffer.from(response.data);
|
||
const pdfData = await pdf(dataBuffer);
|
||
return pdfData.text;
|
||
}
|
||
catch (error) {
|
||
console.error('PDF 解析失败:', error);
|
||
throw new Error('PDF 解析失败');
|
||
}
|
||
}
|
||
async updateTime(groupId) {
|
||
await this.chatGroupEntity.update(groupId, {
|
||
updatedAt: new Date(),
|
||
});
|
||
}
|
||
async del(body, req) {
|
||
const { groupId } = body;
|
||
const { id } = req.user;
|
||
const g = await this.chatGroupEntity.findOne({
|
||
where: { id: groupId, userId: id },
|
||
});
|
||
if (!g) {
|
||
throw new common_1.HttpException('非法操作、您在删除一个非法资源！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
const r = await this.chatGroupEntity.update({ id: groupId }, { isDelete: true });
|
||
if (r.affected) {
|
||
return '删除成功';
|
||
}
|
||
else {
|
||
throw new common_1.HttpException('删除失败！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
}
|
||
async delAll(req) {
|
||
const { id } = req.user;
|
||
const r = await this.chatGroupEntity.update({ userId: id, isSticky: false, isDelete: false }, { isDelete: true });
|
||
if (r.affected) {
|
||
return '删除成功';
|
||
}
|
||
else {
|
||
throw new common_1.HttpException('删除失败！', common_1.HttpStatus.BAD_REQUEST);
|
||
}
|
||
}
|
||
async getGroupInfoFromId(id) {
|
||
if (!id)
|
||
return;
|
||
const groupInfo = await this.chatGroupEntity.findOne({ where: { id } });
|
||
if (groupInfo) {
|
||
const { pdfTextContent } = groupInfo, rest = __rest(groupInfo, ["pdfTextContent"]);
|
||
return rest;
|
||
}
|
||
}
|
||
async getGroupPdfText(groupId) {
|
||
const groupInfo = await this.chatGroupEntity.findOne({
|
||
where: { id: groupId },
|
||
});
|
||
if (groupInfo) {
|
||
return groupInfo.pdfTextContent;
|
||
}
|
||
}
|
||
};
|
||
ChatGroupService = __decorate([
|
||
(0, common_1.Injectable)(),
|
||
__param(0, (0, typeorm_1.InjectRepository)(chatGroup_entity_1.ChatGroupEntity)),
|
||
__param(1, (0, typeorm_1.InjectRepository)(app_entity_1.AppEntity)),
|
||
__metadata("design:paramtypes", [typeorm_2.Repository,
|
||
typeorm_2.Repository,
|
||
models_service_1.ModelsService])
|
||
], ChatGroupService);
|
||
exports.ChatGroupService = ChatGroupService;
|