From b0b373d2fb1d27577c56b30918e10008995e21de Mon Sep 17 00:00:00 2001
From: Yi <yxiaoisme@gmail.com>
Date: Wed, 8 Jan 2025 14:50:14 +0800
Subject: [PATCH] fix: revert the modelInList check in the model icon component

---
 .../account-setting/model-provider-page/model-icon/index.tsx  | 4 +---
 .../model-parameter-modal/agent-model-trigger.tsx             | 1 -
 2 files changed, 1 insertion(+), 4 deletions(-)

diff --git a/web/app/components/header/account-setting/model-provider-page/model-icon/index.tsx b/web/app/components/header/account-setting/model-provider-page/model-icon/index.tsx
index 4c3f53a00b..0576672a4f 100644
--- a/web/app/components/header/account-setting/model-provider-page/model-icon/index.tsx
+++ b/web/app/components/header/account-setting/model-provider-page/model-icon/index.tsx
@@ -13,14 +13,12 @@ type ModelIconProps = {
   modelName?: string
   className?: string
   isDeprecated?: boolean
-  isInModelList?: boolean
 }
 const ModelIcon: FC<ModelIconProps> = ({
   provider,
   className,
   modelName,
   isDeprecated = false,
-  isInModelList = true,
 }) => {
   const language = useLanguage()
   if (provider?.provider.includes('openai') && modelName?.includes('gpt-4o'))
@@ -28,7 +26,7 @@ const ModelIcon: FC<ModelIconProps> = ({
   if (provider?.provider.includes('openai') && modelName?.startsWith('gpt-4'))
     return <div className='flex items-center justify-center'><OpenaiViolet className={cn('w-5 h-5', className)}/></div>
 
-  if (provider?.icon_small && isInModelList) {
+  if (provider?.icon_small) {
     return (
       <div className={`flex items-center justify-center w-5 h-5 ${isDeprecated ? 'opacity-50' : ''}`}>
         <img
diff --git a/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx b/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
index f38440837c..716914118d 100644
--- a/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
+++ b/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
@@ -112,7 +112,6 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
             provider={currentProvider || modelProvider}
             modelName={currentModel?.model || modelId}
             isDeprecated={hasDeprecated}
-            isInModelList={inModelList}
           />
           <ModelDisplay
             currentModel={currentModel}