feat: 初步支持绘图api

This commit is contained in:
Rock Chin
2022-12-27 22:52:53 +08:00
parent bdfaecea1c
commit 74fb8ef15c
5 changed files with 38 additions and 5 deletions

View File

@@ -6,6 +6,8 @@ from func_timeout import func_set_timeout
import logging
import openai
from mirai import Image
import config
import pkg.openai.session
@@ -128,6 +130,17 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str) ->
reply_str += "\n当前使用:{}".format(using_key_name)
reply = [reply_str]
elif cmd == 'draw':
if len(params) == 0:
reply = ["[bot]err:请输入图片描述文字"]
else:
session = pkg.openai.session.get_session(session_name)
res = session.draw_image(" ".join(params))
logging.debug("draw_image result:{}".format(res))
reply = [Image(url=res['data'][0]['url'])]
except Exception as e:
mgr.notify_admin("{}指令执行失败:{}".format(session_name, e))
logging.exception(e)