update ai msg ui

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']

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

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(),
 
     }
 })