Stop and disconnect on stop event

trypromptly · Oct 23, 2024 · dd0964f · dd0964f
1 parent b76a7d7
commit dd0964f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/llmstack/server/consumers.py b/llmstack/server/consumers.py
@@ -137,6 +137,9 @@ async def disconnect(self, close_code):
         if self._app_runner:
             await self._app_runner.stop()
 
+    async def stop(self):
+        await self.close()
+
     async def _respond_to_event(self, text_data):
         json_data = json.loads(text_data)
         client_request_id = json_data.get("id", None)
@@ -227,7 +230,7 @@ async def _respond_to_event(self, text_data):
 
             # TODO: Implement delete asset
         elif event == "stop":
-            self.disconnect()
+            await self.stop()
 
     async def _respond_to_event_old(self, text_data):
         from llmstack.apps.apis import AppViewSet