shen/bot-on-anything
1
0
Fork 0
mirror of https://github.com/zhayujie/bot-on-anything.git synced 2026-01-19 01:21:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update link_ai_bot.py

Browse Source
This commit is contained in:
henrikclh
2024-08-24 15:01:34 +08:00
committed by GitHub
parent 1c38c509dc
commit 65d3151e62
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
model/linkai/link_ai_bot.py
View File

@@ -48,16 +48,13 @@ class LinkAIBot(Model):
elif context.get('type', None) == 'IMAGE_CREATE':
ok, res = self.create_img(query, 0)
if ok:
#return [res]
reply_content = res
return [res]
else:
#return res
reply_content = "failed to create images"
return res
# return reply
# else:
# # reply = Reply(ReplyType.ERROR, "Bot不支持处理{}类型的消息".format(context.type))
# return reply
return reply_content
def _chat(self, query, context, retry_count=0):
"""
@@ -121,10 +118,15 @@ class LinkAIBot(Model):
reply_content += knowledge_suffix
# image process
if response["choices"][0].get("img_urls"):
reply_content = response["choices"][0].get("text_content", "") + " " + " ".join(response["choices"][0].get("img_urls"))
else:
reply_content = response["choices"][0]["message"]["content"]
if 'send' in type(context['channel']).__dict__: # 通道实例所属类的定义是否有send方法
thread = threading.Thread(target=self._send_image, args=(context['channel'], context, response["choices"][0].get("img_urls")))
thread.start()
if response["choices"][0].get("text_content"):
reply_content = response["choices"][0].get("text_content")
else:
reply_content = response["choices"][0].get("text_content", "") + " " + " ".join(response["choices"][0].get("img_urls")) # 图像生成时候需要合并文本和图片url
reply_content = self._process_url(reply_content)
# thread = threading.Thread(target=self._send_image, args=(context['channel'], context, response["choices"][0].get("img_urls")))
# thread.start()
# if response["choices"][0].get("text_content"):