Spaces:
Runtime error
Runtime error
File size: 3,611 Bytes
3b6afc0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
import { Plugin, GPTIcon, AnthropicIcon } from '~/components/svg';
import { useAuthContext } from '~/hooks/AuthContext';
import { cn } from '~/utils';
const getIcon = (props) => {
const { size = 30, isCreatedByUser, button, model, message = true } = props;
// eslint-disable-next-line react-hooks/rules-of-hooks
const { user } = useAuthContext();
if (isCreatedByUser) {
return (
<div
title={user?.name || 'User'}
style={{
width: size,
height: size,
}}
className={'relative flex items-center justify-center' + props?.className}
>
<img
className="rounded-sm"
src={
user?.avatar ||
`https://api.dicebear.com/6.x/initials/svg?seed=${
user?.name || 'User'
}&fontFamily=Verdana&fontSize=36`
}
alt="avatar"
/>
</div>
);
} else if (!isCreatedByUser) {
const { endpoint, error } = props;
let icon, bg, name;
if (endpoint === 'azureOpenAI') {
const { chatGptLabel } = props;
icon = <GPTIcon size={size * 0.7} />;
bg = 'linear-gradient(0.375turn, #61bde2, #4389d0)';
name = chatGptLabel || 'ChatGPT';
} else if (endpoint === 'openAI' || (endpoint === 'gptPlugins' && message)) {
const { chatGptLabel } = props;
icon = <GPTIcon size={size * 0.7} />;
bg =
model && model.toLowerCase().startsWith('gpt-4')
? '#AB68FF'
: chatGptLabel
? `rgba(16, 163, 127, ${button ? 0.75 : 1})`
: `rgba(16, 163, 127, ${button ? 0.75 : 1})`;
name = chatGptLabel || 'ChatGPT';
} else if (endpoint === 'gptPlugins' && !message) {
icon = <Plugin size={size * 0.7} />;
bg = `rgba(69, 89, 164, ${button ? 0.75 : 1})`;
name = 'Plugins';
} else if (endpoint === 'google') {
const { modelLabel } = props;
icon = <img src="/assets/google-palm.svg" alt="Palm Icon" />;
name = modelLabel || 'PaLM2';
} else if (endpoint === 'anthropic') {
const { modelLabel } = props;
icon = <AnthropicIcon size={size * 0.7} />;
bg = '#d09a74';
name = modelLabel || 'Claude';
} else if (endpoint === 'bingAI') {
const { jailbreak } = props;
if (jailbreak) {
icon = <img src="/assets/bingai-jb.png" alt="Bing Icon"/>;
name = 'Sydney';
} else {
icon = <img src="/assets/bingai.png" alt="Sydney Icon"/>;
name = 'BingAI';
}
} else if (endpoint === 'chatGPTBrowser') {
icon = <GPTIcon size={size * 0.7} />;
bg =
model && model.toLowerCase().startsWith('gpt-4')
? '#AB68FF'
: `rgba(0, 163, 255, ${button ? 0.75 : 1})`;
name = 'ChatGPT';
} else if (endpoint === null) {
icon = <GPTIcon size={size * 0.7} />;
bg = 'grey';
name = 'N/A';
} else {
icon = <GPTIcon size={size * 0.7} />;
bg = 'grey';
name = 'UNKNOWN';
}
return (
<div
title={name}
style={{
background: bg || 'transparent',
width: size,
height: size,
}}
className={cn(
'relative flex items-center justify-center rounded-sm text-white ',
props?.className ?? '',
)}
>
{icon}
{error && (
<span className="absolute right-0 top-[20px] -mr-2 flex h-4 w-4 items-center justify-center rounded-full border border-white bg-red-500 text-[10px] text-white">
!
</span>
)}
</div>
);
}
};
export default getIcon;
|