From de48a1c7e9b6850bd89e958af91a5b5779e59a08 Mon Sep 17 00:00:00 2001
From: AkaraChen <akarachen@outlook.com>
Date: Fri, 27 Dec 2024 16:48:16 +0800
Subject: [PATCH] fix: type error

---
 web/app/components/workflow/nodes/agent/node.tsx | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/web/app/components/workflow/nodes/agent/node.tsx b/web/app/components/workflow/nodes/agent/node.tsx
index 3cfde2e64d..626f38b11f 100644
--- a/web/app/components/workflow/nodes/agent/node.tsx
+++ b/web/app/components/workflow/nodes/agent/node.tsx
@@ -72,10 +72,15 @@ const AgentNode: FC<NodeProps<AgentNodeType>> = (props) => {
     >
       {models.map((model) => {
         return <ModelSelector
-        // TODO: ugly
-          key={model?.param || Math.random()}
+          key={model.param}
           modelList={[]}
-          defaultModel={model || undefined}
+          defaultModel={
+            'provider' in model
+              ? {
+                provider: model.provider,
+                model: model.model,
+              }
+              : undefined}
           readonly
         />
       })}