fix: terminate system should kill cortex process (#955)

janhq · Aug 1, 2024 · 49c428a · 49c428a
1 parent efa55f1
commit 49c428a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cortex-js/src/infrastructure/controllers/system.controller.ts b/cortex-js/src/infrastructure/controllers/system.controller.ts
@@ -30,13 +30,15 @@ import {
 } from 'rxjs';
 import { ResourcesManagerService } from '../services/resources-manager/resources-manager.service';
 import { ResourceEvent } from '@/domain/models/resource.interface';
+import { CortexUsecases } from '@/usecases/cortex/cortex.usecases';
 
 @ApiTags('System')
 @Controller('system')
 export class SystemController {
   constructor(
     private readonly downloadManagerService: DownloadManagerService,
     private readonly modelsUsecases: ModelsUsecases,
+    private readonly cortexUsecases: CortexUsecases,
     private readonly eventEmitter: EventEmitter2,
     private readonly resourcesManagerService: ResourcesManagerService,
   ) {}
@@ -47,6 +49,7 @@ export class SystemController {
   })
   @Delete()
   async delete() {
+    await this.cortexUsecases.stopCortex().catch(() => {});
     process.exit(0);
   }