mirror of
https://github.com/vastxie/99AI.git
synced 2025-11-11 11:13:43 +08:00
v4.1.0
This commit is contained in:
vastxie
264
AIWebQuickDeploy/dist/modules/models/models.service.js
vendored
Normal file
264
AIWebQuickDeploy/dist/modules/models/models.service.js
vendored
Normal file
@@ -0,0 +1,264 @@
|
||||
"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); }
|
||||
};
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.ModelsService = void 0;
|
||||
const status_constant_1 = require("../../common/constants/status.constant");
|
||||
const utils_1 = require("../../common/utils");
|
||||
const common_1 = require("@nestjs/common");
|
||||
const typeorm_1 = require("@nestjs/typeorm");
|
||||
const typeorm_2 = require("typeorm");
|
||||
const models_entity_1 = require("./models.entity");
|
||||
let ModelsService = class ModelsService {
|
||||
constructor(modelsEntity) {
|
||||
this.modelsEntity = modelsEntity;
|
||||
this.modelTypes = [];
|
||||
this.modelMaps = {};
|
||||
this.keyList = {};
|
||||
this.keyPoolMap = {};
|
||||
this.keyPoolIndexMap = {};
|
||||
}
|
||||
async onModuleInit() {
|
||||
await this.initCalcKey();
|
||||
}
|
||||
async initCalcKey() {
|
||||
this.keyPoolMap = {};
|
||||
this.keyPoolIndexMap = {};
|
||||
this.keyList = {};
|
||||
this.modelMaps = {};
|
||||
this.modelTypes = [];
|
||||
const allKeys = await this.modelsEntity.find({ where: { status: true } });
|
||||
const keyTypes = allKeys.reduce((pre, cur) => {
|
||||
if (!pre[cur.keyType]) {
|
||||
pre[cur.keyType] = [cur];
|
||||
}
|
||||
else {
|
||||
pre[cur.keyType].push(cur);
|
||||
}
|
||||
return pre;
|
||||
}, {});
|
||||
this.modelTypes = Object.keys(keyTypes).map((keyType) => {
|
||||
return { label: status_constant_1.ModelsMapCn[keyType], val: keyType };
|
||||
});
|
||||
this.modelMaps = keyTypes;
|
||||
this.keyList = {};
|
||||
allKeys.forEach((keyDetail) => {
|
||||
const { keyType, model } = keyDetail;
|
||||
if (!this.keyPoolMap[model])
|
||||
this.keyPoolMap[model] = [];
|
||||
this.keyPoolMap[model].push(keyDetail);
|
||||
if (!this.keyPoolIndexMap[model])
|
||||
this.keyPoolIndexMap[model] = 0;
|
||||
if (!this.keyList[keyType])
|
||||
this.keyList[keyType] = {};
|
||||
if (!this.keyList[keyType][model])
|
||||
this.keyList[keyType][model] = [];
|
||||
this.keyList[keyType][model].push(keyDetail);
|
||||
});
|
||||
}
|
||||
async lockKey(keyId, remark, keyStatus = -1) {
|
||||
const res = await this.modelsEntity.update({ id: keyId }, { status: false, keyStatus, remark });
|
||||
common_1.Logger.error(`key: ${keyId} 欠费或被官方封禁导致不可用，已被系统自动锁定`);
|
||||
this.initCalcKey();
|
||||
}
|
||||
async getCurrentModelKeyInfo(model) {
|
||||
let modelKeyInfo = await this.modelsEntity.findOne({
|
||||
where: { model: model },
|
||||
});
|
||||
if (!modelKeyInfo) {
|
||||
return null;
|
||||
}
|
||||
return modelKeyInfo;
|
||||
}
|
||||
async getSpecialModelKeyInfo(modelPrefix) {
|
||||
const matchingModels = await this.modelsEntity.find({
|
||||
where: { model: (0, typeorm_2.Like)(`${modelPrefix}%`) },
|
||||
});
|
||||
if (matchingModels.length === 0) {
|
||||
throw new common_1.HttpException('未找到匹配的模型，请重新选择模型！', common_1.HttpStatus.BAD_REQUEST);
|
||||
}
|
||||
const firstMatchModel = matchingModels[0];
|
||||
const modifiedModelName = firstMatchModel.model.replace(modelPrefix, '');
|
||||
const modifiedModel = Object.assign(Object.assign({}, firstMatchModel), { model: modifiedModelName });
|
||||
return modifiedModel;
|
||||
}
|
||||
async getBaseConfig() {
|
||||
if (!this.modelTypes.length || !Object.keys(this.modelMaps).length)
|
||||
return;
|
||||
const { keyType, modelName, model, deductType, deduct, isFileUpload, modelAvatar, modelDescription, } = this.modelMaps[1][0];
|
||||
return {
|
||||
modelInfo: {
|
||||
keyType,
|
||||
modelName,
|
||||
model,
|
||||
deductType,
|
||||
deduct,
|
||||
isFileUpload,
|
||||
modelAvatar,
|
||||
modelDescription,
|
||||
},
|
||||
};
|
||||
}
|
||||
async setModel(params) {
|
||||
try {
|
||||
if (isNaN(params.timeout)) {
|
||||
params.timeout = null;
|
||||
}
|
||||
const { id } = params;
|
||||
params.status && (params.keyStatus = 1);
|
||||
if (id) {
|
||||
const res = await this.modelsEntity.update({ id }, params);
|
||||
await this.initCalcKey();
|
||||
return res.affected > 0;
|
||||
}
|
||||
else {
|
||||
const { keyType, key } = params;
|
||||
if (Number(keyType !== 1)) {
|
||||
const res = await this.modelsEntity.save(params);
|
||||
await this.initCalcKey();
|
||||
return res;
|
||||
}
|
||||
else {
|
||||
const data = key.map((k) => {
|
||||
try {
|
||||
const data = JSON.parse(JSON.stringify(params));
|
||||
data.key = k;
|
||||
if (isNaN(data.timeout)) {
|
||||
data.timeout = null;
|
||||
}
|
||||
return data;
|
||||
}
|
||||
catch (error) {
|
||||
console.log('parse error: ', error);
|
||||
}
|
||||
});
|
||||
const res = await this.modelsEntity.save(data);
|
||||
await this.initCalcKey();
|
||||
return res;
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (error) {
|
||||
console.log('error: ', error);
|
||||
}
|
||||
}
|
||||
async delModel({ id }) {
|
||||
if (!id) {
|
||||
throw new common_1.HttpException('缺失必要参数！', common_1.HttpStatus.BAD_REQUEST);
|
||||
}
|
||||
const m = await this.modelsEntity.findOne({ where: { id } });
|
||||
if (!m) {
|
||||
throw new common_1.HttpException('当前账号不存在！', common_1.HttpStatus.BAD_REQUEST);
|
||||
}
|
||||
const res = await this.modelsEntity.delete({ id });
|
||||
await this.initCalcKey();
|
||||
return res;
|
||||
}
|
||||
async queryModels(req, params) {
|
||||
const { role } = req.user;
|
||||
const { keyType, key, status, model, page = 1, size = 10 } = params;
|
||||
let where = {};
|
||||
keyType && (where.keyType = keyType);
|
||||
model && (where.model = model);
|
||||
status && (where.status = Number(status) === 1 ? true : false);
|
||||
key && (where.key = (0, typeorm_2.Like)(`%${key}%`));
|
||||
const [rows, count] = await this.modelsEntity.findAndCount({
|
||||
where: where,
|
||||
order: {
|
||||
modelOrder: 'ASC',
|
||||
},
|
||||
skip: (page - 1) * size,
|
||||
take: size,
|
||||
});
|
||||
if (role !== 'super') {
|
||||
rows.forEach((item) => {
|
||||
item.key && (item.key = (0, utils_1.hideString)(item.key));
|
||||
});
|
||||
}
|
||||
return { rows, count };
|
||||
}
|
||||
async modelsList() {
|
||||
const cloneModelMaps = JSON.parse(JSON.stringify(this.modelMaps));
|
||||
Object.keys(cloneModelMaps).forEach((key) => {
|
||||
cloneModelMaps[key] = cloneModelMaps[key]
|
||||
.filter((t) => t.keyStatus === 1)
|
||||
.sort((a, b) => a.modelOrder - b.modelOrder);
|
||||
cloneModelMaps[key] = Array.from(cloneModelMaps[key]
|
||||
.map((t) => {
|
||||
const { modelName, keyType, model, deduct, deductType, maxRounds, modelAvatar, isFileUpload, modelDescription, } = t;
|
||||
return {
|
||||
modelName,
|
||||
keyType,
|
||||
model,
|
||||
deduct,
|
||||
deductType,
|
||||
maxRounds,
|
||||
modelAvatar,
|
||||
isFileUpload,
|
||||
modelDescription,
|
||||
};
|
||||
})
|
||||
.reduce((map, obj) => map.set(obj.modelName, obj), new Map())
|
||||
.values());
|
||||
});
|
||||
return {
|
||||
modelTypeList: this.modelTypes,
|
||||
modelMaps: cloneModelMaps,
|
||||
};
|
||||
}
|
||||
async getMjInfo() {
|
||||
const modelInfo = await this.modelsEntity.findOne({
|
||||
where: { model: 'midjourney' },
|
||||
});
|
||||
if (modelInfo) {
|
||||
return {
|
||||
modelName: modelInfo.modelName,
|
||||
model: modelInfo.model,
|
||||
deduct: modelInfo.deduct,
|
||||
deductType: modelInfo.deductType,
|
||||
};
|
||||
}
|
||||
else {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
async saveUseLog(id, useToken) {
|
||||
await this.modelsEntity
|
||||
.createQueryBuilder()
|
||||
.update(models_entity_1.ModelsEntity)
|
||||
.set({
|
||||
useCount: () => 'useCount + 1',
|
||||
useToken: () => `useToken + ${useToken}`,
|
||||
})
|
||||
.where('id = :id', { id })
|
||||
.execute();
|
||||
}
|
||||
async getAllKey() {
|
||||
return await this.modelsEntity.find();
|
||||
}
|
||||
async queryModelType(params) {
|
||||
return 1;
|
||||
}
|
||||
async setModelType(params) {
|
||||
return 1;
|
||||
}
|
||||
async delModelType(params) {
|
||||
return 1;
|
||||
}
|
||||
};
|
||||
ModelsService = __decorate([
|
||||
(0, common_1.Injectable)(),
|
||||
__param(0, (0, typeorm_1.InjectRepository)(models_entity_1.ModelsEntity)),
|
||||
__metadata("design:paramtypes", [typeorm_2.Repository])
|
||||
], ModelsService);
|
||||
exports.ModelsService = ModelsService;
|
||||
Reference in New Issue
Block a user