fix: did not cache workers and clusters in model files

Author: hibig

src/atoms/models.ts

@@ -20,17 +20,30 @@ export const getRequestId = () => {
   return store.get(requestIdAtom);
 };
 
+// store for cluster list: res.items from api
 export const clusterListAtom = atom<
   {
     label: string;
     value: number;
+    provider: string;
+    state: string;
+    is_default: boolean;
+    workers: number;
+    ready_workers: number;
+    gpus: number;
   }[]
 >([]);
 
+// store for worker list: res.items from api
 export const workerListAtom = atom<
   {
     label: string;
     value: number;
+    cluster_id: number;
+    state: string;
+    id: number;
+    labels: Record<string, any>;
+    name: string;
   }[]
 >([]);
 

src/pages/llmodels/hooks/use-form-initial-values.ts

@@ -153,6 +153,8 @@ export const useGenerateWorkerOptions = () => {
       { state: string; labels: Record<string, string>; cluster_id: number }
     >[]
   >([]);
+  const [, setClusterListAtom] = useAtom(clusterListAtom);
+  const [, setWorkerListAtom] = useAtom(workerListAtom);
 
   const generateCascaderWorkerOptions = (
     workerList: WorkerListItem[],
@@ -201,29 +203,31 @@ export const useGenerateWorkerOptions = () => {
     const data = await getDataList();
     const [workerList, clusterList] = data;
     generateCascaderWorkerOptions(workerList, clusterList);
-    setWorkersList(
-      workerList.map((item) => ({
-        cluster_id: item.cluster_id,
-        state: item.state,
-        label: item.name,
-        value: item.id,
-        id: item.id,
-        labels: item.labels || {},
-        name: item.name
-      }))
-    );
-    setClusterList(
-      clusterList.map((item) => ({
-        label: item.name,
-        value: item.id,
-        provider: item.provider as string,
-        state: item.state,
-        is_default: item.is_default,
-        workers: item.workers,
-        ready_workers: item.ready_workers,
-        gpus: item.gpus
-      }))
-    );
+
+    const workerOptions = workerList.map((item) => ({
+      cluster_id: item.cluster_id,
+      state: item.state,
+      label: item.name,
+      value: item.id,
+      id: item.id,
+      labels: item.labels || {},
+      name: item.name
+    }));
+    const clusterOptions = clusterList.map((item) => ({
+      label: item.name,
+      value: item.id,
+      provider: item.provider as string,
+      state: item.state,
+      is_default: item.is_default,
+      workers: item.workers,
+      ready_workers: item.ready_workers,
+      gpus: item.gpus
+    }));
+
+    setWorkersList(workerOptions);
+    setClusterList(clusterOptions);
+    setWorkerListAtom(workerOptions);
+    setClusterListAtom(clusterOptions);
   };
   return {
     getWorkerOptionList,
@@ -280,7 +284,12 @@ export default function useFormInitialValues() {
       const list =
         data.items?.map((item) => ({
           label: item.name,
-          value: item.id
+          value: item.id,
+          cluster_id: item.cluster_id,
+          state: item.state,
+          id: item.id,
+          labels: item.labels || {},
+          name: item.name
         })) || [];
       setWorkerList(data.items);
       setWorkerListAtom(list);