fix odoo 13 gpt

app_chatgpt/models/ai_robot.py

@@ -1,7 +1,13 @@
 # -*- coding: utf-8 -*-
 
-import requests
+import requests, json
+import openai
 from odoo import api, fields, models, _
+from odoo.exceptions import UserError
+from .lib.WordsSearch import WordsSearch
+
+import logging
+_logger = logging.getLogger(__name__)
 
 
 class AiRobot(models.Model):
@@ -9,9 +15,10 @@ class AiRobot(models.Model):
     _description = 'Gpt Robot'
     _order = 'sequence, name'
 
-    name = fields.Char(string='Name', translate=True)
-    provider = fields.Selection(string="AI Provider", selection=[('openai', 'OpenAI')], required=True, default='openai')
+    name = fields.Char(string='Name', translate=True, required=True)
+    provider = fields.Selection(string="AI Provider", selection=[('openai', 'OpenAI'), ('azure', 'Azure')], required=True, default='openai')
     ai_model = fields.Selection(string="AI Model", selection=[
+        ('gpt-4', 'Chatgpt 4'),
         ('gpt-3.5-turbo', 'Chatgpt 3.5 Turbo'),
         ('gpt-3.5-turbo-0301', 'Chatgpt 3.5 Turbo on 20230301'),
         ('text-davinci-003', 'Chatgpt 3 Davinci'),
@@ -20,6 +27,7 @@ class AiRobot(models.Model):
         ('dall-e2', 'Dall-E Image'),
     ], required=True, default='gpt-3.5-turbo',
                              help="""
+GPT-4: Can understand Image, generate natural language or code.
 GPT-3.5: A set of models that improve on GPT-3 and can understand as well as generate natural language or code
 DALL·E: A model that can generate and edit images given a natural language prompt
 Whisper: A model that can convert audio into text
@@ -30,10 +38,126 @@ 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=300)
+    endpoint = fields.Char('End Point', default='https://api.openai.com/v1/chat/completions')
+    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)
+    sensitive_words = fields.Text('Sensitive Words Plus', help='Sensitive word filtering. Separate keywords with a carriage return.')
+    is_filtering = fields.Boolean('Filter Sensitive Words', default=False, help='Use base Filter in dir models/lib/sensi_words.txt')
 
     def action_disconnect(self):
         requests.delete('https://chatgpt.com/v1/disconnect')
+        
+    def get_ai(self, data, partner_name='odoo', *args):
+        #     通用方法
+        self.ensure_one()
+        if hasattr(self, 'get_%s' % self.provider):
+            return getattr(self, 'get_%s' % self.provider)(data, partner_name, *args)
+        else:
+            return _('No robot provider found')
 
+    def get_openai(self, data, partner_name='odoo', *args):
+        self.ensure_one()
+        headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
+        R_TIMEOUT = 300
+        o_url = self.endpoint or "https://api.openai.com/v1/chat/completions"
+    
+        # 以下处理 open ai
+        #     获取模型信息
+        # list_model = requests.get("https://api.openai.com/v1/models", headers=headers)
+        # model_info = requests.get("https://api.openai.com/v1/models/%s" % ai_model, headers=headers)
+        if self.ai_model == 'dall-e2':
+            # todo: 处理 图像引擎，主要是返回参数到聊天中
+            # image_url = response['data'][0]['url']
+            # https://platform.openai.com/docs/guides/images/introduction
+            pdata = {
+                "prompt": data,
+                "n": 3,
+                "size": "1024x1024",
+            }
+            return '建设中'
+        elif self.ai_model in ['gpt-3.5-turbo', 'gpt-3.5-turbo-0301']:
+            pdata = {
+                "model": self.ai_model,
+                "messages": [{"role": "user", "content": data}],
+                "temperature": 0.9,
+                "max_tokens": self.max_length or 1000,
+                "top_p": 1,
+                "frequency_penalty": 0.0,
+                "presence_penalty": 0.6,
+                "user": partner_name,
+                "stop": ["Human:", "AI:"]
+            }
+            _logger.warning('=====================open input pdata: %s' % pdata)
+            response = requests.post(o_url, data=json.dumps(pdata), headers=headers, timeout=R_TIMEOUT)
+            res = response.json()
+            if 'choices' in res:
+                # for rec in res:
+                #     res = rec['message']['content']
+                res = '\n'.join([x['message']['content'] for x in res['choices']])
+                res = self.filter_sensitive_words(res)
+                return res
+        else:
+            pdata = {
+                "model": self.ai_model,
+                "prompt": data,
+                "temperature": 0.9,
+                "max_tokens": self.max_length or 1000,
+                "top_p": 1,
+                "frequency_penalty": 0.0,
+                "presence_penalty": 0.6,
+                "user": partner_name,
+                "stop": ["Human:", "AI:"]
+            }
+            response = requests.post(o_url, data=json.dumps(pdata), headers=headers, timeout=R_TIMEOUT)
+            res = response.json()
+            if 'choices' in res:
+                res = '\n'.join([x['text'] for x in res['choices']])
+                res = self.filter_sensitive_words(res)
+                return res
+    
+        return "获取结果超时，请重新跟我聊聊。"
+
+    def get_azure(self, data, partner_name='odoo', *args):
+        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."))
+        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=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:"])
+
+        _logger.warning('=====================azure input data: %s' % data)
+        if 'choices' in response:
+            res = response['choices'][0]['text'].replace(' .', '.').strip()
+            res = self.filter_sensitive_words(res)
+            return res
+
+    @api.onchange('provider')
+    def _onchange_provider(self):
+        if self.provider == 'openai':
+            self.endpoint = 'https://api.openai.com/v1/chat/completions'
+        elif self.provider == 'azure':
+            self.endpoint = 'https://odoo.openai.azure.com'
+            
+    def filter_sensitive_words(self, data):
+        if self.is_filtering:
+            search = WordsSearch()
+            s = self.sensitive_words
+            search.SetKeywords(s.split('\n'))
+            result = search.Replace(text=data)
+            return result
+        else:
+            return data