feat: delete for me

Author: Daksh

stream_chat/async_chat/client.py

@@ -355,8 +355,16 @@ async def update_message_partial(
         data.update(options)
         return await self.put(f"messages/{message_id}", data=data)
 
-    async def delete_message(self, message_id: str, **options: Any) -> StreamResponse:
-        return await self.delete(f"messages/{message_id}", options)
+    async def delete_message(self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any) -> StreamResponse:
+        if delete_for_me and not deleted_by:
+            raise ValueError("deleted_by is required when delete_for_me is True")
+        
+        data = options.copy()
+        if delete_for_me:
+            data["delete_for_me"] = True
+            data["deleted_by"] = deleted_by
+        
+        return await self.delete(f"messages/{message_id}", data)
 
     async def undelete_message(self, message_id: str, user_id: str) -> StreamResponse:
         return await self.post(

stream_chat/base/client.py

@@ -560,10 +560,16 @@ def update_message_partial(
 
     @abc.abstractmethod
     def delete_message(
-        self, message_id: str, **options: Any
+        self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any
     ) -> Union[StreamResponse, Awaitable[StreamResponse]]:
         """
         Deletes a message.
+        
+        Args:
+            message_id: The ID of the message to delete
+            delete_for_me: If True, deletes the message only for the specified user
+            deleted_by: The user ID who is deleting the message (required when delete_for_me is True)
+            **options: Additional options to pass to the delete request
         """
         pass
 

stream_chat/client.py

@@ -342,8 +342,16 @@ def update_message_partial(
         data.update(options)
         return self.put(f"messages/{message_id}", data=data)
 
-    def delete_message(self, message_id: str, **options: Any) -> StreamResponse:
-        return self.delete(f"messages/{message_id}", options)
+    def delete_message(self, message_id: str, delete_for_me: bool = False, deleted_by: str = None, **options: Any) -> StreamResponse:
+        if delete_for_me and not deleted_by:
+            raise ValueError("deleted_by is required when delete_for_me is True")
+        
+        data = options.copy()
+        if delete_for_me:
+            data["delete_for_me"] = True
+            data["deleted_by"] = deleted_by
+        
+        return self.delete(f"messages/{message_id}", data)
 
     def undelete_message(self, message_id: str, user_id: str) -> StreamResponse:
         return self.post(

stream_chat/tests/async_chat/test_client.py

@@ -462,6 +462,28 @@ async def test_delete_message(
         )
         await client.delete_message(msg_id, hard=True)
 
+    async def test_delete_message_for_me(
+        self, client: StreamChatAsync, channel: Channel, random_user: Dict
+    ):
+        """Test deleting a message for a specific user (delete for me functionality)"""
+        msg_id = str(uuid.uuid4())
+        await channel.send_message(
+            {"id": msg_id, "text": "helloworld"}, random_user["id"]
+        )
+        
+        # Delete message for the user
+        response = await client.delete_message(
+            msg_id, 
+            delete_for_me=True, 
+            deleted_by=random_user["id"]
+        )
+        
+        # Verify the request succeeded
+        assert response.is_ok()
+        
+        # Verify the message is still visible to other users but deleted for the specified user
+        # (This would require additional API calls to verify, but we're just testing the request succeeds)
+
     async def test_flag_message(
         self, client: StreamChatAsync, channel: Channel, random_user, server_user: Dict
     ):

stream_chat/tests/test_client.py

@@ -509,6 +509,24 @@ def test_delete_message(self, client: StreamChat, channel, random_user: Dict):
         channel.send_message({"id": msg_id, "text": "helloworld"}, random_user["id"])
         client.delete_message(msg_id, hard=True)
 
+    def test_delete_message_for_me(self, client: StreamChat, channel, random_user: Dict):
+        """Test deleting a message for a specific user (delete for me functionality)"""
+        msg_id = str(uuid.uuid4())
+        channel.send_message({"id": msg_id, "text": "helloworld"}, random_user["id"])
+        
+        # Delete message for the user
+        response = client.delete_message(
+            msg_id, 
+            delete_for_me=True, 
+            deleted_by=random_user["id"]
+        )
+        
+        # Verify the request succeeded
+        assert response.is_ok()
+        
+        # Verify the message is still visible to other users but deleted for the specified user
+        # (This would require additional API calls to verify, but we're just testing the request succeeds)
+
     def test_flag_message(
         self, client: StreamChat, channel, random_user, server_user: Dict
     ):