This commit is contained in:
vastxie
2024-04-20 22:15:08 +08:00
parent 133b89076e
commit 1c9e023c08
221 changed files with 6107 additions and 6478 deletions

View File

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

View File

@@ -98,6 +98,10 @@ __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);
ModelsEntity = __decorate([
(0, typeorm_1.Entity)({ name: 'models' })
], ModelsEntity);

View File

@@ -12,13 +12,12 @@ 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])],
imports: [typeorm_1.TypeOrmModule.forFeature([models_entity_1.ModelsEntity])],
controllers: [models_controller_1.ModelsController],
providers: [models_service_1.ModelsService],
exports: [models_service_1.ModelsService]

View File

@@ -13,17 +13,15 @@ var __param = (this && this.__param) || function (paramIndex, decorator) {
};
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");
const status_constant_1 = require("../../common/constants/status.constant");
const utils_1 = require("../../common/utils");
const modelType_entity_1 = require("./modelType.entity");
let ModelsService = class ModelsService {
constructor(modelsEntity, modelsTypeEntity) {
constructor(modelsEntity) {
this.modelsEntity = modelsEntity;
this.modelsTypeEntity = modelsTypeEntity;
this.modelTypes = [];
this.modelMaps = {};
this.keyList = {};
@@ -97,12 +95,8 @@ let ModelsService = class ModelsService {
async getBaseConfig(appId) {
if (!this.modelTypes.length || !Object.keys(this.modelMaps).length)
return;
const modelTypeInfo = this.modelTypes[0];
if (!modelTypeInfo)
return;
const { keyType, modelName, model, deductType, deduct, isFileUpload } = this.modelMaps[modelTypeInfo.val][0];
const { keyType, modelName, model, deductType, deduct, isFileUpload } = this.modelMaps[1][0];
return {
modelTypeInfo,
modelInfo: { keyType, modelName, model, deductType, deduct, isFileUpload }
};
}
@@ -238,8 +232,6 @@ let ModelsService = class ModelsService {
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])
__metadata("design:paramtypes", [typeorm_2.Repository])
], ModelsService);
exports.ModelsService = ModelsService;