From 1c82e3870a55e7c5439a9aca3791d02e5fc9b4a8 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Fri, 8 Mar 2024 11:06:32 +0800
Subject: [PATCH] feat: not choose model hide in node

---
 web/app/components/workflow/nodes/llm/node.tsx     | 13 ++++++++-----
 .../workflow/nodes/question-classifier/node.tsx    | 14 +++++++++-----
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/web/app/components/workflow/nodes/llm/node.tsx b/web/app/components/workflow/nodes/llm/node.tsx
index 69372a4a54..521ac05dd4 100644
--- a/web/app/components/workflow/nodes/llm/node.tsx
+++ b/web/app/components/workflow/nodes/llm/node.tsx
@@ -14,13 +14,16 @@ const Node: FC<NodeProps<LLMNodeType>> = ({
   const {
     textGenerationModelList,
   } = useTextGenerationCurrentProviderAndModelAndModelList()
+  const hasSetModel = provider && modelId
   return (
     <div className='px-3'>
-      <ModelSelector
-        defaultModel={(provider || modelId) ? { provider, model: modelId } : undefined}
-        modelList={textGenerationModelList}
-        readonly
-      />
+      {hasSetModel && (
+        <ModelSelector
+          defaultModel={{ provider, model: modelId }}
+          modelList={textGenerationModelList}
+          readonly
+        />
+      )}
     </div>
   )
 }
diff --git a/web/app/components/workflow/nodes/question-classifier/node.tsx b/web/app/components/workflow/nodes/question-classifier/node.tsx
index 431c1ba810..8680880d8e 100644
--- a/web/app/components/workflow/nodes/question-classifier/node.tsx
+++ b/web/app/components/workflow/nodes/question-classifier/node.tsx
@@ -22,13 +22,17 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => {
   const {
     textGenerationModelList,
   } = useTextGenerationCurrentProviderAndModelAndModelList()
+  const hasSetModel = provider && modelId
+
   return (
     <div className='px-3'>
-      <ModelSelector
-        defaultModel={(provider || modelId) ? { provider, model: modelId } : undefined}
-        modelList={textGenerationModelList}
-        readonly
-      />
+      {hasSetModel && (
+        <ModelSelector
+          defaultModel={{ provider, model: modelId }}
+          modelList={textGenerationModelList}
+          readonly
+        />
+      )}
       <div className='mt-2 space-y-0.5'>
         {topics.map((topic, index) => (
           <div