From 294570b4525b1a26d8808a50d46feddd580623fd Mon Sep 17 00:00:00 2001
From: ivan deng <guohuadeng@hotmail.com>
Date: Fri, 17 Mar 2023 19:38:20 +0800
Subject: [PATCH] fix gpt

---
 app_chatgpt/models/ai_robot.py     | 17 +++++++++++++----
 app_chatgpt/models/mail_channel.py |  1 +
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/app_chatgpt/models/ai_robot.py b/app_chatgpt/models/ai_robot.py
index 5c5c7110..475a9555 100644
--- a/app_chatgpt/models/ai_robot.py
+++ b/app_chatgpt/models/ai_robot.py
@@ -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")
     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')
-    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')
     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')
 
     def get_openai(self, data):
+        self.ensure_one()
+        # only for azure
         openai.api_type = self.provider
         if not self.endpoint:
             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."))
         openai.api_version = self.api_version
         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,
-                                            presence_penalty=0, stop=["Human:", "AI:"])
+        response = openai.Completion.create(
+            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:
             res = response['choices'][0]['text'].replace(' .', '.').strip()
             return res
diff --git a/app_chatgpt/models/mail_channel.py b/app_chatgpt/models/mail_channel.py
index a4aebc79..d0f2f6b2 100644
--- a/app_chatgpt/models/mail_channel.py
+++ b/app_chatgpt/models/mail_channel.py
@@ -70,6 +70,7 @@ class Channel(models.Model):
                     res = '\n'.join([x['text'] for x in res['choices']])
                     return res
             elif provider == 'azure':
+                _logger.warning('=====================azure input data: %s' % data)
                 res = gpt_id.get_openai(data)
                 return res
         #     获取模型信息