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

fix gpt

Browse Source
This commit is contained in:
ivan deng
2023-03-17 19:38:20 +08:00
parent a061c7a840
commit 294570b452
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app_chatgpt/models/ai_robot.py
View File

@@ -34,9 +34,9 @@ GPT-3 A set of models that can understand and generate natural language
""") """)
openapi_api_key = fields.Char(string="API Key", help="Provide the API key here") openapi_api_key = fields.Char(string="API Key", help="Provide the API key here")
temperature = fields.Float(string='Temperature', default=0.9) temperature = fields.Float(string='Temperature', default=0.9)
max_length = fields.Integer('Max Length', default=100) max_length = fields.Integer('Max Length', default=300)
endpoint = fields.Char('End Point') endpoint = fields.Char('End Point')
engine = fields.Char('Engine', default='odooapp') engine = fields.Char('Engine', help='If use Azure, Please input the Model deployment name.')
api_version = fields.Char('API Version', default='2022-12-01') api_version = fields.Char('API Version', default='2022-12-01')
sequence = fields.Integer('Sequence', help="Determine the display order", default=10) sequence = fields.Integer('Sequence', help="Determine the display order", default=10)
@@ -44,6 +44,8 @@ GPT-3 A set of models that can understand and generate natural language
requests.delete('https://chatgpt.com/v1/disconnect') requests.delete('https://chatgpt.com/v1/disconnect')
def get_openai(self, data): def get_openai(self, data):
self.ensure_one()
# only for azure
openai.api_type = self.provider openai.api_type = self.provider
if not self.endpoint: if not self.endpoint:
raise UserError(_("Please Set your AI robot's endpoint first.")) raise UserError(_("Please Set your AI robot's endpoint first."))
@@ -52,8 +54,15 @@ GPT-3 A set of models that can understand and generate natural language
raise UserError(_("Please Set your AI robot's API Version first.")) raise UserError(_("Please Set your AI robot's API Version first."))
openai.api_version = self.api_version openai.api_version = self.api_version
openai.api_key = self.openapi_api_key openai.api_key = self.openapi_api_key
response = openai.Completion.create(engine=self.engine, prompt=data, temperature=self.temperature, max_tokens=self.max_length, top_p=0.5, frequency_penalty=0, response = openai.Completion.create(
presence_penalty=0, stop=["Human:", "AI:"]) engine=self.engine,
prompt=data,
temperature=self.temperature or 0.9,
max_tokens=self.max_length or 600,
top_p=0.5,
frequency_penalty=0,
presence_penalty=0, stop=["Human:", "AI:"])
if 'choices' in response: if 'choices' in response:
res = response['choices'][0]['text'].replace(' .', '.').strip() res = response['choices'][0]['text'].replace(' .', '.').strip()
return res return res

1
app_chatgpt/models/mail_channel.py
View File

@@ -70,6 +70,7 @@ class Channel(models.Model):
res = '\n'.join([x['text'] for x in res['choices']]) res = '\n'.join([x['text'] for x in res['choices']])
return res return res
elif provider == 'azure': elif provider == 'azure':
_logger.warning('=====================azure input data: %s' % data)
res = gpt_id.get_openai(data) res = gpt_id.get_openai(data)
return res return res
# 获取模型信息 # 获取模型信息