add deepseek

app_deepseek/models/mail_channel.py

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+
+import logging
+import datetime
+from odoo import api, fields, models, tools, _
+from odoo.exceptions import UserError
+from odoo.osv import expression
+_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=0):
+        gpt_id = False
+        if self.ai_partner_id:
+            ai_user = self.env['res.users'].search([('partner_id', '=', self.ai_partner_id.id)], limit=1)
+            gpt_id = ai_user.mapped('gpt_id')
+        if gpt_id and gpt_id.provider == 'deepseek':
+            context_history = []
+            afterTime = fields.Datetime.now() - datetime.timedelta(minutes=minutes)
+            message_model = self.env['mail.message'].sudo()
+            domain = [('res_id', '=', channel_id),
+                  ('model', '=', 'mail.channel'),
+                  ('message_type', '!=', 'user_notification'),
+                  ('parent_id', '!=', False),
+                  ('is_ai', '=', True),
+                  ('body', '!=', '<p>%s</p>' % _('Response Timeout, please speak again.')),
+                  ('body', '!=', _('温馨提示：您发送的内容含有敏感词，请修改内容后再向我发送。'))]
+            if self.channel_type in ['group', 'channel']:
+                # 群聊增加时间限制，当前找所有人，不限制 author_id
+                domain = expression.AND([domain, [('date', '>=', afterTime)]])
+            else:
+                domain = expression.AND([domain, [('author_id', '=', answer_id.id)]])
+            if chat_count == 0:
+                ai_msg_list = []
+            else:
+                ai_msg_list = message_model.with_context(tz='UTC').search(domain, order="id asc", 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()
+                ai_content = str(ai_msg.body).replace("<p>", "").replace("</p>", "").replace("<p>", "")
+                if ai_msg.author_id.sudo().gpt_id and answer_id.sudo().gpt_id and ai_msg.author_id.sudo().gpt_id == answer_id.sudo().gpt_id:
+                    context_history.insert(0, {
+                        'role': 'assistant',
+                        'content': ai_content,
+                    })
+                if not user_msg.author_id.gpt_id:
+                    user_content = user_msg.body.replace("<p>", "").replace("</p>", "").replace('@%s' % answer_id.name, '').lstrip()
+                    context_history.append({
+                        'role': 'user',
+                        'content': user_content,
+                    })
+                    context_history.append({
+                        'role': 'assistant',
+                        'content': ai_content,
+                    })
+            return context_history
+        else:
+            return super(Channel, self).get_openai_context(channel_id, author_id, answer_id, minutes, chat_count)