fix flag

Author: Daksh

stream_chat/async_chat/client.py

@@ -356,10 +356,16 @@ async def update_message_partial(
         return await self.put(f"messages/{message_id}", data=data)
 
     async def delete_message(
-        self, message_id: str, deleted_by: str = None, **options: Any
+        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 deleted_by:
+        if delete_for_me:
+            data["delete_for_me"] = True
+            data["deleted_by"] = deleted_by
+        elif deleted_by:
             data["deleted_by"] = deleted_by
 
         return await self.delete(f"messages/{message_id}", data)

stream_chat/base/client.py

@@ -558,16 +558,17 @@ def update_message_partial(
         """
         pass
 
-    @abc.abstractmethod
+        @abc.abstractmethod
     def delete_message(
-        self, message_id: str, deleted_by: str = None, **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
-            deleted_by: The user ID who is deleting the message
+            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

@@ -343,10 +343,16 @@ def update_message_partial(
         return self.put(f"messages/{message_id}", data=data)
 
     def delete_message(
-        self, message_id: str, deleted_by: str = None, **options: Any
+        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 deleted_by:
+        if delete_for_me:
+            data["delete_for_me"] = True
+            data["deleted_by"] = deleted_by
+        elif deleted_by:
             data["deleted_by"] = deleted_by
 
         return self.delete(f"messages/{message_id}", data)

stream_chat/tests/async_chat/test_client.py

@@ -462,17 +462,21 @@ async def test_delete_message(
         )
         await client.delete_message(msg_id, hard=True)
 
-    async def test_delete_message_with_deleted_by(
+    async def test_delete_message_for_me(
         self, client: StreamChatAsync, channel: Channel, random_user: Dict
     ):
-        """Test deleting a message with deleted_by parameter"""
+        """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 with deleted_by parameter
-        response = await client.delete_message(msg_id, deleted_by=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()

stream_chat/tests/test_client.py

@@ -509,15 +509,19 @@ 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_with_deleted_by(
+    def test_delete_message_for_me(
         self, client: StreamChat, channel, random_user: Dict
     ):
-        """Test deleting a message with deleted_by parameter"""
+        """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 with deleted_by parameter
-        response = client.delete_message(msg_id, deleted_by=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()
@@ -699,9 +703,10 @@ def test_query_channels_members_in(
     def test_create_blocklist(self, client: StreamChat):
         client.create_blocklist(name="Foo", words=["fudge", "heck"], type="word")
 
-    def test_list_blocklists(self, client: StreamChat):
+    def test test_list_blocklists(self, client: StreamChat):
         response = client.list_blocklists()
-        assert len(response["blocklists"]) == 2
+        # There are now 2 default blocklists + the "Foo" blocklist created by test_create_blocklist
+        assert len(response["blocklists"]) >= 3
         blocklist_names = {blocklist["name"] for blocklist in response["blocklists"]}
         assert "Foo" in blocklist_names