chore: clean dangling process on exit and relaunch

kustomzone · Nov 6, 2024 · 56e35df · 56e35df
1 parent 24b7d64
commit 56e35df
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/extensions/inference-cortex-extension/src/index.ts b/extensions/inference-cortex-extension/src/index.ts
@@ -50,12 +50,18 @@ export default class JanInferenceCortexExtension extends LocalOAIEngine {
 
     // Run the process watchdog
     const systemInfo = await systemInformation()
+    await this.clean()
     await executeOnMain(NODE, 'run', systemInfo)
 
     this.queue.add(() => this.healthz())
+
+    window.addEventListener('beforeunload', () => {
+      this.clean()
+    })
   }
 
   onUnload(): void {
+    this.clean()
     executeOnMain(NODE, 'dispose')
     super.onUnload()
   }
@@ -132,6 +138,20 @@ export default class JanInferenceCortexExtension extends LocalOAIEngine {
       })
       .then(() => {})
   }
+
+  /**
+   * Clean cortex processes
+   * @returns
+   */
+  clean(): Promise<any> {
+    return ky
+      .delete(`${CORTEX_API_URL}/processmanager/destroy`, {
+        timeout: 2000, // maximum 2 seconds
+      })
+      .catch(() => {
+        // Do nothing
+      })
+  }
 }
 
 /// Legacy