Method delete

backend/app/operators/assistant/chat.py:51–81

(self, **kwargs)

        return chat

    async def delete(self, **kwargs) -> ModelEntity:
        kwargs = self._check_kwargs(object_id_required=True, **kwargs)
        assistant_id = kwargs.get("assistant_id")
        chat_id = kwargs.get("chat_id")

        async with self.postgres_pool.get_db_connection() as conn:
            async with conn.transaction():
                # get assistant
                assistant = await assistant_ops.get(
                    postgres_conn=conn,
                    assistant_id=assistant_id,
                )

                # get chat
                chat = await self.get(
                    postgres_conn=conn,
                    assistant_id=assistant_id,
                    chat_id=chat_id,
                )

                # delete chat
                await self._delete_entity(conn, chat)

                # update assistant num_chats
                await assistant_ops.update(
                    postgres_conn=conn,
                    assistant_id=assistant_id,
                    update_dict={"num_chats": assistant.num_chats - 1},
                )

        return assistant


chat_ops = ChatModelOperator(