audrius/app-odoo
1
0
Fork 0
mirror of https://github.com/guohuadeng/app-odoo.git synced 2025-02-23 04:11:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

update app_chatgpt: get_ai_post

Browse Source
This commit is contained in:
Chill
2024-03-29 16:10:38 +08:00
parent fd562a013d
commit 1f97a0f00c
3 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app_chatgpt/models/ai_robot.py
View File

@@ -171,12 +171,15 @@ GPT-3 A set of models that can understand and generate natural language
res_post, usage, is_ai = self.get_ai_post(res, author_id, answer_id, param)
return res
def get_ai_post(self, res, author_id=False, answer_id=False, param={}):
def get_ai_post(self, res, author_id=False, answer_id=False, param=None):
# hook，高级版要替代
if res and author_id or isinstance(res, list) or isinstance(res, dict):
# TODO: and type(res) == openai.Completion
# 返回是个对象，那么就是ai
# if isinstance(res, dict):
if param is None:
param = {}
if not res or not author_id or (not isinstance(res, list) and not isinstance(res, dict)):
return res, False, False
# 返回是个对象，那么就是ai
usage = content = data = None
try:
if self.provider == 'openai':
# openai 格式处理
usage = res['usage']
@@ -184,15 +187,15 @@ GPT-3 A set of models that can understand and generate natural language
# _logger.warning('===========Ai响应:%s' % content)
elif self.provider == 'azure':
# azure 格式
usage = json.loads(json.dumps(res['usage']))
content = json.loads(json.dumps(res['choices'][0]['message']['content']))
usage = res['usage']
content = res['choices'][0]['message']['content']
else:
usage = False
content = res
data = content.replace(' .', '.').strip()
return data, usage, True
else:
# 直接返回错误语句，那么就是非ai
except Exception as e:
_logger.error('==========app_chatgpt get_ai_post Error: %s' % e)
return res, False, False
def get_ai_system(self, content=None):