File size: 2,459 Bytes
88e64ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
import openai
from .Extension import Extension
# 拓展的配置信息，用于ai理解拓展的功能 *必填*
ext_config:dict = {
"name": "paint", # 拓展名称，用于标识拓展
"arguments": {
'style': 'str', # 绘画风格
'content': 'str', # 绘画内容描述
},
"description": "paint a picture.",
# 参考词，用于上下文参考使用，为空则每次都会被参考(消耗token)
"refer_word": ['paint', '画', '图'],
# 每次消息回复中最大调用次数，不填则默认为99
"max_call_times_per_msg": 1,
# 作者信息
"author": "OREOREO",
# 版本
"version": "0.0.1",
# 拓展简介
"intro": "绘图",
}
class CustomExtension(Extension):
async def call(self, arg_dict: dict, ctx_data: dict) -> dict:
""" 当拓展被调用时执行的函数 *由拓展自行实现*
参数:
arg_dict: dict, 由ai解析的参数字典 {参数名: 参数值(类型为str)}
"""
custom_config:dict = self.get_custom_config() # 获取yaml中的配置信息
# 从arg_dict中获取参数
content = arg_dict.get('content', None)
style = arg_dict.get('style', None)
if style is None:
style = ',anime style, colored-pencil'
response = openai.Image.create(
prompt= content + ',' + style + ', high detail.',
n=1,
size="1024x1024"
)
image_url = response['data'][0]['url']
res = response
if image_url is None:
return {
'text': "[ext_paint] 图片生成错误...",
'image': None, # 图片url
'voice': None, # 语音url
}
elif "rejected" in res:
# 返回的信息将会被发送到会话中
return {
'text': "[ext_paint] 抱歉，这个图违反了ai生成规定，可能是太色了吧", # 文本信息
'image': None, # 图片url
'voice': None, # 语音url
}
else:
# 返回的信息将会被发送到会话中
return {
'text': None, # 文本信息
'image': image_url, # 图片url
'voice': None, # 语音url
}
def __init__(self, custom_config: dict):
super().__init__(ext_config.copy(), custom_config) |