v4.5 (#403)

projects/app/src/service/core/ai/model.ts

@@ -0,0 +1,68 @@
+import {
+  defaultChatModels,
+  defaultCQModels,
+  defaultExtractModels,
+  defaultQAModels,
+  defaultQGModels,
+  defaultVectorModels
+} from '@/constants/model';
+
+export const getChatModel = (model?: string) => {
+  return (
+    (global.chatModels || defaultChatModels).find((item) => item.model === model) ||
+    defaultChatModels[0]
+  );
+};
+export const getQAModel = (model?: string) => {
+  return (
+    (global.qaModels || defaultQAModels).find((item) => item.model === model) ||
+    global.qaModels?.[0] ||
+    defaultQAModels[0]
+  );
+};
+export const getCQModel = (model?: string) => {
+  return (
+    (global.cqModels || defaultCQModels).find((item) => item.model === model) ||
+    global.cqModels?.[0] ||
+    defaultCQModels[0]
+  );
+};
+export const getExtractModel = (model?: string) => {
+  return (
+    (global.extractModels || defaultExtractModels).find((item) => item.model === model) ||
+    global.extractModels?.[0] ||
+    defaultExtractModels[0]
+  );
+};
+export const getQGModel = (model?: string) => {
+  return (
+    (global.qgModels || defaultQGModels).find((item) => item.model === model) ||
+    global.qgModels?.[0] ||
+    defaultQGModels[0]
+  );
+};
+
+export const getVectorModel = (model?: string) => {
+  return (
+    global.vectorModels.find((item) => item.model === model) ||
+    global.vectorModels?.[0] ||
+    defaultVectorModels[0]
+  );
+};
+
+export enum ModelTypeEnum {
+  chat = 'chat',
+  qa = 'qa',
+  cq = 'cq',
+  extract = 'extract',
+  qg = 'qg',
+  vector = 'vector'
+}
+export const getModelMap = {
+  [ModelTypeEnum.chat]: getChatModel,
+  [ModelTypeEnum.qa]: getQAModel,
+  [ModelTypeEnum.cq]: getCQModel,
+  [ModelTypeEnum.extract]: getExtractModel,
+  [ModelTypeEnum.qg]: getQGModel,
+  [ModelTypeEnum.vector]: getVectorModel
+};

projects/app/src/service/core/app/module.ts

@@ -0,0 +1,12 @@
+import { FlowModuleTypeEnum } from '@/constants/flow';
+import { AppModuleItemType } from '@/types/app';
+
+export const getChatModelNameListByModules = (modules: AppModuleItemType[]): string[] => {
+  const chatModules = modules.filter((item) => item.flowType === FlowModuleTypeEnum.chatNode);
+  return chatModules
+    .map((item) => {
+      const model = item.inputs.find((input) => input.key === 'model')?.value;
+      return global.chatModels.find((item) => item.model === model)?.name || '';
+    })
+    .filter((item) => item);
+};