diff --git a/web/src/components/image.tsx b/web/src/components/image.tsx
new file mode 100644
index 000000000..7cb068968
--- /dev/null
+++ b/web/src/components/image.tsx
@@ -0,0 +1,19 @@
+import { api_host } from '@/utils/api';
+
+interface IImage {
+ id: string;
+ className: string;
+}
+
+const Image = ({ id, className, ...props }: IImage) => {
+ return (
+
+ );
+};
+
+export default Image;
diff --git a/web/src/pages/add-knowledge/components/knowledge-chunk/components/chunk-card/index.tsx b/web/src/pages/add-knowledge/components/knowledge-chunk/components/chunk-card/index.tsx
index 8936c1d9d..89af5283b 100644
--- a/web/src/pages/add-knowledge/components/knowledge-chunk/components/chunk-card/index.tsx
+++ b/web/src/pages/add-knowledge/components/knowledge-chunk/components/chunk-card/index.tsx
@@ -1,7 +1,7 @@
+import Image from '@/components/image';
import { IChunk } from '@/interfaces/database/knowledge';
import { api_host } from '@/utils/api';
import { Card, Checkbox, CheckboxProps, Flex, Popover, Switch } from 'antd';
-
import { useState } from 'react';
import styles from './index.less';
@@ -13,22 +13,6 @@ interface IProps {
handleCheckboxClick: (chunkId: string, checked: boolean) => void;
}
-interface IImage {
- id: string;
- className: string;
-}
-// Pass onMouseEnter and onMouseLeave to img tag using props
-const Image = ({ id, className, ...props }: IImage) => {
- return (
-
- );
-};
-
const ChunkCard = ({
item,
checked,
diff --git a/web/src/pages/add-knowledge/components/knowledge-dataset/knowledge-upload-file/index.tsx b/web/src/pages/add-knowledge/components/knowledge-dataset/knowledge-upload-file/index.tsx
index d6a0da610..db475e22d 100644
--- a/web/src/pages/add-knowledge/components/knowledge-dataset/knowledge-upload-file/index.tsx
+++ b/web/src/pages/add-knowledge/components/knowledge-dataset/knowledge-upload-file/index.tsx
@@ -28,7 +28,7 @@ import {
UploadProps,
} from 'antd';
import classNames from 'classnames';
-import { ReactElement, useEffect, useRef, useState } from 'react';
+import { ReactElement, useEffect, useMemo, useRef, useState } from 'react';
import { Nullable } from 'typings';
import { Link, useDispatch, useNavigate, useSelector } from 'umi';
@@ -63,6 +63,24 @@ const UploaderItem = ({
const documentId = file?.response?.id;
+ const parserList = useMemo(() => {
+ return parserArray.map((x) => {
+ const arr = x.split(':');
+ return { value: arr[0], label: arr[1] };
+ });
+ }, [parserArray]);
+
+ const saveParser = (parserId: string) => {
+ dispatch({
+ type: 'kFModel/document_change_parser',
+ payload: {
+ parser_id: parserId,
+ doc_id: documentId,
+ parser_config: parserConfig,
+ },
+ });
+ };
+
const onChange = (e: RadioChangeEvent) => {
const val = e.target.value;
setValue(val);
@@ -72,12 +90,12 @@ const UploaderItem = ({
const content = (
- {parserArray.map(
+ {parserList.map(
(
x, // value is lowercase, key is uppercase
) => (
-
- {x}
+
+ {x.label}
),
)}
@@ -92,17 +110,6 @@ const UploaderItem = ({
}
};
- const saveParser = (parserId: string) => {
- dispatch({
- type: 'kFModel/document_change_parser',
- payload: {
- parser_id: parserId,
- doc_id: documentId,
- parser_config: parserConfig,
- },
- });
- };
-
useEffect(() => {
setValue(defaultParserId);
}, [defaultParserId]);
diff --git a/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.less b/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.less
index 46ef353dc..1059af550 100644
--- a/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.less
+++ b/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.less
@@ -33,4 +33,11 @@
font-size: 12px;
font-weight: 500;
}
+ .image {
+ width: 100px;
+ }
+ .imagePreview {
+ display: block;
+ width: 260px;
+ }
}
diff --git a/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.tsx b/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.tsx
index dbe7b22aa..fac22905a 100644
--- a/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.tsx
+++ b/web/src/pages/add-knowledge/components/knowledge-testing/testing-result/index.tsx
@@ -1,6 +1,15 @@
import { ReactComponent as SelectedFilesCollapseIcon } from '@/assets/svg/selected-files-collapse.svg';
+import Image from '@/components/image';
import { ITestingChunk } from '@/interfaces/database/knowledge';
-import { Card, Collapse, Flex, Pagination, PaginationProps, Space } from 'antd';
+import {
+ Card,
+ Collapse,
+ Flex,
+ Pagination,
+ PaginationProps,
+ Popover,
+ Space,
+} from 'antd';
import { useDispatch, useSelector } from 'umi';
import { TestingModelState } from '../model';
import styles from './index.less';
@@ -92,7 +101,22 @@ const TestingResult = ({ handleTesting }: IProps) => {
>
{chunks.map((x) => (
}>
- {x.content_with_weight}
+
+ {x.img_id && (
+
+ }
+ >
+
+
+ )}
+ {x.content_with_weight}
+
))}