mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 03:15:06 +08:00
38 lines
1014 B
Python
38 lines
1014 B
Python
import logging
|
|
|
|
import mirai
|
|
|
|
from .. import aamgr
|
|
import config
|
|
|
|
|
|
@aamgr.AbstractCommandNode.register(
|
|
parent=None,
|
|
name="draw",
|
|
description="使用DALL·E生成图片",
|
|
usage="!draw <图片提示语>",
|
|
aliases=[],
|
|
privilege=1
|
|
)
|
|
class DrawCommand(aamgr.AbstractCommandNode):
|
|
@classmethod
|
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
|
import pkg.openai.session
|
|
|
|
reply = []
|
|
|
|
if len(ctx.params) == 0:
|
|
reply = ["[bot]err: 未提供图片描述文字"]
|
|
else:
|
|
session = pkg.openai.session.get_session(ctx.session_name)
|
|
|
|
res = session.draw_image(" ".join(ctx.params))
|
|
|
|
logging.debug("draw_image result:{}".format(res))
|
|
reply = [mirai.Image(url=res['data'][0]['url'])]
|
|
if not (hasattr(config, 'include_image_description')
|
|
and not config.include_image_description):
|
|
reply.append(" ".join(ctx.params))
|
|
|
|
return True, reply
|