update to 15 gpt

app_chatgpt/models/res_partner.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+
+from odoo import fields, models, api
+
+
+class ResPartner(models.Model):
+    _inherit = "res.partner"
+
+    gpt_id = fields.Many2one('ai.robot', string='Bind to Ai', ondelete='set null')
+
+    is_chat_private = fields.Boolean('Allow Chat Private', default=False)
+
+    @api.model
+    def im_search(self, name, limit=20):
+        users = self.env['res.users'].search([
+            ('id', '!=', self.env.user.id),
+            ('name', 'ilike', name),
+            ('active', '=', True),
+            ('share', '=', False),
+            ('is_chat_private', '=', True)
+        ], order='gpt_id, name, id', limit=limit)
+        return list(users.partner_id.mail_partner_format().values())
+
+    def mail_partner_format(self, fields=None):
+        # 直接覆盖原生，增加 gpt_id 字段
+        partners_format = dict()
+        if not fields:
+            fields = {'id': True, 'name': True, 'email': True, 'active': True, 'im_status': True, 'gpt_id': 0, 'user': {}}
+        for partner in self:
+            data = {}
+            if 'id' in fields:
+                data['id'] = partner.id
+            if 'name' in fields:
+                name = partner.name
+                # 英文不好分，暂时不隐名
+                # if not partner.related_user_id.gpt_id:
+                #     name = partner.name[0] + '*' * (len(partner.name) - 1)
+                data['name'] = name
+            if 'email' in fields:
+                data['email'] = partner.email
+            if 'active' in fields:
+                data['active'] = partner.active
+            if 'im_status' in fields:
+                data['im_status'] = partner.im_status
+            if 'gpt_id' in fields:
+                data['gpt_id'] = partner.gpt_id.id if partner.gpt_id else 0
+            if 'user' in fields:
+                internal_users = partner.user_ids - partner.user_ids.filtered('share')
+                main_user = internal_users[0] if len(internal_users) > 0 else partner.user_ids[0] if len(partner.user_ids) > 0 else self.env['res.users']
+                data['user'] = {
+                    "id": main_user.id,
+                    "isInternalUser": not main_user.share,
+                } if main_user else [('clear',)]
+            # if 'guest' in self.env.context or not self.env.user.has_group('base.group_erp_manager'):
+            # 完全不显示 邮箱
+            data.pop('email', None)
+            partners_format[partner] = data
+        return partners_format