v4.4.5-3 (#357)

projects/app/src/pages/api/chat/init.ts

@@ -6,7 +6,7 @@ import { authUser } from '@/service/utils/auth';
 import { ChatItemType } from '@/types/chat';
 import { authApp } from '@/service/utils/auth';
 import type { ChatSchema } from '@/types/mongoSchema';
-import { getGuideModules, getChatModelNameList } from '@/components/ChatBox/utils';
+import { getChatModelNameList, getGuideModule } from '@/components/ChatBox/utils';
 import { TaskResponseKeyEnum } from '@/constants/chat';
 
 /* 初始化我的聊天框，需要身份验证 */
@@ -81,7 +81,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         chatId,
         appId,
         app: {
-          ...getGuideModules(app.modules),
+          userGuideModule: getGuideModule(app.modules),
           chatModels: getChatModelNameList(app.modules),
           name: app.name,
           avatar: app.avatar,

projects/app/src/pages/api/core/ai/agent/createQuestionGuide.ts

@@ -0,0 +1,77 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { CreateQuestionGuideProps } from '@/api/core/ai/agent/type';
+import { getAIChatApi } from '@fastgpt/core/aiApi/config';
+import { Prompt_QuestionGuide } from '@/prompts/core/agent';
+import { pushQuestionGuideBill } from '@/service/common/bill/push';
+import { defaultQGModel } from '@/pages/api/system/getInitData';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    await connectToDatabase();
+    const { messages } = req.body as CreateQuestionGuideProps;
+    const { user } = await authUser({ req, authToken: true, authApiKey: true, authBalance: true });
+
+    if (!user) {
+      throw new Error('user not found');
+    }
+    const qgModel = global.qgModel || defaultQGModel;
+
+    const chatAPI = getAIChatApi(user.openaiAccount);
+
+    const { data } = await chatAPI.createChatCompletion({
+      model: qgModel.model,
+      temperature: 0,
+      max_tokens: 200,
+      messages: [
+        ...messages,
+        {
+          role: 'user',
+          content: Prompt_QuestionGuide
+        }
+      ],
+      stream: false
+    });
+
+    const answer = data.choices?.[0].message?.content || '';
+    const totalTokens = data.usage?.total_tokens || 0;
+
+    const start = answer.indexOf('[');
+    const end = answer.lastIndexOf(']');
+
+    if (start === -1 || end === -1) {
+      return jsonRes(res, {
+        data: []
+      });
+    }
+
+    const jsonStr = answer
+      .substring(start, end + 1)
+      .replace(/(\\n|\\)/g, '')
+      .replace(/  /g, '');
+
+    try {
+      jsonRes(res, {
+        data: JSON.parse(jsonStr)
+      });
+
+      pushQuestionGuideBill({
+        tokens: totalTokens,
+        userId: user._id
+      });
+
+      return;
+    } catch (error) {
+      return jsonRes(res, {
+        data: []
+      });
+    }
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

projects/app/src/pages/api/support/outLink/init.ts

@@ -4,7 +4,7 @@ import { connectToDatabase, OutLink, User } from '@/service/mongo';
 import type { InitShareChatResponse } from '@/api/response/chat';
 import { authApp } from '@/service/utils/auth';
 import { HUMAN_ICON } from '@/constants/chat';
-import { getChatModelNameList, getGuideModules } from '@/components/ChatBox/utils';
+import { getChatModelNameList, getGuideModule } from '@/components/ChatBox/utils';
 import { authShareChatInit } from '@/service/support/outLink/auth';
 
 /* init share chat window */
@@ -46,7 +46,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       data: {
         userAvatar: user?.avatar || HUMAN_ICON,
         app: {
-          ...getGuideModules(app.modules),
+          userGuideModule: getGuideModule(app.modules),
           chatModels: getChatModelNameList(app.modules),
           name: app.name,
           avatar: app.avatar,

projects/app/src/pages/api/system/getInitData.ts

@@ -97,6 +97,14 @@ export const defaultCQModel: FunctionModelItemType = {
   prompt: '',
   functionCall: true
 };
+export const defaultQGModel: FunctionModelItemType = {
+  model: 'gpt-3.5-turbo',
+  name: 'FastAI-4k',
+  maxToken: 4000,
+  price: 1.5,
+  prompt: '',
+  functionCall: false
+};
 
 const defaultVectorModels: VectorModelItemType[] = [
   {
@@ -130,6 +138,7 @@ export async function getInitConfig() {
     global.qaModel = res.QAModel || defaultQAModel;
     global.extractModel = res.ExtractModel || defaultExtractModel;
     global.cqModel = res.CQModel || defaultCQModel;
+    global.qgModel = res.QGModel || defaultQGModel;
     global.vectorModels = res.VectorModels || defaultVectorModels;
   } catch (error) {
     setDefaultData();
@@ -143,6 +152,9 @@ export function setDefaultData() {
   global.chatModels = defaultChatModels;
   global.qaModel = defaultQAModel;
   global.vectorModels = defaultVectorModels;
+  global.extractModel = defaultExtractModel;
+  global.cqModel = defaultCQModel;
+  global.qgModel = defaultQGModel;
 }
 
 export function getSystemVersion() {