Remove unused code

Author: ajhai

llmstack/play/messages.py

@@ -8,7 +8,6 @@
 
 class MessageType(StrEnum):
     BEGIN = "BEGIN"
-    BOOKKEEPING = "BOOKKEEPING"
     CONTENT = "CONTENT"
     CONTENT_STREAM_CHUNK = "CONTENT_STREAM_CHUNK"
     CONTENT_STREAM_BEGIN = "CONTENT_STREAM_BEGIN"

llmstack/server/consumers.py

@@ -239,132 +239,6 @@ async def _respond_to_event(self, text_data):
         elif event == "stop":
             await self.stop()
 
-    async def _respond_to_event_old(self, text_data):
-        from llmstack.apps.apis import AppViewSet
-        from llmstack.apps.models import AppSessionFiles
-
-        json_data = json.loads(text_data)
-        id = json_data.get("id", None)
-        event = json_data.get("event", None)
-        request_uuid = str(uuid.uuid4())
-        self._session_id = self._session_id or json_data.get(
-            "session_id",
-            None,
-        )
-        self._user = self.scope.get("user", None)
-        self._session = self.scope.get("session", None)
-
-        # if event == "run":
-        #     try:
-        #         request = await _build_request_from_input({"input": input, "stream": True}, self.scope)
-        #         if is_ratelimited_fn(request, self._respond_to_event):
-        #             raise Ratelimited("Rate limit reached.")
-
-        #         output_stream, self._coordinator_ref = await self._run_app(request_uuid=request_uuid, request=request)
-        #         # Generate a uuid for the response
-        #         response_id = str(uuid.uuid4())
-
-        #         async for output in output_stream:
-        #             if "errors" in output or "session" in output:
-        #                 if "session" in output:
-        #                     self._session_id = output["session"]["id"]
-        #                 await self.send(text_data=json.dumps({**output, **{"reply_to": id}}))
-        #             else:
-        #                 await self.send(
-        #                     text_data=json.dumps(
-        #                         {"output": output, "reply_to": id, "id": response_id, "request_id": request_uuid}
-        #                     )
-        #                 )
-
-        #         await self.send(
-        #             text_data=json.dumps(
-        #                 {"event": "done", "reply_to": id, "id": response_id, "request_id": request_uuid}
-        #             )
-        #         )
-        #     except Ratelimited:
-        #         await self.send(
-        #             text_data=json.dumps({"event": "ratelimited", "reply_to": id, "request_id": request_uuid})
-        #         )
-        #     except UsageLimitReached:
-        #         await self.send(
-        #             text_data=json.dumps({"event": "usagelimited", "reply_to": id, "request_id": request_uuid})
-        #         )
-        #     except Exception as e:
-        #         logger.exception(e)
-        #         await self.send(text_data=json.dumps({"errors": [str(e)], "reply_to": id, "request_id": request_uuid}))
-
-        if event == "init":
-            # Create a new session and return the session id
-            self._session_id = await AppViewSet().init_app_async(self.app_id)
-            await self.send(text_data=json.dumps({"session": {"id": self._session_id}, "request_id": request_uuid}))
-
-        if event == "create_asset":
-            try:
-                # Create an asset in the session. Returns asset info for the other side to upload content to
-                session_created = False
-                if not self._session_id:
-                    self._session_id = await AppViewSet().init_app_async(self.app_id)
-                    session_created = True
-
-                asset_data = json_data.get("data", {})
-                asset_metadata = {
-                    "file_name": asset_data.get("file_name", str(uuid.uuid4())),
-                    "mime_type": asset_data.get("mime_type", "application/octet-stream"),
-                    "app_uuid": self.app_id,
-                    "username": (
-                        self._user.username
-                        if self._user and not self._user.is_anonymous
-                        else self._session.get("_prid", "")
-                    ),
-                }
-
-                asset = await sync_to_async(AppSessionFiles.create_asset)(
-                    asset_metadata, self._session_id, streaming=asset_data.get("streaming", False)
-                )
-
-                if not asset:
-                    await self.send(
-                        text_data=json.dumps(
-                            {
-                                "errors": ["Failed to create asset"],
-                                "reply_to": id,
-                                "request_id": request_uuid,
-                                "asset_request_id": id,
-                            }
-                        )
-                    )
-                    return
-
-                output = {
-                    "asset": asset.objref,
-                    "reply_to": id,
-                    "request_id": request_uuid,
-                    "asset_request_id": id,
-                }
-
-                if session_created:
-                    output["session"] = {"id": self._session_id}
-
-                await self.send(text_data=json.dumps(output))
-            except Exception as e:
-                logger.exception(e)
-                await self.send(
-                    text_data=json.dumps(
-                        {"errors": [str(e)], "reply_to": id, "request_id": request_uuid, "asset_request_id": id}
-                    )
-                )
-
-        if event == "delete_asset":
-            # Delete an asset in the session
-            if not self._session_id:
-                return
-
-            # TODO: Implement delete asset
-
-        if event == "stop":
-            if self._coordinator_ref:
-                self._coordinator_ref.stop()
-
     async def receive(self, text_data):
         self._event_response_task = run_coro_in_new_loop(self._respond_to_event(text_data), name="respond_to_event")