mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-01 06:22:05 +08:00
Feat: Limit view with more knowledge when list knowledge so many (#6093)
### What problem does this PR solve? Limit view with more knowledge when list knowledge so many. ### Type of change - [x] Refactoring
This commit is contained in:
parent
7f701a5756
commit
8495036ff9
@ -3,10 +3,10 @@ import { useFetchKnowledgeList } from '@/hooks/knowledge-hooks';
|
|||||||
import { IRetrievalNode } from '@/interfaces/database/flow';
|
import { IRetrievalNode } from '@/interfaces/database/flow';
|
||||||
import { UserOutlined } from '@ant-design/icons';
|
import { UserOutlined } from '@ant-design/icons';
|
||||||
import { Handle, NodeProps, Position } from '@xyflow/react';
|
import { Handle, NodeProps, Position } from '@xyflow/react';
|
||||||
import { Avatar, Flex } from 'antd';
|
import { Avatar, Button, Flex } from 'antd';
|
||||||
import classNames from 'classnames';
|
import classNames from 'classnames';
|
||||||
import { get } from 'lodash';
|
import { get } from 'lodash';
|
||||||
import { useMemo } from 'react';
|
import { useMemo, useState } from 'react';
|
||||||
import { LeftHandleStyle, RightHandleStyle } from './handle-icon';
|
import { LeftHandleStyle, RightHandleStyle } from './handle-icon';
|
||||||
import styles from './index.less';
|
import styles from './index.less';
|
||||||
import NodeHeader from './node-header';
|
import NodeHeader from './node-header';
|
||||||
@ -20,6 +20,7 @@ export function RetrievalNode({
|
|||||||
const knowledgeBaseIds: string[] = get(data, 'form.kb_ids', []);
|
const knowledgeBaseIds: string[] = get(data, 'form.kb_ids', []);
|
||||||
const { theme } = useTheme();
|
const { theme } = useTheme();
|
||||||
const { list: knowledgeList } = useFetchKnowledgeList(true);
|
const { list: knowledgeList } = useFetchKnowledgeList(true);
|
||||||
|
const [showAllKnowledge, setShowAllKnowledge] = useState(false);
|
||||||
const knowledgeBases = useMemo(() => {
|
const knowledgeBases = useMemo(() => {
|
||||||
return knowledgeBaseIds.map((x) => {
|
return knowledgeBaseIds.map((x) => {
|
||||||
const item = knowledgeList.find((y) => x === y.id);
|
const item = knowledgeList.find((y) => x === y.id);
|
||||||
@ -31,6 +32,13 @@ export function RetrievalNode({
|
|||||||
});
|
});
|
||||||
}, [knowledgeList, knowledgeBaseIds]);
|
}, [knowledgeList, knowledgeBaseIds]);
|
||||||
|
|
||||||
|
const displayedKnowledgeBases = showAllKnowledge
|
||||||
|
? knowledgeBases
|
||||||
|
: knowledgeBases.slice(0, 3);
|
||||||
|
function showAllItem(e: any) {
|
||||||
|
e.stopPropagation(); // Prevent event from bubbling to parent
|
||||||
|
setShowAllKnowledge(!showAllKnowledge);
|
||||||
|
}
|
||||||
return (
|
return (
|
||||||
<section
|
<section
|
||||||
className={classNames(
|
className={classNames(
|
||||||
@ -66,7 +74,7 @@ export function RetrievalNode({
|
|||||||
})}
|
})}
|
||||||
></NodeHeader>
|
></NodeHeader>
|
||||||
<Flex vertical gap={8}>
|
<Flex vertical gap={8}>
|
||||||
{knowledgeBases.map((knowledge) => {
|
{displayedKnowledgeBases.map((knowledge) => {
|
||||||
return (
|
return (
|
||||||
<div className={styles.nodeText} key={knowledge.id}>
|
<div className={styles.nodeText} key={knowledge.id}>
|
||||||
<Flex align={'center'} gap={6}>
|
<Flex align={'center'} gap={6}>
|
||||||
@ -82,6 +90,25 @@ export function RetrievalNode({
|
|||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
})}
|
})}
|
||||||
|
{knowledgeBases.length > 3 && (
|
||||||
|
<div className={styles.nodeText}>
|
||||||
|
<Button
|
||||||
|
type="link"
|
||||||
|
size="small"
|
||||||
|
onClick={showAllItem}
|
||||||
|
style={{
|
||||||
|
padding: 0,
|
||||||
|
height: 'auto',
|
||||||
|
lineHeight: '1.5',
|
||||||
|
textAlign: 'left',
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{showAllKnowledge
|
||||||
|
? 'Hide'
|
||||||
|
: `Show more ${knowledgeBases.length - 3} knowledge`}
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</Flex>
|
</Flex>
|
||||||
</section>
|
</section>
|
||||||
);
|
);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user