audrius/app-odoo
1
0
Fork 0
mirror of https://github.com/guohuadeng/app-odoo.git synced 2025-02-23 04:11:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

app_ai_kimi增加kimi支持

Browse Source
This commit is contained in:
Chill
2025-02-10 18:00:09 +08:00
parent b2f9348d87
commit 279e925c4c
2 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app_chatgpt/models/ai_robot.py
View File

@@ -133,6 +133,10 @@ GPT-3 A set of models that can understand and generate natural language
# hook，都正常 # hook，都正常
return False return False
def get_msg_files_content(self, message):
# hook
return False
def get_ai(self, data, author_id=False, answer_id=False, param={}): def get_ai(self, data, author_id=False, answer_id=False, param={}):
# 通用方法 # 通用方法
# author_id: 请求的 partner_id 对象 # author_id: 请求的 partner_id 对象

17
app_chatgpt/models/discuss_channel.py
View File

@@ -167,7 +167,6 @@ class Channel(models.Model):
def _notify_thread(self, message, msg_vals=False, **kwargs): def _notify_thread(self, message, msg_vals=False, **kwargs):
rdata = super(Channel, self)._notify_thread(message, msg_vals=msg_vals, **kwargs) rdata = super(Channel, self)._notify_thread(message, msg_vals=msg_vals, **kwargs)
# print(f'rdata:{rdata}')
answer_id = self.env['res.partner'] answer_id = self.env['res.partner']
user_id = self.env['res.users'] user_id = self.env['res.users']
author_id = msg_vals.get('author_id') author_id = msg_vals.get('author_id')
@@ -265,6 +264,7 @@ class Channel(models.Model):
if not api_key: if not api_key:
_logger.warning(_("ChatGPT Robot【%s】have not set open api key.")) _logger.warning(_("ChatGPT Robot【%s】have not set open api key."))
return rdata return rdata
try: try:
openapi_context_timeout = int(self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openapi_context_timeout')) or 60 openapi_context_timeout = int(self.env['ir.config_parameter'].sudo().get_param('app_chatgpt.openapi_context_timeout')) or 60
except: except:
@@ -294,6 +294,11 @@ class Channel(models.Model):
if hasattr(channel, 'is_private') and channel.description: if hasattr(channel, 'is_private') and channel.description:
messages.append({"role": "system", "content": channel.description}) messages.append({"role": "system", "content": channel.description})
if message.attachment_ids:
file_content = ai.get_msg_files_content(message)
if file_content:
messages.append({"role": "system", "content": file_content})
try: try:
c_history = self.get_openai_context(channel.id, author_id, answer_id, openapi_context_timeout, chat_count) c_history = self.get_openai_context(channel.id, author_id, answer_id, openapi_context_timeout, chat_count)
if c_history: if c_history:
@@ -301,11 +306,11 @@ class Channel(models.Model):
messages.append({"role": "user", "content": msg}) messages.append({"role": "user", "content": msg})
msg_len = sum(len(str(m)) for m in messages) msg_len = sum(len(str(m)) for m in messages)
# 接口最大接收 8430 Token # 接口最大接收 8430 Token
if msg_len * 2 > ai.max_send_char: # if msg_len * 2 > ai.max_send_char:
messages = [] # messages = []
if hasattr(channel, 'is_private') and channel.description: # if hasattr(channel, 'is_private') and channel.description:
messages.append({"role": "system", "content": channel.description}) # messages.append({"role": "system", "content": channel.description})
messages.append({"role": "user", "content": msg}) # messages.append({"role": "user", "content": msg})
msg_len = sum(len(str(m)) for m in messages) msg_len = sum(len(str(m)) for m in messages)
if msg_len * 2 > ai.max_send_char: if msg_len * 2 > ai.max_send_char:
new_msg = channel.with_user(user_id).message_post(body=_('您所发送的提示词已超长。'), message_type='comment', new_msg = channel.with_user(user_id).message_post(body=_('您所发送的提示词已超长。'), message_type='comment',