feat: 模型介绍和温度调整。完善聊天页提示

src/pages/api/model/create.ts

@@ -3,12 +3,21 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { ModelStatusEnum, OpenAiList } from '@/constants/model';
+import { ModelStatusEnum, ModelList, ChatModelNameEnum } from '@/constants/model';
+import type { ServiceName } from '@/types/mongoSchema';
 import { Model } from '@/service/models/model';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, serviceModelName, serviceModelCompany = 'openai' } = req.body;
+    const {
+      name,
+      serviceModelName,
+      serviceModelCompany = 'openai'
+    } = req.body as {
+      name: string;
+      serviceModelName: `${ChatModelNameEnum}`;
+      serviceModelCompany: ServiceName;
+    };
     const { authorization } = req.headers;
 
     if (!authorization) {
@@ -22,10 +31,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 凭证校验
     const userId = await authToken(authorization);
 
-    const modelItem = OpenAiList.find((item) => item.model === serviceModelName);
+    const modelItem = ModelList[serviceModelCompany].find(
+      (item) => item.model === serviceModelName
+    );
 
     if (!modelItem) {
-      throw new Error('模型错误');
+      throw new Error('模型不存在');
     }
 
     await connectToDatabase();
@@ -43,8 +54,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const authCount = await Model.countDocuments({
       userId
     });
-    if (authCount >= 10) {
-      throw new Error('上限 10 个模型');
+    if (authCount >= 20) {
+      throw new Error('上限 20 个模型');
     }
 
     // 创建模型

src/pages/api/model/detail.tsx

@@ -3,7 +3,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { Model } from '@/service/models/model';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -26,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 根据 userId 获取模型信息
-    const model: ModelType | null = await Model.findOne({
+    const model = await Model.findOne<ModelSchema>({
       userId,
       _id: modelId
     });

src/pages/api/model/getTrainings.ts

@@ -6,7 +6,7 @@ import formidable from 'formidable';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
 import { join } from 'path';
 import fs from 'fs';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import type { OpenAIApi } from 'openai';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { httpsAgent } from '@/service/utils/tools';

src/pages/api/model/putTrainStatus.ts

@@ -3,7 +3,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Training } from '@/service/mongo';
 import { getOpenAIApi } from '@/service/utils/chat';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { TrainingItemType } from '@/types/training';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { OpenAiTuneStatusEnum } from '@/service/constants/training';
@@ -26,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取模型
-    const model: ModelType | null = await Model.findById(modelId);
+    const model = await Model.findById<ModelSchema>(modelId);
 
     if (!model || model.status !== 'training') {
       throw new Error('模型不在训练中');

src/pages/api/model/train.ts

@@ -7,7 +7,7 @@ import formidable from 'formidable';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
 import { join } from 'path';
 import fs from 'fs';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import type { OpenAIApi } from 'openai';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { httpsAgent } from '@/service/utils/tools';
@@ -38,7 +38,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取模型的状态
-    const model: ModelType | null = await Model.findById(modelId);
+    const model = await Model.findById<ModelSchema>(modelId);
 
     if (!model || model.status !== 'running') {
       throw new Error('模型正忙');

src/pages/api/model/update.ts

@@ -8,7 +8,8 @@ import type { ModelUpdateParams } from '@/types/model';
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, service, security, systemPrompt } = req.body as ModelUpdateParams;
+    const { name, service, security, systemPrompt, intro, temperature } =
+      req.body as ModelUpdateParams;
     const { modelId } = req.query as { modelId: string };
     const { authorization } = req.headers;
 
@@ -33,8 +34,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       },
       {
         name,
-        service,
         systemPrompt,
+        intro,
+        temperature,
+        service,
         security
       }
     );