diff --git a/app_chatgpt/models/mail_channel.py b/app_chatgpt/models/mail_channel.py
index e16cd950..a8d654de 100644
--- a/app_chatgpt/models/mail_channel.py
+++ b/app_chatgpt/models/mail_channel.py
@@ -67,6 +67,7 @@ class Channel(models.Model):
         if res:
             res = res.replace('\n', '<br/>')
             new_msg = channel.with_user(user_id).message_post(body=res, message_type='comment', subtype_xmlid='mail.mt_comment', parent_id=message.id)
+            new_msg.write({'is_ai': True})
             if usage:
                 prompt_tokens = usage['prompt_tokens']
                 completion_tokens = usage['completion_tokens']
diff --git a/app_chatgpt/models/mail_message.py b/app_chatgpt/models/mail_message.py
index 3de5ae14..a6e2d2c4 100644
--- a/app_chatgpt/models/mail_message.py
+++ b/app_chatgpt/models/mail_message.py
@@ -10,6 +10,8 @@ class Message(models.Model):
     ai_completion_tokens = fields.Integer('AI Completion Tokens')
     cost_tokens = fields.Integer('Cost Tokens')
 
+    is_ai = fields.Boolean('Is Ai', default=False)
+
     def _message_add_reaction(self, content):
         super(Message, self)._message_add_reaction(content)
         if self.create_uid.gpt_id:
@@ -24,4 +26,5 @@ class Message(models.Model):
             message['human_prompt_tokens'] = message_sudo.human_prompt_tokens
             message['ai_completion_tokens'] = message_sudo.ai_completion_tokens
             message['cost_tokens'] = message_sudo.cost_tokens
+            message['is_ai'] = message_sudo.is_ai
         return message_values
diff --git a/app_chatgpt/static/src/models/message.js b/app_chatgpt/static/src/models/message.js
index 4c003c52..4d715392 100644
--- a/app_chatgpt/static/src/models/message.js
+++ b/app_chatgpt/static/src/models/message.js
@@ -15,12 +15,16 @@ registerPatch({
             if ('ai_completion_tokens' in data) {
                 data2.ai_completion_tokens = data.ai_completion_tokens;
             }
+            if ('is_ai' in data) {
+                data2.is_ai = data.is_ai;
+            }
             return data2;
         },
     },
     fields: {
         human_prompt_tokens: attr(),
         ai_completion_tokens: attr(),
+        is_ai: attr(),
 
     }
 })
\ No newline at end of file