# -*- coding: utf-8 -*- import json import logging import requests from odoo import models, fields, api, _ from odoo.exceptions import UserError _logger = logging.getLogger(__name__) class AiRule(models.Model): """AI 规则配置 — 控制 AI 回复行为、限制范围、对接技能接口""" _name = 'ai.rule' _description = 'AI 规则' _order = 'priority, id' create_date = fields.Datetime(string='创建时间', readonly=True) write_date = fields.Datetime(string='最后更新', readonly=True) name = fields.Char(string='规则名称', required=True) rule_type = fields.Selection([ ('instruction', '回复指令'), ('restriction', '限制规则'), ('skill', '技能接口'), ('method', '调用模型方法'), ('active_guide', '主动感知'), ], string='规则类型', required=True, default='instruction', help='回复指令：注入到 AI 系统提示词中，控制回复风格和内容。\n' '限制规则：告诉 AI 哪些话题不能回答。\n' '技能接口：对接外部 API，AI 可调用获取专业数据。\n' '调用模型方法：直接调用 Odoo 模型方法（如撤回消息），无需 HTTP 接口。\n' '主动感知：在用户操作（如流程结束）时主动弹出引导消息。' ) active = fields.Boolean(string='启用', default=True) priority = fields.Integer(string='优先级', default=10, help='数字越小优先级越高') # ---- 适用范围 ---- provider_ids = fields.Many2many( 'ai.provider', 'ai_rule_provider_rel', 'rule_id', 'provider_id', string='适用服务商', help='留空则对所有服务商生效', ) group_ids = fields.Many2many( 'res.groups', 'ai_rule_group_rel', 'rule_id', 'group_id', string='适用用户组', help='留空则对所有用户生效', ) # ---- 回复指令 / 限制规则 内容 ---- content = fields.Text( string='规则内容', help='回复指令类型：写给 AI 的系统指令，如"所有回复不超过 200 字"。\n' '限制规则类型：描述不允许回答的话题，如"不得回答薪资相关问题"。', ) # ---- 技能接口配置 ---- skill_name = fields.Char( string='技能名称', help='AI 工具调用时的函数名，如 query_weather, get_stock_price', ) skill_description = fields.Text( string='技能描述', help='告诉 AI 什么时候调用这个技能，如"当用户询问天气时调用"', ) skill_parameters = fields.Text( string='参数定义 (JSON Schema)', help='OpenAI 兼容的 JSON Schema 格式，定义技能需要的参数', default='{\n "type": "object",\n "properties": {},\n "required": []\n}', ) skill_endpoint = fields.Char( string='接口地址', help='技能对接的 API URL，如 https://api.example.com/weather', ) skill_method = fields.Selection([ ('GET', 'GET'), ('POST', 'POST'), ], string='请求方式', default='POST') skill_headers = fields.Text( string='请求头 (JSON)', help='额外请求头，如 {"Authorization": "Bearer xxx"}', default='{}', ) skill_timeout = fields.Integer( string='接口超时(秒)', default=10, ) # ---- 数据写入权限 ---- can_write = fields.Boolean( string='允许修改数据', default=False, help='勾选后，匹配此规则时 AI 可以调用 create/write/unlink 工具修改 Odoo 数据。' '只要有一条适用规则勾选了此项，AI 就拥有写入权限。', ) # ---- 调用模型方法配置 ---- skill_model = fields.Char( string='目标模型', help='方法所在模型的技术名称，如 attendance.message.log。仅 method 类型使用。', ) skill_method_name = fields.Char( string='方法名称', help='要调用的模型方法名，如 action_recall_message。仅 method 类型使用。', ) # ---- 技能接口执行 ---- def execute_skill(self, arguments): """调用技能接口，返回结果字典""" self.ensure_one() if self.rule_type != 'skill' or not self.skill_endpoint: return {'error': '该规则不是技能接口类型或未配置接口地址'} try: headers = json.loads(self.skill_headers or '{}') except json.JSONDecodeError: headers = {} headers.setdefault('Content-Type', 'application/json') timeout = max(self.skill_timeout or 10, 1) try: if self.skill_method == 'GET': resp = requests.get( self.skill_endpoint, params=arguments, headers=headers, timeout=timeout, ) else: resp = requests.post( self.skill_endpoint, json=arguments, headers=headers, timeout=timeout, ) resp.raise_for_status() return resp.json() except requests.exceptions.Timeout: return {'error': f'技能接口超时（{timeout}秒）'} except requests.exceptions.RequestException as e: _logger.exception('Skill %s call failed: %s', self.skill_name, e) return {'error': f'技能接口调用失败：{e}'} except (ValueError, json.JSONDecodeError): return {'error': '技能接口返回格式错误（非 JSON）'} def get_tool_schema(self): """将技能规则转换为 OpenAI 兼容的 tool schema（skill 和 method 类型通用）""" self.ensure_one() try: params = json.loads(self.skill_parameters or '{}') except json.JSONDecodeError: params = {'type': 'object', 'properties': {}} return { 'type': 'function', 'function': { 'name': self.skill_name or f'skill_{self.id}', 'description': self.skill_description or self.name, 'parameters': params, }, } def execute_method(self, arguments): """执行模型方法调用（rule_type='method'）。搜索目标模型记录，逐条调用指定方法。""" self.ensure_one() if self.rule_type != 'method': return {'error': '该规则不是调用模型方法类型'} if not self.skill_model or self.skill_model not in self.env: return {'error': f'未配置目标模型或模型不存在：{self.skill_model}'} if not self.skill_method_name: return {'error': '未配置方法名称'} Model = self.env[self.skill_model] method = getattr(Model, self.skill_method_name, None) if not method: return {'error': f'模型 {self.skill_model} 中没有方法 {self.skill_method_name}'} try: domain = arguments.get('domain') or [] ids = arguments.get('ids') or [] if ids: records = Model.browse(ids).exists() else: records = Model.search(domain) if not records: return {'success': True, 'message': '没有找到符合条件的记录', 'count': 0} results = [] for rec in records: try: res = method(rec) results.append({'id': rec.id, 'result': str(res) if res else None}) except Exception as e: results.append({'id': rec.id, 'error': str(e)}) return { 'success': True, 'count': len(records), 'results': results, } except Exception as e: _logger.exception('Method rule %s failed: %s', self.skill_name, e) return {'error': f'{type(e).__name__}: {e}'} # ---- 主动感知配置 ---- guide_model = fields.Char( string='触发模型', help='监听哪个模型，如 yuthon.notice', ) guide_field = fields.Char( string='触发字段', help='监听哪个字段变化，如 work_end', ) guide_field_value = fields.Char( string='触发值', help='字段值等于什么时触发，如 True', ) guide_message = fields.Char( string='提醒内容', help='主动弹出时显示的文字，支持变量：{record_name}, {user_name}', ) guide_button_label = fields.Char( string='按钮文字', default='确认执行', ) guide_method_name = fields.Char( string='后台方法名', help='按钮点击后调用的后台方法名，如 notice_send_to_all。后台自动传入 record_id（当前记录ID）', ) guide_method_desc = fields.Char( string='方法说明', ) def _do_test_guide(self): """测试主动感知：模拟触发，创建引导消息（内部方法，由控制器调用）""" self.ensure_one() if self.rule_type != 'active_guide': raise UserError('仅主动感知类型的规则可以测试') # 获取或创建用户对话 conv = self.env['ai.conversation'].search([ ('create_uid', '=', self.env.uid), ], limit=1, order='write_date desc') if not conv: provider = self.env['ai.provider'].search([('active', '=', True)], limit=1) if not provider: raise UserError('请先配置 AI 服务商') conv = self.env['ai.conversation'].create({ 'name': 'AI 对话', 'provider_id': provider.id, }) # 组装测试消息（替换变量为测试值） message = self.guide_message or '【测试】主动感知已触发' message = message.replace('{record_name}', '[测试记录]') message = message.replace('{user_name}', self.env.user.name) message = message.replace('{record_id}', '[测试]') # 组装 action config（按钮点击后的动作） action_config = { 'type': 'execute_tool', 'tool': self.guide_method_name or '', 'params': { 'record_id': 0, 'model': self.guide_model or '', } } # 创建 AI 消息（带 action 按钮） self.env['ai.message'].create({ 'conversation_id': conv.id, 'role': 'assistant', 'content': f'🧪 {message}', 'action': json.dumps({ 'type': 'guide_button', 'label': self.guide_button_label or '确认执行', 'action': action_config, }, ensure_ascii=False), }) return conv.id def action_test_guide(self): """按钮调用：创建测试消息，返回 conv_id 供 JS 打开聊天框""" try: conv_id = self._do_test_guide() return {'conv_id': conv_id, 'message': '测试消息已创建，请查看聊天框'} except UserError as e: return {'error': str(e)}