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

src/pages/api/chat/chatGpt.ts

@@ -8,6 +8,7 @@ import { ChatItemType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { PassThrough } from 'stream';
+import { ModelList } from '@/constants/model';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -56,6 +57,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       });
     }
 
+    // 计算温度
+    const modelConstantsData = ModelList['openai'].find(
+      (item) => item.model === model.service.modelName
+    );
+    if (!modelConstantsData) {
+      throw new Error('模型异常');
+    }
+    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
+
     // 获取 chatAPI
     const chatAPI = getOpenAIApi(userApiKey);
     let startTime = Date.now();
@@ -63,8 +73,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const chatResponse = await chatAPI.createChatCompletion(
       {
         model: model.service.chatModel,
-        temperature: 1,
-        // max_tokens: model.security.contentMaxLen,
+        temperature: temperature,
+        max_tokens: modelConstantsData.maxToken,
         messages: formatPrompts,
         stream: true
       },

src/pages/api/chat/gpt3.ts

@@ -5,6 +5,7 @@ import { connectToDatabase } from '@/service/mongo';
 import { getOpenAIApi, authChat } from '@/service/utils/chat';
 import { ChatItemType } from '@/types/chat';
 import { httpsAgent } from '@/service/utils/tools';
+import { ModelList } from '@/constants/model';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -27,13 +28,22 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     // prompt处理
     const formatPrompt = prompt.map((item) => `${item.value}\n\n###\n\n`).join('');
 
+    // 计算温度
+    const modelConstantsData = ModelList['openai'].find(
+      (item) => item.model === model.service.modelName
+    );
+    if (!modelConstantsData) {
+      throw new Error('模型异常');
+    }
+    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
+
     // 发送请求
     const response = await chatAPI.createCompletion(
       {
         model: model.service.modelName,
         prompt: formatPrompt,
-        temperature: 0.5,
-        max_tokens: model.security.contentMaxLen,
+        temperature: temperature,
+        max_tokens: modelConstantsData.maxToken,
         top_p: 1,
         frequency_penalty: 0,
         presence_penalty: 0.6,

src/pages/api/chat/init.ts

@@ -47,6 +47,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         modelId: model._id,
         name: model.name,
         avatar: model.avatar,
+        intro: model.intro,
         secret: model.security,
         chatModel: model.service.chatModel,
         history: chat.content

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
       }
     );