diff --git a/web/app/components/datasets/common/document-picker/index.tsx b/web/app/components/datasets/common/document-picker/index.tsx index 624286f71d..7d13890cd2 100644 --- a/web/app/components/datasets/common/document-picker/index.tsx +++ b/web/app/components/datasets/common/document-picker/index.tsx @@ -13,6 +13,8 @@ import s from '@/app/components/datasets/documents/style.module.css' import cn from '@/utils/classnames' import SearchInput from '@/app/components/base/search-input' import { GeneralType, ParentChildType } from '@/app/components/base/icons/src/public/knowledge' +import { useDocumentList } from '@/service/knowledge/use-document' +import Loading from '@/app/components/base/loading' type Props = { value: { @@ -34,6 +36,17 @@ const DocumentPicker: FC = ({ processMode, parentMode, } = value + const [query, setQuery] = useState('') + + const { data } = useDocumentList({ + datasetId: 'b8443630-cd2d-4fb7-aa65-21af8a858a7d', + query: { + keyword: query, + page: 1, + limit: 20, + }, + }) + const documentsList = data?.data const localExtension = extension?.toLowerCase() || name?.split('.')?.pop()?.toLowerCase() const isParentChild = processMode === 'hierarchical' const TypeIcon = isParentChild ? ParentChildType : GeneralType @@ -44,8 +57,6 @@ const DocumentPicker: FC = ({ }] = useBoolean(false) const ArrowIcon = open ? RiArrowDownSLine : RiArrowUpSLine - const [query, setQuery] = useState('') - return ( = ({ +
-
- {['EOS R3 Tech Sheet.pdf', 'Steve Jobs The Man Who Thought DifferentSteve Jobs The Man Who Thought DifferentSteve Jobs The Man Who Thought Different'].map(item => ( -
-
-
{item}
+ {documentsList + ? ( +
+ {documentsList.map(item => ( +
{ + onChange(item) + setOpen(false) + } + } + > +
+
{item.name}
+
+ ))}
- ))} -
+ ) + : (
+ +
)}
+ ) diff --git a/web/app/components/datasets/documents/detail/index.tsx b/web/app/components/datasets/documents/detail/index.tsx index 79723907e2..95144c2769 100644 --- a/web/app/components/datasets/documents/detail/index.tsx +++ b/web/app/components/datasets/documents/detail/index.tsx @@ -48,7 +48,7 @@ export const DocumentTitle: FC = ({ extension, name, process processMode, parentMode: parent_mode, }} - onChange={() => { }} + onChange={(doc) => { console.log(doc) }} />
) diff --git a/web/service/knowledge/use-dateset.ts b/web/service/knowledge/use-dateset.ts new file mode 100644 index 0000000000..e69de29bb2 diff --git a/web/service/knowledge/use-document.ts b/web/service/knowledge/use-document.ts new file mode 100644 index 0000000000..909f33ac67 --- /dev/null +++ b/web/service/knowledge/use-document.ts @@ -0,0 +1,25 @@ +import { + useQuery, +} from '@tanstack/react-query' +import { get } from '../base' +import type { SimpleDocumentDetail } from '@/models/datasets' + +const NAME_SPACE = 'knowledge/document' + +const useDocumentListKey = [NAME_SPACE, 'documentList'] +export const useDocumentList = (payload: { + datasetId: string + query: { + keyword: string + page: number + limit: number + } +}) => { + const { query, datasetId } = payload + return useQuery<{ data: SimpleDocumentDetail[] }>({ + queryKey: [...useDocumentListKey, datasetId, query], + queryFn: () => get<{ data: SimpleDocumentDetail[] }>(`/datasets/${datasetId}/documents`, { + params: query, + }), + }) +} diff --git a/web/service/knowledge/use-hit-testing.ts b/web/service/knowledge/use-hit-testing.ts new file mode 100644 index 0000000000..e69de29bb2 diff --git a/web/service/knowledge/use-import.ts b/web/service/knowledge/use-import.ts new file mode 100644 index 0000000000..e69de29bb2 diff --git a/web/service/knowledge/use-segement.ts b/web/service/knowledge/use-segement.ts new file mode 100644 index 0000000000..e69de29bb2