# -*- coding: utf-8 -*- import openai import requests, json import datetime # from transformers import TextDavinciTokenizer, TextDavinciModel from odoo import api, fields, models, tools, _ from odoo.exceptions import UserError import logging _logger = logging.getLogger(__name__) class Channel(models.Model): _inherit = 'mail.channel' def get_openai_context(self, channel_id, author_id, answer_id, minutes=30, chat_count=1): # 上下文处理，要处理群的方式，以及独聊的方式 # azure新api 处理 context_history = [] afterTime = fields.Datetime.now() - datetime.timedelta(minutes=minutes) message_model = self.env['mail.message'].sudo() # 处理消息： 取最新问题 + 上 chat_count=1次的交互，将之前的交互按时间顺序拼接。 # 注意： ai 每一次回复都有 parent_id 来处理连续性 # 私聊处理 # todo: 更好的处理方式 domain = [('res_id', '=', channel_id), ('model', '=', 'mail.channel'), ('message_type', '!=', 'user_notification'), ('parent_id', '!=', False), ('author_id', '=', answer_id.id), ('body', '!=', '

%s

' % _('Response Timeout, please speak again.')), ('body', '!=', _('温馨提示：您发送的内容含有敏感词，请修改内容后再向我发送。'))] if self.channel_type in ['group', 'channel']: # 群聊增加时间限制，当前找所有人，不限制 author_id domain += [('date', '>=', afterTime)] ai_msg_list = message_model.with_context(tz='UTC').search(domain, order="id desc", limit=chat_count) for ai_msg in ai_msg_list: # 判断这个 ai_msg 是不是ai发，有才 insert。 判断 user_msg 是不是 user发的，有才 insert user_msg = ai_msg.parent_id.sudo() if ai_msg.author_id.sudo().gpt_id: ai_content = str(ai_msg.body).replace("

", "").replace("

", "").replace("

", "") context_history.insert(0, { 'role': 'assistant', 'content': ai_content, }) if not user_msg.author_id.gpt_id: user_content = user_msg.description.replace("

", "").replace("

", "").replace('@%s' % answer_id.name, '').lstrip() context_history.insert(0, { 'role': 'user', 'content': user_content, }) return context_history def get_ai_config(self, ai): # 勾子，用于取ai 配置 return {} def get_ai_response(self, ai, messages, channel, user_id, message): author_id = message.create_uid.partner_id answer_id = user_id.partner_id # todo: 只有个人配置的群聊才给配置 param = self.get_ai_config(ai) res, usage, is_ai = ai.get_ai(messages, author_id, answer_id, param) if res: res = res.replace('\n', '
') new_msg = channel.with_user(user_id).message_post(body=res, message_type='comment', subtype_xmlid='mail.mt_comment', parent_id=message.id) if usage: prompt_tokens = usage['prompt_tokens'] completion_tokens = usage['completion_tokens'] total_tokens = usage['total_tokens'] new_msg.write({ 'human_prompt_tokens': prompt_tokens, 'ai_completion_tokens': completion_tokens, 'cost_tokens': total_tokens, }) def _notify_thread(self, message, msg_vals=False, **kwargs): rdata = super(Channel, self)._notify_thread(message, msg_vals=msg_vals, **kwargs) # print(f'rdata:{rdata}') answer_id = self.env['res.partner'] user_id = self.env['res.users'] author_id = msg_vals.get('author_id') ai = self.env['ai.robot'].sudo() channel = self.env['mail.channel'] channel_type = self.channel_type messages = [] # 不处理 一般notify，但处理欢迎 if '
joined the channel
'): return rdata if channel_type == 'chat': channel_partner_ids = self.channel_partner_ids answer_id = channel_partner_ids - message.author_id user_id = answer_id.mapped('user_ids').sudo().filtered(lambda r: r.gpt_id)[:1] if user_id and answer_id.gpt_id: gpt_policy = user_id.gpt_policy gpt_wl_partners = user_id.gpt_wl_partners is_allow = message.author_id.id in gpt_wl_partners.ids if gpt_policy == 'all' or (gpt_policy == 'limit' and is_allow): ai = answer_id.sudo().gpt_id elif channel_type in ['group', 'channel']: # partner_ids = @ ids partner_ids = list(msg_vals.get('partner_ids')) if hasattr(self, 'ai_partner_id') and self.ai_partner_id: # 当有主id时，使用主id if self.ai_partner_id.id in partner_ids: partner_ids = [self.ai_partner_id.id] if partner_ids: # 常规群聊 @ partners = self.env['res.partner'].search([('id', 'in', partner_ids)]) # user_id = user, who has binded gpt robot user_id = partners.mapped('user_ids').sudo().filtered(lambda r: r.gpt_id)[:1] elif message.body == _('
joined the channel
'): # 欢迎的情况 partners = self.channel_partner_ids.sudo().filtered(lambda r: r.gpt_id)[:1] user_id = partners.mapped('user_ids')[:1] elif self.member_count == 2: # 处理独聊频道 if hasattr(self, 'is_private') and not self.is_private: # 2个人的非私有频道不处理 pass else: partners = self.channel_partner_ids.sudo().filtered(lambda r: r.gpt_id)[:1] user_id = partners.mapped('user_ids')[:1] elif not message.author_id.gpt_id: # 没有@时，默认第一个robot # robot = self.env.ref('app_chatgpt.chatgpt_robot') # 临时用azure if hasattr(self, 'ai_partner_id') and self.ai_partner_id: # 当有主id时，使用主id user_id = self.ai_partner_id.mapped('user_ids')[:1] else: # 使用群里的第一个robot partners = self.channel_partner_ids.sudo().filtered(lambda r: r.gpt_id)[:1] user_id = partners.mapped('user_ids')[:1] if user_id: # todo: 此处理不判断，将此处逻辑迁移至 get_ai_pre， 非ai回复的直接内容注意设置为 is_ai=false gpt_policy = user_id.gpt_policy gpt_wl_partners = user_id.gpt_wl_partners is_allow = message.author_id.id in gpt_wl_partners.ids answer_id = user_id.partner_id if gpt_policy == 'all' or (gpt_policy == 'limit' and is_allow): ai = user_id.sudo().gpt_id elif user_id.gpt_id and not is_allow: # 暂时有限用户的Ai raise UserError(_('此Ai暂时未开放，请联系管理员。')) chatgpt_channel_id = self.env.ref('app_chatgpt.channel_chatgpt') if message.body == _('
joined the channel
'): msg = _("Please warmly welcome our new partner %s and send him the best wishes.") % message.author_id.name else: # 不能用 preview， 如果用 : 提示词则 preview信息丢失 plaintext_ct = tools.html_to_inner_content(message.body) msg = plaintext_ct.replace('@%s' % answer_id.name, '').lstrip() if not msg: return rdata # api_key = self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openapi_api_key') # ai处理，不要自问自答 if ai and answer_id != message.author_id: api_key = ai.openapi_api_key if not api_key: _logger.warning(_("ChatGPT Robot【%s】have not set open api key.")) return rdata try: openapi_context_timeout = int(self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openapi_context_timeout')) or 60 except: openapi_context_timeout = 60 sync_config = self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openai_sync_config') openai.api_key = api_key # 非4版本，取0次。其它取3 次历史 if '4' in ai.ai_model: chat_count = 1 if self.chat_count > 0 else 0 else: chat_count = self.chat_count if author_id != answer_id.id and self.channel_type == 'chat': # 私聊 _logger.info(f'私聊:author_id:{author_id},partner_chatgpt.id:{answer_id.id}') channel = self.env[msg_vals.get('model')].browse(msg_vals.get('res_id')) elif author_id != answer_id.id and msg_vals.get('model', '') == 'mail.channel' and msg_vals.get('res_id', 0) == chatgpt_channel_id.id: # todo: 公开的群聊，当前只开1个，后续更多 _logger.info(f'频道群聊:author_id:{author_id},partner_chatgpt.id:{answer_id.id}') channel = chatgpt_channel_id elif author_id != answer_id.id and msg_vals.get('model', '') == 'mail.channel' and self.channel_type in ['group', 'channel']: # 高级用户自建的话题 channel = self.env[msg_vals.get('model')].browse(msg_vals.get('res_id')) if hasattr(channel, 'is_private') and channel.description: messages.append({"role": "system", "content": channel.description}) try: c_history = self.get_openai_context(channel.id, author_id, answer_id, openapi_context_timeout, chat_count) if c_history: messages += c_history messages.append({"role": "user", "content": msg}) msg_len = sum(len(str(m)) for m in messages) # 接口最大接收 8430 Token if msg_len * 2 >= 8000: messages = [{"role": "user", "content": msg}] if sync_config == 'sync': self.get_ai_response(ai, messages, channel, user_id, message) else: self.with_delay().get_ai_response(ai, messages, channel, user_id, message) except Exception as e: raise UserError(_(e)) return rdata def _message_post_after_hook(self, message, msg_vals): if message.author_id.gpt_id: if msg_vals['body'] not in [_('Response Timeout, please speak again.'), _('温馨提示：您发送的内容含有敏感词，请修改内容后再向我发送。')]: message.is_ai = True return super(Channel, self)._message_post_after_hook(message, msg_vals)