mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
38 lines
1003 B
Python
38 lines
1003 B
Python
import logging
|
|
|
|
import mirai
|
|
|
|
from .. import aamgr
|
|
from ....utils import context
|
|
|
|
|
|
@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)]
|
|
config = context.get_config_manager().data
|
|
if config['include_image_description']:
|
|
reply.append(" ".join(ctx.params))
|
|
|
|
return True, reply
|