4.6.5- CoreferenceResolution Module (#631)

packages/service/core/ai/audio/speech.ts

@@ -1,14 +1,12 @@
 import type { NextApiResponse } from 'next';
 import { getAIApi } from '../config';
-import { defaultAudioSpeechModels } from '../../../../global/core/ai/model';
-import { UserModelSchema } from '@fastgpt/global/support/user/type';
 
 export async function text2Speech({
   res,
   onSuccess,
   onError,
   input,
-  model = defaultAudioSpeechModels[0].model,
+  model,
   voice,
   speed = 1
 }: {

packages/service/core/ai/functions/queryExtension.ts

@@ -0,0 +1,59 @@
+import { replaceVariable } from '@fastgpt/global/common/string/tools';
+import { getAIApi } from '../config';
+
+const prompt = `
+您的任务是生成根据用户问题，从不同角度，生成两个不同版本的问题，以便可以从矢量数据库检索相关文档。例如：
+问题: FastGPT如何使用？
+OUTPUT: ["FastGPT使用教程。","怎么使用FastGPT？"]
+-------------------
+问题: FastGPT如何收费？
+OUTPUT: ["FastGPT收费标准。","FastGPT是如何计费的？"]
+-------------------
+问题: 怎么FastGPT部署？
+OUTPUT: ["FastGPT的部署方式。","如何部署FastGPT？"]
+-------------------
+问题 question: {{q}}
+OUTPUT: 
+`;
+
+export const searchQueryExtension = async ({ query, model }: { query: string; model: string }) => {
+  const ai = getAIApi(undefined, 480000);
+
+  const result = await ai.chat.completions.create({
+    model,
+    temperature: 0,
+    messages: [
+      {
+        role: 'user',
+        content: replaceVariable(prompt, { q: query })
+      }
+    ],
+    stream: false
+  });
+
+  const answer = result.choices?.[0]?.message?.content || '';
+  if (!answer) {
+    return {
+      queries: [query],
+      model,
+      inputTokens: 0,
+      responseTokens: 0
+    };
+  }
+
+  try {
+    return {
+      queries: JSON.parse(answer) as string[],
+      model,
+      inputTokens: result.usage?.prompt_tokens || 0,
+      responseTokens: result.usage?.completion_tokens || 0
+    };
+  } catch (error) {
+    return {
+      queries: [query],
+      model,
+      inputTokens: 0,
+      responseTokens: 0
+    };
+  }
+};

packages/service/core/plugin/controller.ts

@@ -3,7 +3,7 @@ import { FlowModuleTemplateType } from '@fastgpt/global/core/module/type';
 import { FlowNodeTypeEnum } from '@fastgpt/global/core/module/node/constant';
 import { plugin2ModuleIO } from '@fastgpt/global/core/module/utils';
 import { PluginSourceEnum } from '@fastgpt/global/core/plugin/constants';
-import type { PluginTemplateType } from '@fastgpt/global/core/plugin/type.d';
+import type { PluginRuntimeType, PluginTemplateType } from '@fastgpt/global/core/plugin/type.d';
 import { ModuleTemplateTypeEnum } from '@fastgpt/global/core/module/constants';
 
 /* 
@@ -41,6 +41,7 @@ const getPluginTemplateById = async (id: string): Promise<PluginTemplateType> =>
     if (!item) return Promise.reject('plugin not found');
     return {
       id: String(item._id),
+      teamId: String(item.teamId),
       name: item.name,
       avatar: item.avatar,
       intro: item.intro,
@@ -74,16 +75,14 @@ export async function getPluginPreviewModule({
 }
 
 /* run plugin time */
-export async function getPluginRuntimeById(id: string): Promise<PluginTemplateType> {
+export async function getPluginRuntimeById(id: string): Promise<PluginRuntimeType> {
   const plugin = await getPluginTemplateById(id);
 
   return {
-    id: plugin.id,
-    source: plugin.source,
-    templateType: plugin.templateType,
+    teamId: plugin.teamId,
     name: plugin.name,
     avatar: plugin.avatar,
-    intro: plugin.intro,
+    showStatus: plugin.showStatus,
     modules: plugin.modules
   };
 }