close connection

huridocs · Oct 28, 2024 · f5d8231 · f5d8231
1 parent e44d40d
commit f5d8231
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/app/api/suggestions/routes.ts b/app/api/suggestions/routes.ts
@@ -30,6 +30,7 @@ async function processTrainFunction(
     return;
   }
 
+  await IX.stop();
   const status = await callback(new ObjectId(req.body.extractorId));
   res.json(status);
 }
@@ -147,6 +148,7 @@ export const suggestionsRoutes = (app: Application) => {
     async (req, res, _next) => {
       const IX = new InformationExtraction();
       await processTrainFunction(IX.trainModel, req, res);
+      await IX.stop();
     }
   );
 
@@ -158,6 +160,7 @@ export const suggestionsRoutes = (app: Application) => {
     async (req, res, _next) => {
       const IX = new InformationExtraction();
       await processTrainFunction(IX.status, req, res);
+      await IX.stop();
     }
   );