mirror of
https://github.com/guohuadeng/app-odoo.git
synced 2025-02-23 04:11:36 +02:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from odoo import fields, models
|
||
|
||
|
||
class Message(models.Model):
|
||
_inherit = "mail.message"
|
||
|
||
human_prompt_tokens = fields.Integer('Human Prompt Tokens')
|
||
ai_completion_tokens = fields.Integer('AI Completion Tokens')
|
||
cost_tokens = fields.Integer('Cost Tokens')
|
||
# 是否ai回复
|
||
is_ai = fields.Boolean('Is Ai', default=False)
|
||
# 得到 ai 响应后，需要特殊处理ai的
|
||
ai2model = fields.Char('Ai Response model')
|
||
ai2id = fields.Integer('Ai Response id')
|
||
|
||
def _message_add_reaction(self, content):
|
||
super(Message, self)._message_add_reaction(content)
|
||
if self.create_uid.gpt_id:
|
||
# 处理反馈
|
||
pass
|
||
|
||
def message_format(self, format_reply=True, msg_vals=None):
|
||
message_values = super(Message, self).message_format(format_reply=format_reply, msg_vals=msg_vals)
|
||
for message in message_values:
|
||
message_sudo = self.browse(message['id']).sudo().with_prefetch(self.ids)
|
||
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
|