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

opt 优化 chatgpt 接口，将错误提示也响应。

Browse Source
This commit is contained in:
Ivan Office
2025-01-08 21:15:32 +08:00
parent e42e065a7d
commit 6818df15e1
3 changed files with 34 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
app_chatgpt/models/ai_robot.py
View File

@@ -3,6 +3,7 @@
import os
from openai import OpenAI
from openai import AzureOpenAI
# from openai.error import OpenAIError
import requests, json
import base64
@@ -192,8 +193,12 @@ GPT-3 A set of models that can understand and generate natural language
# _logger.warning('===========Ai响应:%s' % content)
elif self.provider == 'azure':
# azure 格式
usage = res['usage']
content = res['choices'][0]['message']['content']
if res.get('error'):
usage = False
content = res.get('error')
else:
usage = res['usage']
content = res['choices'][0]['message']['content']
else:
usage = False
content = res
@@ -338,20 +343,28 @@ GPT-3 A set of models that can understand and generate natural language
api_key=self.openapi_api_key,
timeout=request_timeout
)
response = client.chat.completions.create(
model=self.engine,
messages=messages,
# 返回的回答数量
n=1,
max_tokens=max_tokens,
temperature=temperature,
top_p=top_p,
frequency_penalty=frequency_penalty,
presence_penalty=presence_penalty,
stop=None,
)
res = response.model_dump()
if 'choices' in res:
try:
response = client.chat.completions.create(
model=self.engine,
messages=messages,
# 返回的回答数量
n=1,
max_tokens=max_tokens,
temperature=temperature,
top_p=top_p,
frequency_penalty=frequency_penalty,
presence_penalty=presence_penalty,
stop=None,
)
res = response.model_dump()
except Exception as e:
# 处理OpenAI相关错误
res = {
'code': e.code,
'usage': False,
'error': str(e),
}
if 'choices' in res or 'error' in res:
return res
else:
_logger.warning('=====================azure output data: %s' % response.json())