From d0e8e61ac66ec732192892d120f0262f80bec864 Mon Sep 17 00:00:00 2001 From: Chill Date: Tue, 11 Feb 2025 16:32:58 +0800 Subject: [PATCH] =?UTF-8?q?[16]=20kimi=E5=9B=BE=E6=96=87=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app_chatgpt/__manifest__.py | 2 +- app_chatgpt/models/ai_robot.py | 34 +++++++------ app_chatgpt/models/mail_channel.py | 62 ++++++++++++++++------- app_chatgpt/static/description/index.html | 2 +- 4 files changed, 65 insertions(+), 35 deletions(-) diff --git a/app_chatgpt/__manifest__.py b/app_chatgpt/__manifest__.py index 31f06bef..f44dcef7 100644 --- a/app_chatgpt/__manifest__.py +++ b/app_chatgpt/__manifest__.py @@ -10,7 +10,7 @@ { 'name': 'ChatGPT4, China Ali,AiGC Center.Ai服务中心,聚合全网Ai', - 'version': '16.0.25.02.10', + 'version': '16.0.25.02.11', 'author': 'odooai.cn', 'company': 'odooai.cn', 'maintainer': 'odooai.cn', diff --git a/app_chatgpt/models/ai_robot.py b/app_chatgpt/models/ai_robot.py index e4d0b417..7b28f8e5 100644 --- a/app_chatgpt/models/ai_robot.py +++ b/app_chatgpt/models/ai_robot.py @@ -133,13 +133,17 @@ GPT-3 A set of models that can understand and generate natural language # hook,都正常 return False + def get_msg_file_content(self, message): + # hook + return False + def get_ai(self, data, author_id=False, answer_id=False, param={}): # 通用方法 # author_id: 请求的 partner_id 对象 # answer_id: 回答的 partner_id 对象 # param,dict 形式的参数 # 调整输出为2个参数:res_post详细内容,is_ai是否ai的响应 - + self.ensure_one() # 前置勾子,一般返回 False,有问题返回响应内容,用于处理敏感词等 res_pre = self.get_ai_pre(data, author_id, answer_id, param) @@ -149,7 +153,7 @@ GPT-3 A set of models that can understand and generate natural language if not hasattr(self, 'get_%s' % self.provider): res = _('No robot provider found') return res, {}, False - + res = getattr(self, 'get_%s' % self.provider)(data, author_id, answer_id, param) # 后置勾子,返回处理后的内容 res_post, usage, is_ai = self.get_ai_post(res, author_id, answer_id, param) @@ -176,7 +180,7 @@ GPT-3 A set of models that can understand and generate natural language # 后置勾子,返回处理后的内容 res_post, usage, is_ai = self.get_ai_post(res, author_id, answer_id, param) return res - + def get_ai_post(self, res, author_id=False, answer_id=False, param=None): # hook,高级版要替代 if param is None: @@ -202,7 +206,7 @@ GPT-3 A set of models that can understand and generate natural language except Exception as e: _logger.error('==========app_chatgpt get_ai_post Error: %s' % e) return res, False, False - + # if res and author_id and isinstance(res, openai.openai_object.OpenAIObject) or isinstance(res, list) or isinstance(res, dict): # # 返回是个对象,那么就是ai # # if isinstance(res, dict): @@ -223,14 +227,14 @@ GPT-3 A set of models that can understand and generate natural language # else: # # 直接返回错误语句,那么就是非ai # return res, False, False - + def get_ai_system(self, content=None): # 获取基础ai角色设定, role system sys_content = content or self.sys_content if sys_content: return {"role": "system", "content": sys_content} return {} - + def get_ai_model_info(self): self.ensure_one() headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"} @@ -238,7 +242,7 @@ GPT-3 A set of models that can understand and generate natural language 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: @@ -280,7 +284,7 @@ GPT-3 A set of models that can understand and generate natural language frequency_penalty = param.get('frequency_penalty') if param.get('frequency_penalty') else self.frequency_penalty presence_penalty = param.get('presence_penalty') if param.get('presence_penalty') else self.presence_penalty request_timeout = param.get('request_timeout') if param.get('request_timeout') else self.ai_timeout - + if self.stop: stop = self.stop.split(',') else: @@ -320,7 +324,7 @@ GPT-3 A set of models that can understand and generate natural language return res else: _logger.warning('=====================openai output data: %s' % response.json()) - + return _("Response Timeout, please speak again.") # if self.ai_model in ['gpt-3.5-turbo', 'gpt-3.5-turbo-0301']: # # 基本与 azure 同,要处理 api_base @@ -401,7 +405,7 @@ GPT-3 A set of models that can understand and generate natural language # return response # else: # _logger.warning('=====================openai output data: %s' % response.json()) - + # return _("Response Timeout, please speak again.") def get_azure(self, data, author_id, answer_id, param={}): @@ -409,10 +413,10 @@ GPT-3 A set of models that can understand and generate natural language # only for azure if not self.endpoint: raise UserError(_("Please Set your AI robot's endpoint first.")) - + if not self.api_version: raise UserError(_("Please Set your AI robot's API Version first.")) - + if self.stop: stop = self.stop.split(',') else: @@ -436,7 +440,7 @@ GPT-3 A set of models that can understand and generate natural language if sys_content: messages.insert(0, sys_content) # 暂时不变 - + client = AzureOpenAI( api_version=self.api_version, azure_endpoint=self.endpoint, @@ -468,7 +472,7 @@ GPT-3 A set of models that can understand and generate natural language self.endpoint = 'https://api.openai.com/v1/chat/completions' elif self.provider == 'azure': self.endpoint = 'https://odoo.openai.azure.com' - + if self.provider: # 取头像 module_path = modules.get_module_path('app_chatgpt', display_warning=False) @@ -477,7 +481,7 @@ GPT-3 A set of models that can understand and generate natural language if path: image_file = tools.file_open(path, 'rb') self.image_avatar = base64.b64encode(image_file.read()) - + @api.onchange('set_ai_model') def _onchange_set_ai_model(self): if self.set_ai_model: diff --git a/app_chatgpt/models/mail_channel.py b/app_chatgpt/models/mail_channel.py index ce186401..92bb4f39 100644 --- a/app_chatgpt/models/mail_channel.py +++ b/app_chatgpt/models/mail_channel.py @@ -80,7 +80,7 @@ class Channel(models.Model): presence_penalty = fields.Float('避免重复词值', default=1, help="-2~2,值越大越少重复词") is_current_channel = fields.Boolean('是否当前用户默认频道', compute='_compute_is_current_channel', help='是否当前用户默认微信对话频道') - + # begin 处理Ai对话 is_ai_conversation = fields.Boolean('Ai Conversation', default=False, help='Set active to make conversation between 2+ Ai Employee. You Just say first word, then Ai robots Auto Chat.') @@ -263,11 +263,11 @@ class Channel(models.Model): # # 暂时有限用户的Ai # raise UserError(_('此Ai暂时未开放,请联系管理员。')) # end: 找ai,增加 ai二人转功能 - + if hasattr(ai, 'is_translator') and ai.is_translator and ai.ai_model == 'translator': return rdata chatgpt_channel_id = self.env.ref('app_chatgpt.channel_chatgpt') - + if message.body == _('
joined the channel
'): msg = _("Please warmly welcome our new partner %s and send him the best wishes.") % message.author_id.name else: @@ -282,12 +282,12 @@ class Channel(models.Model): sync_config = self._context.get('app_ai_sync_config') else: sync_config = self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openai_sync_config') - + if self._context.get('app_ai_chat_padding_time'): padding_time = int(self._context.get('app_ai_chat_padding_time')) else: padding_time = int(self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.ai_chat_padding_time')) - + # api_key = self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openapi_api_key') # ai处理,不要自问自答 if ai and answer_id != message.author_id: @@ -305,7 +305,7 @@ class Channel(models.Model): chat_count = 1 else: chat_count = self.chat_count or 3 - + if author_id != answer_id.id and self.channel_type == 'chat': # 私聊 _logger.info(f'私聊:author_id:{author_id},partner_chatgpt.id:{answer_id.id}') @@ -317,25 +317,51 @@ class Channel(models.Model): elif author_id != answer_id.id and msg_vals.get('model', '') == 'mail.channel' and self.channel_type in ['group', 'channel']: # 高级用户自建的话题 channel = self.env[msg_vals.get('model')].browse(msg_vals.get('res_id')) - + try: # 处理提示词 - sys_content = channel.description + add_sys_content - messages.append({"role": "system", "content": sys_content}) + sys_content = '%s%s' % (channel.description if channel.description else "", add_sys_content if add_sys_content else "") + if len(sys_content): + messages.append({"role": "system", "content": sys_content}) c_history = self.get_openai_context(channel.id, author_id, answer_id, openapi_context_timeout, chat_count) if c_history: messages += c_history - messages.append({"role": "user", "content": msg}) + if message.attachment_ids: + attachment = message.attachment_ids[:1] + file_content = ai.get_msg_file_content(message) + if not file_content: + messages.append({"role": "user", "content": msg}) + if attachment.mimetype in ['image/jpeg', 'image/png', 'image/gif', 'image/bmp', 'image/webp']: + messages.append({ + "role": "user", + "content": [ + { + "type": "image_url", + "image_url": { + "url": file_content, + }, + }, + { + "type": "text", + "text": msg + } + ] + }) + else: + messages.append({"role": "system", "content": file_content}) + messages.append({"role": "user", "content": msg}) + else: + messages.append({"role": "user", "content": msg}) msg_len = sum(len(str(m)) for m in messages) # 接口最大接收 8430 Token - if msg_len * 2 > ai.max_send_char: - messages = [] - messages.append({"role": "user", "content": msg}) - msg_len = sum(len(str(m)) for m in messages) - if msg_len * 2 > ai.max_send_char: - new_msg = channel.with_user(user_id).message_post(body=_('您所发送的提示词已超长。'), message_type='comment', - subtype_xmlid='mail.mt_comment', - parent_id=message.id) + # if msg_len * 2 > ai.max_send_char: + # messages = [] + # messages.append({"role": "user", "content": msg}) + # msg_len = sum(len(str(m)) for m in messages) + # if msg_len * 2 > ai.max_send_char: + # new_msg = channel.with_user(user_id).message_post(body=_('您所发送的提示词已超长。'), message_type='comment', + # subtype_xmlid='mail.mt_comment', + # parent_id=message.id) # if msg_len * 2 >= 8000: # messages = [{"role": "user", "content": msg}] diff --git a/app_chatgpt/static/description/index.html b/app_chatgpt/static/description/index.html index 7e8e1000..e9c8d8dc 100644 --- a/app_chatgpt/static/description/index.html +++ b/app_chatgpt/static/description/index.html @@ -4,7 +4,7 @@

Latest ChatGPT AI Center. GPT 3.5, Ali Ai, Baidu Ai, Multi Robot Support. Chat and Training

Support GPT-4o, chatgpt 4 32k. 3.5 turbo, text-davinci, Integration All ChatGpt Api.

-

Latest update: v16.0.25.02.10

+

Latest update: v16.0.25.02.11

Add Alibaba Qwen support(search 'app_ai_ali'), update chatgpt api