From fbc2453f8f98d5ad8706382af895e9aebc06842b Mon Sep 17 00:00:00 2001 From: ivan deng Date: Wed, 12 Apr 2023 14:22:36 +0800 Subject: [PATCH] =?UTF-8?q?pre=20#I6SC9C=20app=5Fchatgpt=E4=BC=98=E5=8C=96?= =?UTF-8?q?=EF=BC=8C=E6=8C=87=E5=AE=9A=E7=94=A8=E6=88=B7=E6=97=B6=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=E4=BD=BF=E7=94=A8=E6=83=85=E5=86=B5=20=E7=BD=AE?= =?UTF-8?q?=E9=A1=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app_chatgpt/models/ai_robot.py | 23 ++++++++++++++++++++--- 1 file changed, 20 insertions(+), 3 deletions(-) diff --git a/app_chatgpt/models/ai_robot.py b/app_chatgpt/models/ai_robot.py index 091d0230..461981a6 100644 --- a/app_chatgpt/models/ai_robot.py +++ b/app_chatgpt/models/ai_robot.py @@ -49,14 +49,31 @@ GPT-3 A set of models that can understand and generate natural language def action_disconnect(self): requests.delete('https://chatgpt.com/v1/disconnect') - def get_ai(self, data, partner_name='odoo', *args): + def get_ai(self, data, sender_id=False, answer_id=False, **kwargs): # 通用方法 + # sender_id: 请求的 partner_id 对象 + # answer_id: 回答的 partner_id 对象 + # kwargs，dict 形式的可变参数 self.ensure_one() + # 前置勾子，一般返回 False，有问题返回响应内容 + res_pre = self.get_ai_pre(data, sender_id, answer_id, **kwargs) + if res_pre: + return res_pre if hasattr(self, 'get_%s' % self.provider): - return getattr(self, 'get_%s' % self.provider)(data, partner_name, *args) + res = getattr(self, 'get_%s' % self.provider)(data, sender_id, answer_id, **kwargs) else: - return _('No robot provider found') + res = _('No robot provider found') + + # 后置勾子，返回处理后的内容，用于处理敏感词等 + res_post = self.get_ai_post(res, sender_id, answer_id, **kwargs) + return res_post + def get_ai_pre(self, data, sender_id=False, answer_id=False, **kwargs): + return False + + def get_ai_post(self, res, sender_id=False, answer_id=False, **kwargs): + return res + def get_ai_model_info(self): self.ensure_one() headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}