From 2d116b1468e4783af17266c2a718061a4c1b8a1c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Fri, 19 May 2023 04:16:43 +0200 Subject: [PATCH] upgraded working with personality workflows --- gpt4all_api/api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/gpt4all_api/api.py b/gpt4all_api/api.py index 945ba8bb..68ea105a 100644 --- a/gpt4all_api/api.py +++ b/gpt4all_api/api.py @@ -253,7 +253,10 @@ class ModelProcess: if self.personality.processor_cfg is not None: if "custom_workflow" in self.personality.processor_cfg: if self.personality.processor_cfg["custom_workflow"]: - self.personality.processor.run_workflow(self._generate, command[1], command[0]) + output = self.personality.processor.run_workflow(self._generate, command[1], command[0]) + self._callback(output) + self.is_generating.value = 0 + continue self._generate(command[0], self._callback) while not self.generation_queue.empty():