diff --git a/app_chatgpt/models/ai_robot.py b/app_chatgpt/models/ai_robot.py index 59bfaeb9..5c5c7110 100644 --- a/app_chatgpt/models/ai_robot.py +++ b/app_chatgpt/models/ai_robot.py @@ -3,6 +3,7 @@ import requests import openai from odoo import api, fields, models, _ +from odoo.exceptions import UserError class AiRobot(models.Model): @@ -34,6 +35,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") temperature = fields.Float(string='Temperature', default=0.9) max_length = fields.Integer('Max Length', default=100) + endpoint = fields.Char('End Point') + engine = fields.Char('Engine', default='odooapp') + api_version = fields.Char('API Version', default='2022-12-01') sequence = fields.Integer('Sequence', help="Determine the display order", default=10) def action_disconnect(self): @@ -41,10 +45,14 @@ GPT-3 A set of models that can understand and generate natural language def get_openai(self, data): openai.api_type = self.provider - openai.api_base = "https://odooapp.openai.azure.com/" - openai.api_version = "2022-12-01" + if not self.endpoint: + raise UserError(_("Please Set your AI robot's endpoint first.")) + openai.api_base = self.endpoint + if not self.api_version: + raise UserError(_("Please Set your AI robot's API Version first.")) + openai.api_version = self.api_version openai.api_key = self.openapi_api_key - response = openai.Completion.create(engine='odooapp', prompt=data, temperature=self.temperature, max_tokens=self.max_length, top_p=0.5, frequency_penalty=0, + response = openai.Completion.create(engine=self.engine, prompt=data, temperature=self.temperature, max_tokens=self.max_length, top_p=0.5, frequency_penalty=0, presence_penalty=0, stop=["Human:", "AI:"]) if 'choices' in response: res = response['choices'][0]['text'].replace(' .', '.').strip() diff --git a/app_chatgpt/views/ai_robot_views.xml b/app_chatgpt/views/ai_robot_views.xml index ea8e5306..790b32f1 100644 --- a/app_chatgpt/views/ai_robot_views.xml +++ b/app_chatgpt/views/ai_robot_views.xml @@ -39,6 +39,9 @@ + + +