From 3e691aa31fb7f5a807b10f64acc89ca82a7e2b0e Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Wed, 5 Mar 2025 14:04:13 +0800
Subject: [PATCH] fix: default model in embedding page

---
 .../components/ground-embedding.tsx           | 34 +++++++++++++------
 .../playground/components/ground-reranker.tsx | 34 +++++++++++++------
 src/pages/playground/hooks/use-init-meta.ts   |  8 +++--
 3 files changed, 51 insertions(+), 25 deletions(-)

diff --git a/src/pages/playground/components/ground-embedding.tsx b/src/pages/playground/components/ground-embedding.tsx
index 45cc8b06..04fbbdd7 100644
--- a/src/pages/playground/components/ground-embedding.tsx
+++ b/src/pages/playground/components/ground-embedding.tsx
@@ -111,11 +111,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
     paramsRef,
     modelMeta,
     formFields
-  } = useInitLLmMeta(props, {
-    defaultValues: {},
-    defaultParamsConfig: [],
-    metaKeys: LLM_METAKEYS
-  });
+  } = useInitLLmMeta(
+    {
+      modelList,
+      isChat: true
+    },
+    {
+      defaultValues: {},
+      defaultParamsConfig: [],
+      metaKeys: LLM_METAKEYS
+    }
+  );
 
   useImperativeHandle(ref, () => {
     return {
@@ -429,11 +435,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
     ];
   }, [outputHeight, scatterData, embeddingData]);
 
-  useEffect(() => {
-    setMessageId();
-    setScatterData([]);
-    setTokenResult(null);
-  }, [parameters.model]);
+  const onValuesChange = useCallback(
+    (changeValues: Record<string, any>, allValues: Record<string, any>) => {
+      if (changeValues.model) {
+        setScatterData([]);
+        setTokenResult(null);
+      } else {
+        handleOnValuesChange(changeValues, allValues);
+      }
+    },
+    [handleOnValuesChange]
+  );
 
   useHotkeys(
     HotKeys.SUBMIT,
@@ -709,7 +721,7 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
         <div className="box">
           <DynamicParams
             ref={formRef}
-            onValuesChange={handleOnValuesChange}
+            onValuesChange={onValuesChange}
             paramsConfig={paramsConfig}
             initialValues={initialValues}
             modelList={modelList}
diff --git a/src/pages/playground/components/ground-reranker.tsx b/src/pages/playground/components/ground-reranker.tsx
index fc19685f..df4ffd55 100644
--- a/src/pages/playground/components/ground-reranker.tsx
+++ b/src/pages/playground/components/ground-reranker.tsx
@@ -129,11 +129,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
     paramsRef,
     modelMeta,
     formFields
-  } = useInitLLmMeta(props, {
-    defaultValues: { top_n: 3 },
-    defaultParamsConfig: fieldConfig,
-    metaKeys: LLM_METAKEYS
-  });
+  } = useInitLLmMeta(
+    {
+      modelList,
+      isChat: true
+    },
+    {
+      defaultValues: { top_n: 3 },
+      defaultParamsConfig: fieldConfig,
+      metaKeys: LLM_METAKEYS
+    }
+  );
 
   useImperativeHandle(ref, () => {
     return {
@@ -425,6 +431,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
     setIsEmptyText(false);
   };
 
+  const onValuesChange = useCallback(
+    (changedValues: any, allValues: any) => {
+      if (changedValues.model) {
+        setTokenResult(null);
+      } else {
+        handleOnValuesChange(changedValues, allValues);
+      }
+    },
+    [handleOnValuesChange]
+  );
+
   useHotkeys(
     HotKeys.SUBMIT,
     (e: any) => {
@@ -437,11 +454,6 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
     }
   );
 
-  useEffect(() => {
-    setMessageId();
-    setTokenResult(null);
-  }, [parameters.model]);
-
   useEffect(() => {
     if (scroller.current) {
       initialize(scroller.current);
@@ -591,7 +603,7 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
         <div className="box">
           <DynamicParams
             ref={formRef}
-            onValuesChange={handleOnValuesChange}
+            onValuesChange={onValuesChange}
             paramsConfig={paramsConfig}
             initialValues={initialValues}
             modelList={modelList}
diff --git a/src/pages/playground/hooks/use-init-meta.ts b/src/pages/playground/hooks/use-init-meta.ts
index 47d16d6f..48037e8d 100644
--- a/src/pages/playground/hooks/use-init-meta.ts
+++ b/src/pages/playground/hooks/use-init-meta.ts
@@ -109,7 +109,7 @@ export const useInitLLmMeta = (
 
   const handleOnModelChange = useCallback(
     (val: string) => {
-      if (!val) return;
+      if (!val || val === parameters.model) return;
       const model = modelList.find((item) => item.value === val);
       const { form: initialData, meta } = extractLLMMeta(model?.meta);
 
@@ -135,7 +135,7 @@ export const useInitLLmMeta = (
       });
       setParamsConfig(config);
     },
-    [modelList, defaultParamsConfig]
+    [modelList, parameters, defaultParamsConfig]
   );
 
   const handleOnValuesChange = useCallback(
@@ -143,8 +143,10 @@ export const useInitLLmMeta = (
       if (changeValues.model) {
         handleOnModelChange(changeValues.model);
         return;
+      } else {
+        setParams(allValues);
+        setInitialValues(allValues);
       }
-      setParams(allValues);
     },
     [handleOnModelChange]
   );