update ai msg ui

2025-02-23 04:11:36 +02:00 · 2023-04-25 11:30:32 +08:00
parent 641b443391
commit 1810c81155
3 changed files with 8 additions and 0 deletions
--- 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']