feat: Add KnowledgeGraphModal #162

web/src/components/chunk-method-modal/hooks.ts

@@ -14,25 +14,59 @@ const ParserListMap = new Map([
       'presentation',
       'one',
       'qa',
+      'knowledge_graph',
     ],
   ],
   [
     ['doc', 'docx'],
-    ['naive', 'resume', 'book', 'laws', 'one', 'qa', 'manual'],
+    [
+      'naive',
+      'resume',
+      'book',
+      'laws',
+      'one',
+      'qa',
+      'manual',
+      'knowledge_graph',
+    ],
   ],
   [
     ['xlsx', 'xls'],
-    ['naive', 'qa', 'table', 'one'],
+    ['naive', 'qa', 'table', 'one', 'knowledge_graph'],
   ],
   [['ppt', 'pptx'], ['presentation']],
   [
     ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'tif', 'tiff', 'webp', 'svg', 'ico'],
     ['picture'],
   ],
-  [['txt'], ['naive', 'resume', 'book', 'laws', 'one', 'qa', 'table']],
-  [['csv'], ['naive', 'resume', 'book', 'laws', 'one', 'qa', 'table']],
-  [['md'], ['naive', 'qa']],
-  [['json'], ['naive']],
+  [
+    ['txt'],
+    [
+      'naive',
+      'resume',
+      'book',
+      'laws',
+      'one',
+      'qa',
+      'table',
+      'knowledge_graph',
+    ],
+  ],
+  [
+    ['csv'],
+    [
+      'naive',
+      'resume',
+      'book',
+      'laws',
+      'one',
+      'qa',
+      'table',
+      'knowledge_graph',
+    ],
+  ],
+  [['md'], ['naive', 'qa', 'knowledge_graph']],
+  [['json'], ['naive', 'knowledge_graph']],
 ]);
 
 const getParserList = (

web/src/components/chunk-method-modal/index.tsx

@@ -119,7 +119,7 @@ const ChunkMethodModal: React.FC<IProps> = ({
       <Space size={[0, 8]} wrap>
         <Form.Item label={t('chunkMethod')} className={styles.chunkMethod}>
           <Select
-            style={{ width: 120 }}
+            style={{ width: 160 }}
             onChange={handleChange}
             value={selectedTag}
             options={parserList}

web/src/hooks/chunk-hooks.ts

@@ -206,3 +206,22 @@ export const useFetchChunk = (chunkId?: string): ResponseType<any> => {
 
   return data;
 };
+
+export const useFetchKnowledgeGraph = (): ResponseType<any> => {
+  const { documentId } = useGetKnowledgeSearchParams();
+
+  const { data } = useQuery({
+    queryKey: ['fetchKnowledgeGraph', documentId],
+    initialData: true,
+    gcTime: 0,
+    queryFn: async () => {
+      const data = await kbService.knowledge_graph({
+        doc_id: documentId,
+      });
+
+      return data;
+    },
+  });
+
+  return data;
+};