add feishuKonwledge dataset

Author: dreamer6680

packages/global/core/dataset/api.d.ts

@@ -26,7 +26,8 @@ export type DatasetUpdateBody = {
   defaultPermission?: DatasetSchemaType['defaultPermission'];
   apiServer?: DatasetSchemaType['apiServer'];
   yuqueServer?: DatasetSchemaType['yuqueServer'];
-  feishuServer?: DatasetSchemaType['feishuServer'];
+  feishuShareServer?: DatasetSchemaType['feishuShareServer'];
+  feishuKnowledgeServer?: DatasetSchemaType['feishuKnowledgeServer'];
   chunkSettings?: DatasetSchemaType['chunkSettings'];
 
   // sync schedule

packages/global/core/dataset/apiDataset.d.ts

@@ -17,11 +17,17 @@ export type APIFileServer = {
   authorization?: string;
   basePath?: string;
 };
-export type FeishuServer = {
+export type FeishuShareServer = {
   appId: string;
   appSecret?: string;
   folderToken: string;
 };
+export type FeishuKnowledgeServer = {
+  appId: string;
+  appSecret: string;
+  basePath?: string;
+};
+
 export type YuqueServer = {
   userId: string;
   token?: string;

packages/global/core/dataset/constants.ts

@@ -7,7 +7,8 @@ export enum DatasetTypeEnum {
   websiteDataset = 'websiteDataset', // depp link
   externalFile = 'externalFile',
   apiDataset = 'apiDataset',
-  feishu = 'feishu',
+  feishuShare = 'feishuShare',
+  feishuKnowledge = 'feishuKnowledge',
   yuque = 'yuque'
 }
 export const DatasetTypeMap = {
@@ -36,9 +37,14 @@ export const DatasetTypeMap = {
     label: i18nT('dataset:api_file'),
     collectionLabel: i18nT('common:File')
   },
-  [DatasetTypeEnum.feishu]: {
-    icon: 'core/dataset/feishuDatasetOutline',
-    label: i18nT('dataset:feishu_dataset'),
+  [DatasetTypeEnum.feishuShare]: {
+    icon: 'core/dataset/feishuShareDatasetOutline',
+    label: i18nT('dataset:feishu_share_dataset'),
+    collectionLabel: i18nT('common:File')
+  },
+  [DatasetTypeEnum.feishuKnowledge]: {
+    icon: 'core/dataset/feishuKnowledgeDatasetOutline',
+    label: i18nT('dataset:feishu_knowledge_dataset'),
     collectionLabel: i18nT('common:File')
   },
   [DatasetTypeEnum.yuque]: {

packages/global/core/dataset/type.d.ts

@@ -12,7 +12,12 @@ import type {
   ChunkSettingModeEnum
 } from './constants';
 import type { DatasetPermission } from '../../support/permission/dataset/controller';
-import type { APIFileServer, FeishuServer, YuqueServer } from './apiDataset';
+import type {
+  APIFileServer,
+  FeishuShareServer,
+  YuqueServer,
+  FeishuKnowledgeServer
+} from './apiDataset';
 import type { SourceMemberType } from 'support/user/type';
 import type { DatasetDataIndexTypeEnum } from './data/constants';
 
@@ -73,8 +78,9 @@ export type DatasetSchemaType = {
 
   inheritPermission: boolean;
   apiServer?: APIFileServer;
-  feishuServer?: FeishuServer;
+  feishuShareServer?: FeishuShareServer;
   yuqueServer?: YuqueServer;
+  feishuKnowledgeServer?: FeishuKnowledgeServer;
 
   // abandon
   autoSync?: boolean;

packages/service/core/dataset/apiDataset/index.ts

@@ -1,27 +1,33 @@
 import type {
   APIFileServer,
   YuqueServer,
-  FeishuServer
+  FeishuShareServer,
+  FeishuKnowledgeServer
 } from '@fastgpt/global/core/dataset/apiDataset';
 import { useApiDatasetRequest } from './api';
 import { useYuqueDatasetRequest } from '../yuqueDataset/api';
-import { useFeishuDatasetRequest } from '../feishuDataset/api';
+import { useFeishuShareDatasetRequest } from '../feishuShareDataset/api';
+import { useFeishuKnowledgeDatasetRequest } from '../feishuKnowledgeDataset/api';
 
 export const getApiDatasetRequest = async (data: {
   apiServer?: APIFileServer;
   yuqueServer?: YuqueServer;
-  feishuServer?: FeishuServer;
+  feishuShareServer?: FeishuShareServer;
+  feishuKnowledgeServer?: FeishuKnowledgeServer;
 }) => {
-  const { apiServer, yuqueServer, feishuServer } = data;
+  const { apiServer, yuqueServer, feishuShareServer, feishuKnowledgeServer } = data;
 
   if (apiServer) {
     return useApiDatasetRequest({ apiServer });
   }
   if (yuqueServer) {
     return useYuqueDatasetRequest({ yuqueServer });
   }
-  if (feishuServer) {
-    return useFeishuDatasetRequest({ feishuServer });
+  if (feishuShareServer) {
+    return useFeishuShareDatasetRequest({ feishuShareServer });
+  }
+  if (feishuKnowledgeServer) {
+    return useFeishuKnowledgeDatasetRequest({ feishuKnowledgeServer });
   }
   return Promise.reject('Can not find api dataset server');
 };

packages/service/core/dataset/collection/utils.ts

@@ -160,7 +160,8 @@ export const syncCollection = async (collection: CollectionWithDatasetType) => {
       type: DatasetSourceReadTypeEnum.apiFile,
       sourceId,
       apiServer: dataset.apiServer,
-      feishuServer: dataset.feishuServer,
+      feishuShareServer: dataset.feishuShareServer,
+      feishuKnowledgeServer: dataset.feishuKnowledgeServer,
       yuqueServer: dataset.yuqueServer
     };
   })();