feat: paragraph

Author: wangdan-fit2cloud

ui/src/api/knowledge/knowledge.ts

@@ -43,9 +43,9 @@ const getKnowledgeList: (
  * 获取全部知识库
  * @param 参数
  */
-// const getAllDataset: (loading?: Ref<boolean>) => Promise<Result<any[]>> = (loading) => {
-//   return get(`${prefix}`, undefined, loading)
-// }
+const getAllKnowledge: (loading?: Ref<boolean>) => Promise<Result<any[]>> = (loading) => {
+  return get(`${prefix}`, undefined, loading)
+}
 
 /**
  * 同步知识库
@@ -260,5 +260,6 @@ export default {
   getLarkDocumentList,
   importLarkDocument,
   generateRelated,
-  delKnowledge
+  delKnowledge,
+  getAllKnowledge
 }

ui/src/api/knowledge/paragraph.ts

@@ -7,7 +7,7 @@ const prefix = '/workspace/' + localStorage.getItem('workspace_id') + '/knowledg
 /**
  * 创建段落
  * @param 参数
- * dataset_id, document_id
+ * knowledge_id, document_id
  * {
       "content": "string",
       "title": "string",
@@ -20,103 +20,185 @@ const prefix = '/workspace/' + localStorage.getItem('workspace_id') + '/knowledg
     }
  */
 const postParagraph: (
-  dataset_id: string,
+  knowledge_id: string,
   document_id: string,
   data: any,
   loading?: Ref<boolean>,
-) => Promise<Result<any>> = (dataset_id, document_id, data, loading) => {
-  return post(`${prefix}/${dataset_id}/document/${document_id}/paragraph`, data, undefined, loading)
+) => Promise<Result<any>> = (knowledge_id, document_id, data, loading) => {
+  return post(
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph`,
+    data,
+    undefined,
+    loading,
+  )
 }
 
 /**
  * 段落列表
- * @param 参数 dataset_id document_id
- * page {
-          "current_page": "string",
-          "page_size": "string",
-        }
+ * @param 参数 knowledge_id document_id
  * param {
           "title": "string",
           "content": "string",
         }
  */
 const getParagraph: (
-  dataset_id: string,
+  knowledge_id: string,
   document_id: string,
   page: pageRequest,
   param: any,
   loading?: Ref<boolean>,
-) => Promise<Result<any>> = (dataset_id, document_id, page, param, loading) => {
+) => Promise<Result<any>> = (knowledge_id, document_id, page, param, loading) => {
   return get(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/${page.current_page}/${page.page_size}`,
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/${page.current_page}/${page.page_size}`,
     param,
     loading,
   )
 }
 
+/**
+ * 修改段落
+ * @param 参数
+ * knowledge_id, document_id, paragraph_id
+ * {
+    "content": "string",
+    "title": "string",
+    "is_active": true,
+      "problem_list": [
+        {
+          "content": "string"
+        }
+      ]
+  }
+ */
+const putParagraph: (
+  knowledge_id: string,
+  document_id: string,
+  paragraph_id: string,
+  data: any,
+  loading?: Ref<boolean>,
+) => Promise<Result<any>> = (knowledge_id, document_id, paragraph_id, data, loading) => {
+  return put(
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/${paragraph_id}`,
+    data,
+    undefined,
+    loading,
+  )
+}
+
 /**
  * 删除段落
- * @param 参数 dataset_id, document_id, paragraph_id
+ * @param 参数 knowledge_id, document_id, paragraph_id
  */
 const delParagraph: (
-  dataset_id: string,
+  knowledge_id: string,
   document_id: string,
   paragraph_id: string,
   loading?: Ref<boolean>,
-) => Promise<Result<boolean>> = (dataset_id, document_id, paragraph_id, loading) => {
+) => Promise<Result<boolean>> = (knowledge_id, document_id, paragraph_id, loading) => {
   return del(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/${paragraph_id}`,
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/${paragraph_id}`,
     undefined,
     {},
     loading,
   )
 }
 
 /**
- * 批量删除段落
- * @param 参数 dataset_id, document_id
+ * 某段落问题列表
+ * @param 参数 knowledge_id，document_id，paragraph_id
  */
-const delMulParagraph: (
-  dataset_id: string,
+const getParagraphProblem: (
+  knowledge_id: string,
   document_id: string,
+  paragraph_id: string,
+) => Promise<Result<any>> = (knowledge_id, document_id, paragraph_id: string) => {
+  return get(`${prefix}/${knowledge_id}/document/${document_id}/paragraph/${paragraph_id}/problem`)
+}
+
+/**
+ * 给某段落创建问题
+ * @param 参数
+ * knowledge_id, document_id, paragraph_id
+ * {
+      content": "string"
+    }
+ */
+const postParagraphProblem: (
+  knowledge_id: string,
+  document_id: string,
+  paragraph_id: string,
   data: any,
   loading?: Ref<boolean>,
-) => Promise<Result<boolean>> = (dataset_id, document_id, data, loading) => {
-  return del(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/_batch`,
-    undefined,
-    { id_list: data },
+) => Promise<Result<any>> = (knowledge_id, document_id, paragraph_id, data: any, loading) => {
+  return post(
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/${paragraph_id}/problem`,
+    data,
+    {},
     loading,
   )
 }
 
+/**
+ * 添加某段落关联问题
+ * @param knowledge_id 数据集id
+ * @param document_id 文档id
+ * @param loading 加载器
+ * @query data {
+ *              paragraph_id 段落id  problem_id 问题id
+ *             }
+ */
+const putAssociationProblem: (
+  knowledge_id: string,
+  document_id: string,
+  data: any,
+  loading?: Ref<boolean>,
+) => Promise<Result<any>> = (knowledge_id, document_id, data, loading) => {
+  return put(
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/association`,
+    {},
+    data,
+    loading,
+  )
+}
 
+/**
+ * 批量删除段落
+ * @param 参数 knowledge_id, document_id
+ */
+const putMulParagraph: (
+  knowledge_id: string,
+  document_id: string,
+  data: any,
+  loading?: Ref<boolean>,
+) => Promise<Result<boolean>> = (knowledge_id, document_id, data, loading) => {
+  return put(
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/batch_delete`,
+    { id_list: data },
+    undefined,
+    loading,
+  )
+}
 
 /**
- * 修改段落
- * @param 参数
- * dataset_id, document_id, paragraph_id
+ * 批量关联问题
+ * @param 参数 knowledge_id, document_id
  * {
-  "content": "string",
-  "title": "string",
-  "is_active": true,
-  "problem_list": [
-    {
-      "id": "string",
-      "content": "string"
-    }
-  ]
+      "paragraph_id_list": [
+        "3fa85f64-5717-4562-b3fc-2c963f66afa6"
+      ],
+      "model_id": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
+      "prompt": "string",
+      "document_id": "3fa85f64-5717-4562-b3fc-2c963f66afa6"
     }
  */
-const putParagraph: (
-  dataset_id: string,
+const putBatchGenerateRelated: (
+  knowledge_id: string,
   document_id: string,
-  paragraph_id: string,
   data: any,
   loading?: Ref<boolean>,
-) => Promise<Result<any>> = (dataset_id, document_id, paragraph_id, data, loading) => {
+) => Promise<Result<boolean>> = (knowledge_id, document_id, data, loading) => {
   return put(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/${paragraph_id}`,
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/batch_generate_related`,
     data,
     undefined,
     loading,
@@ -125,76 +207,62 @@ const putParagraph: (
 
 /**
  * 批量迁移段落
- * @param 参数 dataset_id,target_dataset_id,
+ * @param 参数 knowledge_id,target_knowledge_id,
  */
 const putMigrateMulParagraph: (
-  dataset_id: string,
+  knowledge_id: string,
   document_id: string,
-  target_dataset_id: string,
+  target_knowledge_id: string,
   target_document_id: string,
   data: any,
   loading?: Ref<boolean>,
 ) => Promise<Result<boolean>> = (
-  dataset_id,
+  knowledge_id,
   document_id,
-  target_dataset_id,
+  target_knowledge_id,
   target_document_id,
   data,
   loading,
 ) => {
   return put(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/migrate/dataset/${target_dataset_id}/document/${target_document_id}`,
-    data,
-    undefined,
-    loading,
-  )
-}
-
-const batchGenerateRelated: (
-  dataset_id: string,
-  document_id: string,
-  data: any,
-  loading?: Ref<boolean>,
-) => Promise<Result<boolean>> = (dataset_id, document_id, data, loading) => {
-  return put(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/batch_generate_related`,
+    `${prefix}/${knowledge_id}/document/${document_id}/paragraph/migrate/dataset/${target_knowledge_id}/document/${target_document_id}`,
     data,
     undefined,
     loading,
   )
 }
 
 /**
- * 创建问题
- * @param 参数
- * dataset_id, document_id, paragraph_id
- * {
-      "id": "string",
-      content": "string"
-    }
+ * 解除某段落关联问题
+ * @param 参数 dataset_id, document_id,
+ * @query data {
+ *            paragraph_id 段落id  problem_id 问题id
+ *         }
  */
-const postProblem: (
+const putDisassociationProblem: (
   dataset_id: string,
   document_id: string,
-  paragraph_id: string,
   data: any,
   loading?: Ref<boolean>,
-) => Promise<Result<any>> = (dataset_id, document_id, paragraph_id, data: any, loading) => {
-  return post(
-    `${prefix}/${dataset_id}/document/${document_id}/paragraph/${paragraph_id}/problem`,
-    data,
+) => Promise<Result<boolean>> = (dataset_id, document_id, data, loading) => {
+  return put(
+    `${prefix}/${dataset_id}/document/${document_id}/paragraph/unassociation`,
     {},
+    data,
     loading,
   )
 }
 
 export default {
+  postParagraph,
   getParagraph,
-  delParagraph,
   putParagraph,
-  postParagraph,
-  delMulParagraph,
+  delParagraph,
+  getParagraphProblem,
+  postParagraphProblem,
+  putAssociationProblem,
+  putMulParagraph,
+  putBatchGenerateRelated,
   putMigrateMulParagraph,
-  batchGenerateRelated,
-  postProblem,
+  putDisassociationProblem
 }