mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-02-04 01:25:07 +08:00
### What problem does this PR solve? Fixes: Fixed model provider issues and improved some features - Removed the old login page - Updated model provider icons - Added RAPTOR modification range parameter ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
@ -1,9 +1,11 @@
|
||||
import { IconMap } from '@/constants/llm';
|
||||
import { IconMap, LLMFactory } from '@/constants/llm';
|
||||
import { cn } from '@/lib/utils';
|
||||
import Icon, { UserOutlined } from '@ant-design/icons';
|
||||
import { IconComponentProps } from '@ant-design/icons/lib/components/Icon';
|
||||
import { Avatar } from 'antd';
|
||||
import { AvatarSize } from 'antd/es/avatar/AvatarContext';
|
||||
import { useMemo } from 'react';
|
||||
import { IconFontFill } from './icon-font';
|
||||
import { useIsDarkTheme } from './theme-provider';
|
||||
|
||||
const importAll = (requireContext: __WebpackModuleApi.RequireContext) => {
|
||||
@ -61,15 +63,45 @@ export const LlmIcon = ({
|
||||
size?: AvatarSize;
|
||||
imgClass?: string;
|
||||
}) => {
|
||||
const icon = IconMap[name as keyof typeof IconMap];
|
||||
const isDark = useIsDarkTheme();
|
||||
const themeIcons = [
|
||||
LLMFactory.FishAudio,
|
||||
LLMFactory.TogetherAI,
|
||||
LLMFactory.Meituan,
|
||||
];
|
||||
let icon = useMemo(() => {
|
||||
const icontemp = IconMap[name as keyof typeof IconMap];
|
||||
if (themeIcons.includes(name as LLMFactory)) {
|
||||
if (isDark) {
|
||||
return icontemp + '-dark';
|
||||
} else {
|
||||
return icontemp + '-bright';
|
||||
}
|
||||
}
|
||||
return icontemp;
|
||||
}, [name, isDark]);
|
||||
|
||||
const svgIcons = [
|
||||
LLMFactory.LocalAI,
|
||||
// LLMFactory.VolcEngine,
|
||||
LLMFactory.MiniMax,
|
||||
LLMFactory.Gemini,
|
||||
LLMFactory.StepFun,
|
||||
// LLMFactory.DeerAPI,
|
||||
];
|
||||
if (svgIcons.includes(name as LLMFactory)) {
|
||||
return (
|
||||
<SvgIcon
|
||||
name={`llm/${icon}`}
|
||||
width={width}
|
||||
height={height}
|
||||
imgClass={imgClass}
|
||||
></SvgIcon>
|
||||
);
|
||||
}
|
||||
|
||||
return icon ? (
|
||||
<SvgIcon
|
||||
name={`llm/${icon}`}
|
||||
width={width}
|
||||
height={height}
|
||||
imgClass={imgClass}
|
||||
></SvgIcon>
|
||||
<IconFontFill name={icon} className={cn('size-8', imgClass)} />
|
||||
) : (
|
||||
<Avatar shape="square" size={size} icon={<UserOutlined />} />
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user