feat: check for driver when returning status

F33RNI · May 15, 2024 · 7214dae · 7214dae
1 parent 6bab1a4
commit 7214dae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/lmao/module_wrapper.py b/src/lmao/module_wrapper.py
@@ -197,7 +197,7 @@ def ask(self, request: Dict) -> Generator[Dict]:
 
         # Clear status back to IDLE
         finally:
-            self.status = STATUS_IDLE
+            self.status = STATUS_IDLE if self._module_class is not None else STATUS_NOT_INITIALIZED
 
     def response_stop(self) -> None:
         """Wrapper for response_stop() function
@@ -211,7 +211,7 @@ def response_stop(self) -> None:
 
         # Clear status back to IDLE
         finally:
-            self.status = STATUS_IDLE
+            self.status = STATUS_IDLE if self._module_class is not None else STATUS_NOT_INITIALIZED
 
     def delete_conversation(self, conversation: Dict) -> None:
         """Wrapper for conversation_delete() function
@@ -231,4 +231,4 @@ def delete_conversation(self, conversation: Dict) -> None:
 
         # Clear status back to IDLE
         finally:
-            self.status = STATUS_IDLE
+            self.status = STATUS_IDLE if self._module_class is not None else STATUS_NOT_INITIALIZED