feat: playground rerank

src/pages/llmodels/components/table-list.tsx

@@ -367,26 +367,29 @@ const Models: React.FC<ModelsProps> = ({
     setOpenAddModal(true);
   };
 
-  const handleSelect = useCallback((val: any, row: ListItem) => {
-    if (val === 'edit') {
-      handleEdit(row);
-    }
-    if (val === 'chat') {
-      handleOpenPlayGround(row);
-    }
-    if (val === 'delete') {
-      handleDelete(row);
-    }
-    if (val === 'embedding') {
-      setEmbeddingParams({
-        params: {
-          input: 'Your text string goes here',
-          model: row.name
-        },
-        show: true
-      });
-    }
-  }, []);
+  const handleSelect = useCallback(
+    (val: any, row: ListItem) => {
+      if (val === 'edit') {
+        handleEdit(row);
+      }
+      if (val === 'chat') {
+        handleOpenPlayGround(row);
+      }
+      if (val === 'delete') {
+        handleDelete(row);
+      }
+      if (val === 'embedding') {
+        setEmbeddingParams({
+          params: {
+            input: 'Your text string goes here',
+            model: row.name
+          },
+          show: true
+        });
+      }
+    },
+    [handleEdit, handleOpenPlayGround, handleDelete, setEmbeddingParams]
+  );
 
   const handleChildSelect = useCallback(
     (val: any, row: ModelInstanceListItem, list: ModelInstanceListItem[]) => {
@@ -524,7 +527,14 @@ const Models: React.FC<ModelsProps> = ({
                   <AutoTooltip ghost>
                     <span className="m-r-5">{text}</span>
                   </AutoTooltip>
-                  {record.embedding_only && (
+                  {record.reranker && (
+                    <span>
+                      <Tag style={{ margin: 0 }} color="geekblue">
+                        Reranker
+                      </Tag>
+                    </span>
+                  )}
+                  {record.embedding_only && !record.reranker && (
                     <span>
                       <Tag style={{ margin: 0 }} color="geekblue">
                         Embedding Only

src/pages/llmodels/components/update-modal.tsx

@@ -72,21 +72,24 @@ const UpdateModal: React.FC<AddModalProps> = (props) => {
     setGpuOptions(list);
   };
 
-  const initFormValue = useMemo(() => {
-    const result = setSourceRepoConfigValue(
-      props.data?.source || '',
-      props.data
-    );
+  useEffect(() => {
+    if (action === PageAction.EDIT && open) {
+      const result = setSourceRepoConfigValue(
+        props.data?.source || '',
+        props.data
+      );
 
-    return {
-      ...result.values,
-      ..._.omit(props.data, result.omits),
-      scheduleType: props.data?.gpu_selector ? 'manual' : 'auto',
-      gpu_selector: props.data?.gpu_selector
-        ? `${props.data?.gpu_selector.worker_name}-${props.data?.gpu_selector.gpu_name}-${props.data?.gpu_selector.gpu_index}`
-        : null
-    };
-  }, [props.data]);
+      const formData = {
+        ...result.values,
+        ..._.omit(props.data, result.omits),
+        scheduleType: props.data?.gpu_selector ? 'manual' : 'auto',
+        gpu_selector: props.data?.gpu_selector
+          ? `${props.data?.gpu_selector.worker_name}-${props.data?.gpu_selector.gpu_name}-${props.data?.gpu_selector.gpu_index}`
+          : null
+      };
+      form.setFieldsValue(formData);
+    }
+  }, [open]);
 
   useEffect(() => {
     setIsGGUF(props.data?.backend === backendOptionsMap.llamaBox);
@@ -245,6 +248,10 @@ const UpdateModal: React.FC<AddModalProps> = (props) => {
     }
   };
 
+  const handleOnClose = () => {
+    onCancel?.();
+  };
+
   useEffect(() => {
     getGPUList();
   }, []);
@@ -256,6 +263,7 @@ const UpdateModal: React.FC<AddModalProps> = (props) => {
       centered={true}
       onOk={handleSumit}
       onCancel={onCancel}
+      onClose={handleOnClose}
       destroyOnClose={true}
       closeIcon={true}
       maskClosable={false}
@@ -294,9 +302,6 @@ const UpdateModal: React.FC<AddModalProps> = (props) => {
             padding: 'var(--ant-modal-content-padding)',
             paddingBlock: 0
           }}
-          initialValues={{
-            ...initFormValue
-          }}
         >
           <Form.Item<FormData>
             name="name"