feat: Related Knowledge

Author: wangdan-fit2cloud

ui/src/api/knowledge/knowledge.ts

@@ -21,6 +21,23 @@ const getKnowledgeByFolder: (data?: any, loading?: Ref<boolean>) => Promise<Resu
   return get(`${prefix}/knowledge`, data, loading)
 }
 
+/**
+ * 知识库列表（无分页）
+ * @param 参数
+ * param  {
+    folder_id: "string",
+    name: "string",
+    tool_type: "string",
+    desc: string,
+  }
+ */
+const getKnowledgeList: (param?: any, loading?: Ref<boolean>) => Promise<Result<any>> = (
+  param,
+  loading,
+) => {
+  return get(`${prefix}/knowledge`, param, loading)
+}
+
 /**
  * 知识库分页列表
  * @param 参数
@@ -31,7 +48,7 @@ const getKnowledgeByFolder: (data?: any, loading?: Ref<boolean>) => Promise<Resu
  desc: string,
  }
  */
-const getKnowledgeList: (
+const getKnowledgeListPage: (
   page: pageRequest,
   param?: any,
   loading?: Ref<boolean>,
@@ -234,14 +251,6 @@ const postWebKnowledge: (data: any, loading?: Ref<boolean>) => Promise<Result<an
   return post(`${prefix}/knowledge/web`, data, undefined, loading)
 }
 
-/**
- * 获取全部知识库
- * @param 参数
- */
-const getAllKnowledge: (loading?: Ref<boolean>) => Promise<Result<any[]>> = (loading) => {
-  return get(`${prefix}`, undefined, loading)
-}
-
 /**
  * 获取飞书文档列表
  * @param knowledge_id
@@ -269,6 +278,7 @@ const importLarkDocument: (
 export default {
   getKnowledgeByFolder,
   getKnowledgeList,
+  getKnowledgeListPage,
   getKnowledgeDetail,
   putKnowledge,
   delKnowledge,
@@ -284,5 +294,4 @@ export default {
 
   getLarkDocumentList,
   importLarkDocument,
-  getAllKnowledge,
 }

ui/src/api/shared/knowledge.ts

@@ -79,6 +79,23 @@ const getKnowledgeByFolder: (data?: any, loading?: Ref<boolean>) => Promise<Resu
   return get(`${prefix}/knowledge`, data, loading)
 }
 
+/**
+ * 知识库列表（无分页）
+ * @param 参数
+ * param  {
+ "folder_id": "string",
+ "name": "string",
+ "tool_type": "string",
+ desc: string,
+ }
+ */
+const getKnowledgeList: (param?: any, loading?: Ref<boolean>) => Promise<Result<any>> = (
+  param,
+  loading,
+) => {
+  return get(`${prefix}/knowledge`, param, loading)
+}
+
 /**
  * 知识库分页列表
  * @param 参数
@@ -89,7 +106,7 @@ const getKnowledgeByFolder: (data?: any, loading?: Ref<boolean>) => Promise<Resu
  desc: string,
  }
  */
-const getKnowledgeList: (
+const getKnowledgeListPage: (
   page: pageRequest,
   param?: any,
   loading?: Ref<boolean>,
@@ -292,14 +309,6 @@ const postWebKnowledge: (data: any, loading?: Ref<boolean>) => Promise<Result<an
   return post(`${prefix}/knowledge/web`, data, undefined, loading)
 }
 
-/**
- * 获取全部知识库
- * @param 参数
- */
-const getAllKnowledge: (loading?: Ref<boolean>) => Promise<Result<any[]>> = (loading) => {
-  return get(`${prefix}/knowledge`, undefined, loading)
-}
-
 /**
  * 获取飞书文档列表
  * @param knowledge_id
@@ -327,6 +336,7 @@ const importLarkDocument: (
 export default {
   getKnowledgeByFolder,
   getKnowledgeList,
+  getKnowledgeListPage,
   getKnowledgeDetail,
   putKnowledge,
   delKnowledge,
@@ -342,7 +352,6 @@ export default {
 
   getLarkDocumentList,
   importLarkDocument,
-  getAllKnowledge,
   getSharedWorkspaceKnowledge,
   getSharedWorkspaceKnowledgePage,
   getSharedAuthorizationKnowledgeGet,

ui/src/layout/components/breadcrumb/index.vue

@@ -169,7 +169,7 @@ function changeMenu(id: string) {
 function getDataset() {
   loading.value = true
   knowledge
-    .asyncGetAllKnowledge()
+    .asyncGetRootKnowledge()
     .then((res: any) => {
       list.value = res.data
       common.saveBreadcrumb(list.value)

ui/src/locales/lang/en-US/views/application.ts

@@ -131,7 +131,6 @@ export default {
     addKnowledge: 'Add Related Knowledge',
     addKnowledgePlaceholder: 'The selected knowledge must use the same embedding model',
     selected: 'Selected',
-    countDataset: 'Knowledge',
 
     selectSearchMode: 'Retrieval Mode',
     vectorSearch: 'Vector Search',

ui/src/locales/lang/zh-CN/views/application.ts

@@ -122,7 +122,6 @@ export default {
     addKnowledge: '添加关联知识库',
     addKnowledgePlaceholder: '所选知识库必须使用相同的 Embedding 模型',
     selected: '已选',
-    countDataset: '个知识库',
 
     selectSearchMode: '检索模式',
     vectorSearch: '向量检索',

ui/src/locales/lang/zh-Hant/views/application.ts

@@ -122,7 +122,6 @@ export default {
     addKnowledge: '新增關聯知識庫',
     addKnowledgePlaceholder: '所選知識庫必須使用相同的 Embedding 模型',
     selected: '已選',
-    countDataset: '個知識庫',
 
     selectSearchMode: '檢索模式',
     vectorSearch: '向量檢索',

ui/src/stores/modules-shared-system/knowledge.ts

@@ -1,8 +1,8 @@
-import {defineStore} from 'pinia'
-import type {knowledgeData} from '@/api/type/knowledge'
-import type {UploadUserFile} from 'element-plus'
+import { defineStore } from 'pinia'
+import type { knowledgeData } from '@/api/type/knowledge'
+import type { UploadUserFile } from 'element-plus'
 import knowledgeApi from '@/api/shared/knowledge'
-import {type Ref} from 'vue'
+import { type Ref } from 'vue'
 
 export interface knowledgeStateTypes {
   baseInfo: knowledgeData | null
@@ -31,10 +31,13 @@ const useKnowledgeStore = defineStore('knowledg', {
     saveDocumentsFile(file: UploadUserFile[]) {
       this.documentsFiles = file
     },
-    async asyncGetAllKnowledge(loading?: Ref<boolean>) {
+    async asyncGetRootKnowledge(loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
+        const params = {
+          folder_id: localStorage.getItem('workspace_id'),
+        }
         knowledgeApi
-          .getAllKnowledge(loading)
+          .getKnowledgeList(params, loading)
           .then((data) => {
             resolve(data)
           })
@@ -43,10 +46,7 @@ const useKnowledgeStore = defineStore('knowledg', {
           })
       })
     },
-    async asyncGetKnowledgeDetail(
-      knowledge_id: string,
-      loading?: Ref<boolean>,
-    ) {
+    async asyncGetKnowledgeDetail(knowledge_id: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         knowledgeApi
           .getKnowledgeDetail(knowledge_id, loading)
@@ -58,11 +58,7 @@ const useKnowledgeStore = defineStore('knowledg', {
           })
       })
     },
-    async asyncSyncKnowledge(
-      id: string,
-      sync_type: string,
-      loading?: Ref<boolean>,
-    ) {
+    async asyncSyncKnowledge(id: string, sync_type: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         knowledgeApi
           .putSyncWebKnowledge(id, sync_type, loading)

ui/src/stores/modules/application.ts

@@ -1,6 +1,5 @@
 import { defineStore } from 'pinia'
 import applicationApi from '@/api/application/application'
-import knowledgeAPI from '@/api/knowledge/knowledge.ts'
 import applicationXpackApi from '@/api/application/application-xpack'
 import { type Ref } from 'vue'
 import { getBrowserLang } from '@/locales/index'
@@ -36,19 +35,6 @@ const useApplicationStore = defineStore('application', {
       })
     },
 
-    async asyncGetApplicationKnowledge(id: string, loading?: Ref<boolean>) {
-      return new Promise((resolve, reject) => {
-        knowledgeAPI
-          .getAllKnowledge(loading)
-          .then((data) => {
-            resolve(data)
-          })
-          .catch((error) => {
-            reject(error)
-          })
-      })
-    },
-
     async asyncGetAccessToken(id: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         const user = useUserStore()

ui/src/stores/modules/knowledge.ts

@@ -1,8 +1,8 @@
-import {defineStore} from 'pinia'
-import type {knowledgeData} from '@/api/type/knowledge'
-import type {UploadUserFile} from 'element-plus'
+import { defineStore } from 'pinia'
+import type { knowledgeData } from '@/api/type/knowledge'
+import type { UploadUserFile } from 'element-plus'
 import knowledgeApi from '@/api/knowledge/knowledge'
-import {type Ref} from 'vue'
+import { type Ref } from 'vue'
 
 export interface knowledgeStateTypes {
   baseInfo: knowledgeData | null
@@ -31,10 +31,13 @@ const useKnowledgeStore = defineStore('knowledge', {
     saveDocumentsFile(file: UploadUserFile[]) {
       this.documentsFiles = file
     },
-    async asyncGetAllKnowledge(loading?: Ref<boolean>) {
+    async asyncGetRootKnowledge(loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
+        const params = {
+          folder_id: localStorage.getItem('workspace_id'),
+        }
         knowledgeApi
-          .getAllKnowledge(loading)
+          .getKnowledgeList(params, loading)
           .then((data) => {
             resolve(data)
           })
@@ -43,10 +46,7 @@ const useKnowledgeStore = defineStore('knowledge', {
           })
       })
     },
-    async asyncGetKnowledgeDetail(
-      knowledge_id: string,
-      loading?: Ref<boolean>,
-    ) {
+    async asyncGetKnowledgeDetail(knowledge_id: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         knowledgeApi
           .getKnowledgeDetail(knowledge_id, loading)
@@ -58,11 +58,7 @@ const useKnowledgeStore = defineStore('knowledge', {
           })
       })
     },
-    async asyncSyncKnowledge(
-      id: string,
-      sync_type: string,
-      loading?: Ref<boolean>,
-    ) {
+    async asyncSyncKnowledge(id: string, sync_type: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         knowledgeApi
           .putSyncWebKnowledge(id, sync_type, loading)

ui/src/views/application/ApplicationSetting.vue

@@ -185,6 +185,7 @@
                             <div class="flex align-center" style="width: 80%">
                               <KnowledgeIcon
                                 :type="relatedObject(knowledgeList, item, 'id')?.type"
+                                class="mr-12"
                               />
 
                               <span
@@ -439,7 +440,7 @@ import { t } from '@/locales'
 import TTSModeParamSettingDialog from './component/TTSModeParamSettingDialog.vue'
 import ReasoningParamSettingDialog from './component/ReasoningParamSettingDialog.vue'
 
-const { model, application } = useStore()
+const { knowledge, model, application } = useStore()
 
 const route = useRoute()
 const {
@@ -628,7 +629,7 @@ function getDetail() {
 }
 
 function getKnowledge() {
-  application.asyncGetApplicationKnowledge(id, knowledgeLoading).then((res: any) => {
+  knowledge.asyncGetRootKnowledge(knowledgeLoading).then((res: any) => {
     knowledgeList.value = res.data
   })
 }
@@ -706,8 +707,7 @@ function refresh() {
 
 onMounted(() => {
   getModel()
-  // todo
-  // getKnowledge()
+  getKnowledge()
   getDetail()
   getSTTModel()
   getTTSModel()