update

app/models/token_relation.py

@@ -9,6 +9,7 @@ class RelationType(str, Enum):
     File = "file"
     Thread = "thread"
     Action = "action"
+    Message = "message"
 
 
 class TokenRelationBase(BaseModel):

app/services/message/message.py

@@ -174,11 +174,12 @@ class MessageService:
     async def delete_message(
         *, session: AsyncSession, thread_id: str, message_id: str
     ) -> Message:
-        await MessageService.get_message(
+        message = await MessageService.get_message(
             session=session, thread_id=thread_id, message_id=message_id
         )
+        await session.delete(message)
         await auth_policy.delete_token_rel(
-            session=session, relation_type=RelationType.Action, relation_id=message_id
+            session=session, relation_type=RelationType.Message, relation_id=message_id
         )
         await session.commit()
         return DeleteResponse(id=message_id, object="message.deleted", deleted=True)