lobe-chat

1
import { memo } from 'react';
2
import { Center } from 'react-layout-kit';
3

4
import FileIcon from '@/components/FileIcon';
5
import RepoIcon from '@/components/RepoIcon';
6
import { KnowledgeType } from '@/types/knowledgeBase';
7

8
interface KnowledgeIconProps {
9
  fileType?: string;
10
  name: string;
11
  size?: number | { file?: number; repo?: number };
12
  type: KnowledgeType;
13
}
14

15
const KnowledgeIcon = memo<KnowledgeIconProps>(({ type, size, fileType, name }) => {
16
  const repoSize = (typeof size === 'object' ? size.repo : size) || 24;
17
  const fileSize = (typeof size === 'object' ? size.file : size) || 24;
18

19
  return type === KnowledgeType.KnowledgeBase ? (
20
    <Center height={repoSize} width={repoSize}>
21
      <RepoIcon size={repoSize / 1.2} />
22
    </Center>
23
  ) : (
24
    <FileIcon fileName={name} fileType={fileType!} size={fileSize} />
25
  );
26
});
27

28
export default KnowledgeIcon;
29