feat: document

Author: wangdan-fit2cloud

ui/src/api/knowledge/document.ts

@@ -21,7 +21,7 @@ Object.defineProperty(prefix, 'value', {
   }
  */
 
-const getDocument: (
+const getDocumentPage: (
   knowledge_id: string,
   page: pageRequest,
   param: any,
@@ -89,10 +89,16 @@ const delDocument: (
 
 const putBatchCancelTask: (
   knowledge_id: string,
+  document_id: string,
   data: any,
   loading?: Ref<boolean>,
-) => Promise<Result<boolean>> = (knowledge_id, data, loading) => {
-  return put(`${prefix.value}/${knowledge_id}/document/cancel_task/_batch`, data, undefined, loading)
+) => Promise<Result<boolean>> = (knowledge_id, document_id, data, loading) => {
+  return put(
+    `${prefix.value}/${knowledge_id}/document/${document_id}/batch_cancel_task`,
+    data,
+    undefined,
+    loading,
+  )
 }
 
 /**
@@ -235,7 +241,13 @@ const putMulDocument: (
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<any>> = (knowledge_id, data, loading) => {
-  return put(`${prefix.value}/${knowledge_id}/document/batch_create`, data, {}, loading, 1000 * 60 * 5)
+  return put(
+    `${prefix.value}/${knowledge_id}/document/batch_create`,
+    data,
+    {},
+    loading,
+    1000 * 60 * 5,
+  )
 }
 
 /**
@@ -277,7 +289,12 @@ const putBatchGenerateRelated: (
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<boolean>> = (knowledge_id, data, loading) => {
-  return put(`${prefix.value}/${knowledge_id}/document/batch_generate_related`, data, undefined, loading)
+  return put(
+    `${prefix.value}/${knowledge_id}/document/batch_generate_related`,
+    data,
+    undefined,
+    loading,
+  )
 }
 
 /**
@@ -293,7 +310,12 @@ const putBatchEditHitHandling: (
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<boolean>> = (knowledge_id, data, loading) => {
-  return put(`${prefix.value}/${knowledge_id}/document/batch_hit_handling`, data, undefined, loading)
+  return put(
+    `${prefix.value}/${knowledge_id}/document/batch_hit_handling`,
+    data,
+    undefined,
+    loading,
+  )
 }
 
 /**
@@ -334,7 +356,12 @@ const putMulSyncDocument: (
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<boolean>> = (knowledge_id, data, loading) => {
-  return put(`${prefix.value}/${knowledge_id}/document/batch_sync`, { id_list: data }, undefined, loading)
+  return put(
+    `${prefix.value}/${knowledge_id}/document/batch_sync`,
+    { id_list: data },
+    undefined,
+    loading,
+  )
 }
 
 /**
@@ -374,8 +401,17 @@ const postQADocument: (
  * 分段预览（上传文档）
  * @param 参数  file:file,limit:number,patterns:array,with_filter:boolean
  */
-const postSplitDocument: (data: any, id: string) => Promise<Result<any>> = (data, id) => {
-  return post(`${prefix.value}/${id}/document/split`, data, undefined, undefined, 1000 * 60 * 60)
+const postSplitDocument: (knowledge_id: string, data: any) => Promise<Result<any>> = (
+  knowledge_id,
+  data,
+) => {
+  return post(
+    `${prefix.value}/${knowledge_id}/document/split`,
+    data,
+    undefined,
+    undefined,
+    1000 * 60 * 60,
+  )
 }
 
 /**
@@ -384,9 +420,10 @@ const postSplitDocument: (data: any, id: string) => Promise<Result<any>> = (data
  * @returns 分段标识列表
  */
 const listSplitPattern: (
+  knowledge_id: string,
   loading?: Ref<boolean>,
-) => Promise<Result<Array<KeyValue<string, string>>>> = (loading) => {
-  return get(`${prefix.value}/document/split_pattern`, {}, loading)
+) => Promise<Result<Array<KeyValue<string, string>>>> = (knowledge_id, loading) => {
+  return get(`${prefix.value}/${knowledge_id}/document/split_pattern`, {}, loading)
 }
 
 /**
@@ -411,7 +448,7 @@ const exportQATemplate: (fileName: string, type: string, loading?: Ref<boolean>)
   type,
   loading,
 ) => {
-  return exportExcel(fileName, `${prefix.value}/document/template/export`, { type }, loading)
+  return exportExcel(fileName, `/workspace/knowledge/document/template/export`, { type }, loading)
 }
 
 /**
@@ -423,7 +460,12 @@ const exportTableTemplate: (fileName: string, type: string, loading?: Ref<boolea
   type,
   loading,
 ) => {
-  return exportExcel(fileName, `${prefix.value}/document/table_template/export`, { type }, loading)
+  return exportExcel(
+    fileName,
+    `/workspace/knowledge/document/table_template/export`,
+    { type },
+    loading,
+  )
 }
 
 /**
@@ -445,13 +487,34 @@ const postWebDocument: (
   return post(`${prefix.value}/${knowledge_id}/document/web`, data, undefined, loading)
 }
 
-const getAllDocument: (knowledge_id: string, loading?: Ref<boolean>) => Promise<Result<any>> = (
-  knowledge_id,
-  loading,
-) => {
-  return get(`${prefix.value}/${knowledge_id}/document`, undefined, loading)
+/**
+ * 飞书导入获得相关文档
+ * @param 参数
+ * {
+ "source_url_list": [
+ "string"
+ ],
+ "selector": "string"
+ }
+ }
+ */
+const getLarkDocumentList: (
+  knowledge_id: string,
+  folder_token: string,
+  data: any,
+  loading?: Ref<boolean>,
+) => Promise<Result<any>> = (knowledge_id, folder_token, data, loading) => {
+  return post(
+    `${prefix.value}/lark/${knowledge_id}/${folder_token}/doc_list`,
+    data,
+    undefined,
+    loading,
+  )
 }
 
+/**
+ * 同步飞书文档
+ */
 const putLarkDocumentSync: (
   knowledge_id: string,
   document_id: string,
@@ -465,16 +528,38 @@ const putLarkDocumentSync: (
   )
 }
 
-const delMulLarkSyncDocument: (
+/**
+ * 批量同步飞书文档
+ */
+const putMulLarkSyncDocument: (
   knowledge_id: string,
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<boolean>> = (knowledge_id, data, loading) => {
   return put(`${prefix.value}/lark/${knowledge_id}/_batch`, { id_list: data }, undefined, loading)
 }
 
+/**
+ * 导入飞书文档
+ */
+const importLarkDocument: (
+  knowledge_id: string,
+  data: any,
+  loading?: Ref<boolean>
+) => Promise<Result<Array<any>>> = (knowledge_id, data, loading) => {
+  return post(`${prefix.value}/lark/${knowledge_id}/import`, data, null, loading)
+}
+
+// todo
+const getAllDocument: (knowledge_id: string, loading?: Ref<boolean>) => Promise<Result<any>> = (
+  knowledge_id,
+  loading,
+) => {
+  return get(`${prefix.value}/${knowledge_id}/document`, undefined, loading)
+}
+
 export default {
-  getDocument,
+  getDocumentPage,
   getDocumentDetail,
   putDocument,
   delDocument,
@@ -499,8 +584,8 @@ export default {
   exportQATemplate,
   exportTableTemplate,
   postWebDocument,
-
-  getAllDocument,
+  getLarkDocumentList,
   putLarkDocumentSync,
-  delMulLarkSyncDocument,
+  putMulLarkSyncDocument,
+  importLarkDocument
 }

ui/src/api/knowledge/knowledge.ts

@@ -255,30 +255,6 @@ const postWebKnowledge: (data: any, loading?: Ref<boolean>) => Promise<Result<an
   return post(`${prefix.value}/web`, data, undefined, loading)
 }
 
-/**
- * 获取飞书文档列表
- * @param knowledge_id
- * @param folder_token
- * @param loading
- * @returns
- */
-const getLarkDocumentList: (
-  knowledge_id: string,
-  folder_token: string,
-  data: any,
-  loading?: Ref<boolean>,
-) => Promise<Result<Array<any>>> = (knowledge_id, folder_token, data, loading) => {
-  return post(`${prefix.value}/lark/${knowledge_id}/${folder_token}/doc_list`, data, null, loading)
-}
-
-const importLarkDocument: (
-  knowledge_id: string,
-  data: any,
-  loading?: Ref<boolean>,
-) => Promise<Result<Array<any>>> = (knowledge_id, data, loading) => {
-  return post(`${prefix.value}/lark/${knowledge_id}/import`, data, null, loading)
-}
-
 const postLarkKnowledge: (data: any, loading?: Ref<boolean>) => Promise<Result<Array<any>>> = (
   data,
   loading,
@@ -303,7 +279,5 @@ export default {
   getKnowledgeModel,
   postWebKnowledge,
 
-  getLarkDocumentList,
-  importLarkDocument,
   postLarkKnowledge,
 }

ui/src/api/knowledge/paragraph.ts

@@ -41,14 +41,14 @@ const postParagraph: (
 }
 
 /**
- * 段落列表
+ * 段落分页列表
  * @param 参数 knowledge_id document_id
  * param {
           "title": "string",
           "content": "string",
         }
  */
-const getParagraph: (
+const getParagraphPage: (
   knowledge_id: string,
   document_id: string,
   page: pageRequest,
@@ -145,6 +145,30 @@ const postParagraphProblem: (
   )
 }
 
+
+/**
+ * 段落调整顺序
+ * @param knowledge_id 数据集id
+ * @param document_id 文档id
+ * @param loading 加载器
+ * @query data {
+ *              paragraph_id 段落id  new_position 新顺序
+ *             }
+ */
+const putAdjustPosition: (
+  knowledge_id: string,
+  document_id: string,
+  data: any,
+  loading?: Ref<boolean>,
+) => Promise<Result<any>> = (knowledge_id, document_id, data, loading) => {
+  return put(
+    `${prefix.value}/${knowledge_id}/document/${document_id}/paragraph/adjust_position`,
+    {},
+    data,
+    loading,
+  )
+}
+
 /**
  * 添加某段落关联问题
  * @param knowledge_id 数据集id
@@ -262,7 +286,7 @@ const putDisassociationProblem: (
 
 export default {
   postParagraph,
-  getParagraph,
+  getParagraphPage,
   putParagraph,
   delParagraph,
   getParagraphProblem,

ui/src/api/knowledge/problem.ts

@@ -33,7 +33,7 @@ const postProblems: (
    }
  */
 
-const getProblems: (
+const getProblemsPage: (
   knowledge_id: string,
   page: pageRequest,
   param: any,
@@ -119,7 +119,7 @@ const putMulProblem: (
 
 export default {
   postProblems,
-  getProblems,
+  getProblemsPage,
   putProblems,
   delProblems,
   getDetailProblems,