mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-02 18:41:01 +08:00
36 lines
766 B
TypeScript
36 lines
766 B
TypeScript
|
|
interface IconProps {
|
|
width?: number;
|
|
height?: number;
|
|
className?: string;
|
|
}
|
|
|
|
// 获取AI模型图标的函数
|
|
export const getModelIcon = (modelType: string, props: IconProps = {}) => {
|
|
// 支持完整ID或类型名
|
|
const type = modelType.includes('_') ? modelType.split('_').pop() : modelType;
|
|
|
|
let iconPath: string | null = null;
|
|
|
|
switch (type) {
|
|
case 'deepseek':
|
|
iconPath = '/icons/deepseek.svg';
|
|
break;
|
|
case 'qwen':
|
|
iconPath = '/icons/qwen.svg';
|
|
break;
|
|
default:
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<img
|
|
src={iconPath}
|
|
alt={`${type} icon`}
|
|
width={props.width || 24}
|
|
height={props.height || 24}
|
|
className={props.className}
|
|
style={{ borderRadius: '50%' }}
|
|
/>
|
|
);
|
|
}; |