detect status

frontend/src/apis/index.ts

@@ -1,4 +1,4 @@
-import commonStore from '../stores/commonStore';
+import commonStore, {ModelStatus} from '../stores/commonStore';
 
 export const readRoot = async () => {
   const port = commonStore.getCurrentModelConfig().apiParameters.apiPort;
@@ -35,3 +35,17 @@ export const updateConfig = async (body: any) => {
     body: JSON.stringify(body)
   });
 };
+
+export const getStatus = async (timeout?: number): Promise<ModelStatus | undefined> => {
+  const controller = new AbortController();
+  if (timeout)
+    setTimeout(() => controller.abort(), timeout);
+
+  const port = commonStore.getCurrentModelConfig().apiParameters.apiPort;
+  let ret: ModelStatus | undefined;
+  await fetch(`http://127.0.0.1:${port}/status`, {signal: controller.signal}).then(r => r.json()).then(data => {
+    ret = data.status;
+  }).catch(() => {
+  });
+  return ret;
+};

frontend/src/startup.ts

@@ -1,11 +1,16 @@
 import commonStore, {defaultModelConfigs} from './stores/commonStore';
 import {ReadJson} from '../wailsjs/go/backend_golang/App';
 import {downloadProgramFiles, LocalConfig, refreshModels} from './utils';
+import {getStatus} from './apis';
 
 export async function startup() {
   downloadProgramFiles();
   initCache();
   await initConfig();
+  getStatus(500).then(status => {
+    if (status)
+      commonStore.setModelStatus(status);
+  });
 }
 
 async function initConfig() {