diff --git a/app_chatgpt/models/__init__.py b/app_chatgpt/models/__init__.py index b8459a87..80d7b07b 100644 --- a/app_chatgpt/models/__init__.py +++ b/app_chatgpt/models/__init__.py @@ -3,5 +3,6 @@ from . import mail_channel from . import res_config_settings from . import ai_robot +from . import res_partner from . import res_users from . import mail_message diff --git a/app_chatgpt/models/res_partner.py b/app_chatgpt/models/res_partner.py new file mode 100644 index 00000000..a4b6787b --- /dev/null +++ b/app_chatgpt/models/res_partner.py @@ -0,0 +1,9 @@ +# -*- coding: utf-8 -*- + +from odoo import fields, models + + +class ResPartner(models.Model): + _inherit = "res.partner" + + gpt_id = fields.Many2one('ai.robot', string='Bind to ChatGpt') diff --git a/app_chatgpt/models/res_users.py b/app_chatgpt/models/res_users.py index 262c5223..2fa6a8be 100644 --- a/app_chatgpt/models/res_users.py +++ b/app_chatgpt/models/res_users.py @@ -6,7 +6,8 @@ from odoo import fields, models class ResUsers(models.Model): _inherit = "res.users" - gpt_id = fields.Many2one('ai.robot', string='Bind to ChatGpt') + # 改为在 partner中设置,用户处绑定 + gpt_id = fields.Many2one('ai.robot', string='Bind to ChatGpt', related='partner_id.gpt_id', inherited=True, readonly=False) gpt_policy = fields.Selection([ ('all', 'All Users'), ('limit', 'Selected Users')