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

NineAI 2.4.2

Browse Source
This commit is contained in:
vastxie
2024-01-17 09:22:28 +08:00
commit bdc48207fc
636 changed files with 41864 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -0,0 +1,40 @@
"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
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;

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

@@ -0,0 +1,84 @@
"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: 'dsadgadaorjoqm', description: '模型key', required: true }),
__metadata("design:type", Object)
], SetModelDto.prototype, "key", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 'dasdasdk2n31323k', description: '模型秘钥', required: false }),
__metadata("design:type", String)
], SetModelDto.prototype, "secret", 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: 'key的权重' }),
__metadata("design:type", Number)
], SetModelDto.prototype, "keyWeight", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 4096, description: '模型支持的最大TOken数量', required: true }),
__metadata("design:type", Number)
], SetModelDto.prototype, "maxModelTokens", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1000, description: '模型支持的最大回复TOken数量', required: true }),
__metadata("design:type", Number)
], SetModelDto.prototype, "maxResponseTokens", 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: true, 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);
exports.SetModelDto = SetModelDto;

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

@@ -0,0 +1,72 @@
"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: 1, description: '温度' }),
__metadata("design:type", Number)
], SetModelTypeDto.prototype, "temperature", 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: '单次扣除金额', 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;

72
dist/modules/models/modelType.entity.js vendored Normal file
View File

@@ -0,0 +1,72 @@
"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.ModelsTypeEntity = void 0;
const typeorm_1 = require("typeorm");
const baseEntity_1 = require("../../common/entity/baseEntity");
let ModelsTypeEntity = class ModelsTypeEntity extends baseEntity_1.BaseEntity {
};
__decorate([
(0, typeorm_1.Column)({ comment: 'key模型类型 1: openai 2: 文心一言 3:清华智谱' }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "keyType", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型名称[给用户看的]' }),
__metadata("design:type", String)
], ModelsTypeEntity.prototype, "modelName", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '是否开放模型: 0:禁用 1：启用', default: 1 }),
__metadata("design:type", Boolean)
], ModelsTypeEntity.prototype, "status", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '绑定使用的模型是？最终调用的' }),
__metadata("design:type", String)
], ModelsTypeEntity.prototype, "model", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型温度0-2直接', default: 0.6 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "temperature", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型的使用次数', default: 0 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "useCount", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型总计使用的token数量', default: 0 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "useToken", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '单词调用扣除的次数', default: 1 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "deduct", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '扣除余额类型 1： 普通模型 2：高级模型', default: 1 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "deductType", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型设置允许用户使用的最大回复Token', default: 2048 }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "maxResponseTokens", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '限制用户上下文可选最大轮次数', nullable: true }),
__metadata("design:type", Number)
], ModelsTypeEntity.prototype, "maxRounds", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '是否为绘画模型Dall-E3', default: 0 }),
__metadata("design:type", Boolean)
], ModelsTypeEntity.prototype, "isDallE3", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '是否为特殊模型、可以提供联想翻译、思维导图等特殊操作', default: 0 }),
__metadata("design:type", Boolean)
], ModelsTypeEntity.prototype, "isUseTool", void 0);
ModelsTypeEntity = __decorate([
(0, typeorm_1.Entity)({ name: 'models_type' })
], ModelsTypeEntity);
exports.ModelsTypeEntity = ModelsTypeEntity;

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

@@ -0,0 +1,131 @@
"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();
}
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)('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;

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

@@ -0,0 +1,100 @@
"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 typeorm_1 = require("typeorm");
const baseEntity_1 = require("../../common/entity/baseEntity");
let ModelsEntity = class ModelsEntity extends baseEntity_1.BaseEntity {
};
__decorate([
(0, typeorm_1.Column)({ comment: 'key模型类型 1: openai 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: '模型的key' }),
__metadata("design:type", String)
], ModelsEntity.prototype, "key", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型的secret', default: null }),
__metadata("design:type", String)
], ModelsEntity.prototype, "secret", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '部分模型的调用token', default: null }),
__metadata("design:type", String)
], ModelsEntity.prototype, "accessToken", 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: '绑定的模型是？' }),
__metadata("design:type", String)
], ModelsEntity.prototype, "model", 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: 1 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "keyWeight", 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: '模型支持的最大Token', default: 1000 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "maxModelTokens", void 0);
__decorate([
(0, typeorm_1.Column)({ comment: '模型设置的最大回复Token', default: 4096 }),
__metadata("design:type", Number)
], ModelsEntity.prototype, "maxResponseTokens", 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: '当前模型的超时时间单位s', default: 200 }),
__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: '备注信息', nullable: true }),
__metadata("design:type", String)
], ModelsEntity.prototype, "remark", 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: '是否是绘画key: 0:不是 1：是', default: 0 }),
__metadata("design:type", Boolean)
], ModelsEntity.prototype, "isDraw", void 0);
ModelsEntity = __decorate([
(0, typeorm_1.Entity)({ name: 'models' })
], ModelsEntity);
exports.ModelsEntity = ModelsEntity;

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

@@ -0,0 +1,27 @@
"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");
const modelType_entity_1 = require("./modelType.entity");
let ModelsModule = class ModelsModule {
};
ModelsModule = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({
imports: [typeorm_1.TypeOrmModule.forFeature([models_entity_1.ModelsEntity, modelType_entity_1.ModelsTypeEntity])],
controllers: [models_controller_1.ModelsController],
providers: [models_service_1.ModelsService],
exports: [models_service_1.ModelsService]
})
], ModelsModule);
exports.ModelsModule = ModelsModule;

245
dist/modules/models/models.service.js vendored Normal file
View File

@@ -0,0 +1,245 @@
"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 common_1 = require("@nestjs/common");
const typeorm_1 = require("@nestjs/typeorm");
const typeorm_2 = require("typeorm");
const models_entity_1 = require("./models.entity");
const status_constant_1 = require("../../common/constants/status.constant");
const baidu_1 = require("../chatgpt/baidu");
const utils_1 = require("../../common/utils");
const modelType_entity_1 = require("./modelType.entity");
let ModelsService = class ModelsService {
constructor(modelsEntity, modelsTypeEntity) {
this.modelsEntity = modelsEntity;
this.modelsTypeEntity = modelsTypeEntity;
this.modelTypes = [];
this.modelMaps = {};
this.keyList = {};
this.keyPoolMap = {};
this.keyPoolIndexMap = {};
}
async onModuleInit() {
await this.initCalcKey();
this.refreshBaiduAccesstoken();
}
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, keyWeight } = keyDetail;
if (!this.keyPoolMap[model])
this.keyPoolMap[model] = [];
for (let index = 0; index < keyWeight; index++) {
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) {
if (!this.keyPoolMap[model]) {
throw new common_1.HttpException('当前调用模型已经被移除、请重新选择模型！', common_1.HttpStatus.BAD_REQUEST);
}
this.keyPoolIndexMap[model]++;
const index = this.keyPoolIndexMap[model];
if (index >= this.keyPoolMap[model].length)
this.keyPoolIndexMap[model] = 0;
const key = this.keyPoolMap[model][this.keyPoolIndexMap[model]];
return key;
}
async getBaseConfig(appId) {
if (!this.modelTypes.length || !Object.keys(this.modelMaps).length)
return;
const modelTypeInfo = appId ? this.modelTypes.find(item => Number(item.val) === 1) : this.modelTypes[0];
if (!modelTypeInfo)
return;
const { keyType, modelName, model, maxModelTokens, maxResponseTokens, deductType, deduct, maxRounds } = this.modelMaps[modelTypeInfo.val][0];
return {
modelTypeInfo,
modelInfo: { keyType, modelName, model, maxModelTokens, maxResponseTokens, topN: 0.8, systemMessage: '', deductType, deduct, maxRounds, rounds: 8 }
};
}
async setModel(params) {
try {
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();
if (keyType === 2) {
this.refreshBaiduAccesstoken();
}
return res;
}
else {
const data = key.map(k => {
try {
const data = JSON.parse(JSON.stringify(params));
data.key = k;
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,
skip: (page - 1) * size,
take: size,
});
if (role !== 'super') {
rows.forEach(item => {
item.key && (item.key = (0, utils_1.hideString)(item.key));
item.secret && (item.secret = (0, utils_1.hideString)(item.secret));
});
}
return { rows, count };
}
async modelsList() {
const cloneModelMaps = JSON.parse(JSON.stringify(this.modelMaps));
Object.keys(cloneModelMaps).forEach(key => {
cloneModelMaps[key] = Array.from(cloneModelMaps[key].map(t => {
const { modelName, model, deduct, deductType, maxRounds } = t;
return { modelName, model, deduct, deductType, maxRounds };
}).reduce((map, obj) => map.set(obj.modelName, obj), new Map()).values());
});
return {
modelTypeList: this.modelTypes,
modelMaps: cloneModelMaps
};
}
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 refreshBaiduAccesstoken() {
const allKeys = await this.modelsEntity.find({ where: { keyType: 2 } });
const keysMap = {};
allKeys.forEach(keyInfo => {
const { key, secret } = keyInfo;
if (!keysMap.key) {
keysMap[key] = [{ keyInfo }];
}
else {
keysMap[key].push(keyInfo);
}
});
Object.keys(keysMap).forEach(async (key) => {
const { secret, id } = keysMap[key][0]['keyInfo'];
const accessToken = await (0, baidu_1.getAccessToken)(key, secret);
await this.modelsEntity.update({ key }, { accessToken });
});
setTimeout(() => {
this.initCalcKey();
}, 1000);
}
async getRandomDrawKey() {
const drawkeys = await this.modelsEntity.find({ where: { isDraw: true, status: true } });
if (!drawkeys.length) {
throw new common_1.HttpException('当前未指定特殊模型KEY、前往后台模型池设置吧！', common_1.HttpStatus.BAD_REQUEST);
}
return (0, utils_1.getRandomItemFromArray)(drawkeys);
}
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)),
__param(1, (0, typeorm_1.InjectRepository)(modelType_entity_1.ModelsTypeEntity)),
__metadata("design:paramtypes", [typeorm_2.Repository,
typeorm_2.Repository])
], ModelsService);
exports.ModelsService = ModelsService;