xinyin025/99AI
1
1
Fork 0
mirror of https://github.com/vastxie/99AI.git synced 2025-11-10 18:53:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

v4.1.0

Browse Source
This commit is contained in:
vastxie
2025-03-04 17:36:53 +08:00
parent cf7dc1d1e7
commit 1e9b9f7ba4
649 changed files with 23183 additions and 1925 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
AIWebQuickDeploy/dist/modules/models/dto/queryModel.dto.js vendored Normal file
View File

@@ -0,0 +1,52 @@
"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.QueryModelDto = void 0;
const swagger_1 = require("@nestjs/swagger");
class QueryModelDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '页码', required: true }),
__metadata("design:type", Number)
], QueryModelDto.prototype, "page", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 10, description: '数量', required: true }),
__metadata("design:type", Number)
], QueryModelDto.prototype, "size", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '模型类型', required: true }),
__metadata("design:type", Number)
], QueryModelDto.prototype, "keyType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: 'dsadgadaorjoqm',
description: '模型key',
required: true,
}),
__metadata("design:type", String)
], QueryModelDto.prototype, "key", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '是否开启当前key对应的模型',
required: true,
}),
__metadata("design:type", Boolean)
], QueryModelDto.prototype, "status", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: 'gpt-3.5',
description: '当前key绑定的模型是多少 需要调用的模型',
required: true,
}),
__metadata("design:type", String)
], QueryModelDto.prototype, "model", void 0);
exports.QueryModelDto = QueryModelDto;

32
AIWebQuickDeploy/dist/modules/models/dto/queryModelType.dto.js vendored Normal file
View File

@@ -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);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.QueryModelTypeDto = void 0;
const swagger_1 = require("@nestjs/swagger");
class QueryModelTypeDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '页码', required: true }),
__metadata("design:type", Number)
], QueryModelTypeDto.prototype, "page", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 10, description: '数量', required: true }),
__metadata("design:type", Number)
], QueryModelTypeDto.prototype, "size", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '模型类型', required: true }),
__metadata("design:type", Number)
], QueryModelTypeDto.prototype, "keyType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '是否开启当前key对应的模型', required: true }),
__metadata("design:type", Boolean)
], QueryModelTypeDto.prototype, "status", void 0);
exports.QueryModelTypeDto = QueryModelTypeDto;

128
AIWebQuickDeploy/dist/modules/models/dto/setModel.dto.js vendored Normal file
View File

@@ -0,0 +1,128 @@
"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.SetModelDto = void 0;
const swagger_1 = require("@nestjs/swagger");
class SetModelDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: 'key id', required: false }),
__metadata("design:type", Number)
], SetModelDto.prototype, "id", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '模型类型', required: true }),
__metadata("design:type", Number)
], SetModelDto.prototype, "keyType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: '默认', description: '模型中文名称', required: true }),
__metadata("design:type", String)
], SetModelDto.prototype, "modelName", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 'sk-', description: '模型key', required: false }),
__metadata("design:type", Object)
], SetModelDto.prototype, "key", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '是否开启当前key对应的模型',
required: true,
}),
__metadata("design:type", Boolean)
], SetModelDto.prototype, "status", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: 'gpt-3.5',
description: '当前key绑定的模型是多少 需要调用的模型',
required: true,
}),
__metadata("design:type", String)
], SetModelDto.prototype, "model", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '模型排序' }),
__metadata("design:type", Number)
], SetModelDto.prototype, "modelOrder", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 'https://***.png', required: false }),
__metadata("design:type", String)
], SetModelDto.prototype, "modelAvatar", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: 4096,
description: '模型支持的最大TOken数量',
required: false,
}),
__metadata("design:type", Number)
], SetModelDto.prototype, "maxModelTokens", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '模型的代理地址',
required: false,
}),
__metadata("design:type", String)
], SetModelDto.prototype, "proxyUrl", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 300, description: '模型超时时间', required: false }),
__metadata("design:type", Number)
], SetModelDto.prototype, "timeout", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: 'key状态', required: false }),
__metadata("design:type", Number)
], SetModelDto.prototype, "keyStatus", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '扣费类型 1： 普通 2： 高级余额',
required: false,
}),
__metadata("design:type", Number)
], SetModelDto.prototype, "deductType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '单次扣除金额', required: false }),
__metadata("design:type", Number)
], SetModelDto.prototype, "deduct", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '最大上下文轮次',
required: false,
}),
__metadata("design:type", Number)
], SetModelDto.prototype, "maxRounds", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '是否设置为绘画Key',
required: false,
}),
__metadata("design:type", Boolean)
], SetModelDto.prototype, "isDraw", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '是否支持文件上传',
required: false,
}),
__metadata("design:type", Number)
], SetModelDto.prototype, "isFileUpload", void 0);
__decorate([
(0, swagger_1.ApiProperty)({
example: true,
description: '是否使用token计费',
required: false,
}),
__metadata("design:type", Boolean)
], SetModelDto.prototype, "isTokenBased", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: 'token计费比例', required: false }),
__metadata("design:type", Number)
], SetModelDto.prototype, "tokenFeeRatio", void 0);
exports.SetModelDto = SetModelDto;

76
AIWebQuickDeploy/dist/modules/models/dto/setModelType.dto.js vendored Normal file
View File

@@ -0,0 +1,76 @@
"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.SetModelTypeDto = void 0;
const swagger_1 = require("@nestjs/swagger");
class SetModelTypeDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: 'model id', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "id", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '模型类型', required: true }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "keyType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: '普通模型', description: '模型中文名称', required: true }),
__metadata("design:type", String)
], SetModelTypeDto.prototype, "modelName", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '是否开启当前key对应的模型', required: true }),
__metadata("design:type", Boolean)
], SetModelTypeDto.prototype, "status", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 'gpt-3.5', description: '当前key绑定的模型是多少 需要调用的模型', required: true }),
__metadata("design:type", String)
], SetModelTypeDto.prototype, "model", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 300, description: '模型超时时间', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "timeout", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '扣费类型 1： 普通 2： 高级余额', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "deductType", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '文件上传类型 0 : 不使用 1: ALL类型 2: 4V类型', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "isFileUpload", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '单次扣除金额', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "deduct", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '排序id 越大越靠前', default: 100 }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "order", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 4000, description: '模型允许用户使用的最大token设置过高意味着单次的上下文会很高控制模型上下文控制使用token数量', required: true }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "maxTokens", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1000, description: '模型支持的最大回复TOken数量', required: true }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "maxResponseTokens", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '最大上下文轮次', required: false }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "maxRounds", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '是否设置为Dall-E3绘画Key', required: false }),
__metadata("design:type", Boolean)
], SetModelTypeDto.prototype, "isDallE3", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: true, description: '是否设置为工具key', required: false }),
__metadata("design:type", Boolean)
], SetModelTypeDto.prototype, "isUseTool", void 0);
exports.SetModelTypeDto = SetModelTypeDto;

141
AIWebQuickDeploy/dist/modules/models/models.controller.js vendored Normal file
View File

@@ -0,0 +1,141 @@
"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.ModelsController = void 0;
const swagger_1 = require("@nestjs/swagger");
const models_service_1 = require("./models.service");
const common_1 = require("@nestjs/common");
const superAuth_guard_1 = require("../../common/auth/superAuth.guard");
const setModel_dto_1 = require("./dto/setModel.dto");
const queryModel_dto_1 = require("./dto/queryModel.dto");
const adminAuth_guard_1 = require("../../common/auth/adminAuth.guard");
const setModelType_dto_1 = require("./dto/setModelType.dto");
const queryModelType_dto_1 = require("./dto/queryModelType.dto");
let ModelsController = class ModelsController {
constructor(modelsService) {
this.modelsService = modelsService;
}
setModel(params) {
return this.modelsService.setModel(params);
}
delModel(params) {
return this.modelsService.delModel(params);
}
queryModels(req, params) {
return this.modelsService.queryModels(req, params);
}
modelsList() {
return this.modelsService.modelsList();
}
getMjInfo() {
return this.modelsService.getMjInfo();
}
baseConfig() {
return this.modelsService.getBaseConfig();
}
queryModelType(params) {
return this.modelsService.queryModelType(params);
}
setModelType(params) {
return this.modelsService.setModelType(params);
}
delModelType(params) {
return this.modelsService.delModelType(params);
}
};
__decorate([
(0, common_1.Post)('setModel'),
(0, swagger_1.ApiOperation)({ summary: '设置模型' }),
(0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [setModel_dto_1.SetModelDto]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "setModel", null);
__decorate([
(0, common_1.Post)('delModel'),
(0, swagger_1.ApiOperation)({ summary: '删除模型' }),
(0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "delModel", null);
__decorate([
(0, common_1.Get)('query'),
(0, swagger_1.ApiOperation)({ summary: '管理端查询模型列表' }),
(0, common_1.UseGuards)(adminAuth_guard_1.AdminAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Req)()),
__param(1, (0, common_1.Query)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Request, queryModel_dto_1.QueryModelDto]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "queryModels", null);
__decorate([
(0, common_1.Get)('list'),
(0, swagger_1.ApiOperation)({ summary: '客户端查询当前所有可以使用的模型' }),
__metadata("design:type", Function),
__metadata("design:paramtypes", []),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "modelsList", null);
__decorate([
(0, common_1.Get)('mjInfo'),
(0, swagger_1.ApiOperation)({ summary: '客户端查询当前所有可以使用的模型' }),
__metadata("design:type", Function),
__metadata("design:paramtypes", []),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "getMjInfo", null);
__decorate([
(0, common_1.Get)('baseConfig'),
(0, swagger_1.ApiOperation)({ summary: '客户端查询当前已经配置模型的基础配置' }),
__metadata("design:type", Function),
__metadata("design:paramtypes", []),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "baseConfig", null);
__decorate([
(0, common_1.Get)('queryModelType'),
(0, swagger_1.ApiOperation)({ summary: '查询模型类型' }),
__param(0, (0, common_1.Query)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [queryModelType_dto_1.QueryModelTypeDto]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "queryModelType", null);
__decorate([
(0, common_1.Post)('setModelType'),
(0, swagger_1.ApiOperation)({ summary: '创建修改模型类型' }),
(0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [setModelType_dto_1.SetModelTypeDto]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "setModelType", null);
__decorate([
(0, common_1.Post)('delModelType'),
(0, swagger_1.ApiOperation)({ summary: '删除模型类型' }),
(0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [Object]),
__metadata("design:returntype", void 0)
], ModelsController.prototype, "delModelType", null);
ModelsController = __decorate([
(0, common_1.Controller)('models'),
__metadata("design:paramtypes", [models_service_1.ModelsService])
], ModelsController);
exports.ModelsController = ModelsController;

118
AIWebQuickDeploy/dist/modules/models/models.entity.js vendored Normal file
View File

@@ -0,0 +1,118 @@
"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.ModelsEntity = void 0;
const baseEntity_1 = require("../../common/entity/baseEntity");
const typeorm_1 = require("typeorm");
let ModelsEntity = class ModelsEntity extends baseEntity_1.BaseEntity {
};
__decorate([
(0, typeorm_1.Column)({ comment: '模型类型 1: 普通对话 2: 绘画 3:高级对话' }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "keyType", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型名称' }),
__metadata("design:type", String)
], ModelsEntity.prototype, "modelName", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '绑定的模型是？' }),
__metadata("design:type", String)
], ModelsEntity.prototype, "model", void 0);
__decorate([
(0, typeorm_1.Column)({ length: 1024, comment: '模型头像', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "modelAvatar", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型排序', default: 1 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "modelOrder", void 0);
__decorate([
(0, typeorm_1.Column)({
comment: '模型上下文支持的最大Token',
default: 2000,
nullable: true,
}),
__metadata("design:type", Number)
], ModelsEntity.prototype, "maxModelTokens", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型上下文最大条数', nullable: true }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "maxRounds", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型上下文最大条数', nullable: true }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "timeout", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型单次调用扣除的次数', default: 1 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "deduct", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型扣除余额类型 1: 普通模型 2: 高级模型', default: 1 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "deductType", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '是否使用token计费: 0:不是 1: 是', default: 0 }),
__metadata("design:type", Boolean)
], ModelsEntity.prototype, "isTokenBased", void 0);
__decorate([
(0, typeorm_1.Column)({
comment: '是否支持文件上传: 0:不是 1: 附件链接格式 2: 4V格式',
default: 0,
}),
__metadata("design:type", Number)
], ModelsEntity.prototype, "isFileUpload", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: 'token计费比例', default: 0 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "tokenFeeRatio", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型附加信息', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "remark", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型的key', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "key", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '使用的状态: 0:禁用 1：启用', default: 1 }),
__metadata("design:type", Boolean)
], ModelsEntity.prototype, "status", void 0);
__decorate([
(0, typeorm_1.Column)({
comment: 'key的状态: 1:有效 -1:被封号 -2: 错误的秘钥 -3: 余额使用完了',
default: 1,
}),
__metadata("design:type", Number)
], ModelsEntity.prototype, "keyStatus", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: 'key的使用次数', default: 0 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "useCount", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: 'key的已经使用的token数量', default: 0 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "useToken", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '当前模型的代理地址', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "proxyUrl", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型频率限制 次/小时', default: 999 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "modelLimits", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型介绍', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "modelDescription", void 0);
ModelsEntity = __decorate([
(0, typeorm_1.Entity)({ name: 'models' })
], ModelsEntity);
exports.ModelsEntity = ModelsEntity;

26
AIWebQuickDeploy/dist/modules/models/models.module.js vendored Normal file
View File

@@ -0,0 +1,26 @@
"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;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ModelsModule = void 0;
const common_1 = require("@nestjs/common");
const models_controller_1 = require("./models.controller");
const models_service_1 = require("./models.service");
const typeorm_1 = require("@nestjs/typeorm");
const models_entity_1 = require("./models.entity");
let ModelsModule = class ModelsModule {
};
ModelsModule = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({
imports: [typeorm_1.TypeOrmModule.forFeature([models_entity_1.ModelsEntity])],
controllers: [models_controller_1.ModelsController],
providers: [models_service_1.ModelsService],
exports: [models_service_1.ModelsService]
})
], ModelsModule);
exports.ModelsModule = ModelsModule;

264
AIWebQuickDeploy/dist/modules/models/models.service.js vendored Normal file
View 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;