fix: image create prefix match bug

This commit is contained in:
zhayujie
2022-12-19 03:06:48 +08:00
parent bf50694658
commit 17b6c63fb9
2 changed files with 1 additions and 3 deletions

View File

@@ -53,7 +53,6 @@ class OpenAIBot(Bot):
return image_url
def edit_img(self, query, src_img):
openai.api_key = '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'
try:
response = openai.Image.create_edit(
image=open(src_img, 'rb'),
@@ -70,7 +69,6 @@ class OpenAIBot(Bot):
return image_url
def migration_img(self, query, src_img):
openai.api_key = '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'
try:
response = openai.Image.create_variation(

View File

@@ -52,7 +52,7 @@ class WechatChannel(Channel):
if len(str_list) == 2:
content = str_list[1].strip()
img_match_prefix = self.check_prefix(content, ["画图"])
img_match_prefix = self.check_prefix(content, conf().get('image_create_prefix'))
if img_match_prefix:
content = content.split(img_match_prefix, 1)[1].strip()
thead_pool.submit(self._do_send_img, content, from_user_id)