fix: 画图报错

This commit is contained in:
Soulter
2024-05-24 13:33:02 +08:00
parent 8da4433e57
commit f00f1e8933
3 changed files with 3 additions and 6 deletions
+1 -1
View File
@@ -217,7 +217,7 @@ class Command:
"help": "帮助",
"keyword": "设置关键词/关键指令回复",
"update": "更新项目",
"nick": "设置机器人昵称",
"nick": "设置机器人唤醒词",
"plugin": "插件安装、卸载和重载",
"web on/off": "LLM 网页搜索能力",
}
+1 -4
View File
@@ -248,9 +248,6 @@ class CommandOpenAIOfficial(Command):
async def draw(self, message: str):
if self.provider is None:
return False, "未启用 OpenAI 官方 API", "draw"
if message.startswith("/"):
message = message[2:]
elif message.startswith(""):
message = message[1:]
message = message.removeprefix("/").removeprefix("")
img_url = await self.provider.image_generate(message)
return True, img_url, "draw"
+1 -1
View File
@@ -415,7 +415,7 @@ class ProviderOpenAIOfficial(Provider):
return False
async def image_generate(self, prompt, session_id, **kwargs) -> str:
async def image_generate(self, prompt: str, session_id: str = None, **kwargs) -> str:
'''
生成图片
'''