mirror of
https://github.com/guohuadeng/app-odoo.git
synced 2025-02-23 04:11:36 +02:00
243 lines
10 KiB
Python
243 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
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):
|
||
_name = 'ai.robot'
|
||
_description = 'Gpt Robot'
|
||
_order = 'sequence, name'
|
||
|
||
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'),
|
||
('code-davinci-002', 'Chatgpt 2 Code Optimized'),
|
||
('text-davinci-002', 'Chatgpt 2 Davinci'),
|
||
('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
|
||
Embeddings: A set of models that can convert text into a numerical form
|
||
CodexLimited: A set of models that can understand and generate code, including translating natural language to code
|
||
Moderation: A fine-tuned model that can detect whether text may be sensitive or unsafe
|
||
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, sender_id=False, answer_id=False, **kwargs):
|
||
# 通用方法
|
||
# sender_id: 请求的 partner_id 对象
|
||
# answer_id: 回答的 partner_id 对象
|
||
# kwargs，dict 形式的可变参数
|
||
self.ensure_one()
|
||
# 前置勾子，一般返回 False，有问题返回响应内容
|
||
res_pre = self.get_ai_pre(data, sender_id, answer_id, **kwargs)
|
||
if res_pre:
|
||
return res_pre
|
||
if hasattr(self, 'get_%s' % self.provider):
|
||
res = getattr(self, 'get_%s' % self.provider)(data, sender_id, answer_id, **kwargs)
|
||
else:
|
||
res = _('No robot provider found')
|
||
|
||
# 后置勾子，返回处理后的内容，用于处理敏感词等
|
||
res_post = self.get_ai_post(res, sender_id, answer_id, **kwargs)
|
||
return res_post
|
||
|
||
def get_ai_pre(self, data, sender_id=False, answer_id=False, **kwargs):
|
||
return False
|
||
|
||
def get_ai_post(self, res, sender_id=False, answer_id=False, **kwargs):
|
||
res = self.filter_sensitive_words(res)
|
||
return res
|
||
|
||
def get_ai_model_info(self):
|
||
self.ensure_one()
|
||
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
|
||
R_TIMEOUT = 300
|
||
o_url = "https://api.openai.com/v1/models/%s" % self.ai_model
|
||
if self.endpoint:
|
||
o_url = self.endpoint.replace("/chat/completions", "") + "/models/%s" % self.ai_model
|
||
|
||
response = requests.get(o_url, headers=headers, timeout=R_TIMEOUT)
|
||
response.close()
|
||
if response:
|
||
res = response.json()
|
||
r_text = json.dumps(res, indent=2)
|
||
else:
|
||
r_text = 'No response.'
|
||
raise UserError(r_text)
|
||
|
||
def get_ai_list_model(self):
|
||
self.ensure_one()
|
||
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
|
||
R_TIMEOUT = 300
|
||
o_url = "https://api.openai.com/v1/models"
|
||
if self.endpoint:
|
||
o_url = self.endpoint.replace("/chat/completions", "") + "/models"
|
||
response = requests.get(o_url, headers=headers, timeout=R_TIMEOUT)
|
||
response.close()
|
||
if response:
|
||
res = response.json()
|
||
r_text = json.dumps(res, indent=2)
|
||
else:
|
||
r_text = 'No response.'
|
||
raise UserError(r_text)
|
||
|
||
def get_openai(self, data, sender_id, answer_id, *args):
|
||
self.ensure_one()
|
||
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
|
||
R_TIMEOUT = 3000
|
||
o_url = self.endpoint or "https://api.openai.com/v1/chat/completions"
|
||
partner_name = 'odoo'
|
||
# if sender_id:
|
||
# partner_name = sender_id.name
|
||
# 以下处理 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)
|
||
try:
|
||
res = response.json()
|
||
if 'usage' in res:
|
||
usage = res['usage']
|
||
prompt_tokens = usage['prompt_tokens']
|
||
completion_tokens = usage['completion_tokens']
|
||
total_tokens = usage['total_tokens']
|
||
vals = {
|
||
'human_prompt_tokens': sender_id.human_prompt_tokens + prompt_tokens,
|
||
'ai_completion_tokens': sender_id.ai_completion_tokens + completion_tokens,
|
||
'tokens_total': sender_id.tokens_total + total_tokens,
|
||
'used_number': sender_id.used_number + 1,
|
||
}
|
||
if not sender_id.first_ask_time:
|
||
ask_date = response.headers.get("Date")
|
||
vals.update({
|
||
'first_ask_time': ask_date
|
||
})
|
||
sender_id.write(vals)
|
||
if 'choices' in res:
|
||
# for rec in res:
|
||
# res = rec['message']['content']
|
||
res = '\n'.join([x['message']['content'] for x in res['choices']])
|
||
return res
|
||
except Exception as e:
|
||
_logger.warning("Get Response Json failed: %s", e)
|
||
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']])
|
||
return res
|
||
|
||
return "获取结果超时，请重新跟我聊聊。"
|
||
|
||
def get_azure(self, data, sender_id, answer_id, *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
|
||
pdata = {
|
||
"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' % pdata)
|
||
response = openai.Completion.create(pdata)
|
||
|
||
if 'choices' in response:
|
||
res = response['choices'][0]['text'].replace(' .', '.').strip()
|
||
return res
|
||
else:
|
||
_logger.warning('=====================azure output data: %s' % response)
|
||
return _('Azure no response')
|
||
|
||
@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
|
||
if s:
|
||
search.SetKeywords(s.split('\n'))
|
||
else:
|
||
search.SetKeywords([])
|
||
data = search.Replace(text=data)
|
||
return data
|
||
else:
|
||
return data
|