audrius/app-odoo
1
0
Fork 0
mirror of https://github.com/guohuadeng/app-odoo.git synced 2025-02-23 04:11:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

update ai msg ui

Browse Source
This commit is contained in:
Chill
2023-04-25 11:30:32 +08:00
parent 641b443391
commit 1810c81155
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app_chatgpt/models/mail_channel.py
View File

@@ -67,6 +67,7 @@ class Channel(models.Model):
if res: if res:
res = res.replace('\n', '<br/>') 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 = 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: if usage:
prompt_tokens = usage['prompt_tokens'] prompt_tokens = usage['prompt_tokens']
completion_tokens = usage['completion_tokens'] completion_tokens = usage['completion_tokens']

3
app_chatgpt/models/mail_message.py
View File

@@ -10,6 +10,8 @@ class Message(models.Model):
ai_completion_tokens = fields.Integer('AI Completion Tokens') ai_completion_tokens = fields.Integer('AI Completion Tokens')
cost_tokens = fields.Integer('Cost Tokens') cost_tokens = fields.Integer('Cost Tokens')
is_ai = fields.Boolean('Is Ai', default=False)
def _message_add_reaction(self, content): def _message_add_reaction(self, content):
super(Message, self)._message_add_reaction(content) super(Message, self)._message_add_reaction(content)
if self.create_uid.gpt_id: if self.create_uid.gpt_id:
@@ -24,4 +26,5 @@ class Message(models.Model):
message['human_prompt_tokens'] = message_sudo.human_prompt_tokens message['human_prompt_tokens'] = message_sudo.human_prompt_tokens
message['ai_completion_tokens'] = message_sudo.ai_completion_tokens message['ai_completion_tokens'] = message_sudo.ai_completion_tokens
message['cost_tokens'] = message_sudo.cost_tokens message['cost_tokens'] = message_sudo.cost_tokens
message['is_ai'] = message_sudo.is_ai
return message_values return message_values

4
app_chatgpt/static/src/models/message.js
View File

@@ -15,12 +15,16 @@ registerPatch({
if ('ai_completion_tokens' in data) { if ('ai_completion_tokens' in data) {
data2.ai_completion_tokens = data.ai_completion_tokens; data2.ai_completion_tokens = data.ai_completion_tokens;
} }
if ('is_ai' in data) {
data2.is_ai = data.is_ai;
}
return data2; return data2;
}, },
}, },
fields: { fields: {
human_prompt_tokens: attr(), human_prompt_tokens: attr(),
ai_completion_tokens: attr(), ai_completion_tokens: attr(),
is_ai: attr(),
} }
}) })